scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-15 04:19:08 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 121

test(proxy): strengthen batch fallback regression assertions

Browse source 
Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
This commit is contained in:
mrinal 2026-10-08 18:19:29 +00:00
parent a04b2437e4
commit 1cba5db774
1 changed files with 8 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
tests/unit/enterprise/proxy/hooks/test_managed_files.py
View file

@ -2450,6 +2450,11 @@ async def test_store_batch_output_file_skips_lookup_for_fallback_written_moments
stored_object = managed_file_table.rows["unified-output"].file_object
assert stored_object is not None
assert (stored_object.bytes, stored_object.litellm_details_fallback) == (836, True)
assert managed_file_table.upsert_calls == []
assert managed_file_table.update_many_calls[0][0] == {
"unified_file_id": "unified-output"
}
assert set(managed_file_table.update_many_calls[0][1]) == {"file_object"}
@pytest.mark.asyncio
@ -3133,6 +3138,9 @@ async def test_afile_retrieve_case3_times_out_on_default_provider_route():
)
retrieve.assert_awaited_once()
assert retrieve.await_args.kwargs["file_id"] == "provider-output"
assert retrieve.await_args.kwargs["max_retries"] == 0
assert "custom_llm_provider" not in retrieve.await_args.kwargs
@pytest.mark.asyncio