From e04c7dc8f1b8ee402d72329fcfff5563b648250e Mon Sep 17 00:00:00 2001 From: kerry Date: Fri, 6 Nov 2026 05:36:49 +0000 Subject: [PATCH] test: resolve current proxy logger in DB metrics span fixture Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com> --- tests/unit/proxy/db/test_log_db_metrics_service_spans.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/tests/unit/proxy/db/test_log_db_metrics_service_spans.py b/tests/unit/proxy/db/test_log_db_metrics_service_spans.py index a7fd830b948..fd10d301fc9 100644 --- a/tests/unit/proxy/db/test_log_db_metrics_service_spans.py +++ b/tests/unit/proxy/db/test_log_db_metrics_service_spans.py @@ -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)