fixed ruff and typecheck

Author: wrisa

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

@@ -36,7 +36,7 @@
 ---
 """
 
-from typing import Any, Callable, Collection, Optional
+from typing import Any, Callable, Collection
 
 from langchain_core.callbacks import BaseCallbackHandler  # type: ignore
 from wrapt import wrap_function_wrapper  # type: ignore

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

@@ -28,7 +28,6 @@ def handler(self, tracer):
             (uuid.uuid4(), True),  # Parent in spans
         ],
     )
-
     def test_create_span(
         self, handler, tracer, parent_run_id, parent_in_spans
     ):
@@ -42,9 +41,7 @@ def test_create_span(
         # Setup parent if needed
         if parent_run_id is not None and parent_in_spans:
             parent_mock_span = unittest.mock.Mock(spec=Span)
-            handler.spans[parent_run_id] = _SpanState(
-                span=parent_mock_span
-            )
+            handler.spans[parent_run_id] = _SpanState(span=parent_mock_span)
 
         with (
             unittest.mock.patch.object(
@@ -81,10 +78,7 @@ def test_create_span(
                 )
                 mock_set_span_in_context.assert_called_once_with(mock_span)
 
-
-    def test_end_span(
-            self, handler
-    ):
+    def test_end_span(self, handler):
         # Arrange
         run_id = uuid.uuid4()
         mock_span = unittest.mock.Mock(spec=Span)
@@ -104,4 +98,4 @@ def test_end_span(
         mock_span.end.assert_called_once()
         child_mock_span.end.assert_called_once()
         assert run_id not in handler.spans
-        assert child_run_id not in handler.spans
+        assert child_run_id not in handler.spans