openlit 1.34.0__py3-none-any.whl → 1.34.1__py3-none-any.whl

openlit/instrumentation/google_ai_studio/async_google_ai_studio.py

@@ -39,26 +39,34 @@ def async_generate(version, environment, application_name,
             start_time = time.time()
             response = await wrapped(*args, **kwargs)
 
-            response = process_chat_response(
-                    instance = instance,
-                    response=response,
-                    request_model=request_model,
-                    pricing_info=pricing_info,
-                    server_port=server_port,
-                    server_address=server_address,
-                    environment=environment,
-                    application_name=application_name,
-                    metrics=metrics,
-                    start_time=start_time,
-                    span=span,
-                    args=args,
-                    kwargs=kwargs,
-                    capture_message_content=capture_message_content,
-                    disable_metrics=disable_metrics,
-                    version=version,
-            )
-
-        return response
+            try:
+                response = process_chat_response(
+                        instance = instance,
+                        response=response,
+                        request_model=request_model,
+                        pricing_info=pricing_info,
+                        server_port=server_port,
+                        server_address=server_address,
+                        environment=environment,
+                        application_name=application_name,
+                        metrics=metrics,
+                        start_time=start_time,
+                        span=span,
+                        args=args,
+                        kwargs=kwargs,
+                        capture_message_content=capture_message_content,
+                        disable_metrics=disable_metrics,
+                        version=version,
+                )
+
+                return response
+
+            except Exception as e:
+                handle_exception(span, e)
+                logger.error("Error in trace creation: %s", e)
+
+                # Return original response
+                return response
 
     return wrapper
 

openlit/instrumentation/google_ai_studio/google_ai_studio.py

@@ -39,26 +39,34 @@ def generate(version, environment, application_name,
             start_time = time.time()
             response = wrapped(*args, **kwargs)
 
-            response = process_chat_response(
-                    instance = instance,
-                    response=response,
-                    request_model=request_model,
-                    pricing_info=pricing_info,
-                    server_port=server_port,
-                    server_address=server_address,
-                    environment=environment,
-                    application_name=application_name,
-                    metrics=metrics,
-                    start_time=start_time,
-                    span=span,
-                    args=args,
-                    kwargs=kwargs,
-                    capture_message_content=capture_message_content,
-                    disable_metrics=disable_metrics,
-                    version=version,
-            )
-
-        return response
+            try:
+                response = process_chat_response(
+                        instance = instance,
+                        response=response,
+                        request_model=request_model,
+                        pricing_info=pricing_info,
+                        server_port=server_port,
+                        server_address=server_address,
+                        environment=environment,
+                        application_name=application_name,
+                        metrics=metrics,
+                        start_time=start_time,
+                        span=span,
+                        args=args,
+                        kwargs=kwargs,
+                        capture_message_content=capture_message_content,
+                        disable_metrics=disable_metrics,
+                        version=version,
+                )
+
+                return response
+
+            except Exception as e:
+                handle_exception(span, e)
+                logger.error("Error in trace creation: %s", e)
+
+                # Return original response
+                return response
 
     return wrapper
 

{openlit-1.34.0.dist-info → openlit-1.34.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: openlit
-Version: 1.34.0
+Version: 1.34.1
 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.0.dist-info → openlit-1.34.1.dist-info}/RECORD

@@ -57,8 +57,8 @@ openlit/instrumentation/embedchain/embedchain.py,sha256=f4hyOr1Xr0RC4PNHRu46aV-j
 openlit/instrumentation/firecrawl/__init__.py,sha256=kyVsAiDBC2djifqT2w1cPRAotiEyEabNvnBeSQxi9N8,1876
 openlit/instrumentation/firecrawl/firecrawl.py,sha256=4X38UrLYeGm3uez-edYA6qEc0nKC3p77yfKgKBBud0A,3826
 openlit/instrumentation/google_ai_studio/__init__.py,sha256=d4aDvCSfDtT2geRbwG5yinu62uPTHaj4PtalimuvG-k,2685
-openlit/instrumentation/google_ai_studio/async_google_ai_studio.py,sha256=kv8X1c3klmFEPTyxj9kJag2D670_YZxxkBhsIrW7JH4,5501
-openlit/instrumentation/google_ai_studio/google_ai_studio.py,sha256=MJQjHF1o6DuXHr1xE2-hp_cgejV65-LpXIGL6bacE4Q,5403
+openlit/instrumentation/google_ai_studio/async_google_ai_studio.py,sha256=eOUrIMsUh2dG8BqMB7oizuQBbntIb4sy2WkYrF3Dzj0,5815
+openlit/instrumentation/google_ai_studio/google_ai_studio.py,sha256=fvwZcWlfjT5eXPmeYA5JMSyPfHfLZKBnzBd0iHQxEec,5717
 openlit/instrumentation/google_ai_studio/utils.py,sha256=s9-5qnIWfXsq2xjYbWToI0RBAUo5lT0yLPOILc6eY1k,11256
 openlit/instrumentation/gpt4all/__init__.py,sha256=cO8mi3hhPDXcNwb9AwQ3-wQ_ydnOeBRwb0cptlQmAM4,1805
 openlit/instrumentation/gpt4all/gpt4all.py,sha256=EYp0njZ1kF56rTAjYZVtufA5W4xTWGzSIntjJ4MEfl4,24185
@@ -130,7 +130,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=ptyo37PY-FHDx_PShEvbdns71cD4YvvXw15bCRXKCKM,13461
-openlit-1.34.0.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-openlit-1.34.0.dist-info/METADATA,sha256=RCozi6Py2qG7G6-x9NlC7T9VYy74W1Ochq2rFKl4PgE,23469
-openlit-1.34.0.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-openlit-1.34.0.dist-info/RECORD,,
+openlit-1.34.1.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+openlit-1.34.1.dist-info/METADATA,sha256=c3xokUNDxGsFVnc_XqD9q4_aodyLpYGPYBA6uXNd3do,23469
+openlit-1.34.1.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+openlit-1.34.1.dist-info/RECORD,,