Update unknown tool retry logic

pydantic_ai_slim/pydantic_ai/_tool_manager.py

@@ -35,6 +35,8 @@ class ToolManager(Generic[AgentDepsT]):
     """The cached tools for this run step."""
     failed_tools: set[str] = field(default_factory=set)
     """Names of tools that failed in this run step."""
+    default_max_retries: int = 1
+    """Default number of times to retry a tool"""
 
     @classmethod
     @contextmanager
@@ -176,7 +178,7 @@ async def _call_tool(
 
             return result
         except (ValidationError, ModelRetry) as e:
-            max_retries = tool.max_retries if tool is not None else 1
+            max_retries = tool.max_retries if tool is not None else self.default_max_retries
             current_retry = self.ctx.retries.get(name, 0)
 
             if current_retry == max_retries:

pydantic_ai_slim/pydantic_ai/agent/__init__.py

@@ -570,7 +570,7 @@ async def main():
                 output_toolset.max_retries = self._max_result_retries
                 output_toolset.output_validators = output_validators
         toolset = self._get_toolset(output_toolset=output_toolset, additional_toolsets=toolsets)
-        tool_manager = ToolManager[AgentDepsT](toolset)
+        tool_manager = ToolManager[AgentDepsT](toolset, default_max_retries=self._max_tool_retries)
 
         # Build the graph
         graph = _agent_graph.build_agent_graph(self.name, self._deps_type, output_type_)