more fixes

Author: dot-agi

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

@@ -20,8 +20,7 @@
 from agentops.semconv.tool import ToolAttributes
 from agentops.semconv.span_kinds import AgentOpsSpanKindValues
 
-from opentelemetry import context as context_api
-from opentelemetry.trace import SpanKind, Status, StatusCode, get_tracer, set_span_in_context
+from opentelemetry.trace import SpanKind, Status, StatusCode, get_tracer, set_span_in_context, get_current
 
 logger = logging.getLogger(__name__)
 
@@ -43,7 +42,7 @@ def _create_tool_span(parent_span: Span, tool_call_data: Dict[str, Any]):
     with tracer.start_as_current_span(
         name=f"tool_call.{tool_call_data['function']['name']}",
         kind=SpanKind.INTERNAL,
-        context=set_span_in_context(context_api.get_current(), parent_span)
+        context=set_span_in_context(parent_span, get_current())
     ) as tool_span:
         # Set the span kind to TOOL
         tool_span.set_attribute("agentops.span.kind", AgentOpsSpanKindValues.TOOL)