scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-11-14 09:03:16 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 214

test: resolve current proxy logger in DB metrics span fixture

Browse source 
Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
This commit is contained in:
kerry 2026-11-06 05:36:49 +00:00
parent 9ffb82e139
commit e04c7dc8f1
1 changed files with 6 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
tests/unit/proxy/db/test_log_db_metrics_service_spans.py
View file

@ -23,7 +23,7 @@ from litellm.integrations.datadog.datadog import DataDogLogger
from litellm.integrations.opentelemetry import OpenTelemetry, OpenTelemetryConfig
from litellm.proxy.db.log_db_metrics import log_db_metrics
from litellm.proxy.db.prisma_client import _PrismaDrainTracker, _TrackedPrismaEngine
from litellm.proxy.proxy_server import proxy_logging_obj
from litellm.proxy import proxy_server
from litellm.integrations.prometheus_services import PrometheusServicesLogger
from prometheus_client import REGISTRY
@ -80,9 +80,12 @@ def rig(monkeypatch: pytest.MonkeyPatch) -> _Rig:
with patch("asyncio.create_task", side_effect=_discard_periodic_flush):
datadog: Final = DataDogLogger()
monkeypatch.setattr(litellm, "service_callback", [otel, datadog, "prometheus_system"])
monkeypatch.setattr(proxy_logging_obj.service_logging_obj, "dd_logger", datadog, raising=False)
monkeypatch.setattr(proxy_server.proxy_logging_obj.service_logging_obj, "dd_logger", datadog, raising=False)
monkeypatch.setattr(
proxy_logging_obj.service_logging_obj, "prometheusServicesLogger", PrometheusServicesLogger(), raising=False
proxy_server.proxy_logging_obj.service_logging_obj,
"prometheusServicesLogger",
PrometheusServicesLogger(),
raising=False,
)
return _Rig(exporter, provider, datadog)