[Temporal - Documentation] Fixing PR errors

Author: scalabreseGD

examples/pydantic_ai_examples/temporal_streaming/agents.py

@@ -4,6 +4,7 @@
 and custom tools for data analysis.
 """
 from datetime import timedelta
+from typing import Any
 
 from temporalio.common import RetryPolicy
 from temporalio.workflow import ActivityConfig
@@ -17,7 +18,7 @@
 from .datamodels import AgentDependencies
 
 
-async def get_mcp_toolsets() -> dict[str, FilteredToolset]:
+async def get_mcp_toolsets() -> dict[str, FilteredToolset[AgentDependencies]]:
     """
     Initialize MCP toolsets for the agent.
 
@@ -34,7 +35,7 @@ async def get_mcp_toolsets() -> dict[str, FilteredToolset]:
     return {'yahoo': yf_server.filtered(lambda ctx, tool_def: True)}
 
 
-async def get_claude_model(parallel_tool_calls: bool = True, **kwargs):
+async def get_claude_model(parallel_tool_calls: bool = True, **kwargs: Any) -> AnthropicModel:
     """
     Create and configure the Claude model.
 
@@ -60,8 +61,8 @@ async def get_claude_model(parallel_tool_calls: bool = True, **kwargs):
     return model
 
 
-async def build_agent(stream_handler: EventStreamHandler,
-                      **kwargs) -> TemporalAgent:
+async def build_agent(stream_handler: EventStreamHandler[AgentDependencies],
+                      **kwargs: Any) -> TemporalAgent[AgentDependencies, str]:
     """
     Build and configure the agent with tools and temporal settings.
 
@@ -78,7 +79,7 @@ async def build_agent(stream_handler: EventStreamHandler,
     agent_name = 'YahooFinanceSearchAgent'
 
     toolsets = await get_mcp_toolsets()
-    agent = Agent(
+    agent: Agent[AgentDependencies, str] = Agent[AgentDependencies, str](
         name=agent_name,
         model=await get_claude_model(**kwargs),
         toolsets=[*toolsets.values()],

examples/pydantic_ai_examples/temporal_streaming/main.py

@@ -28,7 +28,9 @@ async def poll_events(workflow_handle: WorkflowHandle[Any, str]) -> None:
         workflow_handle: Handle to the running workflow.
     """
     while True:
-        event = await workflow_handle.query('event_stream', result_type=Union[EventStream | None])
+        event: Union[EventStream, None] = await workflow_handle.query('event_stream',
+                                                                      result_type=Union[
+                                                                          EventStream, None])  # type: ignore[misc]
         if event is None:
             await asyncio.sleep(0.1)
             continue
@@ -55,6 +57,7 @@ async def main() -> None:
     """
     # Connect to Temporal server
     client = await Client.connect(
+        # target_host='localhost:7233',
         target_host='localhost:7233',
         plugins=[PydanticAIPlugin()],
     )
@@ -77,7 +80,7 @@ async def main() -> None:
     ):
         # Execute the workflow
         workflow_id = f'yahoo-finance-search-{uuid.uuid4()}'
-        workflow_handle: WorkflowHandle[Any, str] = await client.start_workflow(
+        workflow_handle: WorkflowHandle[Any, str] = await client.start_workflow(  # type: ignore[misc]
             'YahooFinanceSearchWorkflow',
             arg='What are the latest financial metrics for Apple (AAPL)?',
             id=workflow_id,

examples/pydantic_ai_examples/temporal_streaming/streaming_handler.py

@@ -24,7 +24,7 @@
 
 
 async def streaming_handler(
-        ctx: RunContext,
+        ctx: RunContext[AgentDependencies],
         event_stream_events: AsyncIterable[AgentStreamEvent],
 ) -> None:
     """
@@ -74,18 +74,25 @@ async def streaming_handler(
                 if len(output_tool_delta['tool_call_id']) == 0:
                     output_tool_delta['tool_call_id'] += event.delta.tool_call_id or ''
                 output_tool_delta['tool_name_delta'] += event.delta.tool_name_delta or ''
-                output_tool_delta['args_delta'] += event.delta.args_delta or ''
+                # Handle args_delta which can be str or dict
+                args_delta = event.delta.args_delta
+                if isinstance(args_delta, str):
+                    output_tool_delta['args_delta'] += args_delta
+                elif isinstance(args_delta, dict):
+                    output_tool_delta['args_delta'] += str(args_delta)
 
     # Add accumulated tool delta output if present
     if len(output_tool_delta['tool_call_id']):
         output += f'\nTool Call Id: {output_tool_delta["tool_call_id"]}'
         output += f'\nTool Name: {output_tool_delta["tool_name_delta"]}'
-        output += f'\nTool Args: {output_tool_delta["args_delta"]}'
+        args_delta_str = str(output_tool_delta["args_delta"])
+        output += f'\nTool Args: {args_delta_str}'
 
     # Send events to workflow if running in an activity
     deps: AgentDependencies = ctx.deps
 
     workflow_id: str = deps.workflow_id
     run_id: str = deps.run_id
-    workflow_handle: WorkflowHandle = activity.client().get_workflow_handle(workflow_id=workflow_id, run_id=run_id)
+    from typing import Any
+    workflow_handle: WorkflowHandle[Any, Any] = activity.client().get_workflow_handle(workflow_id=workflow_id, run_id=run_id)  # type: ignore[misc]
     await workflow_handle.signal('append_event', arg=EventStream(kind=EventKind.EVENT, content=output))

examples/pydantic_ai_examples/temporal_streaming/utils.py

@@ -7,7 +7,7 @@
 import yaml
 
 
-def recursively_modify_api_key(conf) -> dict[str, Any]:
+def recursively_modify_api_key(conf: dict[str, Any]) -> dict[str, Any]:
     """
     Recursively replace API key placeholders with environment variable values.
 
@@ -22,16 +22,17 @@ def recursively_modify_api_key(conf) -> dict[str, Any]:
         A copy of the configuration with API keys replaced by environment variable values.
     """
 
-    def inner(_conf):
+    def inner(_conf: dict[str, Any]) -> None:
         for key, value in _conf.items():
             if isinstance(value, dict):
-                inner(value)
+                inner(value)  # type: ignore[arg-type]
             elif isinstance(value, list):
-                if len(value) > 0 and isinstance(value[0], dict):
-                    for item in value:
-                        inner(item)
+                if len(value) > 0 and isinstance(value[0], dict):  # type: ignore[misc]
+                    item: dict[str, Any]
+                    for item in value:  # type: ignore[assignment,misc]
+                        inner(item)  # type: ignore[arg-type]
                 else:
-                    _conf[key] = [os.environ.get(str(v), v) for v in value]
+                    _conf[key] = [str(os.environ.get(str(v), v)) for v in value]  # type: ignore[misc]
             elif isinstance(value, str):
                 _conf[key] = os.environ.get(value, value)
             else:
@@ -42,7 +43,7 @@ def inner(_conf):
     return copy_conf
 
 
-def read_config_yml(path) -> dict[str, Any]:
+def read_config_yml(path: str) -> dict[str, Any]:
     """
     Read and process a YAML configuration file.
 

examples/pydantic_ai_examples/temporal_streaming/workflow.py

@@ -44,7 +44,7 @@ async def run(self, user_prompt: str) -> str:
             The agent's final output.
         """
         # Retrieve environment variables from configuration
-        wf_vars = await workflow.execute_activity(
+        wf_vars = await workflow.execute_activity(  # type: ignore[misc]
             activity='retrieve_env_vars',
             start_to_close_timeout=timedelta(seconds=10),
             result_type=dict[str, Any],