diff --git a/openai_agents/run_worker.py b/openai_agents/run_worker.py index 9dbaef91..123d4a28 100644 --- a/openai_agents/run_worker.py +++ b/openai_agents/run_worker.py @@ -5,13 +5,11 @@ from temporalio.client import Client from temporalio.contrib.openai_agents import ( - ModelActivity, ModelActivityParameters, - OpenAIAgentsTracingInterceptor, - set_open_ai_agent_temporal_overrides, + OpenAIAgentsPlugin ) -from temporalio.contrib.pydantic import pydantic_data_converter from temporalio.worker import Worker +from temporalio.client import ClientConfig from openai_agents.workflows.agents_as_tools_workflow import AgentsAsToolsWorkflow from openai_agents.workflows.customer_service_workflow import CustomerServiceWorkflow @@ -22,34 +20,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__":