openlit 1.34.22__py3-none-any.whl → 1.34.23__py3-none-any.whl

openlit/instrumentation/langchain_community/__init__.py

@@ -0,0 +1,74 @@
+"""Initializer of Auto Instrumentation of LangChain Community Functions"""
+from typing import Collection
+import importlib.metadata
+from opentelemetry.instrumentation.instrumentor import BaseInstrumentor
+from wrapt import wrap_function_wrapper
+
+from openlit.instrumentation.langchain_community.langchain_community import (
+    general_wrap,
+)
+from openlit.instrumentation.langchain_community.async_langchain_community import (
+    async_general_wrap,
+)
+
+_instruments = ("langchain-community >= 0.2.0",)
+
+WRAPPED_METHODS = [
+    {
+        "package": "langchain_community.document_loaders.base",
+        "object": "BaseLoader.load",
+        "endpoint": "langchain_community.retrieve.load",
+        "wrapper": general_wrap,
+    },
+    {
+        "package": "langchain_community.document_loaders.base",
+        "object": "BaseLoader.aload",
+        "endpoint": "langchain_community.retrieve.load",
+        "wrapper": async_general_wrap,
+    },
+    {
+        "package": "langchain_text_splitters.base",
+        "object": "TextSplitter.split_documents",
+        "endpoint": "langchain_community.retrieve.split_documents",
+        "wrapper": general_wrap,
+    },
+    {
+        "package": "langchain_text_splitters.base",
+        "object": "TextSplitter.create_documents",
+        "endpoint": "langchain_community.retrieve.create_documents",
+        "wrapper": general_wrap,
+    },
+]
+
+class LangChainCommunityInstrumentor(BaseInstrumentor):
+    """
+    An instrumentor for LangChain Community client library.
+    """
+
+    def instrumentation_dependencies(self) -> Collection[str]:
+        return _instruments
+
+    def _instrument(self, **kwargs):
+        version = importlib.metadata.version("langchain-community")
+        environment = kwargs.get("environment", "default")
+        application_name = kwargs.get("application_name", "default")
+        tracer = kwargs.get("tracer")
+        pricing_info = kwargs.get("pricing_info", {})
+        capture_message_content = kwargs.get("capture_message_content", False)
+        metrics = kwargs.get("metrics_dict")
+        disable_metrics = kwargs.get("disable_metrics")
+
+        for wrapped_method in WRAPPED_METHODS:
+            wrap_package = wrapped_method.get("package")
+            wrap_object = wrapped_method.get("object")
+            gen_ai_endpoint = wrapped_method.get("endpoint")
+            wrapper = wrapped_method.get("wrapper")
+            wrap_function_wrapper(
+                wrap_package,
+                wrap_object,
+                wrapper(gen_ai_endpoint, version, environment, application_name,
+                       tracer, pricing_info, capture_message_content, metrics, disable_metrics),
+            )
+
+    def _uninstrument(self, **kwargs):
+        pass

openlit/instrumentation/langchain_community/async_langchain_community.py

@@ -0,0 +1,49 @@
+"""
+Module for monitoring async LangChain Community operations.
+"""
+
+import logging
+from opentelemetry import trace
+from opentelemetry.trace import Status, StatusCode
+
+from openlit.__helpers import handle_exception
+from openlit.instrumentation.langchain_community.utils import process_general_response
+
+# Initialize logger for LangChain Community instrumentation
+logger = logging.getLogger(__name__)
+
+def async_general_wrap(gen_ai_endpoint, version, environment, application_name, tracer, pricing_info,
+    capture_message_content, metrics, disable_metrics):
+    """
+    Generates a telemetry wrapper for GenAI operations.
+    """
+
+    async def wrapper(wrapped, instance, args, kwargs):
+        """
+        Wraps the GenAI operation call.
+        """
+
+        # Prepare server address and port
+        server_address = "127.0.0.1"
+        server_port = "80"
+
+        # Get the parent span from the tracer
+        with tracer.start_as_current_span(gen_ai_endpoint, kind=trace.SpanKind.CLIENT) as span:
+            try:
+                # Call the original async function
+                response = await wrapped(*args, **kwargs)
+
+                # Process the response using the utility function
+                response = process_general_response(
+                    response, gen_ai_endpoint, server_port, server_address,
+                    environment, application_name, span, version
+                )
+
+                span.set_status(Status(StatusCode.OK))
+
+            except Exception as e:
+                handle_exception(span, e)
+
+            return response
+
+    return wrapper

openlit/instrumentation/langchain_community/langchain_community.py

@@ -0,0 +1,49 @@
+"""
+Module for monitoring LangChain Community operations.
+"""
+
+import logging
+from opentelemetry import trace
+from opentelemetry.trace import Status, StatusCode
+
+from openlit.__helpers import handle_exception
+from openlit.instrumentation.langchain_community.utils import process_general_response
+
+# Initialize logger for LangChain Community instrumentation
+logger = logging.getLogger(__name__)
+
+def general_wrap(gen_ai_endpoint, version, environment, application_name, tracer, pricing_info,
+    capture_message_content, metrics, disable_metrics):
+    """
+    Generates a telemetry wrapper for GenAI operations.
+    """
+
+    def wrapper(wrapped, instance, args, kwargs):
+        """
+        Wraps the GenAI operation call.
+        """
+
+        # Prepare server address and port
+        server_address = "127.0.0.1"
+        server_port = "80"
+
+        # Get the parent span from the tracer
+        with tracer.start_as_current_span(gen_ai_endpoint, kind=trace.SpanKind.CLIENT) as span:
+            try:
+                # Call the original function
+                response = wrapped(*args, **kwargs)
+
+                # Process the response using the utility function
+                response = process_general_response(
+                    response, gen_ai_endpoint, server_port, server_address,
+                    environment, application_name, span, version
+                )
+
+                span.set_status(Status(StatusCode.OK))
+
+            except Exception as e:
+                handle_exception(span, e)
+
+            return response
+
+    return wrapper

openlit/instrumentation/langchain_community/utils.py

@@ -0,0 +1,69 @@
+"""
+Utility functions for LangChain Community instrumentation.
+"""
+
+from opentelemetry.trace import Status, StatusCode
+from openlit.semcov import SemanticConvention
+
+def process_general_response(response, gen_ai_endpoint, server_port, server_address,
+                            environment, application_name, span, version="1.0.0"):
+    """
+    Process general LangChain Community operations (document loading, text splitting) and generate telemetry.
+    
+    Args:
+        response: The response object from the LangChain Community operation
+        gen_ai_endpoint: The endpoint identifier for the operation
+        server_port: Server port (empty for community operations)
+        server_address: Server address (empty for community operations)
+        environment: Environment name
+        application_name: Application name
+        span: OpenTelemetry span
+        version: Version string
+    
+    Returns:
+        The original response object
+    """
+
+    # Set span attributes for general operations
+    span.set_attribute(SemanticConvention.GEN_AI_SYSTEM, SemanticConvention.GEN_AI_SYSTEM_LANGCHAIN)
+    span.set_attribute(SemanticConvention.GEN_AI_ENDPOINT, gen_ai_endpoint)
+    span.set_attribute(SemanticConvention.GEN_AI_OPERATION, SemanticConvention.GEN_AI_OPERATION_TYPE_FRAMEWORK)
+    span.set_attribute(SemanticConvention.GEN_AI_SDK_VERSION, version)
+    span.set_attribute(SemanticConvention.GEN_AI_ENVIRONMENT, environment)
+    span.set_attribute(SemanticConvention.GEN_AI_APPLICATION_NAME, application_name)
+
+    # Try to extract source information for document loading operations
+    if gen_ai_endpoint and "retrieve.load" in gen_ai_endpoint:
+        try:
+            if hasattr(response, "__iter__") and len(response) > 0:
+                # For document loaders, try to get source from first document
+                first_doc = response[0]
+                if hasattr(first_doc, "metadata") and isinstance(first_doc.metadata, dict):
+                    source = first_doc.metadata.get("source", "unknown")
+                    span.set_attribute(SemanticConvention.GEN_AI_RETRIEVAL_SOURCE, source)
+
+                # Count number of documents loaded
+                span.set_attribute("gen_ai.retrieval.documents.count", len(response))
+        except (AttributeError, KeyError, IndexError, TypeError):
+            # If we cant extract metadata, just continue without it
+            pass
+
+    # For text splitting operations
+    elif gen_ai_endpoint and ("split_documents" in gen_ai_endpoint or "create_documents" in gen_ai_endpoint):
+        try:
+            if hasattr(response, "__iter__") and len(response) > 0:
+                # Count number of text chunks created
+                span.set_attribute("gen_ai.text_splitter.chunks.count", len(response))
+
+                # Try to get average chunk size
+                total_chars = sum(len(doc.page_content) for doc in response if hasattr(doc, "page_content"))
+                if total_chars > 0:
+                    avg_chunk_size = total_chars // len(response)
+                    span.set_attribute("gen_ai.text_splitter.avg_chunk_size", avg_chunk_size)
+        except (AttributeError, TypeError):
+            # If we cant extract chunk information, just continue without it
+            pass
+
+    span.set_status(Status(StatusCode.OK))
+
+    return response

{openlit-1.34.22.dist-info → openlit-1.34.23.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: openlit
-Version: 1.34.22
+Version: 1.34.23
 Summary: OpenTelemetry-native Auto instrumentation library for monitoring LLM Applications and GPUs, facilitating the integration of observability into your GenAI-driven projects
 License: Apache-2.0
 Keywords: OpenTelemetry,otel,otlp,llm,tracing,openai,anthropic,claude,cohere,llm monitoring,observability,monitoring,gpt,Generative AI,chatGPT,gpu

{openlit-1.34.22.dist-info → openlit-1.34.23.dist-info}/RECORD

@@ -1,5 +1,5 @@
 openlit/__helpers.py,sha256=UmW0E97r7UUps2rkdQ-S_lXhaeH3Txp-umyAOon45mk,16458
-openlit/__init__.py,sha256=ris6-GY0ePSbK_jvawHTXymGClVF7yeKdIT95IRBl18,24086
+openlit/__init__.py,sha256=tYsR2OkDOq7Zf3wBOd3kFfosqUKgZxljMm6mZagMUiI,24296
 openlit/evals/__init__.py,sha256=nJe99nuLo1b5rf7pt9U9BCdSDedzbVi2Fj96cgl7msM,380
 openlit/evals/all.py,sha256=oWrue3PotE-rB5WePG3MRYSA-ro6WivkclSHjYlAqGs,7154
 openlit/evals/bias_detection.py,sha256=mCdsfK7x1vX7S3psC3g641IMlZ-7df3h-V6eiICj5N8,8154
@@ -12,8 +12,10 @@ openlit/guard/prompt_injection.py,sha256=3e4DKxB7QDzM-xPCpwEuureiH_2s_OTJ9BSckkn
 openlit/guard/restrict_topic.py,sha256=KTuWa7XeMsV4oXxOrD1CYZV0wXWxTfA0H3p_6q_IOsk,6444
 openlit/guard/sensitive_topic.py,sha256=RgVw_laFERv0nNdzBsAd2_3yLomMOK-gVq-P7oj1bTk,5552
 openlit/guard/utils.py,sha256=6hE3rCRjFXYjKRQYUo8YsqUSlvod48nOWp8MwoQEYdw,7670
-openlit/instrumentation/ag2/__init__.py,sha256=KgyLJBmwAxRWu7Z0S8FDDK4TZ13EFoAAIalvG5Oq4wc,1839
-openlit/instrumentation/ag2/ag2.py,sha256=eNQziyeZl4396GsIp5qI1Dne2KcnQMmhftW7joKQvNU,6934
+openlit/instrumentation/ag2/__init__.py,sha256=vHKx7ybxwtNMGoIyU3lPp8t33m3lWyA_B-1HtJoW0v4,1880
+openlit/instrumentation/ag2/ag2.py,sha256=ebsBWrXf7apjLumTB4hWw19-q9V0zacyDcJ-7oZoxew,3955
+openlit/instrumentation/ag2/async_ag2.py,sha256=5OnM7tos-T5O6mmxhOVI1EO3J47wodC_HSJPWpvmauM,4031
+openlit/instrumentation/ag2/utils.py,sha256=HwOqUQE4HqHLDLGf8nIPc_aiOVoV7iz30piw31F9pys,7073
 openlit/instrumentation/ai21/__init__.py,sha256=tKX643fwxPWPJq1EXEZd0Xpd6B0jl_ViPFmJ87f5B08,2539
 openlit/instrumentation/ai21/ai21.py,sha256=zyQMfCLcOFG1tQWrZmGeMaVAmj8MtCUeXQtPHmlUAO0,6533
 openlit/instrumentation/ai21/async_ai21.py,sha256=q1Dhxru4tUJu0U1Px3PptNqrSGW0-VfRGcqkLKFR8vQ,6659
@@ -76,9 +78,14 @@ openlit/instrumentation/haystack/haystack.py,sha256=kPkuCJDrccNgAg3yDAHbvEytzyfM
 openlit/instrumentation/julep/__init__.py,sha256=g-hwXjvXAb5IDs5DR_P8rKsnD4beB9tupAzuuviQT3k,3216
 openlit/instrumentation/julep/async_julep.py,sha256=fPUOGAOIxelBt4cw-PGp5zj_B1nfj5ly3Dj0kelw3-s,5327
 openlit/instrumentation/julep/julep.py,sha256=6rGgDOB7UzisgYsm12vnTy39dl9HAlRVBVByEr2116g,5330
-openlit/instrumentation/langchain/__init__.py,sha256=cNlumZ8fwLMlGVFMjNEndOIzooD4FQEOINX9tGVksII,3853
-openlit/instrumentation/langchain/async_langchain.py,sha256=rdk3INGcsxsfzZcoJo0yYtc8A0tQbWevF_mzf9IPrqg,18341
-openlit/instrumentation/langchain/langchain.py,sha256=zgfzfOIDsaRoVgWl1T4XX2CLO7ttGOD15TagZtYQ-vE,17012
+openlit/instrumentation/langchain/__init__.py,sha256=idjeMAL8tCf1KimrS82D4RERbicSxBj82e8WNuaZWs8,2996
+openlit/instrumentation/langchain/async_langchain.py,sha256=5RtaBLifJoDYBPL3d53dT2GDmDzOh5oqyZeJIXAmWxg,3426
+openlit/instrumentation/langchain/langchain.py,sha256=6jO5QAZz_jYyauEyQ76nbTpiNrTLPwLNPKzXmlBn75Y,3336
+openlit/instrumentation/langchain/utils.py,sha256=ermEFuOY9Djh4Np4EHeh7XRzZc-B24A_CPLqkJhvzpY,10470
+openlit/instrumentation/langchain_community/__init__.py,sha256=DGNxMj6RAMQtTFD0plU826D3G-KupROwexN4GjmAFmk,2717
+openlit/instrumentation/langchain_community/async_langchain_community.py,sha256=BX6ErjSX9-RXBxB5cFwDrhVKpb3OGzwpzzw5VPMpp80,1590
+openlit/instrumentation/langchain_community/langchain_community.py,sha256=J-sN5eGC7r-OkPAU-lnbdG7-b_jtYs0esmFy51xdFIk,1560
+openlit/instrumentation/langchain_community/utils.py,sha256=v_QWgSbGZm4PIiInz6Ul4dz3h0lrWfLOX1Hj9pDERUI,3334
 openlit/instrumentation/letta/__init__.py,sha256=K8PtRKxuueyqEYE3LzxWJ74IieNKSI6dmk9sNRd8Mt0,3031
 openlit/instrumentation/letta/letta.py,sha256=SCIpJ4tdB1l1BmeQx4raaTS4MQO5X15pLvS4PepEKBE,8481
 openlit/instrumentation/litellm/__init__.py,sha256=D47yfDLLEKpkaRAy7_Yif70kj88AGqLQYZAABpTN4sE,2284
@@ -143,7 +150,7 @@ openlit/otel/events.py,sha256=VrMjTpvnLtYRBHCiFwJojTQqqNpRCxoD4yJYeQrtPsk,3560
 openlit/otel/metrics.py,sha256=GM2PDloBGRhBTkHHkYaqmOwIAQkY124ZhW4sEqW1Fgk,7086
 openlit/otel/tracing.py,sha256=tjV2bEbEDPUB1Z46gE-UsJsb04sRdFrfbhIDkxViZc0,3103
 openlit/semcov/__init__.py,sha256=8oIh2VC667NDh8FA3M-ESusHmeus1sgDUD8binx_nAc,13519
-openlit-1.34.22.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-openlit-1.34.22.dist-info/METADATA,sha256=ADqRel4c0tZ3_W6h7dUSHh93x56YyU8sHYdjwQZlJVg,23470
-openlit-1.34.22.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-openlit-1.34.22.dist-info/RECORD,,
+openlit-1.34.23.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+openlit-1.34.23.dist-info/METADATA,sha256=I28zXVmrGZFQxfl8ayXqVX2YSHYTX0R5upGyteI5tLg,23470
+openlit-1.34.23.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+openlit-1.34.23.dist-info/RECORD,,