scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-05 08:07:05 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 137

fix(batches): address Greptile type-discipline feedback

Browse source 
Add Final to the reasoning_tokens local var and type the
_batch_cost_result test helper's parameters, per review feedback on
PR #37208.
This commit is contained in:
mubashir1osmani 2026-08-18 19:37:07 -04:00
parent 682032dd1b
commit ff833a5872
2 changed files with 167 additions and 356 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
litellm/batches/batch_utils.py
View file

@ -165,7 +165,9 @@ def _classify_output_line_stats(
custom_llm_provider=custom_llm_provider,
call_type=CallTypes.aretrieve_batch.value,
)
reasoning_tokens = usage.completion_tokens_details.reasoning_tokens if usage.completion_tokens_details else None
reasoning_tokens: Final = (
usage.completion_tokens_details.reasoning_tokens if usage.completion_tokens_details else None
)
yield _BatchOutputLineStats(
cost=line_cost,
prompt_tokens=usage.prompt_tokens,

519
tests/proxy_unit_tests/test_check_batch_cost.py
View file

File diff suppressed because it is too large Load diff