phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-10-08 21:04:39 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix test

Browse source
This commit is contained in:
Swapna Lekkala 2025-10-06 10:32:02 -07:00
parent 4e4db89df6
commit 1b68392f92
1 changed files with 4 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -915,7 +915,7 @@ def test_openai_vector_store_file_batch_create_and_retrieve(compat_client_with_e
# Create multiple files # Create multiple files
file_ids = [] file_ids = []
for i in range(3): for i in range(2):
with BytesIO(f"This is batch test file {i}".encode()) as file_buffer: with BytesIO(f"This is batch test file {i}".encode()) as file_buffer:
file_buffer.name = f"batch_test_{i}.txt" file_buffer.name = f"batch_test_{i}.txt"
file = compat_client.files.create(file=file_buffer, purpose="assistants") file = compat_client.files.create(file=file_buffer, purpose="assistants")
@ -936,7 +936,7 @@ def test_openai_vector_store_file_batch_create_and_retrieve(compat_client_with_e
assert hasattr(batch, "created_at") assert hasattr(batch, "created_at")
# Wait for batch processing to complete # Wait for batch processing to complete
max_retries = 30 # 30 seconds max wait max_retries = 60 # 60 seconds max wait (increased for file processing delays)
retries = 0 retries = 0
retrieved_batch = None retrieved_batch = None
while retries < max_retries: while retries < max_retries:
@ -981,7 +981,7 @@ def test_openai_vector_store_file_batch_list_files(compat_client_with_empty_stor
) )
# Wait for batch processing to complete # Wait for batch processing to complete
max_retries = 30 # 30 seconds max wait max_retries = 60 # 60 seconds max wait (increased for file processing delays)
retries = 0 retries = 0
while retries < max_retries: while retries < max_retries:
retrieved_batch = compat_client.vector_stores.file_batches.retrieve( retrieved_batch = compat_client.vector_stores.file_batches.retrieve(
@ -1098,7 +1098,6 @@ def test_openai_vector_store_file_batch_retrieve_contents(compat_client_with_emp
file_data = [ file_data = [
("test_file_1.txt", b"This is the content of test file 1"), ("test_file_1.txt", b"This is the content of test file 1"),
("test_file_2.txt", b"This is the content of test file 2"), ("test_file_2.txt", b"This is the content of test file 2"),
("test_file_3.txt", b"This is the content of test file 3"),
] ]
file_ids = [] file_ids = []
@ -1115,7 +1114,7 @@ def test_openai_vector_store_file_batch_retrieve_contents(compat_client_with_emp
) )
# Wait for batch processing to complete # Wait for batch processing to complete
max_retries = 30 # 30 seconds max wait max_retries = 60 # 60 seconds max wait (increased for file processing delays)
retries = 0 retries = 0
while retries < max_retries: while retries < max_retries:
retrieved_batch = compat_client.vector_stores.file_batches.retrieve( retrieved_batch = compat_client.vector_stores.file_batches.retrieve(