add coverage-n

Author: Kludex

pydantic_ai_slim/pydantic_ai/builtin_tools.py

@@ -2,7 +2,7 @@
 
 from abc import ABC
 from dataclasses import dataclass
-from typing import Any, Literal
+from typing import Literal
 
 from typing_extensions import TypedDict
 
@@ -19,7 +19,7 @@ class AbstractBuiltinTool(ABC):
     """
 
     # TODO(Marcelo): We need to handle custom tool definitions per model base.
-    def handle_custom_tool_definition(self, model: str) -> Any: ...
+    # def handle_custom_tool_definition(self, model: str) -> Any: ...
 
 
 @dataclass
@@ -97,5 +97,5 @@ class CodeExecutionTool(AbstractBuiltinTool):
 
     Supported by:
     * Anthropic
-    * Gemini
+    * Google
     """

pydantic_ai_slim/pydantic_ai/models/google.py

@@ -229,10 +229,8 @@ def _get_tools(self, model_request_parameters: ModelRequestParameters) -> list[T
         for tool in model_request_parameters.builtin_tools:
             if isinstance(tool, WebSearchTool):
                 tools.append(ToolDict(google_search=GoogleSearchDict()))
-            elif isinstance(tool, CodeExecutionTool):
+            elif isinstance(tool, CodeExecutionTool):  # pragma: no branch
                 tools.append(ToolDict(code_execution=ToolCodeExecutionDict()))
-            else:
-                raise UserError(f'Unsupported builtin tool: {tool}')
         return tools or None
 
     def _get_tool_config(
@@ -511,10 +509,10 @@ def _content_model_response(m: ModelResponse) -> ContentDict:
             # please open an issue. The below code is the code to send thinking to the provider.
             # parts.append({'text': item.content, 'thought': True})
             pass
-        elif isinstance(item, ServerToolCallPart):
+        elif isinstance(item, ServerToolCallPart):  # pragma: no cover
             # Never returned from google
             pass
-        elif isinstance(item, ServerToolReturnPart):
+        elif isinstance(item, ServerToolReturnPart):  # pragma: no cover
             # Never returned from google
             pass
         else:

pydantic_ai_slim/pydantic_ai/models/openai.py

@@ -832,7 +832,7 @@ def _get_tools(self, model_request_parameters: ModelRequestParameters) -> list[r
     def _get_builtin_tools(self, model_request_parameters: ModelRequestParameters) -> list[responses.ToolParam]:
         tools: list[responses.ToolParam] = []
         for tool in model_request_parameters.builtin_tools:
-            if isinstance(tool, WebSearchTool):
+            if isinstance(tool, WebSearchTool):  # pragma: no branch
                 web_search_tool = responses.WebSearchToolParam(
                     type='web_search_preview', search_context_size=tool.search_context_size
                 )