consistency for data converter use

Author: jssmith

openai_agents/run_agents_as_tools_workflow.py

@@ -2,14 +2,19 @@
 
 from temporalio.client import Client
 from temporalio.common import WorkflowIDReusePolicy
+from temporalio.contrib.openai_agents.open_ai_data_converter import (
+    open_ai_data_converter,
+)
 
-# from openai_agents.adapters.open_ai_converter import open_ai_data_converter
 from openai_agents.workflows.agents_as_tools_workflow import AgentsAsToolsWorkflow
 
 
 async def main():
     # Create client connected to server at the given address
-    client = await Client.connect("localhost:7233")
+    client = await Client.connect(
+        "localhost:7233",
+        data_converter=open_ai_data_converter,
+    )
 
     # Execute a workflow
     result = await client.execute_workflow(

openai_agents/run_customer_service_client.py

@@ -1,24 +1,21 @@
 import argparse
 import asyncio
 
-from temporalio import workflow
 from temporalio.client import (
     Client,
     WorkflowQueryRejectedError,
     WorkflowUpdateFailedError,
 )
 from temporalio.common import QueryRejectCondition, WorkflowIDReusePolicy
+from temporalio.contrib.openai_agents.open_ai_data_converter import (
+    open_ai_data_converter,
+)
 from temporalio.service import RPCError, RPCStatusCode
 
-with workflow.unsafe.imports_passed_through():
-    from temporalio.contrib.openai_agents.open_ai_data_converter import (
-        open_ai_data_converter,
-    )
-
-    from openai_agents.workflows.customer_service_workflow import (
-        CustomerServiceWorkflow,
-        ProcessUserMessageInput,
-    )
+from openai_agents.workflows.customer_service_workflow import (
+    CustomerServiceWorkflow,
+    ProcessUserMessageInput,
+)
 
 
 async def main():

openai_agents/run_hello_world_workflow.py

@@ -1,16 +1,13 @@
 import asyncio
 
-from temporalio import workflow
 from temporalio.client import Client
 from temporalio.common import WorkflowIDReusePolicy
+from temporalio.contrib.openai_agents.open_ai_data_converter import (
+    open_ai_data_converter,
+)
 
 from openai_agents.workflows.hello_world_workflow import HelloWorldAgent
 
-with workflow.unsafe.imports_passed_through():
-    from temporalio.contrib.openai_agents.open_ai_data_converter import (
-        open_ai_data_converter,
-    )
-
 
 async def main():
     # Create client connected to server at the given address

openai_agents/run_research_workflow.py

@@ -2,6 +2,9 @@
 
 from temporalio.client import Client
 from temporalio.common import WorkflowIDReusePolicy
+from temporalio.contrib.openai_agents.open_ai_data_converter import (
+    open_ai_data_converter,
+)
 
 from openai_agents.workflows.research_bot_workflow import ResearchWorkflow
 
@@ -10,6 +13,7 @@ async def main():
     # Create client connected to server at the given address
     client = await Client.connect(
         "localhost:7233",
+        data_converter=open_ai_data_converter,
     )
 
     # Execute a workflow

openai_agents/run_tools_workflow.py

@@ -2,13 +2,19 @@
 
 from temporalio.client import Client
 from temporalio.common import WorkflowIDReusePolicy
+from temporalio.contrib.openai_agents.open_ai_data_converter import (
+    open_ai_data_converter,
+)
 
 from openai_agents.workflows.tools_workflow import ToolsWorkflow
 
 
 async def main():
     # Create client connected to server at the given address
-    client = await Client.connect("localhost:7233")
+    client = await Client.connect(
+        "localhost:7233",
+        data_converter=open_ai_data_converter,
+    )
 
     # Execute a workflow
     result = await client.execute_workflow(