scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-10 03:28:53 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 84

style(cost): mark cast call sites for type discipline gate

Browse source 
Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
This commit is contained in:
Devin AI 2026-09-18 20:43:11 +00:00
parent 42c27d11dc
commit bb661f8305
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
litellm/litellm_core_utils/llm_cost_calc/tool_call_cost_tracking.py
View file

@ -37,7 +37,7 @@ from litellm.types.utils import (
def _output_item_field(output_item: object, field: str) -> object:
if isinstance(output_item, dict):
fields: Final[Mapping[str, object]] = cast(Mapping[str, object], output_item)
fields: Final[Mapping[str, object]] = cast(Mapping[str, object], output_item) # cast-ok: narrowed by isinstance
return fields.get(field)
return getattr(output_item, field, None)
@ -266,7 +266,7 @@ class StandardBuiltInToolCostTracking:
def _handle_image_generation_cost(response_object: object, custom_llm_provider: str | None) -> float:
if not isinstance(response_object, ResponsesAPIResponse):
return 0.0
output: Final[list[object]] = cast(list[object], response_object.output)
output: Final[list[object]] = cast(list[object], response_object.output) # cast-ok: narrowed by isinstance
return sum(
StandardBuiltInToolCostTracking._image_generation_call_cost(output_item, custom_llm_provider)
for output_item in output