openlit 1.33.8__py3-none-any.whl → 1.33.9__py3-none-any.whl

openlit/instrumentation/openai/__init__.py

@@ -5,20 +5,12 @@ import importlib.metadata
 from opentelemetry.instrumentation.instrumentor import BaseInstrumentor
 from wrapt import wrap_function_wrapper
 
-from openlit.instrumentation.openai.openai import chat_completions, embedding, finetune
+from openlit.instrumentation.openai.openai import chat_completions, embedding
 from openlit.instrumentation.openai.openai import image_generate, image_variatons, audio_create
 from openlit.instrumentation.openai.async_openai import async_chat_completions, async_embedding
-from openlit.instrumentation.openai.async_openai import async_finetune
 from openlit.instrumentation.openai.async_openai import async_image_generate, async_image_variatons
 from openlit.instrumentation.openai.async_openai import async_audio_create
 
-from openlit.instrumentation.openai.azure_openai import azure_chat_completions, azure_completions
-from openlit.instrumentation.openai.azure_openai import azure_image_generate, azure_embedding
-from openlit.instrumentation.openai.async_azure_openai import azure_async_chat_completions
-from openlit.instrumentation.openai.async_azure_openai import azure_async_completions
-from openlit.instrumentation.openai.async_azure_openai import azure_async_image_generate
-from openlit.instrumentation.openai.async_azure_openai import azure_async_embedding
-
 _instruments = ("openai >= 1.1.1",)
 
 class OpenAIInstrumentor(BaseInstrumentor):
@@ -40,7 +32,7 @@ class OpenAIInstrumentor(BaseInstrumentor):
         wrap_function_wrapper(
             "openai.resources.chat.completions",  
             "Completions.create",  
-            chat_wrapper(version, environment, application_name,
+            chat_completions(version, environment, application_name,
                          tracer, pricing_info, trace_content,
                          metrics, disable_metrics),
         )
@@ -48,7 +40,7 @@ class OpenAIInstrumentor(BaseInstrumentor):
         wrap_function_wrapper(
             "openai.resources.chat.completions",  
             "AsyncCompletions.create",  
-            async_chat_wrapper(version, environment, application_name,
+            async_chat_completions(version, environment, application_name,
                                tracer, pricing_info, trace_content,
                                metrics, disable_metrics),
         )
@@ -56,7 +48,7 @@ class OpenAIInstrumentor(BaseInstrumentor):
         wrap_function_wrapper(
             "openai.resources.images",  
             "Images.generate",  
-            image_generate_wrapper(version, environment, application_name,
+            image_generate(version, environment, application_name,
                                    tracer, pricing_info, trace_content,
                                    metrics, disable_metrics),
         )
@@ -64,7 +56,7 @@ class OpenAIInstrumentor(BaseInstrumentor):
         wrap_function_wrapper(
             "openai.resources.images",  
             "AsyncImages.generate",  
-            async_image_generate_wrapper(version, environment, application_name,
+            async_image_generate(version, environment, application_name,
                                          tracer, pricing_info, trace_content,
                                          metrics, disable_metrics),
         )
@@ -72,7 +64,7 @@ class OpenAIInstrumentor(BaseInstrumentor):
         wrap_function_wrapper(
             "openai.resources.embeddings",  
             "Embeddings.create",  
-            embedding_wrapper(version, environment, application_name,
+            embedding(version, environment, application_name,
                               tracer, pricing_info, trace_content,
                               metrics, disable_metrics),
         )
@@ -80,24 +72,7 @@ class OpenAIInstrumentor(BaseInstrumentor):
         wrap_function_wrapper(
             "openai.resources.embeddings",  
             "AsyncEmbeddings.create",  
-            async_embedding_wrapper(version, environment, application_name,
-                                    tracer, pricing_info, trace_content,
-                                    metrics, disable_metrics),
-        )
-
-        wrap_function_wrapper(
-            "openai.resources.completions",  
-            "Completions.create",  
-            azure_completions("azure_openai.completions", version, environment, application_name,
-                              tracer, pricing_info, trace_content,
-                              metrics, disable_metrics),
-        )
-
-        wrap_function_wrapper(
-            "openai.resources.completions",  
-            "AsyncCompletions.create",  
-            azure_async_completions("azure_openai.completions", version,
-                                    environment, application_name,
+            async_embedding(version, environment, application_name,
                                     tracer, pricing_info, trace_content,
                                     metrics, disable_metrics),
         )
@@ -105,7 +80,7 @@ class OpenAIInstrumentor(BaseInstrumentor):
         wrap_function_wrapper(
             "openai.resources.images",  
             "Images.create_variation",  
-            image_variatons("openai.images.variations", version,
+            image_variatons(version,
                             environment, application_name,
                             tracer, pricing_info, trace_content,
                             metrics, disable_metrics),
@@ -114,7 +89,7 @@ class OpenAIInstrumentor(BaseInstrumentor):
         wrap_function_wrapper(
             "openai.resources.images",  
             "AsyncImages.create_variation",  
-            async_image_variatons("openai.images.variations", version,
+            async_image_variatons(version,
                                   environment, application_name,
                                   tracer, pricing_info, trace_content,
                                   metrics, disable_metrics),
@@ -123,7 +98,7 @@ class OpenAIInstrumentor(BaseInstrumentor):
         wrap_function_wrapper(
             "openai.resources.audio.speech",  
             "Speech.create",  
-            audio_create("openai.audio.speech", version, environment, application_name,
+            audio_create(version, environment, application_name,
                          tracer, pricing_info, trace_content,
                          metrics, disable_metrics),
         )
@@ -131,205 +106,11 @@ class OpenAIInstrumentor(BaseInstrumentor):
         wrap_function_wrapper(
             "openai.resources.audio.speech",  
             "AsyncSpeech.create",  
-            async_audio_create("openai.audio.speech", version, environment, application_name,
+            async_audio_create(version, environment, application_name,
                                tracer, pricing_info, trace_content,
                                metrics, disable_metrics),
         )
 
-        wrap_function_wrapper(
-            "openai.resources.fine_tuning.jobs",  
-            "Jobs.create",  
-            finetune("openai.audio.speech", version, environment, application_name,
-                     tracer, pricing_info, trace_content,
-                     metrics, disable_metrics),
-        )
-
-        wrap_function_wrapper(
-            "openai.resources.fine_tuning.jobs",  
-            "AsyncJobs.create",  
-            async_finetune("openai.fine_tuning.jo", version, environment, application_name,
-                           tracer, pricing_info, trace_content,
-                           metrics, disable_metrics),
-        )
-
     @staticmethod
     def _uninstrument(self, **kwargs):
         pass
-
-def chat_wrapper(version, environment, application_name, tracer, pricing_info, trace_content,
-                 metrics, disable_metrics):
-    """
-    Decorator for making a custom wrapper execute conditionally,
-    based on whether the instance is for Azure OpenAI or not.
-    """
-    def wrapper(wrapped, instance, args, kwargs):
-        # Default to using the standard OpenAI chat completions
-        completion_func = chat_completions("openai.chat.completions", version, environment,
-                                           application_name, tracer, pricing_info, trace_content,
-                                           metrics, disable_metrics)
-
-        # Check if it's an Azure instance by inspecting `base_url`
-        try:
-            base_url = getattr(instance, 'base_url', '')
-            if 'azure.com' in base_url:
-                # Switch to the Azure-specific chat completions logic
-                completion_func = azure_chat_completions("azure_openai.chat.completions",
-                                                         version, environment, application_name,
-                                                         tracer, pricing_info, trace_content,
-                                                         metrics, disable_metrics)
-        except AttributeError:
-            pass  # base_url attribute not found, proceed with the default
-
-        # Execute the selected completion function
-        return completion_func(wrapped, instance, args, kwargs)
-
-    return wrapper
-
-def async_chat_wrapper(version, environment, application_name, tracer, pricing_info,
-                       trace_content, metrics, disable_metrics):
-    """
-    Decorator for making a custom wrapper execute conditionally,
-    based on whether the instance is for Azure OpenAI or not.
-    """
-    def wrapper(wrapped, instance, args, kwargs):
-        # Default to using the standard OpenAI chat completions
-        completion_func = async_chat_completions("openai.chat.completions", version, environment,
-                                                 application_name, tracer,
-                                                 pricing_info, trace_content,
-                                                 metrics, disable_metrics)
-
-        # Check if it's an Azure instance by inspecting `base_url`
-        try:
-            base_url = getattr(instance, 'base_url', '')
-            if 'azure.com' in base_url:
-                # Switch to the Azure-specific chat completions logic
-                completion_func = azure_async_chat_completions("azure_openai.chat.completions",
-                                                               version, environment,
-                                                               application_name, tracer,
-                                                               pricing_info, trace_content,
-                                                               metrics, disable_metrics)
-        except AttributeError:
-            pass  # base_url attribute not found, proceed with the default
-
-        # Execute the selected completion function
-        return completion_func(wrapped, instance, args, kwargs)
-
-    return wrapper
-
-def image_generate_wrapper(version, environment, application_name, tracer, pricing_info,
-                           trace_content, metrics, disable_metrics):
-    """
-    Decorator for making a custom wrapper execute conditionally,
-    based on whether the instance is for Azure OpenAI or not.
-    """
-    def wrapper(wrapped, instance, args, kwargs):
-        # Default to using the standard OpenAI chat completions
-        completion_func = image_generate("openai.images.generate", version, environment,
-                                         application_name, tracer, pricing_info, trace_content,
-                                         metrics, disable_metrics)
-
-        # Check if it's an Azure instance by inspecting `base_url`
-        try:
-            base_url = getattr(instance, 'base_url', '')
-            if 'azure.com' in base_url:
-                # Switch to the Azure-specific chat completions logic
-                completion_func = azure_image_generate("azure_openai.images.generate",
-                                                       version, environment, application_name,
-                                                       tracer, pricing_info, trace_content,
-                                                       metrics, disable_metrics)
-        except AttributeError:
-            pass  # base_url attribute not found, proceed with the default
-
-        # Execute the selected completion function
-        return completion_func(wrapped, instance, args, kwargs)
-
-    return wrapper
-
-def async_image_generate_wrapper(version, environment, application_name, tracer,
-                                 pricing_info, trace_content, metrics, disable_metrics):
-    """
-    Decorator for making a custom wrapper execute conditionally,
-    based on whether the instance is for Azure OpenAI or not.
-    """
-    def wrapper(wrapped, instance, args, kwargs):
-        # Default to using the standard OpenAI chat completions
-        completion_func = async_image_generate("openai.images.generate", version,
-                                               environment, application_name, tracer,
-                                               pricing_info, trace_content,
-                                               metrics, disable_metrics)
-
-        # Check if it's an Azure instance by inspecting `base_url`
-        try:
-            base_url = getattr(instance, 'base_url', '')
-            if 'azure.com' in base_url:
-                # Switch to the Azure-specific chat completions logic
-                completion_func = azure_async_image_generate("azure_openai.images.generate",
-                                                             version, environment,
-                                                             application_name, tracer,
-                                                             pricing_info, trace_content,
-                                                             metrics, disable_metrics)
-        except AttributeError:
-            pass  # base_url attribute not found, proceed with the default
-
-        # Execute the selected completion function
-        return completion_func(wrapped, instance, args, kwargs)
-
-    return wrapper
-
-def embedding_wrapper(version, environment, application_name, tracer, pricing_info,
-                      trace_content, metrics, disable_metrics):
-    """
-    Decorator for making a custom wrapper execute conditionally,
-    based on whether the instance is for Azure OpenAI or not.
-    """
-    def wrapper(wrapped, instance, args, kwargs):
-        # Default to using the standard OpenAI chat completions
-        completion_func = embedding("openai.embeddings", version, environment,
-                                    application_name, tracer, pricing_info, trace_content,
-                                    metrics, disable_metrics)
-
-        # Check if it's an Azure instance by inspecting `base_url`
-        try:
-            base_url = getattr(instance, 'base_url', '')
-            if 'azure.com' in base_url:
-                # Switch to the Azure-specific chat completions logic
-                completion_func = azure_embedding("azure_openai.embeddings",
-                                                  version, environment, application_name,
-                                                  tracer, pricing_info, trace_content,
-                                                  metrics, disable_metrics)
-        except AttributeError:
-            pass  # base_url attribute not found, proceed with the default
-
-        # Execute the selected completion function
-        return completion_func(wrapped, instance, args, kwargs)
-
-    return wrapper
-
-def async_embedding_wrapper(version, environment, application_name, tracer,
-                            pricing_info, trace_content, metrics, disable_metrics):
-    """
-    Decorator for making a custom wrapper execute conditionally,
-    based on whether the instance is for Azure OpenAI or not.
-    """
-    def wrapper(wrapped, instance, args, kwargs):
-        # Default to using the standard OpenAI chat completions
-        completion_func = async_embedding("openai.embeddings", version, environment,
-                                          application_name, tracer, pricing_info, trace_content,
-                                          metrics, disable_metrics)
-
-        # Check if it's an Azure instance by inspecting `base_url`
-        try:
-            base_url = getattr(instance, 'base_url', '')
-            if 'azure.com' in base_url:
-                # Switch to the Azure-specific chat completions logic
-                completion_func = azure_async_embedding("azure_openai.embeddings", version,
-                                                        environment, application_name, tracer,
-                                                        pricing_info, trace_content,
-                                                        metrics, disable_metrics)
-        except AttributeError:
-            pass  # base_url attribute not found, proceed with the default
-
-        # Execute the selected completion function
-        return completion_func(wrapped, instance, args, kwargs)
-
-    return wrapper