fix: lint

Author: argaj

instrumentation-genai/opentelemetry-instrumentation-google-genai/src/opentelemetry/instrumentation/google_genai/generate_content.py

@@ -221,15 +221,15 @@ def _add_request_options_to_span(
         span.set_attribute(key, value)
 
 
-def _get_gen_ai_request_attributes(config: Union[GenerateContentConfigOrDict, None]) -> dict[str, Any]:
+def _get_gen_ai_request_attributes(
+    config: Union[GenerateContentConfigOrDict, None],
+) -> dict[str, Any]:
     if not config:
         return {}
     attributes: dict[str, Any] = {}
     config = _coerce_config_to_object(config)
     if config.seed:
-        attributes[gen_ai_attributes.GEN_AI_REQUEST_SEED] = (
-            config.seed
-        )
+        attributes[gen_ai_attributes.GEN_AI_REQUEST_SEED] = config.seed
     if config.candidate_count:
         attributes[gen_ai_attributes.GEN_AI_REQUEST_CHOICE_COUNT] = (
             config.candidate_count
@@ -400,7 +400,9 @@ def process_completion(
         config: Optional[GenerateContentConfigOrDict] = None,
     ):
         self._update_response(response)
-        self._maybe_log_completion_details(request, response.candidates or [], config)
+        self._maybe_log_completion_details(
+            request, response.candidates or [], config
+        )
 
     def process_error(self, e: Exception):
         self._error_type = str(e.__class__.__name__)
@@ -499,9 +501,7 @@ def _maybe_log_completion_details(
         input_messages = to_input_messages(
             contents=transformers.t_contents(request)
         )
-        output_messages = to_output_messages(
-            candidates=candidates
-        )
+        output_messages = to_output_messages(candidates=candidates)
 
         span = trace.get_current_span()
         event = Event(
@@ -515,12 +515,10 @@ def _maybe_log_completion_details(
             span=span,
             log_record=event,
         )
-        completion_details_attributes = (
-            _create_completion_details_attributes(
-                input_messages,
-                output_messages,
-                system_instructions,
-            )
+        completion_details_attributes = _create_completion_details_attributes(
+            input_messages,
+            output_messages,
+            system_instructions,
         )
         if self._content_recording_enabled in [
             ContentCapturingMode.EVENT_ONLY,
@@ -536,7 +534,12 @@ def _maybe_log_completion_details(
             ContentCapturingMode.SPAN_ONLY,
             ContentCapturingMode.SPAN_AND_EVENT,
         ]:
-            span.set_attributes({k: gen_ai_json_dumps(v) for k, v in completion_details_attributes.items()})
+            span.set_attributes(
+                {
+                    k: gen_ai_json_dumps(v)
+                    for k, v in completion_details_attributes.items()
+                }
+            )
             span.set_attributes(attributes)
 
     def _maybe_log_system_instruction(
@@ -844,7 +847,9 @@ def instrumented_generate_content_stream(
                 helper.process_error(error)
                 raise
             finally:
-                helper._maybe_log_completion_details(contents, candidates, config)
+                helper._maybe_log_completion_details(
+                    contents, candidates, config
+                )
                 helper.finalize_processing()
 
     return instrumented_generate_content_stream
@@ -982,7 +987,9 @@ async def _response_async_generator_wrapper():
                         helper.process_error(error)
                         raise
                     finally:
-                        helper._maybe_log_completion_details(contents, candidates, config)
+                        helper._maybe_log_completion_details(
+                            contents, candidates, config
+                        )
                         helper.finalize_processing()
 
             return _response_async_generator_wrapper()

instrumentation-genai/opentelemetry-instrumentation-google-genai/src/opentelemetry/instrumentation/google_genai/message.py

@@ -15,8 +15,8 @@
 from __future__ import annotations
 
 import logging
-from enum import Enum
 from dataclasses import dataclass
+from enum import Enum
 from typing import Literal
 
 from google.genai import types as genai_types

instrumentation-genai/opentelemetry-instrumentation-google-genai/src/opentelemetry/instrumentation/google_genai/otel_wrapper.py

@@ -18,7 +18,7 @@
 
 import google.genai
 
-from opentelemetry._logs import LogRecord, Logger, LoggerProvider
+from opentelemetry._logs import Logger, LoggerProvider, LogRecord
 from opentelemetry.metrics import Meter, MeterProvider
 from opentelemetry.semconv._incubating.metrics import gen_ai_metrics
 from opentelemetry.semconv.schemas import Schemas
@@ -40,9 +40,7 @@
 
 
 class OTelWrapper:
-    def __init__(
-        self, tracer: Tracer, logger: Logger, meter: Meter
-    ):
+    def __init__(self, tracer: Tracer, logger: Logger, meter: Meter):
         self._tracer = tracer
         self._logger = logger
         self._meter = meter