update span name

Author: kritinv

deepeval/openai_agents/callback_handler.py

@@ -56,8 +56,8 @@ def on_span_start(self, span: Span) -> None:
         observer = Observer(span_type=span_type, func_name="NA")
         if span_type == "llm":
             observer.observe_kwargs["model"] = "temporary model"
-        observer.custom_update_span_attributes = (
-            lambda span_type: custom_update_span_attributes(span_type, span.span_data)
+        observer.update_span_properties = (
+            lambda span_type: update_span_properties(span_type, span.span_data)
         )
         self.span_observers[span.span_id] = observer
         observer.__enter__()

deepeval/openai_agents/extractors.py

@@ -46,34 +46,34 @@ def _check_openai_agents_available():
             "openai-agents is required for this integration. Install it via your package manager"
         )
 
-def custom_update_span_attributes(span: BaseSpan, span_data: SpanData):
+def update_span_properties(span: BaseSpan, span_data: SpanData):
     _check_openai_agents_available()
     # LLM Span
     if isinstance(span_data, ResponseSpanData):
-        update_attributes_from_response_span_data(span, span_data)
+        update_span_properties_from_response_span_data(span, span_data)
     elif isinstance(span_data, GenerationSpanData):
-        udpate_attributes_from_generation_span_data(span, span_data)
+        update_span_properties_from_generation_span_data(span, span_data)
     # Tool Span
     elif isinstance(span_data, FunctionSpanData):
-        update_attributes_from_function_span_data(span, span_data)
+        update_span_properties_from_function_span_data(span, span_data)
     elif isinstance(span_data, MCPListToolsSpanData):
-        update_attributes_from_mcp_list_tool_span_data(span, span_data)
+        update_span_properties_from_mcp_list_tool_span_data(span, span_data)
     # Agent Span 
     elif isinstance(span_data, AgentSpanData):
-        update_attributes_from_agent_span_data(span, span_data)
+        update_span_properties_from_agent_span_data(span, span_data)
     # Custom Span
     elif isinstance(span_data, HandoffSpanData):
-        update_attributes_from_handoff_span_data(span, span_data)
+        update_span_properties_from_handoff_span_data(span, span_data)
     elif isinstance(span_data, CustomSpanData):
-        update_attributes_from_custom_span_data(span, span_data)
+        update_span_properties_from_custom_span_data(span, span_data)
     elif isinstance(span_data, GuardrailSpanData):
-        update_attributes_from_guardrail_span_data(span, span_data)
+        update_span_properties_from_guardrail_span_data(span, span_data)
 
 ########################################################
 ### LLM Span ###########################################
 ########################################################
 
-def update_attributes_from_response_span_data(
+def update_span_properties_from_response_span_data(
     span: LlmSpan, 
     span_data: ResponseSpanData,
 ):
@@ -114,7 +114,7 @@ def update_attributes_from_response_span_data(
     span.output = output
     span.name = "LLM Generation"
 
-def udpate_attributes_from_generation_span_data(
+def update_span_properties_from_generation_span_data(
     span: LlmSpan, 
     generation_span_data: GenerationSpanData,
 ):
@@ -146,7 +146,7 @@ def udpate_attributes_from_generation_span_data(
 ### Tool Span ##########################################
 ########################################################
 
-def update_attributes_from_function_span_data(
+def update_span_properties_from_function_span_data(
     span: ToolSpan, 
     function_span_data: FunctionSpanData,
 ):
@@ -159,7 +159,7 @@ def update_attributes_from_function_span_data(
     span.name = "Function tool: " + function_span_data.name if function_span_data.name else "Function tool"
     span.description = "Function tool"
 
-def update_attributes_from_mcp_list_tool_span_data(
+def update_span_properties_from_mcp_list_tool_span_data(
     span: ToolSpan, 
     mcp_list_tool_span_data: MCPListToolsSpanData,
 ) :
@@ -177,7 +177,7 @@ def update_attributes_from_mcp_list_tool_span_data(
 ### Agent Span #########################################
 ########################################################
 
-def update_attributes_from_agent_span_data(
+def update_span_properties_from_agent_span_data(
     span: AgentSpan, 
     agent_span_data: AgentSpanData
 ):
@@ -197,7 +197,7 @@ def update_attributes_from_agent_span_data(
 ### Custom Span #######################################
 ########################################################
 
-def update_attributes_from_handoff_span_data(
+def update_span_properties_from_handoff_span_data(
     span: AgentSpan, 
     handoff_span_data: HandoffSpanData
 ):
@@ -211,15 +211,15 @@ def update_attributes_from_handoff_span_data(
     span.input = None
     span.output = None
 
-def update_attributes_from_custom_span_data(
+def update_span_properties_from_custom_span_data(
     span: BaseSpan, 
     custom_span_data: CustomSpanData
 ):
     # Update Span
     span.name = custom_span_data.name
     span.metadata = {"data": custom_span_data.data}
 
-def update_attributes_from_guardrail_span_data(
+def update_span_properties_from_guardrail_span_data(
     span: BaseSpan, 
     guardrail_span_data: GuardrailSpanData
 ):

deepeval/tracing/tracing.py

@@ -704,7 +704,7 @@ def __init__(
         )
         self._progress = _progress
         self._pbar_callback_id = _pbar_callback_id
-        self.custom_update_span_attributes: Optional[Callable] = None
+        self.update_span_properties: Optional[Callable] = None
 
     def __enter__(self):
         """Enter the tracer context, creating a new span and setting up parent-child relationships."""
@@ -774,8 +774,8 @@ def __exit__(self, exc_type, exc_val, exc_tb):
         else:
             current_span.status = TraceSpanStatus.SUCCESS
 
-        if self.custom_update_span_attributes is not None:
-            self.custom_update_span_attributes(current_span)
+        if self.update_span_properties is not None:
+            self.update_span_properties(current_span)
         else:
             self.update_span_attributes(current_span)
 

tests/integrations/openai_agents/streaming_agent.py

@@ -1,7 +1,8 @@
 import asyncio
 from openai.types.responses import ResponseTextDeltaEvent
 from agents import Agent, Runner
-
+'
+'
 async def streaming_agent():
     agent = Agent(
         name="Joker",
@@ -10,4 +11,4 @@ async def streaming_agent():
     result = Runner.run_streamed(agent, input="Please tell me 5 jokes.")
     async for event in result.stream_events():
         if event.type == "raw_response_event" and isinstance(event.data, ResponseTextDeltaEvent):
-            print(event.data.delta, end="", flush=True)
+            print(event.data.delta, end="", flush=True)