improve tracing for openai agents

openai_agents/run_customer_service_client.py

@@ -32,12 +32,13 @@ async def main():
     # Query the workflow for the chat history
     # If the workflow is not open, start a new one
     start = False
+    history = []
     try:
         history = await handle.query(
             CustomerServiceWorkflow.get_chat_history,
             reject_condition=QueryRejectCondition.NOT_OPEN,
         )
-    except WorkflowQueryRejectedError as e:
+    except WorkflowQueryRejectedError:
         start = True
     except RPCError as e:
         if e.status == RPCStatusCode.NOT_FOUND:

openai_agents/run_hello_world_workflow.py

@@ -4,7 +4,6 @@
 from temporalio.contrib.pydantic import pydantic_data_converter
 
 from openai_agents.workflows.hello_world_workflow import HelloWorldAgent
-from openai_agents.workflows.research_bot_workflow import ResearchWorkflow
 
 
 async def main():

openai_agents/run_worker.py

@@ -7,6 +7,7 @@
 from temporalio.contrib.openai_agents import (
     ModelActivity,
     ModelActivityParameters,
+    OpenAIAgentsTracingInterceptor,
     set_open_ai_agent_temporal_overrides,
 )
 from temporalio.contrib.pydantic import pydantic_data_converter
@@ -46,6 +47,7 @@ async def main():
                 ModelActivity().invoke_model_activity,
                 get_weather,
             ],
+            interceptors=[OpenAIAgentsTracingInterceptor()],
         )
         await worker.run()
 

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", 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)