Add DynamicToolset validation and export TemporalDynamicToolset

Author: qyinm

pydantic_ai_slim/pydantic_ai/durable_exec/temporal/__init__.py

@@ -14,6 +14,7 @@
 
 from ...exceptions import UserError
 from ._agent import TemporalAgent
+from ._dynamic_toolset import TemporalDynamicToolset
 from ._function_toolset import TemporalFunctionToolset
 from ._logfire import LogfirePlugin
 from ._mcp import TemporalMCPToolset
@@ -28,6 +29,7 @@
     'AgentPlugin',
     'TemporalRunContext',
     'TemporalWrapperToolset',
+    'TemporalDynamicToolset',
     'TemporalFunctionToolset',
     'TemporalMCPToolset',
     'PydanticAIWorkflow',

pydantic_ai_slim/pydantic_ai/durable_exec/temporal/_agent.py

@@ -50,7 +50,7 @@ def _validate_temporal_toolsets(toolsets: Sequence[AbstractToolset[AgentDepsT]],
     """Validate that all leaf toolsets requiring temporal wrapping are properly wrapped.
 
     This function recursively traverses the toolset hierarchy and checks that any leaf
-    toolsets that need temporal wrapping (FunctionToolset, MCPServer, FastMCPToolset)
+    toolsets that need temporal wrapping (FunctionToolset, MCPServer, FastMCPToolset, DynamicToolset)
     are wrapped in a TemporalWrapperToolset.
 
     Args:
@@ -73,10 +73,15 @@ def validate_toolset(t: AbstractToolset[AgentDepsT]) -> AbstractToolset[AgentDep
         if isinstance(t, TemporalWrapperToolset):
             return t
 
-        # Check if this is a FunctionToolset that needs wrapping
         if isinstance(t, FunctionToolset):
             raise UserError(error_msg)
 
+        # Check if this is a DynamicToolset that needs wrapping
+        from pydantic_ai.toolsets._dynamic import DynamicToolset
+
+        if isinstance(t, DynamicToolset):
+            raise UserError(error_msg)
+
         # Check if this is an MCPServer that needs wrapping
         try:
             from pydantic_ai.mcp import MCPServer