skipped telemetry for other than ChatOpenAI and ChatBedrock. Added test for the same.

Author: wrisa

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

@@ -14,7 +14,7 @@
 
 """
 Langchain instrumentation supporting `ChatOpenAI` and `ChatBedrock`, it can be enabled by
-using ``LangChainInstrumentor``.
+using ``LangChainInstrumentor``. Other providers/LLMs may be supported in the future and telemetry for them is skipped for now.
 
 Usage
 -----

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

@@ -54,6 +54,12 @@ def on_chat_model_start(
         metadata: dict[str, Any] | None,
         **kwargs: Any,
     ) -> None:
+        # Other providers/LLMs may be supported in the future and telemetry for them is skipped for now.
+        if serialized and "name" in serialized:
+            name = serialized.get("name")
+            if name not in ("ChatOpenAI", "ChatBedrock"):
+                return
+
         if "invocation_params" in kwargs:
             params = (
                 kwargs["invocation_params"].get("params")
@@ -74,6 +80,10 @@ def on_chat_model_start(
                 request_model = model
                 break
 
+        # Skip telemetry for unsupported request models
+        if request_model == "unknown":
+            return
+
         span = self.span_manager.create_chat_span(
             run_id=run_id,
             parent_run_id=parent_run_id,

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

@@ -7,6 +7,7 @@
 import pytest
 import yaml
 from langchain_aws import ChatBedrock
+from langchain_google_genai import ChatGoogleGenerativeAI
 from langchain_openai import ChatOpenAI
 
 from opentelemetry.instrumentation.langchain import LangChainInstrumentor
@@ -51,6 +52,13 @@ def fixture_us_amazon_nova_lite_v1_0():
     yield llm
 
 
+@pytest.fixture(scope="function", name="gemini")
+def fixture_gemini():
+    llm_model_value = "gemini-2.5-pro"
+    llm = ChatGoogleGenerativeAI(model=llm_model_value, api_key="test_key")
+    yield llm
+
+
 @pytest.fixture(scope="function", name="span_exporter")
 def fixture_span_exporter():
     exporter = InMemorySpanExporter()

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

@@ -53,6 +53,23 @@ def test_us_amazon_nova_lite_v1_0_bedrock_llm_call(
     assert_bedrock_completion_attributes(spans[0], result)
 
 
+# span_exporter, start_instrumentation, gemini are coming from fixtures defined in conftest.py
+@pytest.mark.vcr()
+def test_gemini(span_exporter, start_instrumentation, gemini):
+    messages = [
+        SystemMessage(content="You are a helpful assistant!"),
+        HumanMessage(content="What is the capital of France?"),
+    ]
+
+    result = gemini.invoke(messages)
+
+    assert result.content.find("The capital of France is **Paris**") != -1
+
+    # verify spans
+    spans = span_exporter.get_finished_spans()
+    assert len(spans) == 0  # No spans should be created for gemini as of now
+
+
 def assert_openai_completion_attributes(
     span: ReadableSpan, response: Optional
 ):