openlit 1.34.20__tar.gz → 1.34.22__tar.gz

{openlit-1.34.20 → openlit-1.34.22}/PKG-INFO

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

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "openlit"
-version = "1.34.20"
+version = "1.34.22"
 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.20 → openlit-1.34.22}/src/openlit/__helpers.py

@@ -402,3 +402,43 @@ def record_embedding_metrics(metrics, gen_ai_operation, gen_ai_system, server_ad
     metrics["genai_requests"].add(1, attributes)
     metrics["genai_prompt_tokens"].add(input_tokens, attributes)
     metrics["genai_cost"].record(cost, attributes)
+
+def record_audio_metrics(metrics, gen_ai_operation, gen_ai_system, server_address, server_port,
+    request_model, response_model, environment, application_name, start_time, end_time, cost):
+    """
+    Record audio-specific metrics for the operation.
+    """
+
+    attributes = create_metrics_attributes(
+        operation=gen_ai_operation,
+        system=gen_ai_system,
+        server_address=server_address,
+        server_port=server_port,
+        request_model=request_model,
+        response_model=response_model,
+        service_name=application_name,
+        deployment_environment=environment,
+    )
+    metrics["genai_client_operation_duration"].record(end_time - start_time, attributes)
+    metrics["genai_requests"].add(1, attributes)
+    metrics["genai_cost"].record(cost, attributes)
+
+def record_image_metrics(metrics, gen_ai_operation, gen_ai_system, server_address, server_port,
+    request_model, response_model, environment, application_name, start_time, end_time, cost):
+    """
+    Record image-specific metrics for the operation.
+    """
+
+    attributes = create_metrics_attributes(
+        operation=gen_ai_operation,
+        system=gen_ai_system,
+        server_address=server_address,
+        server_port=server_port,
+        request_model=request_model,
+        response_model=response_model,
+        service_name=application_name,
+        deployment_environment=environment,
+    )
+    metrics["genai_client_operation_duration"].record(end_time - start_time, attributes)
+    metrics["genai_requests"].add(1, attributes)
+    metrics["genai_cost"].record(cost, attributes)

{openlit-1.34.20 → openlit-1.34.22}/src/openlit/instrumentation/openai/__init__.py

@@ -1,147 +1,142 @@
-# pylint: disable=useless-return, bad-staticmethod-argument, disable=duplicate-code
 """Initializer of Auto Instrumentation of OpenAI Functions"""
 from typing import Collection
 import importlib.metadata
 from opentelemetry.instrumentation.instrumentor import BaseInstrumentor
 from wrapt import wrap_function_wrapper
 
-from openlit.instrumentation.openai.openai import chat_completions, embedding, responses, chat_completions_parse
-from openlit.instrumentation.openai.openai import image_generate, image_variatons, audio_create
-from openlit.instrumentation.openai.async_openai import (async_chat_completions, async_embedding,
-                                                         async_chat_completions_parse)
-from openlit.instrumentation.openai.async_openai import async_image_generate, async_image_variatons
-from openlit.instrumentation.openai.async_openai import async_audio_create, async_responses
+from openlit.instrumentation.openai.openai import (
+    chat_completions, embedding, responses, chat_completions_parse,
+    image_generate, image_variatons, audio_create
+)
+from openlit.instrumentation.openai.async_openai import (
+    async_chat_completions, async_embedding, async_chat_completions_parse,
+    async_image_generate, async_image_variations, async_audio_create, async_responses
+)
 
 _instruments = ("openai >= 1.92.0",)
 
 class OpenAIInstrumentor(BaseInstrumentor):
-    """An instrumentor for OpenAI's client library."""
+    """
+    An instrumentor for OpenAI client library.
+    """
 
     def instrumentation_dependencies(self) -> Collection[str]:
         return _instruments
 
     def _instrument(self, **kwargs):
-        application_name = kwargs.get("application_name")
-        environment = kwargs.get("environment")
+        version = importlib.metadata.version("openai")
+        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")
-        pricing_info = kwargs.get("pricing_info")
-        capture_message_content = kwargs.get("capture_message_content")
         disable_metrics = kwargs.get("disable_metrics")
-        version = importlib.metadata.version("openai")
 
+        # chat completions
         wrap_function_wrapper(
             "openai.resources.chat.completions",  
             "Completions.create",  
             chat_completions(version, environment, application_name,
-                         tracer, pricing_info, capture_message_content,
-                         metrics, disable_metrics),
+                tracer, pricing_info, capture_message_content, metrics, disable_metrics),
         )
 
         wrap_function_wrapper(
             "openai.resources.chat.completions",  
             "AsyncCompletions.create",  
             async_chat_completions(version, environment, application_name,
-                               tracer, pricing_info, capture_message_content,
-                               metrics, disable_metrics),
+                tracer, pricing_info, capture_message_content, metrics, disable_metrics),
         )
 
+        # chat completions parse
         wrap_function_wrapper(
-            "openai.resources.responses.responses",  
-            "Responses.create",  
-            responses(version, environment, application_name,
-                         tracer, pricing_info, capture_message_content,
-                         metrics, disable_metrics),
+            "openai.resources.chat.completions",
+            "Completions.parse",
+            chat_completions_parse(version, environment, application_name,
+                tracer, pricing_info, capture_message_content, metrics, disable_metrics),
         )
 
         wrap_function_wrapper(
-            "openai.resources.responses.responses",  
-            "AsyncResponses.create",  
-            async_responses(version, environment, application_name,
-                         tracer, pricing_info, capture_message_content,
-                         metrics, disable_metrics),
+            "openai.resources.chat.completions",
+            "AsyncCompletions.parse",
+            async_chat_completions_parse(version, environment, application_name,
+                tracer, pricing_info, capture_message_content, metrics, disable_metrics),
         )
 
+        # responses
         wrap_function_wrapper(
-            "openai.resources.images",  
-            "Images.generate",  
-            image_generate(version, environment, application_name,
-                                   tracer, pricing_info, capture_message_content,
-                                   metrics, disable_metrics),
+            "openai.resources.responses.responses",  
+            "Responses.create",  
+            responses(version, environment, application_name,
+                tracer, pricing_info, capture_message_content, metrics, disable_metrics),
         )
 
         wrap_function_wrapper(
-            "openai.resources.images",  
-            "AsyncImages.generate",  
-            async_image_generate(version, environment, application_name,
-                                         tracer, pricing_info, capture_message_content,
-                                         metrics, disable_metrics),
+            "openai.resources.responses.responses",  
+            "AsyncResponses.create",  
+            async_responses(version, environment, application_name,
+                tracer, pricing_info, capture_message_content, metrics, disable_metrics),
         )
 
+        # embeddings
         wrap_function_wrapper(
             "openai.resources.embeddings",  
             "Embeddings.create",  
             embedding(version, environment, application_name,
-                              tracer, pricing_info, capture_message_content,
-                              metrics, disable_metrics),
+                tracer, pricing_info, capture_message_content, metrics, disable_metrics),
         )
 
         wrap_function_wrapper(
             "openai.resources.embeddings",  
             "AsyncEmbeddings.create",  
             async_embedding(version, environment, application_name,
-                                    tracer, pricing_info, capture_message_content,
-                                    metrics, disable_metrics),
+                tracer, pricing_info, capture_message_content, metrics, disable_metrics),
+        )
+
+        # image generation
+        wrap_function_wrapper(
+            "openai.resources.images",  
+            "Images.generate",  
+            image_generate(version, environment, application_name,
+                tracer, pricing_info, capture_message_content, metrics, disable_metrics),
         )
 
+        wrap_function_wrapper(
+            "openai.resources.images",  
+            "AsyncImages.generate",  
+            async_image_generate(version, environment, application_name,
+                tracer, pricing_info, capture_message_content, metrics, disable_metrics),
+        )
+
+        # image variations
         wrap_function_wrapper(
             "openai.resources.images",  
             "Images.create_variation",  
-            image_variatons(version,
-                            environment, application_name,
-                            tracer, pricing_info, capture_message_content,
-                            metrics, disable_metrics),
+            image_variatons(version, environment, application_name,
+                tracer, pricing_info, capture_message_content, metrics, disable_metrics),
         )
 
         wrap_function_wrapper(
             "openai.resources.images",  
             "AsyncImages.create_variation",  
-            async_image_variatons(version,
-                                  environment, application_name,
-                                  tracer, pricing_info, capture_message_content,
-                                  metrics, disable_metrics),
+            async_image_variations(version, environment, application_name,
+                tracer, pricing_info, capture_message_content, metrics, disable_metrics),
         )
 
+        # audio generation
         wrap_function_wrapper(
             "openai.resources.audio.speech",  
             "Speech.create",  
             audio_create(version, environment, application_name,
-                         tracer, pricing_info, capture_message_content,
-                         metrics, disable_metrics),
+                tracer, pricing_info, capture_message_content, metrics, disable_metrics),
         )
 
         wrap_function_wrapper(
             "openai.resources.audio.speech",  
             "AsyncSpeech.create",  
             async_audio_create(version, environment, application_name,
-                               tracer, pricing_info, capture_message_content,
-                               metrics, disable_metrics),
-        )
-
-        wrap_function_wrapper(
-            "openai.resources.chat.completions",
-            "Completions.parse",
-            chat_completions_parse(version, environment, application_name, tracer, pricing_info,
-                                   capture_message_content, metrics, disable_metrics),
-        )
-
-        wrap_function_wrapper(
-            "openai.resources.chat.completions",
-            "AsyncCompletions.parse",
-            async_chat_completions_parse(version, environment, application_name, tracer, pricing_info,
-                                         capture_message_content, metrics, disable_metrics),
+                tracer, pricing_info, capture_message_content, metrics, disable_metrics),
         )
 
-    @staticmethod
     def _uninstrument(self, **kwargs):
         pass