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

get_cost_for_built_in_tools

Browse source
This commit is contained in:
Ishaan Jaff 2025-03-22 17:06:34 -07:00
parent beb68037d0
commit 904795443d
1 changed files with 15 additions and 12 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -38,24 +38,27 @@ class StandardBuiltInToolCostTracking:
model_info = litellm.get_model_info(
model=model, custom_llm_provider=custom_llm_provider
)
if (
standard_built_in_tools_params is not None
and standard_built_in_tools_params.get("web_search_options", None)
is not None
):
return StandardBuiltInToolCostTracking.get_cost_for_web_search(
web_search_options=standard_built_in_tools_params.get(
"web_search_options", None
),
model_info=model_info,
)
if isinstance(response_object, ModelResponse):
if (
standard_built_in_tools_params is not None
and standard_built_in_tools_params.get("web_search_options", None)
is not None
):
return StandardBuiltInToolCostTracking.get_cost_for_web_search(
web_search_options=standard_built_in_tools_params.get(
"web_search_options", None
),
model_info=model_info,
)
elif StandardBuiltInToolCostTracking.response_includes_annotations(
if StandardBuiltInToolCostTracking.response_includes_annotations(
response_object
):
return StandardBuiltInToolCostTracking.get_default_cost_for_web_search(
model_info
)
return 0.0
@staticmethod