phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-24 18:24:20 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

test_e2e_batches_files
All checks were successful
Read Version from pyproject.toml / read-version (push) Successful in 14s
Details

Browse source
This commit is contained in:
Ishaan Jaff 2025-03-06 08:42:17 -08:00
parent 3d87bf9fe6
commit cca71b41b3
1 changed files with 4 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

6
tests/openai_misc_endpoints_tests/test_openai_batches_endpoint.py
View file

@ -101,7 +101,7 @@ def await_batch_completion(batch_id: str, custom_llm_provider: str):
)
if batch.status == "completed":
print(f"Batch {batch_id} completed.")
return
return batch.id
tries += 1
print(f"waiting for batch to complete... (attempt {tries}/{max_tries})")
@ -173,9 +173,11 @@ def test_e2e_batches_files(custom_llm_provider):
# azure takes very long to complete a batch
return
else:
await_batch_completion(
response_batch_id = await_batch_completion(
batch_id=batch_id, custom_llm_provider=custom_llm_provider
)
if response_batch_id is None:
return
write_content_to_file(
batch_id=batch_id,