test lint

Author: JWinermaSplunk

util/opentelemetry-util-genai/src/opentelemetry/util/genai/emitters/metrics.py

@@ -16,9 +16,9 @@
 from ..types import (
     AgentInvocation,
     EmbeddingInvocation,
-    RetrievalInvocation,
     Error,
     LLMInvocation,
+    RetrievalInvocation,
     ToolCall,
     Workflow,
 )
@@ -353,4 +353,4 @@ def _record_retrieval_metrics(
 
         self._retrieval_duration_histogram.record(
             duration, attributes=metric_attrs, context=context
-        )
+        )

util/opentelemetry-util-genai/src/opentelemetry/util/genai/emitters/span.py

@@ -29,7 +29,6 @@
     GEN_AI_RETRIEVAL_DOCUMENTS_RETRIEVED,
     GEN_AI_RETRIEVAL_QUERY_TEXT,
     GEN_AI_RETRIEVAL_TOP_K,
-    GEN_AI_RETRIEVAL_TYPE,
     GEN_AI_STEP_ASSIGNED_AGENT,
     GEN_AI_STEP_NAME,
     GEN_AI_STEP_OBJECTIVE,
@@ -49,9 +48,9 @@
     AgentInvocation,
     ContentCapturingMode,
     EmbeddingInvocation,
-    RetrievalInvocation,
     Error,
     LLMInvocation,
+    RetrievalInvocation,
     Step,
     ToolCall,
     Workflow,
@@ -849,4 +848,4 @@ def _error_retrieval(
                 token.__exit__(None, None, None)  # type: ignore[misc]
             except Exception:
                 pass
-        span.end()
+        span.end()

util/opentelemetry-util-genai/src/opentelemetry/util/genai/handler.py

@@ -87,11 +87,11 @@ def genai_debug_log(*_args: Any, **_kwargs: Any) -> None:  # type: ignore
     AgentInvocation,
     ContentCapturingMode,
     EmbeddingInvocation,
-    RetrievalInvocation,
     Error,
     EvaluationResult,
     GenAI,
     LLMInvocation,
+    RetrievalInvocation,
     Step,
     ToolCall,
     Workflow,

util/opentelemetry-util-genai/src/opentelemetry/util/genai/types.py

@@ -326,7 +326,7 @@ class RetrievalInvocation(GenAI):
         default="retrieval",
         metadata={"semconv": GenAIAttributes.GEN_AI_OPERATION_NAME},
     )
-    
+
     # Recommended attributes
     retriever_type: Optional[str] = field(
         default=None,
@@ -344,13 +344,13 @@ class RetrievalInvocation(GenAI):
         default=None,
         metadata={"semconv": "gen_ai.retrieval.documents_retrieved"},
     )
-    
+
     # Opt-in attribute
     results: list[dict[str, Any]] = field(
         default_factory=list,
         metadata={"semconv": "gen_ai.retrieval.documents"},
     )
-    
+
     # Additional utility fields (not in semantic conventions)
     query_vector: Optional[list[float]] = None
 

util/opentelemetry-util-genai/tests/test_retrieval_invocation.py

@@ -1,6 +1,7 @@
 """Tests for RetrievalInvocation lifecycle and telemetry."""
 
 import pytest
+
 from opentelemetry.sdk.trace import TracerProvider
 from opentelemetry.sdk.trace.export import SimpleSpanProcessor
 from opentelemetry.sdk.trace.export.in_memory_span_exporter import (
@@ -11,7 +12,6 @@
 )
 from opentelemetry.util.genai.attributes import (
     GEN_AI_RETRIEVAL_DOCUMENTS_RETRIEVED,
-    GEN_AI_RETRIEVAL_QUERY_TEXT,
     GEN_AI_RETRIEVAL_TOP_K,
     GEN_AI_RETRIEVAL_TYPE,
 )