Update samples for 1.15 (#220)

* update for plugins

* formatting

* reference main branch

* cleanup

* switch to plugins on the runners

* Update dependency to 1.15.0 release

---------

Co-authored-by: Tim Conley <timothy.conley@temporal.io>

Author: jssmith

openai_agents/run_agents_as_tools_workflow.py

@@ -1,7 +1,7 @@
 import asyncio
 
 from temporalio.client import Client
-from temporalio.contrib.pydantic import pydantic_data_converter
+from temporalio.contrib.openai_agents import OpenAIAgentsPlugin
 
 from openai_agents.workflows.agents_as_tools_workflow import AgentsAsToolsWorkflow
 
@@ -10,7 +10,9 @@ async def main():
     # Create client connected to server at the given address
     client = await Client.connect(
         "localhost:7233",
-        data_converter=pydantic_data_converter,
+        plugins=[
+            OpenAIAgentsPlugin(),
+        ],
     )
 
     # Execute a workflow

openai_agents/run_customer_service_client.py

@@ -7,7 +7,7 @@
     WorkflowUpdateFailedError,
 )
 from temporalio.common import QueryRejectCondition
-from temporalio.contrib.pydantic import pydantic_data_converter
+from temporalio.contrib.openai_agents import OpenAIAgentsPlugin
 from temporalio.service import RPCError, RPCStatusCode
 
 from openai_agents.workflows.customer_service_workflow import (
@@ -24,7 +24,9 @@ async def main():
     # Create client connected to server at the given address
     client = await Client.connect(
         "localhost:7233",
-        data_converter=pydantic_data_converter,
+        plugins=[
+            OpenAIAgentsPlugin(),
+        ],
     )
 
     handle = client.get_workflow_handle(args.conversation_id)

openai_agents/run_hello_world_workflow.py

@@ -1,7 +1,7 @@
 import asyncio
 
 from temporalio.client import Client
-from temporalio.contrib.pydantic import pydantic_data_converter
+from temporalio.contrib.openai_agents import OpenAIAgentsPlugin
 
 from openai_agents.workflows.hello_world_workflow import HelloWorldAgent
 
@@ -10,7 +10,9 @@ async def main():
     # Create client connected to server at the given address
     client = await Client.connect(
         "localhost:7233",
-        data_converter=pydantic_data_converter,
+        plugins=[
+            OpenAIAgentsPlugin(),
+        ],
     )
 
     # Execute a workflow

openai_agents/run_research_workflow.py

@@ -1,7 +1,7 @@
 import asyncio
 
 from temporalio.client import Client
-from temporalio.contrib.pydantic import pydantic_data_converter
+from temporalio.contrib.openai_agents import OpenAIAgentsPlugin
 
 from openai_agents.workflows.research_bot_workflow import ResearchWorkflow
 
@@ -10,7 +10,9 @@ async def main():
     # Create client connected to server at the given address
     client = await Client.connect(
         "localhost:7233",
-        data_converter=pydantic_data_converter,
+        plugins=[
+            OpenAIAgentsPlugin(),
+        ],
     )
 
     # Execute a workflow

openai_agents/run_tools_workflow.py

@@ -1,7 +1,7 @@
 import asyncio
 
 from temporalio.client import Client
-from temporalio.contrib.pydantic import pydantic_data_converter
+from temporalio.contrib.openai_agents import OpenAIAgentsPlugin
 
 from openai_agents.workflows.tools_workflow import ToolsWorkflow
 
@@ -10,7 +10,9 @@ async def main():
     # Create client connected to server at the given address
     client = await Client.connect(
         "localhost:7233",
-        data_converter=pydantic_data_converter,
+        plugins=[
+            OpenAIAgentsPlugin(),
+        ],
     )
 
     # Execute a workflow

openai_agents/run_worker.py

@@ -4,13 +4,7 @@
 from datetime import timedelta
 
 from temporalio.client import Client
-from temporalio.contrib.openai_agents import (
-    ModelActivity,
-    ModelActivityParameters,
-    OpenAIAgentsTracingInterceptor,
-    set_open_ai_agent_temporal_overrides,
-)
-from temporalio.contrib.pydantic import pydantic_data_converter
+from temporalio.contrib.openai_agents import ModelActivityParameters, OpenAIAgentsPlugin
 from temporalio.worker import Worker
 
 from openai_agents.workflows.agents_as_tools_workflow import AgentsAsToolsWorkflow
@@ -22,34 +16,33 @@
 
 
 async def main():
-    with set_open_ai_agent_temporal_overrides(
-        model_params=ModelActivityParameters(
-            start_to_close_timeout=timedelta(seconds=60),
-        ),
-    ):
-        # Create client connected to server at the given address
-        client = await Client.connect(
-            "localhost:7233",
-            data_converter=pydantic_data_converter,
-        )
-
-        worker = Worker(
-            client,
-            task_queue="openai-agents-task-queue",
-            workflows=[
-                HelloWorldAgent,
-                ToolsWorkflow,
-                ResearchWorkflow,
-                CustomerServiceWorkflow,
-                AgentsAsToolsWorkflow,
-            ],
-            activities=[
-                ModelActivity().invoke_model_activity,
-                get_weather,
-            ],
-            interceptors=[OpenAIAgentsTracingInterceptor()],
-        )
-        await worker.run()
+    # Create client connected to server at the given address
+    client = await Client.connect(
+        "localhost:7233",
+        plugins=[
+            OpenAIAgentsPlugin(
+                model_params=ModelActivityParameters(
+                    start_to_close_timeout=timedelta(seconds=120)
+                )
+            ),
+        ],
+    )
+
+    worker = Worker(
+        client,
+        task_queue="openai-agents-task-queue",
+        workflows=[
+            HelloWorldAgent,
+            ToolsWorkflow,
+            ResearchWorkflow,
+            CustomerServiceWorkflow,
+            AgentsAsToolsWorkflow,
+        ],
+        activities=[
+            get_weather,
+        ],
+    )
+    await worker.run()
 
 
 if __name__ == "__main__":

pyproject.toml

@@ -6,7 +6,7 @@ authors = [{ name = "Temporal Technologies Inc", email = "sdk@temporal.io" }]
 requires-python = ">=3.10"
 readme = "README.md"
 license = "MIT"
-dependencies = ["temporalio>=1.14.1,<2"]
+dependencies = ["temporalio>=1.15.0,<2"]
 
 [project.urls]
 Homepage = "https://github.com/temporalio/samples-python"
@@ -55,8 +55,8 @@ open-telemetry = [
     "opentelemetry-exporter-otlp-proto-grpc",
 ]
 openai-agents = [
-    "openai-agents >= 0.0.19",
-    "temporalio[openai-agents] >= 1.14.1",
+    "openai-agents[litellm] >= 0.2.3",
+    "temporalio[openai-agents] >= 1.15.0",
 ]
 pydantic-converter = ["pydantic>=2.10.6,<3"]
 sentry = ["sentry-sdk>=1.11.0,<2"]