format

Author: donald-pinckney

tests/openai_agents/basic/test_agent_lifecycle_workflow.py

@@ -2,7 +2,11 @@
 from concurrent.futures import ThreadPoolExecutor
 
 from temporalio.client import Client
-from temporalio.contrib.openai_agents.testing import AgentEnvironment, ResponseBuilders, TestModel
+from temporalio.contrib.openai_agents.testing import (
+    AgentEnvironment,
+    ResponseBuilders,
+    TestModel,
+)
 from temporalio.worker import Worker
 
 from openai_agents.basic.workflows.agent_lifecycle_workflow import (

tests/openai_agents/basic/test_dynamic_system_prompt_workflow.py

@@ -2,7 +2,11 @@
 from concurrent.futures import ThreadPoolExecutor
 
 from temporalio.client import Client
-from temporalio.contrib.openai_agents.testing import AgentEnvironment, ResponseBuilders, TestModel
+from temporalio.contrib.openai_agents.testing import (
+    AgentEnvironment,
+    ResponseBuilders,
+    TestModel,
+)
 from temporalio.worker import Worker
 
 from openai_agents.basic.workflows.dynamic_system_prompt_workflow import (
@@ -12,7 +16,11 @@
 
 def dynamic_system_prompt_test_model():
     return TestModel.returning_responses(
-        [ResponseBuilders.output_message("Style: haiku\nResponse: The weather is cloudy with a chance of meatballs.")]
+        [
+            ResponseBuilders.output_message(
+                "Style: haiku\nResponse: The weather is cloudy with a chance of meatballs."
+            )
+        ]
     )
 
 

tests/openai_agents/basic/test_hello_world_workflow.py

@@ -2,12 +2,12 @@
 from concurrent.futures import ThreadPoolExecutor
 
 from temporalio.client import Client
-from temporalio.worker import Worker
 from temporalio.contrib.openai_agents.testing import (
     AgentEnvironment,
     ResponseBuilders,
     TestModel,
 )
+from temporalio.worker import Worker
 
 from openai_agents.basic.workflows.hello_world_workflow import HelloWorldAgent
 

tests/openai_agents/basic/test_lifecycle_workflow.py

@@ -2,7 +2,11 @@
 from concurrent.futures import ThreadPoolExecutor
 
 from temporalio.client import Client
-from temporalio.contrib.openai_agents.testing import AgentEnvironment, ResponseBuilders, TestModel
+from temporalio.contrib.openai_agents.testing import (
+    AgentEnvironment,
+    ResponseBuilders,
+    TestModel,
+)
 from temporalio.worker import Worker
 
 from openai_agents.basic.workflows.lifecycle_workflow import LifecycleWorkflow

tests/openai_agents/basic/test_local_image_workflow.py

@@ -2,7 +2,11 @@
 from concurrent.futures import ThreadPoolExecutor
 
 from temporalio.client import Client
-from temporalio.contrib.openai_agents.testing import AgentEnvironment, ResponseBuilders, TestModel
+from temporalio.contrib.openai_agents.testing import (
+    AgentEnvironment,
+    ResponseBuilders,
+    TestModel,
+)
 from temporalio.worker import Worker
 
 from openai_agents.basic.activities.image_activities import read_image_as_base64

tests/openai_agents/basic/test_non_strict_output_workflow.py

@@ -2,7 +2,11 @@
 from concurrent.futures import ThreadPoolExecutor
 
 from temporalio.client import Client
-from temporalio.contrib.openai_agents.testing import AgentEnvironment, ResponseBuilders, TestModel
+from temporalio.contrib.openai_agents.testing import (
+    AgentEnvironment,
+    ResponseBuilders,
+    TestModel,
+)
 from temporalio.worker import Worker
 
 from openai_agents.basic.workflows.non_strict_output_workflow import (
@@ -12,7 +16,10 @@
 
 def non_strict_output_test_model():
     return TestModel.returning_responses(
-        [ResponseBuilders.output_message('{"jokes": {"1": "Why do programmers prefer dark mode? Because light attracts bugs!", "2": "How many programmers does it take to change a light bulb? None, that\'s a hardware problem.", "3": "Why do Java developers wear glasses? Because they can\'t C#!"}}')
+        [
+            ResponseBuilders.output_message(
+                '{"jokes": {"1": "Why do programmers prefer dark mode? Because light attracts bugs!", "2": "How many programmers does it take to change a light bulb? None, that\'s a hardware problem.", "3": "Why do Java developers wear glasses? Because they can\'t C#!"}}'
+            )
         ]
     )
 

tests/openai_agents/basic/test_previous_response_id_workflow.py

@@ -2,7 +2,11 @@
 from concurrent.futures import ThreadPoolExecutor
 
 from temporalio.client import Client
-from temporalio.contrib.openai_agents.testing import AgentEnvironment, ResponseBuilders, TestModel
+from temporalio.contrib.openai_agents.testing import (
+    AgentEnvironment,
+    ResponseBuilders,
+    TestModel,
+)
 from temporalio.worker import Worker
 
 from openai_agents.basic.workflows.previous_response_id_workflow import (
@@ -14,7 +18,9 @@ def previous_response_id_test_model():
     return TestModel.returning_responses(
         [
             ResponseBuilders.output_message("The capital of France is Paris."),
-            ResponseBuilders.output_message("Paris has a population of approximately 2.1 million people within the city proper.")
+            ResponseBuilders.output_message(
+                "Paris has a population of approximately 2.1 million people within the city proper."
+            ),
         ]
     )
 

tests/openai_agents/basic/test_remote_image_workflow.py

@@ -2,15 +2,23 @@
 from concurrent.futures import ThreadPoolExecutor
 
 from temporalio.client import Client
-from temporalio.contrib.openai_agents.testing import AgentEnvironment, ResponseBuilders, TestModel
+from temporalio.contrib.openai_agents.testing import (
+    AgentEnvironment,
+    ResponseBuilders,
+    TestModel,
+)
 from temporalio.worker import Worker
 
 from openai_agents.basic.workflows.remote_image_workflow import RemoteImageWorkflow
 
 
 def remote_image_test_model():
     return TestModel.returning_responses(
-        [ResponseBuilders.output_message("I can see the Golden Gate Bridge, a beautiful suspension bridge in San Francisco.")]
+        [
+            ResponseBuilders.output_message(
+                "I can see the Golden Gate Bridge, a beautiful suspension bridge in San Francisco."
+            )
+        ]
     )
 
 

tests/openai_agents/basic/test_tools_workflow.py

@@ -2,7 +2,11 @@
 from concurrent.futures import ThreadPoolExecutor
 
 from temporalio.client import Client
-from temporalio.contrib.openai_agents.testing import AgentEnvironment, ResponseBuilders, TestModel
+from temporalio.contrib.openai_agents.testing import (
+    AgentEnvironment,
+    ResponseBuilders,
+    TestModel,
+)
 from temporalio.worker import Worker
 
 from openai_agents.basic.activities.get_weather_activity import get_weather
@@ -11,8 +15,12 @@
 
 def tools_test_model():
     return TestModel.returning_responses(
-        [ResponseBuilders.tool_call('{"city": "New York"}', "get_weather"),
-         ResponseBuilders.output_message("The weather in New York is sunny with a temperature of 75°F.")]
+        [
+            ResponseBuilders.tool_call('{"city": "New York"}', "get_weather"),
+            ResponseBuilders.output_message(
+                "The weather in New York is sunny with a temperature of 75°F."
+            ),
+        ]
     )