phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

fix - debug otel

Browse source
This commit is contained in:
Ishaan Jaff 2024-06-04 09:39:37 -07:00
parent e96d2e3b1b
commit af32ba0154
1 changed files with 11 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

12
litellm/integrations/opentelemetry.py
View file

@ -4,7 +4,6 @@ from dataclasses import dataclass
from litellm.integrations.custom_logger import CustomLogger from litellm.integrations.custom_logger import CustomLogger
LITELLM_TRACER_NAME = "litellm" LITELLM_TRACER_NAME = "litellm"
LITELLM_RESOURCE = {"service.name": "litellm"} LITELLM_RESOURCE = {"service.name": "litellm"}
@ -41,6 +40,17 @@ class OpenTelemetry(CustomLogger):
trace.set_tracer_provider(provider) trace.set_tracer_provider(provider)
self.tracer = trace.get_tracer(LITELLM_TRACER_NAME) self.tracer = trace.get_tracer(LITELLM_TRACER_NAME)
if bool(os.getenv("DEBUG_OTEL", False)) is True:
# Set up logging
import logging
logging.basicConfig(level=logging.DEBUG)
logger = logging.getLogger(__name__)
# Enable OpenTelemetry logging
otel_exporter_logger = logging.getLogger("opentelemetry.sdk.trace.export")
otel_exporter_logger.setLevel(logging.DEBUG)
def log_success_event(self, kwargs, response_obj, start_time, end_time): def log_success_event(self, kwargs, response_obj, start_time, end_time):
self._handle_sucess(kwargs, response_obj, start_time, end_time) self._handle_sucess(kwargs, response_obj, start_time, end_time)