Adding changes for linting fix

Author: shuning.chen

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

@@ -1,3 +1,5 @@
+import time
+
 from langchain.agents import create_agent
 
 
@@ -135,6 +137,5 @@ def get_weather(city: str) -> str:
     {"session_id": "12345"},
 )
 
-# sleep for 150s to allow evals to finish 
-import time
+# sleep for 150s to allow evals to finish
 time.sleep(150)

instrumentation-genai/opentelemetry-instrumentation-langchain/examples/manual/multi-agent.py

@@ -25,7 +25,8 @@
 try:  # pragma: no cover - optional dependency
     from opentelemetry.util.genai.handler import get_telemetry_handler
 except Exception:  # pragma: no cover
-    get_telemetry_handler = lambda **_: None  # type: ignore
+    def get_telemetry_handler(**_):
+        return None  # type: ignore
 
 UTILS_PATH = Path(__file__).resolve().parent
 if __package__ in (None, ""):
@@ -66,7 +67,8 @@ def multi_agent_demo(llm: ChatOpenAI) -> None:
     try:
         from typing import Annotated, TypedDict
 
-        from langchain_core.messages import AIMessage, BaseMessage, HumanMessage, SystemMessage
+        from langchain_core.messages import AIMessage, BaseMessage
+        from langchain_core.runnables import RunnableLambda
         from langchain_core.tools import tool
         from langgraph.graph import END, StateGraph
         from langgraph.graph.message import add_messages

instrumentation-genai/opentelemetry-instrumentation-langchain/examples/manual/single-agent-codex.py

@@ -11,8 +11,6 @@
 """
 
 import os
-import sys
-from pathlib import Path
 from typing import Any, Tuple
 
 from .test_utils_splunk import DEFAULT_CACHE_FILE, create_cisco_chat_llm  # type: ignore
@@ -60,7 +58,6 @@ def _simple_agent_demo_inner(llm: ChatOpenAI) -> None:
     try:
         from typing import Annotated, TypedDict
 
-        from langchain_core.messages import AIMessage
         from langchain_core.tools import tool
         from langgraph.graph import END, StateGraph
         from langgraph.graph.message import add_messages

instrumentation-genai/opentelemetry-instrumentation-langchain/examples/manual/single-agent.py

@@ -3,8 +3,7 @@
 from pathlib import Path
 from typing import Any, Tuple
 
-from langchain_core.messages import BaseMessage, HumanMessage, SystemMessage
-from langchain_core.runnables import RunnableLambda
+from langchain_core.messages import BaseMessage, HumanMessage
 from langchain_openai import ChatOpenAI
 from opentelemetry import _events, _logs, metrics, trace
 from opentelemetry.exporter.otlp.proto.grpc._log_exporter import OTLPLogExporter
@@ -54,9 +53,7 @@ def _simple_agent_demo_inner(llm: ChatOpenAI) -> None:
     try:
         from typing import Annotated, TypedDict
 
-        from langchain_core.messages import AIMessage
         from langchain_core.tools import tool
-        from langgraph.graph import END, StateGraph
         from langgraph.graph.message import add_messages
         from langgraph.prebuilt import create_react_agent
     except ImportError:  # pragma: no cover - optional dependency

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

@@ -23,6 +23,13 @@
 from wrapt import wrap_function_wrapper
 from opentelemetry.util.genai.handler import get_telemetry_handler
 
+__all__ = [
+    "LangchainInstrumentor",
+    "get_event_logger",
+    "__version__",
+    "Meters",
+]
+
 logger = logging.getLogger(__name__)
 
 _instruments = ("langchain-core > 0.1.0", )
@@ -360,7 +367,7 @@ def __call__(
         args,
         kwargs,
     ) -> None:
-        run_manager = kwargs.get("run_manager")
+        _run_manager = kwargs.get("run_manager")  # noqa: F841
 
         ### FIXME: this was disabled to allow migration to util-genai and needs to be fixed
         # if run_manager:

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

@@ -8,20 +8,22 @@
 from uuid import uuid4
 
 import pytest
-try:  # pragma: no cover - optional dependency in CI
-    from langchain_core.messages import HumanMessage
-except ModuleNotFoundError:  # pragma: no cover - allow running subset without langchain_core
-    HumanMessage = None  # type: ignore[assignment]
+
+# Add package source to path if needed
 _PACKAGE_SRC = Path(__file__).resolve().parents[1] / "src"
 if _PACKAGE_SRC.exists():
     sys.path.insert(0, str(_PACKAGE_SRC))
 
-from opentelemetry.sdk.trace import TracerProvider
+try:  # pragma: no cover - optional dependency in CI
+    from langchain_core.messages import HumanMessage
+except ModuleNotFoundError:  # pragma: no cover - allow running subset without langchain_core
+    HumanMessage = None  # type: ignore[assignment]
 
-from opentelemetry.instrumentation.langchain.callback_handler import (
+from opentelemetry.sdk.trace import TracerProvider  # noqa: E402
+from opentelemetry.instrumentation.langchain.callback_handler import (  # noqa: E402
     LangchainCallbackHandler,
 )
-from opentelemetry.util.genai.types import Step, ToolCall
+from opentelemetry.util.genai.types import Step, ToolCall  # noqa: E402
 
 LANGCHAIN_CORE_AVAILABLE = HumanMessage is not None
 

util/opentelemetry-util-genai/src/opentelemetry/util/genai/handler.py

@@ -53,7 +53,6 @@
 import time
 from typing import Any, Optional
 
-
 try:
     from opentelemetry.util.genai.debug import genai_debug_log
 except Exception:  # pragma: no cover - fallback if debug module missing
@@ -63,19 +62,14 @@ def genai_debug_log(*_args: Any, **_kwargs: Any) -> None:  # type: ignore
 
 
 from opentelemetry import _events as _otel_events
-from opentelemetry import metrics as _metrics
 from opentelemetry import trace as _trace_mod
 from opentelemetry._logs import Logger, LoggerProvider, get_logger
 from opentelemetry.metrics import MeterProvider, get_meter
-from opentelemetry.semconv._incubating.attributes import (
-    gen_ai_attributes as GenAI,
-)
 from opentelemetry.semconv.schemas import Schemas
 from opentelemetry.trace import (
     TracerProvider,
     get_tracer,
 )
-
 from opentelemetry.util.genai.emitters.configuration import (
     build_emitter_pipeline,
 )
@@ -98,10 +92,10 @@ def genai_debug_log(*_args: Any, **_kwargs: Any) -> None:  # type: ignore
     Workflow,
 )
 from opentelemetry.util.genai.utils import (
-    get_content_capturing_mode,
     _is_truthy_env,
     _load_completion_callbacks,
-    _parse_callback_filter
+    _parse_callback_filter,
+    get_content_capturing_mode,
 )
 from opentelemetry.util.genai.version import __version__