update with precommit fixes

Author: nagkumar91

instrumentation-genai/opentelemetry-instrumentation-langchain/src/opentelemetry/instrumentation/langchain/callback_handler.py

@@ -15,8 +15,8 @@
 from __future__ import annotations
 
 from typing import Any
-from uuid import UUID
 from urllib.parse import urlparse
+from uuid import UUID
 
 from langchain_core.callbacks import BaseCallbackHandler  # type: ignore
 from langchain_core.messages import BaseMessage  # type: ignore
@@ -107,7 +107,8 @@ def on_chat_model_start(
             GenAI.GenAiProviderNameValues.AZURE_AI_INFERENCE.value,
         ):
             span.set_attribute(
-                AZURE_RESOURCE_PROVIDER_NAMESPACE, "Microsoft.CognitiveServices"
+                AZURE_RESOURCE_PROVIDER_NAMESPACE,
+                "Microsoft.CognitiveServices",
             )
 
         self._apply_request_attributes(span, params, metadata)

instrumentation-genai/opentelemetry-instrumentation-langchain/tests/test_callback_handler.py

@@ -19,7 +19,6 @@
     AZURE_RESOURCE_PROVIDER_NAMESPACE,
 )
 from opentelemetry.semconv._incubating.attributes.openai_attributes import (
-    OPENAI_REQUEST_SERVICE_TIER,
     OPENAI_RESPONSE_SERVICE_TIER,
     OPENAI_RESPONSE_SYSTEM_FINGERPRINT,
 )
@@ -62,9 +61,7 @@ def test_provider_and_server_metadata_extracted():
 
     span = exporter.get_finished_spans()[0]
     assert span.name == "chat gpt-4o"
-    assert (
-        span.attributes[GenAI.GEN_AI_PROVIDER_NAME] == "azure.ai.openai"
-    )
+    assert span.attributes[GenAI.GEN_AI_PROVIDER_NAME] == "azure.ai.openai"
     assert (
         span.attributes[AZURE_RESOURCE_PROVIDER_NAMESPACE]
         == "Microsoft.CognitiveServices"
@@ -112,9 +109,7 @@ def test_llm_end_sets_response_metadata():
     assert span.attributes[GenAI.GEN_AI_RESPONSE_MODEL] == "gpt-4-0125"
     assert span.attributes[GenAI.GEN_AI_RESPONSE_ID] == "chatcmpl-test"
     assert span.attributes[OPENAI_RESPONSE_SERVICE_TIER] == "premium"
-    assert (
-        span.attributes[OPENAI_RESPONSE_SYSTEM_FINGERPRINT] == "fp-test"
-    )
+    assert span.attributes[OPENAI_RESPONSE_SYSTEM_FINGERPRINT] == "fp-test"
 
 
 def test_choice_count_not_set_when_one():
@@ -138,6 +133,4 @@ def test_choice_count_not_set_when_one():
 
     handler.span_manager.end_span(run_id)
     span = exporter.get_finished_spans()[0]
-    assert (
-        GenAI.GEN_AI_REQUEST_CHOICE_COUNT not in span.attributes
-    )
+    assert GenAI.GEN_AI_REQUEST_CHOICE_COUNT not in span.attributes

instrumentation-genai/opentelemetry-instrumentation-langchain/tests/test_llm_call.py

@@ -7,8 +7,7 @@
 from opentelemetry.instrumentation.langchain.callback_handler import (
     OpenTelemetryLangChainCallbackHandler,
 )
-from opentelemetry.sdk.trace import ReadableSpan
-from opentelemetry.sdk.trace import TracerProvider
+from opentelemetry.sdk.trace import ReadableSpan, TracerProvider
 from opentelemetry.sdk.trace.export import SimpleSpanProcessor
 from opentelemetry.sdk.trace.export.in_memory_span_exporter import (
     InMemorySpanExporter,
@@ -100,9 +99,7 @@ def assert_openai_completion_attributes(
     )
     assert span.attributes[gen_ai_attributes.GEN_AI_REQUEST_MAX_TOKENS] == 100
     assert span.attributes[gen_ai_attributes.GEN_AI_REQUEST_TEMPERATURE] == 0.1
-    assert (
-        span.attributes[gen_ai_attributes.GEN_AI_PROVIDER_NAME] == "openai"
-    )
+    assert span.attributes[gen_ai_attributes.GEN_AI_PROVIDER_NAME] == "openai"
     assert gen_ai_attributes.GEN_AI_RESPONSE_ID in span.attributes
     assert span.attributes[gen_ai_attributes.GEN_AI_REQUEST_TOP_P] == 0.9
     assert (
@@ -190,7 +187,9 @@ def test_azure_chat_sets_provider_and_server_attributes():
     exporter = InMemorySpanExporter()
     provider = TracerProvider()
     provider.add_span_processor(SimpleSpanProcessor(exporter))
-    handler = OpenTelemetryLangChainCallbackHandler(provider.get_tracer(__name__))
+    handler = OpenTelemetryLangChainCallbackHandler(
+        provider.get_tracer(__name__)
+    )
 
     run_id = uuid4()
 
@@ -215,18 +214,14 @@ def test_azure_chat_sets_provider_and_server_attributes():
     span = exporter.get_finished_spans()[0]
 
     assert span.name == "chat gpt-4o"
-    assert (
-        span.attributes[gen_ai_attributes.GEN_AI_REQUEST_MODEL] == "gpt-4o"
-    )
+    assert span.attributes[gen_ai_attributes.GEN_AI_REQUEST_MODEL] == "gpt-4o"
     assert (
         span.attributes[gen_ai_attributes.GEN_AI_PROVIDER_NAME]
         == "azure.ai.openai"
     )
     assert span.attributes["server.address"] == "example.openai.azure.com"
     assert span.attributes["server.port"] == 443
-    assert (
-        span.attributes[gen_ai_attributes.GEN_AI_REQUEST_CHOICE_COUNT] == 2
-    )
+    assert span.attributes[gen_ai_attributes.GEN_AI_REQUEST_CHOICE_COUNT] == 2
     assert span.attributes[OPENAI_REQUEST_SERVICE_TIER] == "default"
     assert (
         span.attributes[AZURE_RESOURCE_PROVIDER_NAMESPACE]