openlit 1.34.4__py3-none-any.whl → 1.34.7__py3-none-any.whl

openlit/instrumentation/google_ai_studio/__init__.py

@@ -1,4 +1,3 @@
-# pylint: disable=useless-return, bad-staticmethod-argument, disable=duplicate-code
 """Initializer of Auto Instrumentation of Google AI Studio Functions"""
 
 from typing import Collection
@@ -25,8 +24,8 @@ class GoogleAIStudioInstrumentor(BaseInstrumentor):
         return _instruments
 
     def _instrument(self, **kwargs):
-        application_name = kwargs.get("application_name", "default_application")
-        environment = kwargs.get("environment", "default_environment")
+        application_name = kwargs.get("application_name", "default")
+        environment = kwargs.get("environment", "default")
         tracer = kwargs.get("tracer")
         metrics = kwargs.get("metrics_dict")
         pricing_info = kwargs.get("pricing_info", {})
@@ -67,5 +66,4 @@ class GoogleAIStudioInstrumentor(BaseInstrumentor):
         )
 
     def _uninstrument(self, **kwargs):
-        # Proper uninstrumentation logic to revert patched methods
         pass

openlit/instrumentation/google_ai_studio/async_google_ai_studio.py

@@ -2,7 +2,6 @@
 Module for monitoring Google AI Studio API calls.
 """
 
-import logging
 import time
 from opentelemetry.trace import SpanKind
 from openlit.__helpers import (
@@ -16,9 +15,6 @@ from openlit.instrumentation.google_ai_studio.utils import (
 )
 from openlit.semcov import SemanticConvention
 
-# Initialize logger for logging potential issues and operations
-logger = logging.getLogger(__name__)
-
 def async_generate(version, environment, application_name,
     tracer, pricing_info, capture_message_content, metrics, disable_metrics):
     """
@@ -61,7 +57,6 @@ def async_generate(version, environment, application_name,
 
             except Exception as e:
                 handle_exception(span, e)
-                logger.error("Error in trace creation: %s", e)
 
             # Return original response
             return response
@@ -144,7 +139,6 @@ def async_generate_stream(version, environment, application_name,
 
                 except Exception as e:
                     handle_exception(self._span, e)
-                    logger.error("Error in trace creation: %s", e)
                 raise
 
     async def wrapper(wrapped, instance, args, kwargs):

openlit/instrumentation/google_ai_studio/google_ai_studio.py

@@ -2,7 +2,6 @@
 Module for monitoring Google AI Studio API calls.
 """
 
-import logging
 import time
 from opentelemetry.trace import SpanKind
 from openlit.__helpers import (
@@ -16,9 +15,6 @@ from openlit.instrumentation.google_ai_studio.utils import (
 )
 from openlit.semcov import SemanticConvention
 
-# Initialize logger for logging potential issues and operations
-logger = logging.getLogger(__name__)
-
 def generate(version, environment, application_name,
     tracer, pricing_info, capture_message_content, metrics, disable_metrics):
     """
@@ -61,7 +57,6 @@ def generate(version, environment, application_name,
 
             except Exception as e:
                 handle_exception(span, e)
-                logger.error("Error in trace creation: %s", e)
 
             # Return original response
             return response
@@ -144,7 +139,6 @@ def generate_stream(version, environment, application_name,
 
                 except Exception as e:
                     handle_exception(self._span, e)
-                    logger.error("Error in trace creation: %s", e)
                 raise
 
     def wrapper(wrapped, instance, args, kwargs):

openlit/instrumentation/google_ai_studio/utils.py

@@ -15,8 +15,7 @@ from openlit.semcov import SemanticConvention
 
 def format_content(messages):
     """
-    Process a list of messages to extract content, categorize them by role,
-    and concatenate all 'content' fields into a single string with role: content format.
+    Process a list of messages to extract content.
     """
 
     formatted_messages = []

openlit/instrumentation/openai/async_openai.py

@@ -733,7 +733,7 @@ def async_chat_completions(version, environment, application_name,
                     formatted_messages = []
                     for message in message_prompt:
                         role = message["role"]
-                        content = message["content"]
+                        content = message.get("content", "")
 
                         if isinstance(content, list):
                             content_str = ", ".join(

openlit/instrumentation/premai/__init__.py

@@ -1,4 +1,3 @@
-# pylint: disable=useless-return, bad-staticmethod-argument, disable=duplicate-code
 """Initializer of Auto Instrumentation of Prem AI Functions"""
 
 from typing import Collection