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

openlit/instrumentation/gpu/__init__.py

@@ -5,7 +5,7 @@ from typing import Collection, Iterable
 import logging
 from functools import partial
 from opentelemetry.instrumentation.instrumentor import BaseInstrumentor
-from opentelemetry.sdk.resources import TELEMETRY_SDK_NAME
+from opentelemetry.sdk.resources import SERVICE_NAME, TELEMETRY_SDK_NAME, DEPLOYMENT_ENVIRONMENT
 from opentelemetry.metrics import get_meter, CallbackOptions, Observation
 from openlit.semcov import SemanticConvetion
 
@@ -139,8 +139,8 @@ class GPUInstrumentor(BaseInstrumentor):
 
                 attributes = {
                     TELEMETRY_SDK_NAME: "openlit",
-                    SemanticConvetion.GEN_AI_APPLICATION_NAME: application_name,
-                    SemanticConvetion.GEN_AI_ENVIRONMENT: environment,
+                    SERVICE_NAME: application_name,
+                    DEPLOYMENT_ENVIRONMENT: environment,
                     SemanticConvetion.GPU_INDEX: str(gpu_index),
                     SemanticConvetion.GPU_UUID: safe_decode(pynvml.nvmlDeviceGetUUID(handle)),
                     SemanticConvetion.GPU_NAME: safe_decode(pynvml.nvmlDeviceGetName(handle))
@@ -199,8 +199,8 @@ class GPUInstrumentor(BaseInstrumentor):
 
                 attributes = {
                     TELEMETRY_SDK_NAME: "openlit",
-                    SemanticConvetion.GEN_AI_APPLICATION_NAME: application_name,
-                    SemanticConvetion.GEN_AI_ENVIRONMENT: environment,
+                    SERVICE_NAME: application_name,
+                    DEPLOYMENT_ENVIRONMENT: environment,
                     # pylint: disable=line-too-long
                     SemanticConvetion.GPU_INDEX: amdsmi.amdsmi_get_xgmi_info(device_handle)['index'],
                     # pylint: disable=line-too-long

openlit/instrumentation/groq/__init__.py

@@ -25,7 +25,7 @@ class GroqInstrumentor(BaseInstrumentor):
         tracer = kwargs.get("tracer")
         metrics = kwargs.get("metrics_dict")
         pricing_info = kwargs.get("pricing_info", {})
-        trace_content = kwargs.get("trace_content", False)
+        capture_message_content = kwargs.get("capture_message_content", False)
         disable_metrics = kwargs.get("disable_metrics")
         version = importlib.metadata.version("groq")
 
@@ -33,16 +33,16 @@ class GroqInstrumentor(BaseInstrumentor):
         wrap_function_wrapper(
             "groq.resources.chat.completions",  
             "Completions.create",  
-            chat("groq.chat.completions", version, environment, application_name,
-                     tracer, pricing_info, trace_content, metrics, disable_metrics),
+            chat(version, environment, application_name,
+                     tracer, pricing_info, capture_message_content, metrics, disable_metrics),
         )
 
         #async
         wrap_function_wrapper(
             "groq.resources.chat.completions",  
             "AsyncCompletions.create",  
-            async_chat("groq.chat.completions", version, environment, application_name,
-                            tracer, pricing_info, trace_content, metrics, disable_metrics),
+            async_chat(version, environment, application_name,
+                            tracer, pricing_info, capture_message_content, metrics, disable_metrics),
         )
 
     def _uninstrument(self, **kwargs):