remove allow_partial from call_tool

Author: petersli

pydantic_ai_slim/pydantic_ai/durable_exec/dbos/_mcp_server.py

@@ -86,6 +86,5 @@ async def call_tool(
         tool_args: dict[str, Any],
         ctx: RunContext[AgentDepsT],
         tool: ToolsetTool[AgentDepsT],
-        allow_partial: bool = False,
     ) -> ToolResult:
         return await self._dbos_wrapped_call_tool_step(name, tool_args, ctx, tool)

pydantic_ai_slim/pydantic_ai/durable_exec/prefect/_function_toolset.py

@@ -42,14 +42,13 @@ async def call_tool(
         tool_args: dict[str, Any],
         ctx: RunContext[AgentDepsT],
         tool: ToolsetTool[AgentDepsT],
-        allow_partial: bool = False,
     ) -> Any:
         """Call a tool, wrapped as a Prefect task with a descriptive name."""
         # Check if this specific tool has custom config or is disabled
         tool_specific_config = self._tool_task_config.get(name, default_task_config)
         if tool_specific_config is None:
             # None means this tool should not be wrapped as a task
-            return await super().call_tool(name, tool_args, ctx, tool, allow_partial=allow_partial)
+            return await super().call_tool(name, tool_args, ctx, tool)
 
         # Merge tool-specific config with default config
         merged_config = self._task_config | tool_specific_config

pydantic_ai_slim/pydantic_ai/durable_exec/prefect/_mcp_server.py

@@ -53,7 +53,6 @@ async def call_tool(
         tool_args: dict[str, Any],
         ctx: RunContext[AgentDepsT],
         tool: ToolsetTool[AgentDepsT],
-        allow_partial: bool = False,
     ) -> ToolResult:
         """Call an MCP tool, wrapped as a Prefect task with a descriptive name."""
         return await self._call_tool_task.with_options(name=f'Call MCP Tool: {name}', **self._task_config)(

pydantic_ai_slim/pydantic_ai/durable_exec/temporal/_function_toolset.py

@@ -111,10 +111,9 @@ async def call_tool(
         tool_args: dict[str, Any],
         ctx: RunContext[AgentDepsT],
         tool: ToolsetTool[AgentDepsT],
-        allow_partial: bool = False,
     ) -> Any:
         if not workflow.in_workflow():
-            return await super().call_tool(name, tool_args, ctx, tool, allow_partial=allow_partial)
+            return await super().call_tool(name, tool_args, ctx, tool)
 
         tool_activity_config = self.tool_activity_config.get(name, {})
         if tool_activity_config is False:
@@ -124,7 +123,7 @@ async def call_tool(
                     f'Temporal activity config for tool {name!r} has been explicitly set to `False` (activity disabled), '
                     'but non-async tools are run in threads which are not supported outside of an activity. Make the tool function async instead.'
                 )
-            return await super().call_tool(name, tool_args, ctx, tool, allow_partial=allow_partial)
+            return await super().call_tool(name, tool_args, ctx, tool)
 
         tool_activity_config = self.activity_config | tool_activity_config
         serialized_run_context = self.run_context_type.serialize_run_context(ctx)

pydantic_ai_slim/pydantic_ai/durable_exec/temporal/_mcp_server.py

@@ -125,10 +125,9 @@ async def call_tool(
         tool_args: dict[str, Any],
         ctx: RunContext[AgentDepsT],
         tool: ToolsetTool[AgentDepsT],
-        allow_partial: bool = False,
     ) -> ToolResult:
         if not workflow.in_workflow():
-            return await super().call_tool(name, tool_args, ctx, tool, allow_partial=allow_partial)
+            return await super().call_tool(name, tool_args, ctx, tool)
 
         tool_activity_config = self.activity_config | self.tool_activity_config.get(name, {})
         serialized_run_context = self.run_context_type.serialize_run_context(ctx)

pydantic_ai_slim/pydantic_ai/mcp.py

@@ -267,7 +267,6 @@ async def call_tool(
         tool_args: dict[str, Any],
         ctx: RunContext[Any],
         tool: ToolsetTool[Any],
-        allow_partial: bool = False,
     ) -> ToolResult:
         if self.tool_prefix:
             name = name.removeprefix(f'{self.tool_prefix}_')

pydantic_ai_slim/pydantic_ai/toolsets/_dynamic.py

@@ -72,10 +72,9 @@ async def call_tool(
         tool_args: dict[str, Any],
         ctx: RunContext[AgentDepsT],
         tool: ToolsetTool[AgentDepsT],
-        allow_partial: bool = False,
     ) -> Any:
         assert self._toolset is not None
-        return await self._toolset.call_tool(name, tool_args, ctx, tool, allow_partial=allow_partial)
+        return await self._toolset.call_tool(name, tool_args, ctx, tool)
 
     def apply(self, visitor: Callable[[AbstractToolset[AgentDepsT]], None]) -> None:
         if self._toolset is None:

pydantic_ai_slim/pydantic_ai/toolsets/abstract.py

@@ -121,7 +121,6 @@ async def call_tool(
         tool_args: dict[str, Any],
         ctx: RunContext[AgentDepsT],
         tool: ToolsetTool[AgentDepsT],
-        allow_partial: bool = False,
     ) -> Any:
         """Call a tool with the given arguments.
 

pydantic_ai_slim/pydantic_ai/toolsets/approval_required.py

@@ -29,9 +29,8 @@ async def call_tool(
         tool_args: dict[str, Any],
         ctx: RunContext[AgentDepsT],
         tool: ToolsetTool[AgentDepsT],
-        allow_partial: bool = False,
     ) -> Any:
         if not ctx.tool_call_approved and self.approval_required_func(ctx, tool.tool_def, tool_args):
             raise ApprovalRequired
 
-        return await super().call_tool(name, tool_args, ctx, tool, allow_partial=allow_partial)
+        return await super().call_tool(name, tool_args, ctx, tool)

pydantic_ai_slim/pydantic_ai/toolsets/combined.py

@@ -89,10 +89,9 @@ async def call_tool(
         tool_args: dict[str, Any],
         ctx: RunContext[AgentDepsT],
         tool: ToolsetTool[AgentDepsT],
-        allow_partial: bool = False,
     ) -> Any:
         assert isinstance(tool, _CombinedToolsetTool)
-        return await tool.source_toolset.call_tool(name, tool_args, ctx, tool.source_tool, allow_partial=allow_partial)
+        return await tool.source_toolset.call_tool(name, tool_args, ctx, tool.source_tool)
 
     def apply(self, visitor: Callable[[AbstractToolset[AgentDepsT]], None]) -> None:
         for toolset in self.toolsets: