cleanup

Author: donald-pinckney

tests/contrib/openai_agents/test_openai.py

@@ -13,7 +13,6 @@
     Sequence,
     Union,
     cast,
-    no_type_check,
 )
 
 import nexusrpc
@@ -552,14 +551,13 @@ async def test_nexus_tool_workflow(
             )
 
 
-@no_type_check
 def research_mock_model():
     responses = [
         ResponseBuilders.output_message(
             '{"searches":[{"query":"best Caribbean surfing spots April","reason":"Identify locations with optimal surfing conditions in the Caribbean during April."},{"query":"top Caribbean islands for hiking April","reason":"Find Caribbean islands with excellent hiking opportunities that are ideal in April."},{"query":"Caribbean water sports destinations April","reason":"Locate Caribbean destinations offering a variety of water sports activities in April."},{"query":"surfing conditions Caribbean April","reason":"Understand the surfing conditions and which islands are suitable for surfing in April."},{"query":"Caribbean adventure travel hiking surfing","reason":"Explore adventure travel options that combine hiking and surfing in the Caribbean."},{"query":"best beaches for surfing Caribbean April","reason":"Identify which Caribbean beaches are renowned for surfing in April."},{"query":"Caribbean islands with national parks hiking","reason":"Find islands with national parks or reserves that offer hiking trails."},{"query":"Caribbean weather April surfing conditions","reason":"Research the weather conditions in April affecting surfing in the Caribbean."},{"query":"Caribbean water sports rentals April","reason":"Look for places where water sports equipment can be rented in the Caribbean during April."},{"query":"Caribbean multi-activity vacation packages","reason":"Look for vacation packages that offer a combination of surfing, hiking, and water sports."}]}'
         )
     ]
-    for i in range(10):
+    for _ in range(10):
         responses.append(
             ModelResponse(
                 output=[

tests/contrib/openai_agents/test_openai_tracing.py

@@ -10,7 +10,7 @@
 from temporalio.contrib.openai_agents.testing import (
     TestModelProvider,
 )
-from tests.contrib.openai_agents.test_openai import ResearchWorkflow, TestResearchModel
+from tests.contrib.openai_agents.test_openai import ResearchWorkflow, research_mock_model
 from tests.helpers import new_worker
 
 
@@ -42,7 +42,7 @@ async def test_tracing(client: Client):
     new_config = client.config()
     new_config["plugins"] = [
         openai_agents.OpenAIAgentsPlugin(
-            model_provider=TestModelProvider(TestResearchModel())
+            model_provider=TestModelProvider(research_mock_model())
         )
     ]
     client = Client(**new_config)