ruff

Signed-off-by: Adrian Cole <[email protected]>

Author: codefromthecrypt

instrumentation/opentelemetry-instrumentation-openai-v2/tests/conftest.py

@@ -2,9 +2,9 @@
 
 import json
 import os
-import yaml
 
 import pytest
+import yaml
 from openai import OpenAI
 
 from opentelemetry.instrumentation.openai_v2 import OpenAIInstrumentor
@@ -152,7 +152,9 @@ class PrettyPrintJSONBody:
     @staticmethod
     def serialize(cassette_dict):
         cassette_dict = convert_body_to_literal(cassette_dict)
-        return yaml.dump(cassette_dict, default_flow_style=False, allow_unicode=True)
+        return yaml.dump(
+            cassette_dict, default_flow_style=False, allow_unicode=True
+        )
 
     @staticmethod
     def deserialize(cassette_string):
@@ -164,6 +166,7 @@ def fixture_vcr(vcr):
     vcr.register_serializer("yaml", PrettyPrintJSONBody)
     return vcr
 
+
 def scrub_response_headers(response):
     """
     This scrubs sensitive response headers. Note they are case-sensitive!

instrumentation/opentelemetry-instrumentation-openai-v2/tests/test_chat_completions.py

@@ -16,7 +16,7 @@
 from typing import Optional
 
 import pytest
-from openai import (OpenAI, APIConnectionError, NotFoundError)
+from openai import APIConnectionError, NotFoundError, OpenAI
 from openai.resources.chat.completions import ChatCompletion
 
 from opentelemetry.sdk.trace import ReadableSpan
@@ -85,7 +85,9 @@ def test_chat_completion_bad_endpoint(span_exporter, instrument_no_content):
         spans[0], llm_model_value, server_address="localhost"
     )
     assert 4242 == spans[0].attributes[ServerAttributes.SERVER_PORT]
-    assert "APIConnectionError" == spans[0].attributes[ErrorAttributes.ERROR_TYPE]
+    assert (
+        "APIConnectionError" == spans[0].attributes[ErrorAttributes.ERROR_TYPE]
+    )
 
 
 @pytest.mark.vcr()