phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-10-03 19:57:35 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix different error in clients

Browse source
This commit is contained in:
Swapna Lekkala 2025-09-30 16:23:17 -07:00
parent fb1d1f2153
commit 943255697e
1 changed files with 9 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

11
tests/integration/vector_io/test_openai_vector_stores.py
View file

@ -1117,15 +1117,22 @@ def test_openai_vector_store_file_batch_error_handling(compat_client_with_empty_
assert batch.file_counts.total == len(file_ids) assert batch.file_counts.total == len(file_ids)
# Invalid files should be marked as failed # Invalid files should be marked as failed
assert batch.file_counts.failed >= 0 # Implementation may vary assert batch.file_counts.failed >= 0 # Implementation may vary
# Determine expected errors based on client type
if isinstance(compat_client, LlamaStackAsLibraryClient):
errors = ValueError
else:
errors = (BadRequestError, OpenAIBadRequestError)
# Test retrieving non-existent batch # Test retrieving non-existent batch
with pytest.raises(ValueError): # Should raise an error for non-existent batch with pytest.raises(errors): # Should raise an error for non-existent batch
compat_client.vector_stores.file_batches.retrieve( compat_client.vector_stores.file_batches.retrieve(
vector_store_id=vector_store.id, vector_store_id=vector_store.id,
batch_id="non_existent_batch_id", batch_id="non_existent_batch_id",
) )
# Test operations on non-existent vector store # Test operations on non-existent vector store
with pytest.raises(ValueError): # Should raise an error for non-existent vector store with pytest.raises(errors): # Should raise an error for non-existent vector store
compat_client.vector_stores.file_batches.create( compat_client.vector_stores.file_batches.create(
vector_store_id="non_existent_vector_store", vector_store_id="non_existent_vector_store",
file_ids=["any_file_id"], file_ids=["any_file_id"],