scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-04 02:31:27 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 70

test(e2e): emit gemini stream usage only on the final chunk

Browse source 
Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
This commit is contained in:
kerry 2026-09-17 01:53:52 +00:00
parent 1de633ac36
commit ef1f306a7d
1 changed files with 2 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
tests/e2e/cost_calculation/scripted_provider.py
View file

@ -750,10 +750,8 @@ def _gemini_body(scenario: Scenario, requested_model: str) -> Mapping[str, objec
def _gemini_sse(scenario: Scenario, requested_model: str) -> bytes:
emit_usage: Final = scenario.stream_usage == "final_chunk"
first: Final = (
_jobj(*((key, value) for key, value in _gemini_body(scenario, requested_model).items() if key != "usageMetadata"))
if scenario.stream_usage == "absent"
else _gemini_body(scenario, requested_model)
first: Final = _jobj(
*((key, value) for key, value in _gemini_body(scenario, requested_model).items() if key != "usageMetadata")
)
return _sse(
(