close tool span

Author: dot-agi

agentops/instrumentation/providers/openai/wrappers/chat.py

@@ -39,22 +39,22 @@ def _create_tool_span(parent_span: Span, tool_call_data: Dict[str, Any]) -> Span
     # Get the tracer for this module
     tracer = get_tracer(__name__)
 
-    # Create a child span for the tool call
-    tool_span = tracer.start_span(
+    # Create a child span for the tool call using a context manager
+    with tracer.start_as_current_span(
         name=f"tool_call.{tool_call_data['function']['name']}",
         kind=SpanKind.INTERNAL,
         context=set_span_in_context(parent_span),
-    )
+    ) as tool_span:
+        # Set the span kind to TOOL
+        tool_span.set_attribute("agentops.span.kind", AgentOpsSpanKindValues.TOOL)
 
-    # Set the span kind to TOOL
-    tool_span.set_attribute("agentops.span.kind", AgentOpsSpanKindValues.TOOL)
+        # Set tool-specific attributes
+        tool_span.set_attribute(ToolAttributes.TOOL_NAME, tool_call_data["function"]["name"])
+        tool_span.set_attribute(ToolAttributes.TOOL_PARAMETERS, tool_call_data["function"]["arguments"])
+        tool_span.set_attribute("tool.call.id", tool_call_data["id"])
+        tool_span.set_attribute("tool.call.type", tool_call_data["type"])
 
-    # Set tool-specific attributes
-    tool_span.set_attribute(ToolAttributes.TOOL_NAME, tool_call_data["function"]["name"])
-    tool_span.set_attribute(ToolAttributes.TOOL_PARAMETERS, tool_call_data["function"]["arguments"])
-    tool_span.set_attribute("tool.call.id", tool_call_data["id"])
-    tool_span.set_attribute("tool.call.type", tool_call_data["type"])
-    return tool_span
+        return tool_span
 
 
 def handle_chat_attributes(