follow openai agents examples more closely

Author: jssmith

openai_agents/workflows/agents_as_tools_workflow.py

@@ -1,4 +1,3 @@
-from agents import gen_trace_id
 from agents import Agent, ItemHelpers, MessageOutputItem, RunConfig, Runner, trace
 
 from temporalio import workflow
@@ -68,8 +67,7 @@ async def run(self, msg: str) -> str:
         config = RunConfig()
 
         # Run the entire orchestration in a single trace
-        trace_id = gen_trace_id()
-        with trace("Orchestrator evaluator", trace_id=trace_id):
+        with trace("Orchestrator evaluator"):
             orchestrator = orchestrator_agent()
             synthesizer = synthesizer_agent()
 

openai_agents/workflows/hello_world_workflow.py

@@ -1,18 +1,15 @@
 from agents import Agent, Runner
-from agents import gen_trace_id, trace
 from temporalio import workflow
 
 
 @workflow.defn
 class HelloWorldAgent:
     @workflow.run
     async def run(self, prompt: str) -> str:
-        trace_id = gen_trace_id()
-        with trace("Hello World", trace_id=trace_id):
-            agent = Agent(
-                name="Assistant",
-                instructions="You only respond in haikus.",
-            )
+        agent = Agent(
+            name="Assistant",
+            instructions="You only respond in haikus.",
+        )
 
-            result = await Runner.run(agent, input=prompt)
-            return result.final_output
+        result = await Runner.run(agent, input=prompt)
+        return result.final_output

openai_agents/workflows/research_agents/research_manager.py

@@ -6,7 +6,7 @@
 
 with workflow.unsafe.imports_passed_through():
     # TODO: Restore progress updates
-    from agents import RunConfig, Runner, custom_span, gen_trace_id, trace
+    from agents import RunConfig, Runner, custom_span, trace
 
     from openai_agents.workflows.research_agents.planner_agent import (
         WebSearchItem,
@@ -28,8 +28,7 @@ def __init__(self):
         self.writer_agent = new_writer_agent()
 
     async def run(self, query: str) -> str:
-        trace_id = gen_trace_id()
-        with trace("Research", trace_id=trace_id):
+        with trace("Research trace"):
             search_plan = await self._plan_searches(query)
             search_results = await self._perform_searches(search_plan)
             report = await self._write_report(query, search_results)

openai_agents/workflows/tools_workflow.py

@@ -2,7 +2,7 @@
 
 from datetime import timedelta
 
-from agents import gen_trace_id, trace
+from agents import  trace
 from agents import Agent, Runner
 from temporalio import workflow
 from temporalio.contrib import openai_agents as temporal_agents
@@ -14,8 +14,7 @@
 class ToolsWorkflow:
     @workflow.run
     async def run(self, question: str) -> str:
-        trace_id = gen_trace_id()
-        with trace("Activity as tool", trace_id=trace_id):
+        with trace("Activity as tool"):
             agent = Agent(
                 name="Hello world",
                 instructions="You are a helpful agent.",