.

Author: kritinv

deepeval/openai_agents/callback_handler.py

@@ -1,6 +1,7 @@
 from deepeval.tracing.tracing import (
     Observer, 
     SpanType, 
+    current_trace_context,
 )
 from deepeval.openai_agents.extractors import *
 
@@ -39,6 +40,11 @@ def on_trace_start(self, trace: Trace) -> None:
         observer.__enter__()
 
     def on_trace_end(self, trace: Trace) -> None:
+        # set thread id if exists
+        current_trace = current_trace_context.get()
+        thread_id = getattr(trace, "group_id", None)
+        current_trace.thread_id = thread_id
+
         observer = self.root_span_observers.pop(trace.trace_id, None)
         if observer:
             observer.__exit__(None, None, None)
@@ -56,6 +62,7 @@ def on_span_start(self, span: Span) -> None:
         self.span_observers[span.span_id] = observer
         observer.__enter__()
 
+
     def on_span_end(self, span: Span) -> None:
         observer = self.span_observers.pop(span.span_id, None)
         if observer:

tests/integrations/openai_agents/test_openai_agent.py

@@ -17,6 +17,7 @@
 from tests.integrations.openai_agents.streaming_agent import streaming_agent
 from tests.integrations.openai_agents.research_agent import research_agent
 from tests.integrations.openai_agents.remote_agent import remote_agent
+from tests.integrations.openai_agents.thread_agent import thread_agent
 from tests.integrations.openai_agents.git_mcp_agent import git_agent
 
 add_trace_processor(DeepEvalTracingProcessor())
@@ -28,12 +29,13 @@
 
     asyncio.run(git_agent())
     # asyncio.run(customer_service_agent())
-    # asyncio.run(research_agent())
-    # asyncio.run(code_interpreter_agent())
-    # asyncio.run(remote_agent())
-    # asyncio.run(streaming_agent())
-    # asyncio.run(streaming_guardrails_agent())
-    # asyncio.run(output_guardrails_agent())
+    asyncio.run(research_agent())
+    asyncio.run(code_interpreter_agent())
+    asyncio.run(remote_agent())
+    asyncio.run(streaming_agent())
+    asyncio.run(streaming_guardrails_agent())
+    asyncio.run(output_guardrails_agent())
+    asyncio.run(thread_agent())
 
     ##########################################
     # Test Async
@@ -51,4 +53,5 @@ async def gather_research_agents():
     async def gather_streaming_agents():
         tasks = [streaming_agent() for _ in range(4)]
         await asyncio.gather(*tasks)
-    # asyncio.run(gather_streaming_agents())
+    # asyncio.run(gather_streaming_agents())
+    

tests/integrations/openai_agents/thread_agent.py

@@ -0,0 +1,19 @@
+
+from agents import Agent, Runner, trace 
+ 
+thread_id = "unique_thread_id"
+ 
+async def thread_agent():
+    agent = Agent(name="Assistant", instructions="Reply very concisely.")
+ 
+    with trace(workflow_name="Conversation", group_id=thread_id):
+        # First turn
+        result = await Runner.run(agent, "What city is the Golden Gate Bridge in?")
+        print(result.final_output)
+        # San Francisco
+ 
+        # Second turn
+        new_input = result.to_input_list() + [{"role": "user", "content": "What state is it in?"}]
+        result = await Runner.run(agent, new_input)
+        print(result.final_output)
+        # California