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

fix time issue with cancel test

Browse source
This commit is contained in:
Swapna Lekkala 2025-10-03 15:45:49 -07:00
parent 581fc8269e
commit e7aba5b35d
1 changed files with 3 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -1049,12 +1049,10 @@ def test_openai_vector_store_file_batch_cancel(compat_client_with_empty_stores,
# Create a vector store
vector_store = compat_client.vector_stores.create(name="batch_cancel_test_store")
# Create a larger batch to ensure we have time to cancel before completion
# Create a large batch to ensure we have time to cancel before completion
file_ids = []
for i in range(50): # Large batch to slow down processing
with BytesIO(
f"This is batch cancel test file {i} with more content to process slowly".encode() * 100
) as file_buffer:
for i in range(100): # Large batch that will take time to process
with BytesIO(f"This is batch cancel test file {i} with substantial content".encode()) as file_buffer:
file_buffer.name = f"batch_cancel_test_{i}.txt"
file = compat_client.files.create(file=file_buffer, purpose="assistants")
file_ids.append(file.id)