From f74640cb99cbd473def18f7bee7958418a46f3e4 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Sat, 22 Mar 2025 17:24:29 -0700 Subject: [PATCH] test_openai_responses_api_web_search_cost_tracking --- .../llm_cost_calc/tool_call_cost_tracking.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/litellm/litellm_core_utils/llm_cost_calc/tool_call_cost_tracking.py b/litellm/litellm_core_utils/llm_cost_calc/tool_call_cost_tracking.py index 629d6c9e311..8f7d14b20fd 100644 --- a/litellm/litellm_core_utils/llm_cost_calc/tool_call_cost_tracking.py +++ b/litellm/litellm_core_utils/llm_cost_calc/tool_call_cost_tracking.py @@ -86,8 +86,9 @@ class StandardBuiltInToolCostTracking: return search_context_pricing.get("search_context_size_medium", 0.0) elif web_search_options.get("search_context_size", None) == "high": return search_context_pricing.get("search_context_size_high", 0.0) - else: - return 0.0 + return StandardBuiltInToolCostTracking.get_default_cost_for_web_search( + model_info + ) @staticmethod def get_default_cost_for_web_search(model_info: ModelInfo) -> float: