test lint

Author: JWinermaSplunk

instrumentation-genai/opentelemetry-instrumentation-langchain/examples/manual/main.py

@@ -261,7 +261,9 @@ def embedding_invocation_demo():
         mixed_vectors = embeddings.embed_documents(mixed_texts)
         print(f"   ✓ Embedded {len(mixed_vectors)} mixed content texts")
         for i, text in enumerate(mixed_texts):
-            print(f"   - Text {i+1}: {text[:40]}... → {len(mixed_vectors[i])}D vector")
+            print(
+                f"   - Text {i + 1}: {text[:40]}... → {len(mixed_vectors[i])}D vector"
+            )
     except Exception as e:
         print(f"   ✗ Error: {e}")
 

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

@@ -129,9 +129,9 @@ def get_entity(self, run_id):
 
 
 @pytest.fixture(name="handler_with_stub")
-def _handler_with_stub_fixture() -> (
-    Tuple[LangchainCallbackHandler, _StubTelemetryHandler]
-):
+def _handler_with_stub_fixture() -> Tuple[
+    LangchainCallbackHandler, _StubTelemetryHandler
+]:
     stub = _StubTelemetryHandler()
     handler = LangchainCallbackHandler(telemetry_handler=stub)
     return handler, stub
@@ -389,9 +389,9 @@ def _invoke_with_env(env_value: Optional[str]):
     invocation_default = _invoke_with_env(None)
     invocation_traceloop = _invoke_with_env("traceloop_compat")
 
-    assert (
-        invocation_default.attributes == invocation_traceloop.attributes
-    ), "Emitter env toggle should not change recorded attributes"
+    assert invocation_default.attributes == invocation_traceloop.attributes, (
+        "Emitter env toggle should not change recorded attributes"
+    )
 
     attrs = invocation_default.attributes
     assert invocation_default.request_model == "gpt-5-nano"

util/opentelemetry-util-genai-traceloop-translator/tests/test_agent_task_message_reconstruction.py

@@ -110,9 +110,9 @@ def test_agent_span_reconstructs_messages(self, setup_tracer_with_handler):
         provider.force_flush()
 
         # Verify that messages were cached (indicating reconstruction happened)
-        assert (
-            span_id in processor._message_cache
-        ), "Messages should be cached for agent span"
+        assert span_id in processor._message_cache, (
+            "Messages should be cached for agent span"
+        )
 
         cached_input, cached_output = processor._message_cache[span_id]
 
@@ -122,31 +122,31 @@ def test_agent_span_reconstructs_messages(self, setup_tracer_with_handler):
 
         # Verify input message format
         input_msg = cached_input[0]
-        assert isinstance(
-            input_msg, InputMessage
-        ), "Should be InputMessage object"
+        assert isinstance(input_msg, InputMessage), (
+            "Should be InputMessage object"
+        )
         assert input_msg.role == "user", "Should have user role"
         assert len(input_msg.parts) == 1, "Should have 1 part"
-        assert isinstance(
-            input_msg.parts[0], Text
-        ), "Part should be Text object"
-        assert (
-            input_msg.parts[0].content == "Plan a trip to Paris"
-        ), "Input content should match"
+        assert isinstance(input_msg.parts[0], Text), (
+            "Part should be Text object"
+        )
+        assert input_msg.parts[0].content == "Plan a trip to Paris", (
+            "Input content should match"
+        )
 
         # Verify output message format
         output_msg = cached_output[0]
-        assert isinstance(
-            output_msg, OutputMessage
-        ), "Should be OutputMessage object"
+        assert isinstance(output_msg, OutputMessage), (
+            "Should be OutputMessage object"
+        )
         assert output_msg.role == "assistant", "Should have assistant role"
         assert len(output_msg.parts) == 1, "Should have 1 part"
-        assert isinstance(
-            output_msg.parts[0], Text
-        ), "Part should be Text object"
-        assert (
-            "Paris" in output_msg.parts[0].content
-        ), "Output should mention Paris"
+        assert isinstance(output_msg.parts[0], Text), (
+            "Part should be Text object"
+        )
+        assert "Paris" in output_msg.parts[0].content, (
+            "Output should mention Paris"
+        )
 
     def test_agent_span_has_genai_attributes(self, setup_tracer_with_handler):
         """Test that agent spans get gen_ai.input.messages and gen_ai.output.messages."""
@@ -176,18 +176,18 @@ def test_agent_span_has_genai_attributes(self, setup_tracer_with_handler):
         assert agent_span.attributes is not None, "Span should have attributes"
 
         # Verify gen_ai attributes are present
-        assert (
-            "gen_ai.input.messages" in agent_span.attributes
-        ), "Should have gen_ai.input.messages"
-        assert (
-            "gen_ai.output.messages" in agent_span.attributes
-        ), "Should have gen_ai.output.messages"
-        assert (
-            "gen_ai.span.kind" in agent_span.attributes
-        ), "Should have gen_ai.span.kind"
-        assert (
-            agent_span.attributes["gen_ai.span.kind"] == "agent"
-        ), "Should preserve agent kind"
+        assert "gen_ai.input.messages" in agent_span.attributes, (
+            "Should have gen_ai.input.messages"
+        )
+        assert "gen_ai.output.messages" in agent_span.attributes, (
+            "Should have gen_ai.output.messages"
+        )
+        assert "gen_ai.span.kind" in agent_span.attributes, (
+            "Should have gen_ai.span.kind"
+        )
+        assert agent_span.attributes["gen_ai.span.kind"] == "agent", (
+            "Should preserve agent kind"
+        )
 
 
 class TestTaskMessageReconstruction:
@@ -232,9 +232,9 @@ def test_task_span_reconstructs_messages(self, setup_tracer_with_handler):
         provider.force_flush()
 
         # Verify that messages were cached (indicating reconstruction happened)
-        assert (
-            span_id in processor._message_cache
-        ), "Messages should be cached for task span"
+        assert span_id in processor._message_cache, (
+            "Messages should be cached for task span"
+        )
 
         cached_input, cached_output = processor._message_cache[span_id]
 
@@ -247,9 +247,9 @@ def test_task_span_reconstructs_messages(self, setup_tracer_with_handler):
             cached_input[0].parts[0].content
             == "Search for flights from Seattle to Paris"
         ), "Input should match"
-        assert (
-            "5 flights" in cached_output[0].parts[0].content
-        ), "Output should mention flights"
+        assert "5 flights" in cached_output[0].parts[0].content, (
+            "Output should mention flights"
+        )
 
     def test_task_span_without_messages_skips_reconstruction(
         self, setup_tracer_with_handler
@@ -267,9 +267,9 @@ def test_task_span_without_messages_skips_reconstruction(
         provider.force_flush()
 
         # Should not crash, and span_id should NOT be in cache
-        assert (
-            span_id not in processor._message_cache
-        ), "Empty task should not be cached"
+        assert span_id not in processor._message_cache, (
+            "Empty task should not be cached"
+        )
 
 
 class TestLLMOperationMessageReconstruction:
@@ -298,9 +298,9 @@ def test_chat_operation_reconstructs_messages(
         provider.force_flush()
 
         # Verify messages were cached
-        assert (
-            span_id in processor._message_cache
-        ), "Messages should be cached for chat operation"
+        assert span_id in processor._message_cache, (
+            "Messages should be cached for chat operation"
+        )
 
         cached_input, cached_output = processor._message_cache[span_id]
         assert len(cached_input) == 1, "Should have 1 input message"
@@ -328,9 +328,9 @@ def test_completion_operation_reconstructs_messages(
 
         provider.force_flush()
 
-        assert (
-            span_id in processor._message_cache
-        ), "Messages should be cached for completion operation"
+        assert span_id in processor._message_cache, (
+            "Messages should be cached for completion operation"
+        )
 
 
 class TestNonLLMSpanSkipsReconstruction:
@@ -351,9 +351,9 @@ def test_workflow_span_skips_reconstruction(
         provider.force_flush()
 
         # Workflow spans should NOT trigger message reconstruction
-        assert (
-            span_id not in processor._message_cache
-        ), "Workflow spans should not cache messages"
+        assert span_id not in processor._message_cache, (
+            "Workflow spans should not cache messages"
+        )
 
     def test_tool_span_skips_reconstruction(self, setup_tracer_with_handler):
         """Test that tool spans skip message reconstruction."""
@@ -367,9 +367,9 @@ def test_tool_span_skips_reconstruction(self, setup_tracer_with_handler):
         provider.force_flush()
 
         # Tool spans should NOT trigger message reconstruction
-        assert (
-            span_id not in processor._message_cache
-        ), "Tool spans should not cache messages"
+        assert span_id not in processor._message_cache, (
+            "Tool spans should not cache messages"
+        )
 
     def test_unknown_span_skips_reconstruction(
         self, setup_tracer_with_handler
@@ -384,9 +384,9 @@ def test_unknown_span_skips_reconstruction(
         provider.force_flush()
 
         # Random spans should NOT trigger message reconstruction
-        assert (
-            span_id not in processor._message_cache
-        ), "Unknown spans should not cache messages"
+        assert span_id not in processor._message_cache, (
+            "Unknown spans should not cache messages"
+        )
 
 
 class TestEdgeCases:
@@ -407,9 +407,9 @@ def test_agent_with_malformed_json(self, setup_tracer_with_handler):
         provider.force_flush()
 
         # Malformed data should not be cached
-        assert (
-            span_id not in processor._message_cache
-        ), "Malformed JSON should not be cached"
+        assert span_id not in processor._message_cache, (
+            "Malformed JSON should not be cached"
+        )
 
     def test_task_with_empty_messages(self, setup_tracer_with_handler):
         """Test that task with empty message arrays is handled."""
@@ -431,12 +431,12 @@ def test_task_with_empty_messages(self, setup_tracer_with_handler):
         # Empty messages should still be cached (as empty lists)
         if span_id in processor._message_cache:
             cached_input, cached_output = processor._message_cache[span_id]
-            assert (
-                cached_input == []
-            ), "Empty input should be cached as empty list"
-            assert (
-                cached_output == []
-            ), "Empty output should be cached as empty list"
+            assert cached_input == [], (
+                "Empty input should be cached as empty list"
+            )
+            assert cached_output == [], (
+                "Empty output should be cached as empty list"
+            )
 
     def test_mixed_span_kinds(self, setup_tracer_with_handler):
         """Test different span kinds in same workflow."""
@@ -473,15 +473,15 @@ def test_mixed_span_kinds(self, setup_tracer_with_handler):
         provider.force_flush()
 
         # Verify caching behavior
-        assert (
-            span_ids["agent"] in processor._message_cache
-        ), "Agent span should cache messages"
-        assert (
-            span_ids["task"] in processor._message_cache
-        ), "Task span should cache messages"
-        assert (
-            span_ids["workflow"] not in processor._message_cache
-        ), "Workflow span should not cache messages"
+        assert span_ids["agent"] in processor._message_cache, (
+            "Agent span should cache messages"
+        )
+        assert span_ids["task"] in processor._message_cache, (
+            "Task span should cache messages"
+        )
+        assert span_ids["workflow"] not in processor._message_cache, (
+            "Workflow span should not cache messages"
+        )
 
 
 if __name__ == "__main__":

util/opentelemetry-util-genai-traceloop-translator/tests/test_args_wrapper_format.py

@@ -57,22 +57,22 @@ def test_args_wrapper_with_messages(self):
         assert len(result) == 1, f"Should have 1 message, got {len(result)}"
 
         message = result[0]
-        assert (
-            message["role"] == "user"
-        ), f"Role should be 'user', got {message['role']}"
-        assert (
-            len(message["parts"]) == 1
-        ), f"Should have 1 part, got {len(message['parts'])}"
+        assert message["role"] == "user", (
+            f"Role should be 'user', got {message['role']}"
+        )
+        assert len(message["parts"]) == 1, (
+            f"Should have 1 part, got {len(message['parts'])}"
+        )
 
         part = message["parts"][0]
-        assert (
-            part["type"] == "text"
-        ), f"Part type should be 'text', got {part['type']}"
+        assert part["type"] == "text", (
+            f"Part type should be 'text', got {part['type']}"
+        )
         assert "Paris" in part["content"], "Content should mention Paris"
         assert "Seattle" in part["content"], "Content should mention Seattle"
-        assert (
-            "boutique hotel" in part["content"]
-        ), "Content should mention boutique hotel"
+        assert "boutique hotel" in part["content"], (
+            "Content should mention boutique hotel"
+        )
 
     def test_args_wrapper_with_multiple_messages(self):
         """Test args wrapper with conversation history."""