Paramaterize OpenAI agents with context type in test

Author: dandavison

tests/contrib/openai_agents/test_openai.py

@@ -114,7 +114,7 @@ class TestHelloModel(StaticTestModel):
 class HelloWorldAgent:
     @workflow.run
     async def run(self, prompt: str) -> str:
-        agent = Agent(
+        agent = Agent[None](
             name="Assistant",
             instructions="You only respond in haikus.",
         )
@@ -358,7 +358,7 @@ class TestNexusWeatherModel(StaticTestModel):
 class ToolsWorkflow:
     @workflow.run
     async def run(self, question: str) -> str:
-        agent: Agent = Agent(
+        agent = Agent[str](
             name="Tools Workflow",
             instructions="You are a helpful agent.",
             tools=[
@@ -390,7 +390,7 @@ async def run(self, question: str) -> str:
 class NexusToolsWorkflow:
     @workflow.run
     async def run(self, question: str) -> str:
-        agent = Agent(
+        agent = Agent[str](
             name="Nexus Tools Workflow",
             instructions="You are a helpful agent.",
             tools=[
@@ -747,25 +747,25 @@ async def test_research_workflow(client: Client, use_local_model: bool):
 
 
 def orchestrator_agent() -> Agent:
-    spanish_agent = Agent(
+    spanish_agent = Agent[None](
         name="spanish_agent",
         instructions="You translate the user's message to Spanish",
         handoff_description="An english to spanish translator",
     )
 
-    french_agent = Agent(
+    french_agent = Agent[None](
         name="french_agent",
         instructions="You translate the user's message to French",
         handoff_description="An english to french translator",
     )
 
-    italian_agent = Agent(
+    italian_agent = Agent[None](
         name="italian_agent",
         instructions="You translate the user's message to Italian",
         handoff_description="An english to italian translator",
     )
 
-    orchestrator_agent = Agent(
+    orchestrator_agent = Agent[None](
         name="orchestrator_agent",
         instructions=(
             "You are a translation agent. You use the tools given to you to translate."