From ded612563dd50145a605b22bf7a481c208d28fd5 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Sat, 22 Mar 2025 18:02:22 -0700 Subject: [PATCH] _get_file_search_tool_call --- .../llm_cost_calc/tool_call_cost_tracking.py | 4 ++++ 1 file changed, 4 insertions(+) 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 723910dbb0c..ca20bce8bfa 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 @@ -137,6 +137,8 @@ class StandardBuiltInToolCostTracking: return WebSearchOptions(**kwargs.get("web_search_options", {})) if "tools" in kwargs: tools = kwargs.get("tools", []) + if tools is None: + return None # Look for web search tool in the tools array for tool in tools: if isinstance(tool, dict): @@ -148,6 +150,8 @@ class StandardBuiltInToolCostTracking: def _get_file_search_tool_call(kwargs: Dict) -> Optional[FileSearchTool]: if "tools" in kwargs: tools = kwargs.get("tools", []) + if tools is None: + return None # Look for web search tool in the tools array for tool in tools: if isinstance(tool, dict):