scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-11 22:51:28 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 123

mypy lint fix

Browse source
This commit is contained in:
Ishaan Jaffer 2025-09-27 12:26:06 -07:00
parent 82978f4c5c
commit 9ea8c72ebc
1 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -123,7 +123,7 @@ class StandardBuiltInToolCostTracking:
model_info = StandardBuiltInToolCostTracking._safe_get_model_info(
model=model, custom_llm_provider=custom_llm_provider
)
file_search_raw = standard_built_in_tools_params.get("file_search", {})
file_search_raw: Any = standard_built_in_tools_params.get("file_search", {})
file_search_usage: Optional[FileSearchTool] = (
FileSearchTool(**file_search_raw) if file_search_raw else None
)
@ -427,7 +427,7 @@ class StandardBuiltInToolCostTracking:
if model_info is None:
return 0.0
search_context_raw = model_info.get("search_context_cost_per_query", {})
search_context_raw: Any = model_info.get("search_context_cost_per_query", {})
search_context_pricing: SearchContextCostPerQuery = (
SearchContextCostPerQuery(**search_context_raw)
if search_context_raw
@ -454,7 +454,7 @@ class StandardBuiltInToolCostTracking:
"""
if model_info is None:
return 0.0
search_context_raw = model_info.get("search_context_cost_per_query", {}) or {}
search_context_raw: Any = model_info.get("search_context_cost_per_query", {}) or {}
search_context_pricing: SearchContextCostPerQuery = (
SearchContextCostPerQuery(**search_context_raw)
if search_context_raw