migrated from local to this repository

Author: haneric00

ai_agent_instrumentation/opentelemetry-instrumentation-langchain-v2/pyproject.toml

@@ -0,0 +1,42 @@
+[build-system]
+requires = ["hatchling"]
+build-backend = "hatchling.build"
+
+[project]
+name = "opentelemetry-instrumentation-langchain-v2"
+dynamic = ["version"]
+description = "OpenTelemetry Official Langchain instrumentation"
+license = "Apache-2.0"
+requires-python = ">=3.9"
+authors = [
+  { name = "OpenTelemetry Authors", email = "[email protected]" },
+]
+classifiers = [
+  "Development Status :: 4 - Beta",
+  "Intended Audience :: Developers",
+  "License :: OSI Approved :: Apache Software License",
+  "Programming Language :: Python",
+  "Programming Language :: Python :: 3",
+  "Programming Language :: Python :: 3.9",
+  "Programming Language :: Python :: 3.10",
+  "Programming Language :: Python :: 3.11",
+  "Programming Language :: Python :: 3.12",
+  "Programming Language :: Python :: 3.13",
+]
+dependencies = [
+  "opentelemetry-api ~= 1.30",
+  "opentelemetry-sdk ~= 1.30",
+  "opentelemetry-instrumentation ~= 0.51b0",
+  "opentelemetry-semantic-conventions ~= 0.51b0"
+]
+[project.optional-dependencies]
+instruments = [
+  "langchain >= 0.3.21",
+]
+
+[tool.hatch.build.targets.wheel]
+packages = ["/src/amazon/opentelemetry"]
+
+[tool.hatch.version]
+path = "version.py"
+

ai_agent_instrumentation/opentelemetry-instrumentation-langchain-v2/src/opentelemetry/instrumentation/langchain_v2/__init__.py

@@ -0,0 +1,56 @@
+from typing import Collection
+from opentelemetry.instrumentation.instrumentor import BaseInstrumentor
+from wrapt import wrap_function_wrapper
+
+from opentelemetry.trace import get_tracer
+from opentelemetry.instrumentation.utils import unwrap
+
+from opentelemetry.instrumentation.langchain_v2.version import __version__
+from opentelemetry.instrumentation.langchain_v2.callback_handler import OpenTelemetryCallbackHandler
+
+__all__ = ["OpenTelemetryCallbackHandler"]
+
+_instruments = ("langchain >= 0.1.0",)
+
+class LangChainInstrumentor(BaseInstrumentor):
+    
+    def instrumentation_dependencies(cls) -> Collection[str]:
+        return _instruments
+    
+    def _instrument(self, **kwargs):
+        tracer_provider = kwargs.get("tracer_provider")
+        tracer = get_tracer(__name__, __version__, tracer_provider)
+
+        otelCallbackHandler = OpenTelemetryCallbackHandler(tracer)
+        
+        wrap_function_wrapper(
+            module="langchain_core.callbacks",
+            name="BaseCallbackManager.__init__",
+            wrapper=_BaseCallbackManagerInitWrapper(otelCallbackHandler),
+        )
+    
+    def _uninstrument(self, **kwargs):
+        unwrap("langchain_core.callbacks", "BaseCallbackManager.__init__")
+        if hasattr(self, "_wrapped"):
+            for module, name in self._wrapped:
+                unwrap(module, name)
+        
+        
+class _BaseCallbackManagerInitWrapper:
+    def __init__(self, callback_handler: "OpenTelemetryCallbackHandler"):
+        self.callback_handler = callback_handler
+        self._wrapped = []
+        
+    def __call__(
+        self,
+        wrapped,
+        instance,
+        args,
+        kwargs,
+    ) -> None:
+        wrapped(*args, **kwargs)
+        for handler in instance.inheritable_handlers:
+            if isinstance(handler, OpenTelemetryCallbackHandler):
+                return None
+        else:
+            instance.add_handler(self.callback_handler, True)