openlit 1.34.32__tar.gz → 1.34.33__tar.gz

{openlit-1.34.32 → openlit-1.34.33}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: openlit
-Version: 1.34.32
+Version: 1.34.33
 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.32 → openlit-1.34.33}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "openlit"
-version = "1.34.32"
+version = "1.34.33"
 description = "OpenTelemetry-native Auto instrumentation library for monitoring LLM Applications and GPUs, facilitating the integration of observability into your GenAI-driven projects"
 authors = ["OpenLIT"]
 license = "Apache-2.0"

openlit-1.34.33/src/openlit/instrumentation/pydantic_ai/__init__.py

@@ -0,0 +1,170 @@
+"""Initializer of Auto Instrumentation of Pydantic AI Functions"""
+
+from typing import Collection
+import importlib.metadata
+from opentelemetry.instrumentation.instrumentor import BaseInstrumentor
+from wrapt import wrap_function_wrapper
+
+from openlit.instrumentation.pydantic_ai.pydantic_ai import (
+    agent_create,
+    agent_run,
+    graph_execution,
+    user_prompt_processing,
+    model_request_processing,
+    tool_calls_processing,
+)
+from openlit.instrumentation.pydantic_ai.async_pydantic_ai import (
+    async_agent_run,
+)
+
+_instruments = ("pydantic-ai >= 0.2.17",)
+
+
+class PydanticAIInstrumentor(BaseInstrumentor):
+    """
+    An instrumentor for Pydantic AI's client library.
+    """
+
+    def instrumentation_dependencies(self) -> Collection[str]:
+        return _instruments
+
+    def _instrument(self, **kwargs):
+        application_name = kwargs.get("application_name", "default_application")
+        environment = kwargs.get("environment", "default_environment")
+        tracer = kwargs.get("tracer")
+        metrics = kwargs.get("metrics_dict")
+        pricing_info = kwargs.get("pricing_info", {})
+        capture_message_content = kwargs.get("capture_message_content", False)
+        disable_metrics = kwargs.get("disable_metrics")
+        version = importlib.metadata.version("pydantic-ai")
+
+        wrap_function_wrapper(
+            "pydantic_ai.agent",
+            "Agent.__init__",
+            agent_create(
+                version,
+                environment,
+                application_name,
+                tracer,
+                pricing_info,
+                capture_message_content,
+                metrics,
+                disable_metrics,
+            ),
+        )
+
+        wrap_function_wrapper(
+            "pydantic_ai.agent",
+            "Agent.run_sync",
+            agent_run(
+                version,
+                environment,
+                application_name,
+                tracer,
+                pricing_info,
+                capture_message_content,
+                metrics,
+                disable_metrics,
+            ),
+        )
+
+        wrap_function_wrapper(
+            "pydantic_ai.agent",
+            "Agent.run",
+            async_agent_run(
+                version,
+                environment,
+                application_name,
+                tracer,
+                pricing_info,
+                capture_message_content,
+                metrics,
+                disable_metrics,
+            ),
+        )
+
+        # Enhanced instrumentation for richer span hierarchy
+        # These wrap internal Pydantic AI graph execution components
+        try:
+            # Agent.iter() - Graph execution iterator
+            wrap_function_wrapper(
+                "pydantic_ai.agent",
+                "Agent.iter",
+                graph_execution(
+                    version,
+                    environment,
+                    application_name,
+                    tracer,
+                    pricing_info,
+                    capture_message_content,
+                    metrics,
+                    disable_metrics,
+                ),
+            )
+        except Exception:
+            # If Agent.iter doesn't exist, skip this instrumentation
+            pass
+
+        try:
+            # UserPromptNode.run() - User prompt processing
+            wrap_function_wrapper(
+                "pydantic_ai._agent_graph",
+                "UserPromptNode.run",
+                user_prompt_processing(
+                    version,
+                    environment,
+                    application_name,
+                    tracer,
+                    pricing_info,
+                    capture_message_content,
+                    metrics,
+                    disable_metrics,
+                ),
+            )
+        except Exception:
+            # If UserPromptNode.run doesn't exist, skip this instrumentation
+            pass
+
+        try:
+            # ModelRequestNode.run() - Model request processing
+            wrap_function_wrapper(
+                "pydantic_ai._agent_graph",
+                "ModelRequestNode.run",
+                model_request_processing(
+                    version,
+                    environment,
+                    application_name,
+                    tracer,
+                    pricing_info,
+                    capture_message_content,
+                    metrics,
+                    disable_metrics,
+                ),
+            )
+        except Exception:
+            # If ModelRequestNode.run doesn't exist, skip this instrumentation
+            pass
+
+        try:
+            # CallToolsNode.run() - Tool calls processing
+            wrap_function_wrapper(
+                "pydantic_ai._agent_graph",
+                "CallToolsNode.run",
+                tool_calls_processing(
+                    version,
+                    environment,
+                    application_name,
+                    tracer,
+                    pricing_info,
+                    capture_message_content,
+                    metrics,
+                    disable_metrics,
+                ),
+            )
+        except Exception:
+            # If CallToolsNode.run doesn't exist, skip this instrumentation
+            pass
+
+    def _uninstrument(self, **kwargs):
+        # Proper uninstrumentation logic to revert patched methods
+        pass

openlit-1.34.33/src/openlit/instrumentation/pydantic_ai/async_pydantic_ai.py

@@ -0,0 +1,38 @@
+"""
+Module for monitoring async Pydantic AI API calls.
+"""
+
+from openlit.instrumentation.pydantic_ai.utils import (
+    common_agent_run_async,
+)
+
+
+def async_agent_run(
+    version,
+    environment,
+    application_name,
+    tracer,
+    pricing_info,
+    capture_message_content,
+    metrics,
+    disable_metrics,
+):
+    """
+    Generates a telemetry wrapper for async GenAI function call
+    """
+
+    async def wrapper(wrapped, instance, args, kwargs):
+        return await common_agent_run_async(
+            wrapped,
+            instance,
+            args,
+            kwargs,
+            tracer,
+            version,
+            environment,
+            application_name,
+            capture_message_content,
+            pricing_info=pricing_info,
+        )
+
+    return wrapper

openlit-1.34.33/src/openlit/instrumentation/pydantic_ai/pydantic_ai.py

@@ -0,0 +1,193 @@
+"""
+Module for monitoring Pydantic AI API calls.
+"""
+
+from openlit.instrumentation.pydantic_ai.utils import (
+    common_agent_run,
+    common_agent_create,
+    common_graph_execution,
+    common_user_prompt_processing,
+    common_model_request_processing,
+    common_tool_calls_processing,
+)
+
+
+def agent_create(
+    version,
+    environment,
+    application_name,
+    tracer,
+    pricing_info,
+    capture_message_content,
+    metrics,
+    disable_metrics,
+):
+    """
+    Generates a telemetry wrapper for GenAI function call
+    """
+
+    def wrapper(wrapped, instance, args, kwargs):
+        return common_agent_create(
+            wrapped,
+            instance,
+            args,
+            kwargs,
+            tracer,
+            version,
+            environment,
+            application_name,
+            capture_message_content,
+        )
+
+    return wrapper
+
+
+def agent_run(
+    version,
+    environment,
+    application_name,
+    tracer,
+    pricing_info,
+    capture_message_content,
+    metrics,
+    disable_metrics,
+):
+    """
+    Generates a telemetry wrapper for GenAI function call
+    """
+
+    def wrapper(wrapped, instance, args, kwargs):
+        return common_agent_run(
+            wrapped,
+            instance,
+            args,
+            kwargs,
+            tracer,
+            version,
+            environment,
+            application_name,
+            capture_message_content,
+            pricing_info=pricing_info,
+        )
+
+    return wrapper
+
+
+def graph_execution(
+    version,
+    environment,
+    application_name,
+    tracer,
+    pricing_info,
+    capture_message_content,
+    metrics,
+    disable_metrics,
+):
+    """
+    Generates a telemetry wrapper for Pydantic AI graph execution
+    """
+
+    def wrapper(wrapped, instance, args, kwargs):
+        return common_graph_execution(
+            wrapped,
+            instance,
+            args,
+            kwargs,
+            tracer,
+            version,
+            environment,
+            application_name,
+            capture_message_content,
+        )
+
+    return wrapper
+
+
+def user_prompt_processing(
+    version,
+    environment,
+    application_name,
+    tracer,
+    pricing_info,
+    capture_message_content,
+    metrics,
+    disable_metrics,
+):
+    """
+    Generates a telemetry wrapper for Pydantic AI user prompt processing
+    """
+
+    def wrapper(wrapped, instance, args, kwargs):
+        return common_user_prompt_processing(
+            wrapped,
+            instance,
+            args,
+            kwargs,
+            tracer,
+            version,
+            environment,
+            application_name,
+            capture_message_content,
+        )
+
+    return wrapper
+
+
+def model_request_processing(
+    version,
+    environment,
+    application_name,
+    tracer,
+    pricing_info,
+    capture_message_content,
+    metrics,
+    disable_metrics,
+):
+    """
+    Generates a telemetry wrapper for Pydantic AI model request processing
+    """
+
+    def wrapper(wrapped, instance, args, kwargs):
+        return common_model_request_processing(
+            wrapped,
+            instance,
+            args,
+            kwargs,
+            tracer,
+            version,
+            environment,
+            application_name,
+            capture_message_content,
+        )
+
+    return wrapper
+
+
+def tool_calls_processing(
+    version,
+    environment,
+    application_name,
+    tracer,
+    pricing_info,
+    capture_message_content,
+    metrics,
+    disable_metrics,
+):
+    """
+    Generates a telemetry wrapper for Pydantic AI tool calls processing
+    """
+
+    def wrapper(wrapped, instance, args, kwargs):
+        return common_tool_calls_processing(
+            wrapped,
+            instance,
+            args,
+            kwargs,
+            tracer,
+            version,
+            environment,
+            application_name,
+            capture_message_content,
+        )
+
+    return wrapper