Fix: is_web_search_tool_chat_completion

litellm/integrations/websearch_interception/handler.py

@@ -17,6 +17,7 @@ from litellm.integrations.custom_logger import CustomLogger
 from litellm.integrations.websearch_interception.tools import (
     get_litellm_web_search_tool,
     is_web_search_tool,
+    is_web_search_tool_chat_completion,
 )
 from litellm.integrations.websearch_interception.transformation import (
     WebSearchTransformation,
@@ -325,11 +326,11 @@ class WebSearchInterceptionLogger(CustomLogger):
             )
             return False, {}
 
-        # Check if tools include any web search tool
-        has_websearch_tool = any(is_web_search_tool(t) for t in (tools or []))
+        # Check if tools include any web search tool (strict check for chat completions)
+        has_websearch_tool = any(is_web_search_tool_chat_completion(t) for t in (tools or []))
         if not has_websearch_tool:
             verbose_logger.debug(
-                "WebSearchInterception: No web search tool in request"
+                "WebSearchInterception: No litellm_web_search tool in request"
             )
             return False, {}
 

litellm/integrations/websearch_interception/tools.py

@@ -49,6 +49,50 @@ def get_litellm_web_search_tool() -> Dict[str, Any]:
     }
 
 
+def is_web_search_tool_chat_completion(tool: Dict[str, Any]) -> bool:
+    """
+    Check if a tool is a web search tool for Chat Completions API (strict check).
+    
+    This is a stricter version that ONLY checks for the exact LiteLLM web search tool name.
+    Use this for Chat Completions API to avoid false positives with user-defined tools.
+
+    Detects ONLY:
+    - LiteLLM standard: name == "litellm_web_search" (Anthropic format)
+    - OpenAI format: type == "function" with function.name == "litellm_web_search"
+
+    Args:
+        tool: Tool dictionary to check
+
+    Returns:
+        True if tool is exactly the LiteLLM web search tool
+
+    Example:
+        >>> is_web_search_tool_chat_completion({"name": "litellm_web_search"})
+        True
+        >>> is_web_search_tool_chat_completion({"type": "function", "function": {"name": "litellm_web_search"}})
+        True
+        >>> is_web_search_tool_chat_completion({"name": "web_search"})
+        False
+        >>> is_web_search_tool_chat_completion({"name": "WebSearch"})
+        False
+    """
+    tool_name = tool.get("name", "")
+    tool_type = tool.get("type", "")
+    
+    # Check for OpenAI format: {"type": "function", "function": {"name": "litellm_web_search"}}
+    if tool_type == "function" and "function" in tool:
+        function_def = tool.get("function", {})
+        function_name = function_def.get("name", "")
+        if function_name == LITELLM_WEB_SEARCH_TOOL_NAME:
+            return True
+
+    # Check for LiteLLM standard tool (Anthropic format)
+    if tool_name == LITELLM_WEB_SEARCH_TOOL_NAME:
+        return True
+
+    return False
+
+
 def is_web_search_tool(tool: Dict[str, Any]) -> bool:
     """
     Check if a tool is a web search tool (native or LiteLLM standard).