mirror of
https://github.com/meta-llama/llama-stack.git
synced 2025-10-03 19:57:35 +00:00
fix test
This commit is contained in:
parent
b3b31bf357
commit
fb1d1f2153
1 changed files with 2 additions and 2 deletions
|
@ -1118,14 +1118,14 @@ def test_openai_vector_store_file_batch_error_handling(compat_client_with_empty_
|
|||
# Invalid files should be marked as failed
|
||||
assert batch.file_counts.failed >= 0 # Implementation may vary
|
||||
# Test retrieving non-existent batch
|
||||
with pytest.raises((BadRequestError, OpenAIBadRequestError)): # Should raise an error for non-existent batch
|
||||
with pytest.raises(ValueError): # Should raise an error for non-existent batch
|
||||
compat_client.vector_stores.file_batches.retrieve(
|
||||
vector_store_id=vector_store.id,
|
||||
batch_id="non_existent_batch_id",
|
||||
)
|
||||
|
||||
# Test operations on non-existent vector store
|
||||
with pytest.raises((BadRequestError, OpenAIBadRequestError)): # Should raise an error for non-existent vector store
|
||||
with pytest.raises(ValueError): # Should raise an error for non-existent vector store
|
||||
compat_client.vector_stores.file_batches.create(
|
||||
vector_store_id="non_existent_vector_store",
|
||||
file_ids=["any_file_id"],
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue