linting

Author: dot-agi

agentops/instrumentation/providers/openai/stream_wrapper.py

@@ -9,7 +9,7 @@
 from typing import Any, AsyncIterator, Iterator
 
 from opentelemetry import context as context_api
-from opentelemetry.trace import Span, SpanKind, Status, StatusCode, set_span_in_context, get_tracer
+from opentelemetry.trace import Span, SpanKind, Status, StatusCode, set_span_in_context
 from opentelemetry.instrumentation.utils import _SUPPRESS_INSTRUMENTATION_KEY
 
 from agentops.logging import logger

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

@@ -20,7 +20,7 @@
 from agentops.semconv.tool import ToolAttributes
 from agentops.semconv.span_kinds import AgentOpsSpanKindValues
 
-from opentelemetry.trace import SpanKind, Status, StatusCode, get_tracer, set_span_in_context, get_current
+from opentelemetry.trace import SpanKind, get_tracer, set_span_in_context, get_current
 
 logger = logging.getLogger(__name__)
 
@@ -30,28 +30,28 @@
 def _create_tool_span(parent_span: Span, tool_call_data: Dict[str, Any]):
     """
     Create a distinct span for each tool call.
-    
+
     Args:
         parent_span: The parent LLM span
         tool_call_data: The tool call data dictionary
     """
     # Get the tracer for this module
     tracer = get_tracer(__name__)
-    
+
     # Create a child span for the tool call
     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, get_current())
+        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)
-        
+
         # 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'])
+        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"])
 
 
 def handle_chat_attributes(
@@ -64,7 +64,7 @@ def handle_chat_attributes(
 
     This function is designed to work with the common wrapper pattern,
     extracting attributes from the method arguments and return value.
-    
+
     Args:
         args: Method arguments (not used in this implementation)
         kwargs: Method keyword arguments
@@ -240,8 +240,8 @@ def handle_chat_attributes(
                                     "type": tool_call.get("type", "function"),
                                     "function": {
                                         "name": function.get("name", ""),
-                                        "arguments": function.get("arguments", "")
-                                    }
+                                        "arguments": function.get("arguments", ""),
+                                    },
                                 }
                                 # Create a child span for this tool call
                                 _create_tool_span(span, tool_call_data)