Skip to content

Commit 8e5ea40

Browse files
committed
replace local adapters with python sdk
1 parent 7d7a6f0 commit 8e5ea40

18 files changed

+292
-390
lines changed

openai_agents/adapters/_heartbeat_decorator.py

Lines changed: 0 additions & 37 deletions
This file was deleted.

openai_agents/adapters/invoke_model_activity.py

Lines changed: 0 additions & 45 deletions
This file was deleted.

openai_agents/adapters/open_ai_converter.py

Lines changed: 0 additions & 111 deletions
This file was deleted.

openai_agents/adapters/temporal_model_provider.py

Lines changed: 0 additions & 106 deletions
This file was deleted.

openai_agents/customer_service_client.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,6 @@
66
from temporalio.common import WorkflowIDReusePolicy, QueryRejectCondition
77
from temporalio.service import RPCError, RPCStatusCode
88

9-
from openai_agents.adapters.open_ai_converter import open_ai_data_converter
109
from openai_agents.workflows.customer_service_workflow import CustomerServiceWorkflow, ProcessUserMessageInput
1110

1211

@@ -16,7 +15,7 @@ async def main():
1615
args = parser.parse_args()
1716

1817
# Create client connected to server at the given address
19-
client = await Client.connect("localhost:7233", data_converter=open_ai_data_converter)
18+
client = await Client.connect("localhost:7233")
2019

2120
handle = client.get_workflow_handle(args.conversation_id)
2221

openai_agents/run_agents_as_tools_workflow.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3,13 +3,13 @@
33
from temporalio.client import Client
44
from temporalio.common import WorkflowIDReusePolicy
55

6-
from openai_agents.adapters.open_ai_converter import open_ai_data_converter
6+
# from openai_agents.adapters.open_ai_converter import open_ai_data_converter
77
from openai_agents.workflows.agents_as_tools_workflow import AgentsAsToolsWorkflow
88

99

1010
async def main():
1111
# Create client connected to server at the given address
12-
client = await Client.connect("localhost:7233", data_converter=open_ai_data_converter)
12+
client = await Client.connect("localhost:7233")
1313

1414
# Execute a workflow
1515
result = await client.execute_workflow(AgentsAsToolsWorkflow.run,

openai_agents/run_hello_world_workflow.py

Lines changed: 12 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -3,22 +3,23 @@
33
from temporalio.client import Client
44
from temporalio.common import WorkflowIDReusePolicy
55

6-
from openai_agents.adapters.open_ai_converter import open_ai_data_converter
76
from openai_agents.workflows.hello_world_workflow import HelloWorldAgent
87

8+
from temporalio.contrib.openai_agents.temporal_openai_agents import (
9+
set_open_ai_agent_temporal_overrides,
10+
)
11+
912

1013
async def main():
1114
# Create client connected to server at the given address
12-
client = await Client.connect("localhost:7233",
13-
data_converter=open_ai_data_converter,
14-
)
15-
16-
# Execute a workflow
17-
result = await client.execute_workflow(HelloWorldAgent.run, "Tell me about recursion in programming.",
18-
id="my-workflow-id", task_queue="my-task-queue",
19-
id_reuse_policy=WorkflowIDReusePolicy.TERMINATE_IF_RUNNING)
20-
21-
print(f"Result: {result}")
15+
client = await Client.connect("localhost:7233")
16+
17+
with set_open_ai_agent_temporal_overrides():
18+
# Execute a workflow
19+
result = await client.execute_workflow(HelloWorldAgent.run, "Tell me about recursion in programming.",
20+
id="my-workflow-id", task_queue="my-task-queue",
21+
id_reuse_policy=WorkflowIDReusePolicy.TERMINATE_IF_RUNNING)
22+
print(f"Result: {result}")
2223

2324

2425
if __name__ == "__main__":

openai_agents/run_research_workflow.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3,15 +3,15 @@
33
from temporalio.client import Client
44
from temporalio.common import WorkflowIDReusePolicy
55

6-
from openai_agents.adapters.open_ai_converter import open_ai_data_converter
6+
# from openai_agents.adapters.open_ai_converter import open_ai_data_converter
77
from openai_agents.workflows.research_bot_workflow import ResearchWorkflow
88

99

1010
async def main():
1111
# Create client connected to server at the given address
1212
client = await Client.connect(
1313
"localhost:7233",
14-
data_converter=open_ai_data_converter
14+
# data_converter=open_ai_data_converter
1515
)
1616

1717
# Execute a workflow

0 commit comments

Comments
 (0)