Linting

Author: tconley1428

temporalio/contrib/openai_agents/_model_parameters.py

@@ -2,7 +2,7 @@
 
 from dataclasses import dataclass
 from datetime import timedelta
-from typing import Optional, Union, Callable, Any
+from typing import Any, Callable, Optional, Union
 
 from agents import Agent, TResponseInputItem
 
@@ -43,7 +43,14 @@ class ModelActivityParameters:
     versioning_intent: Optional[VersioningIntent] = None
     """Versioning intent for the activity."""
 
-    summary_override: Optional[Union[str, Callable[[Agent[Any], Optional[str], Union[str, list[TResponseInputItem]]], str]]] = None
+    summary_override: Optional[
+        Union[
+            str,
+            Callable[
+                [Agent[Any], Optional[str], Union[str, list[TResponseInputItem]]], str
+            ],
+        ]
+    ] = None
     """Summary for the activity execution."""
 
     priority: Priority = Priority.default

temporalio/contrib/openai_agents/_openai_runner.py

@@ -1,17 +1,18 @@
 import json
 import typing
-from dataclasses import replace
-from typing import Any, Union
+from typing import Any, Optional, Union
 
 from agents import (
     Agent,
+    Handoff,
     RunConfig,
+    RunContextWrapper,
     RunResult,
     RunResultStreaming,
     SQLiteSession,
     TContext,
     Tool,
-    TResponseInputItem, Handoff, RunContextWrapper,
+    TResponseInputItem,
 )
 from agents.run import DEFAULT_AGENT_RUNNER, DEFAULT_MAX_TURNS, AgentRunner
 from pydantic_core import to_json
@@ -77,7 +78,7 @@ async def run(
         if run_config is None:
             run_config = RunConfig()
 
-        def model_name(agent: Agent[Any]) -> str:
+        def model_name(agent: Agent[Any]) -> Optional[str]:
             name = run_config.model or agent.model
             if name is not None and not isinstance(name, str):
                 print("Name: ", name, " Agent: ", agent)
@@ -106,10 +107,14 @@ def convert_agent(agent: Agent[Any]) -> None:
                     convert_agent(handoff)
                 elif isinstance(handoff, Handoff):
                     original_invoke = handoff.on_invoke_handoff
-                    async def on_invoke(context: RunContextWrapper[Any], args: str) -> Agent[Any]:
+
+                    async def on_invoke(
+                        context: RunContextWrapper[Any], args: str
+                    ) -> Agent[Any]:
                         handoff_agent = await original_invoke(context, args)
                         convert_agent(handoff_agent)
                         return handoff_agent
+
                     handoff.on_invoke_handoff = on_invoke
 
         convert_agent(starting_agent)

temporalio/contrib/openai_agents/_temporal_model_stub.py

@@ -11,6 +11,7 @@
 from typing import Any, AsyncIterator, Union, cast
 
 from agents import (
+    Agent,
     AgentOutputSchema,
     AgentOutputSchemaBase,
     CodeInterpreterTool,
@@ -25,7 +26,7 @@
     ModelTracing,
     Tool,
     TResponseInputItem,
-    WebSearchTool, Agent,
+    WebSearchTool,
 )
 from agents.items import TResponseStreamEvent
 from openai.types.responses.response_prompt_param import ResponsePromptParam
@@ -70,6 +71,7 @@ async def get_response(
         prompt: Optional[ResponsePromptParam],
     ) -> ModelResponse:
         print("Model stub invocation:", self.model_name)
+
         def make_tool_info(tool: Tool) -> ToolInput:
             if isinstance(
                 tool,
@@ -138,8 +140,15 @@ def make_tool_info(tool: Tool) -> ToolInput:
         )
 
         if self.model_params.summary_override:
-            summary = self.model_params.summary_override if isinstance(self.model_params.summary_override, str) else (
-                self.model_params.summary_override(self.agent, system_instructions, input))
+            summary = (
+                self.model_params.summary_override
+                if isinstance(self.model_params.summary_override, str)
+                else (
+                    self.model_params.summary_override(
+                        self.agent, system_instructions, input
+                    )
+                )
+            )
         else:
             summary = self.agent.name
 

tests/contrib/openai_agents/test_openai.py

@@ -916,7 +916,6 @@ class CustomerServiceModel(StaticTestModel):
     ]
 
 
-
 @workflow.defn
 class CustomerServiceWorkflow:
     def __init__(self, input_items: list[TResponseInputItem] = []):
@@ -995,9 +994,7 @@ async def test_customer_service_workflow(client: Client, use_local_model: bool):
             model_params=ModelActivityParameters(
                 start_to_close_timeout=timedelta(seconds=30)
             ),
-            model_provider= provider
-            if use_local_model
-            else None,
+            model_provider=provider if use_local_model else None,
         )
     ]
     client = Client(**new_config)
@@ -2047,7 +2044,7 @@ async def test_hosted_mcp_tool(client: Client, use_local_model):
 
 
 class AssertDifferentModelProvider(ModelProvider):
-    model_names = set()
+    model_names: set[Optional[str]] = set()
 
     def __init__(self, model: Model):
         self._model = model
@@ -2065,6 +2062,7 @@ class MultipleModelsModel(StaticTestModel):
         ),
     ]
 
+
 @workflow.defn
 class MultipleModelWorkflow:
     @workflow.run
@@ -2078,7 +2076,7 @@ async def run(self):
             name="Lazy Assistant",
             model="gpt-4o-mini",
             instructions="You delegate all your work to another agent.",
-            handoffs=[underling]
+            handoffs=[underling],
         )
         result = await Runner.run(
             starting_agent=starting_agent,
@@ -2095,7 +2093,7 @@ async def test_multiple_models(client: Client):
             model_params=ModelActivityParameters(
                 start_to_close_timeout=timedelta(seconds=120)
             ),
-            model_provider=provider
+            model_provider=provider,
         )
     ]
     client = Client(**new_config)