phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-26 11:14:04 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

use callback_settings when intializing otel

Browse source
This commit is contained in:
Ishaan Jaff 2024-09-09 16:05:48 -07:00
parent 0f0a5c1758
commit e07b2ce6ea
4 changed files with 51 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

8
litellm/proxy/common_utils/callback_utils.py
View file

@ -17,7 +17,7 @@ def initialize_callbacks_on_proxy(
litellm_settings: dict,
callback_specific_params: dict = {},
):
from litellm.proxy.proxy_server import prisma_client
from litellm.proxy.proxy_server import callback_settings, prisma_client
verbose_proxy_logger.debug(
f"{blue_color_code}initializing callbacks={value} on proxy{reset_color_code}"
@ -34,7 +34,11 @@ def initialize_callbacks_on_proxy(
from litellm.integrations.opentelemetry import OpenTelemetry
from litellm.proxy import proxy_server
open_telemetry_logger = OpenTelemetry()
_otel_settings = {}
if "otel" in callback_settings:
_otel_settings = callback_settings["otel"]
open_telemetry_logger = OpenTelemetry(**_otel_settings)
# Add Otel as a service callback
if "otel" not in litellm.service_callback: