WIP Decorators support

instrumentation-genai/opentelemetry-instrumentation-langchain/.gitignore

@@ -0,0 +1,168 @@
+# Byte-compiled / optimized / DLL files
+__pycache__/
+*.py[cod]
+*$py.class
+
+# C extensions
+*.so
+
+# Distribution / packaging
+.Python
+build/
+develop-eggs/
+dist/
+downloads/
+eggs/
+.eggs/
+lib/
+lib64/
+parts/
+sdist/
+var/
+wheels/
+pip-wheel-metadata/
+share/python-wheels/
+*.egg-info/
+.installed.cfg
+*.egg
+MANIFEST
+
+# PyInstaller
+#  Usually these files are written by a python script from a template
+#  before PyInstaller builds the exe, so as to inject date/other infos into it.
+*.manifest
+*.spec
+
+# Installer logs
+pip-log.txt
+pip-delete-this-directory.txt
+
+# Unit test / coverage reports
+htmlcov/
+.tox/
+.nox/
+.coverage
+.coverage.*
+.cache
+nosetests.xml
+coverage.xml
+*.cover
+*.py,cover
+.hypothesis/
+.pytest_cache/
+
+# Translations
+*.mo
+*.pot
+
+# Django stuff:
+*.log
+local_settings.py
+db.sqlite3
+db.sqlite3-journal
+
+# Flask stuff:
+instance/
+.webassets-cache
+
+# Scrapy stuff:
+.scrapy
+
+# Sphinx documentation
+docs/_build/
+
+# PyBuilder
+target/
+
+# Jupyter Notebook
+.ipynb_checkpoints
+
+# IPython
+profile_default/
+ipython_config.py
+
+# pyenv
+.python-version
+
+# pipenv
+#   According to pypa/pipenv#598, it is recommended to include Pipfile.lock in version control.
+#   However, in case of collaboration, if having platform-specific dependencies or dependencies
+#   having no cross-platform support, pipenv may install dependencies that don't work, or not
+#   install all needed dependencies.
+#Pipfile.lock
+
+# PEP 582; used by e.g. github.com/David-OConnor/pyflow
+__pypackages__/
+
+# Celery stuff
+celerybeat-schedule
+celerybeat.pid
+
+# SageMath parsed files
+*.sage.py
+
+# Environments
+.env
+.venv
+env/
+venv/
+ENV/
+env.bak/
+venv.bak/
+
+# Spyder project settings
+.spyderproject
+.spyproject
+
+# Rope project settings
+.ropeproject
+
+# mkdocs documentation
+/site
+
+# mypy
+.mypy_cache/
+.dmypy.json
+dmypy.json
+
+# Pyre type checker
+.pyre/
+
+# Mac files
+.DS_Store
+
+# Environment variables
+.env
+
+# sqlite database files
+*.db
+*.db-shm
+*.db-wal
+
+# PNG files
+*.png
+
+demo/
+
+.ruff_cache
+
+.vscode/
+
+output/
+
+.terraform.lock.hcl
+.terraform/
+foo.sh
+tfplan
+tfplan.txt
+tfplan.json
+terraform_output.json
+
+
+# IntelliJ / PyCharm
+.idea
+
+
+*.txt
+
+.dockerconfigjson

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

@@ -0,0 +1,71 @@
+import os
+
+from dotenv import load_dotenv
+from langchain_core.messages import HumanMessage, SystemMessage
+from langchain_openai import ChatOpenAI
+
+from opentelemetry import _events, _logs, metrics, trace
+from opentelemetry.exporter.otlp.proto.grpc._log_exporter import (
+    OTLPLogExporter,
+)
+from opentelemetry.exporter.otlp.proto.grpc.metric_exporter import (
+    OTLPMetricExporter,
+)
+from opentelemetry.exporter.otlp.proto.grpc.trace_exporter import (
+    OTLPSpanExporter,
+)
+
+# from opentelemetry.genai.sdk.decorators import llm
+from opentelemetry.sdk._events import EventLoggerProvider
+from opentelemetry.sdk._logs import LoggerProvider
+from opentelemetry.sdk._logs.export import BatchLogRecordProcessor
+from opentelemetry.sdk.metrics import MeterProvider
+from opentelemetry.sdk.metrics.export import PeriodicExportingMetricReader
+from opentelemetry.sdk.trace import TracerProvider
+from opentelemetry.sdk.trace.export import BatchSpanProcessor
+from opentelemetry.util.genai.decorators import llm
+
+# configure tracing
+trace.set_tracer_provider(TracerProvider())
+trace.get_tracer_provider().add_span_processor(
+    BatchSpanProcessor(OTLPSpanExporter())
+)
+
+metric_reader = PeriodicExportingMetricReader(OTLPMetricExporter())
+metrics.set_meter_provider(MeterProvider(metric_readers=[metric_reader]))
+
+# configure logging and events
+_logs.set_logger_provider(LoggerProvider())
+_logs.get_logger_provider().add_log_record_processor(
+    BatchLogRecordProcessor(OTLPLogExporter())
+)
+_events.set_event_logger_provider(EventLoggerProvider())
+
+# Load environment variables from .env file
+load_dotenv()
+
+
+@llm(name="invoke_langchain_model")
+def invoke_model(messages):
+    # Get API key from environment variable or set a placeholder
+    api_key = os.getenv("OPENAI_API_KEY")
+    if not api_key:
+        raise ValueError("OPENAI_API_KEY environment variable must be set")
+
+    llm = ChatOpenAI(model="gpt-3.5-turbo", api_key=api_key)
+    result = llm.invoke(messages)
+    return result
+
+
+def main():
+    messages = [
+        SystemMessage(content="You are a helpful assistant!"),
+        HumanMessage(content="What is the capital of France?"),
+    ]
+
+    result = invoke_model(messages)
+    print("LLM output:\n", result)
+
+
+if __name__ == "__main__":
+    main()

instrumentation-genai/opentelemetry-instrumentation-langchain/examples/manual/.env

@@ -5,4 +5,7 @@ OPENAI_API_KEY=sk-YOUR_API_KEY
 # OTEL_EXPORTER_OTLP_ENDPOINT=http://localhost:4317
 # OTEL_EXPORTER_OTLP_PROTOCOL=grpc
 
+# Change to 'false' to hide prompt and completion content
+OTEL_INSTRUMENTATION_LANGCHAIN_CAPTURE_MESSAGE_CONTENT=true
+
 OTEL_SERVICE_NAME=opentelemetry-python-langchain-manual

instrumentation-genai/opentelemetry-instrumentation-langchain/examples/manual/README.rst

@@ -1,3 +1,4 @@
+
 OpenTelemetry Langcahin Instrumentation Example
 ============================================
 

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

@@ -6,6 +6,7 @@
     OTLPSpanExporter,
 )
 from opentelemetry.instrumentation.langchain import LangChainInstrumentor
+
 from opentelemetry.sdk.trace import TracerProvider
 from opentelemetry.sdk.trace.export import BatchSpanProcessor
 

instrumentation-genai/opentelemetry-instrumentation-langchain/examples/manual/requirements.txt

@@ -1,7 +1,8 @@
+
 langchain==0.3.21
 langchain_openai
 opentelemetry-sdk>=1.31.0
 opentelemetry-exporter-otlp-proto-grpc>=1.31.0
 
 # Uncomment after lanchain instrumetation is released
-# opentelemetry-instrumentation-langchain~=2.0b0.dev
+# opentelemetry-instrumentation-langchain~=2.0b0.dev

instrumentation-genai/opentelemetry-instrumentation-langchain/examples/zero-code/README.rst

@@ -1,3 +1,4 @@
+
 OpenTelemetry Langchain Zero-Code Instrumentation Example
 ======================================================
 

instrumentation-genai/opentelemetry-instrumentation-langchain/examples/zero-code/requirements.txt

@@ -1,3 +1,4 @@
+
 langchain==0.3.21
 langchain_openai
 opentelemetry-sdk>=1.31.0

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

@@ -29,6 +29,7 @@
         SystemMessage(content="You are a helpful assistant!"),
         HumanMessage(content="What is the capital of France?"),
     ]
+
     result = llm.invoke(messages)
     LangChainInstrumentor().uninstrument()
 
@@ -66,7 +67,6 @@ def __init__(
 
     def instrumentation_dependencies(self) -> Collection[str]:
         return _instruments
-
     def _instrument(self, **kwargs: Any):
         """
         Enable Langchain instrumentation.

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

@@ -0,0 +1,33 @@
+# Copyright The OpenTelemetry Authors
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+
+class Config:
+    """
+    Shared static config for LangChain OTel instrumentation.
+    """
+
+    # Logger to handle exceptions during instrumentation
+    exception_logger = None
+
+    # Globally suppress instrumentation
+    _suppress_instrumentation = False
+
+    @classmethod
+    def suppress_instrumentation(cls, suppress: bool = True):
+        cls._suppress_instrumentation = suppress
+
+    @classmethod
+    def is_instrumentation_suppressed(cls) -> bool:
+        return cls._suppress_instrumentation