From fb1d1f2153275431faf57a200a7204fc453fbffa Mon Sep 17 00:00:00 2001 From: Swapna Lekkala Date: Tue, 30 Sep 2025 16:11:56 -0700 Subject: [PATCH] fix test --- tests/integration/vector_io/test_openai_vector_stores.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/integration/vector_io/test_openai_vector_stores.py b/tests/integration/vector_io/test_openai_vector_stores.py index 8d1f8ab25..457200111 100644 --- a/tests/integration/vector_io/test_openai_vector_stores.py +++ b/tests/integration/vector_io/test_openai_vector_stores.py @@ -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"],