copilot recs

Author: dot-agi

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

@@ -21,14 +21,14 @@
 from agentops.semconv.span_kinds import AgentOpsSpanKindValues
 
 from opentelemetry import context as context_api
-from opentelemetry.trace import SpanKind, Status, StatusCode, get_tracer
+from opentelemetry.trace import SpanKind, Status, StatusCode, get_tracer, set_span_in_context
 
 logger = logging.getLogger(__name__)
 
 LLM_REQUEST_TYPE = LLMRequestTypeValues.CHAT
 
 
-def _create_tool_span(parent_span, tool_call_data):
+def _create_tool_span(parent_span: Span, tool_call_data: Dict[str, Any]):
     """
     Create a distinct span for each tool call.
     
@@ -43,7 +43,7 @@ def _create_tool_span(parent_span, tool_call_data):
     with tracer.start_as_current_span(
         name=f"tool_call.{tool_call_data['function']['name']}",
         kind=SpanKind.INTERNAL,
-        context=context_api.set_value("current_span", parent_span)
+        context=set_span_in_context(context_api.get_current(), parent_span)
     ) as tool_span:
         # Set the span kind to TOOL
         tool_span.set_attribute("agentops.span.kind", AgentOpsSpanKindValues.TOOL)
@@ -53,9 +53,6 @@ def _create_tool_span(parent_span, tool_call_data):
         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 status to OK for successful tool call creation
-        tool_span.set_status(Status(StatusCode.OK))
 
 
 def handle_chat_attributes(