From 6f1c06f7ae36bc386be4dbe3f1babd57fe4dae50 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Fri, 25 Oct 2024 18:21:17 +0400 Subject: [PATCH] fix test audit logs --- tests/local_testing/test_audit_logs_proxy.py | 1 + 1 file changed, 1 insertion(+) diff --git a/tests/local_testing/test_audit_logs_proxy.py b/tests/local_testing/test_audit_logs_proxy.py index 48187e9b2..275d48670 100644 --- a/tests/local_testing/test_audit_logs_proxy.py +++ b/tests/local_testing/test_audit_logs_proxy.py @@ -50,6 +50,7 @@ from starlette.datastructures import URL from litellm.proxy.management_helpers.audit_logs import create_audit_log_for_update from litellm.proxy._types import LiteLLM_AuditLogs, LitellmTableNames from litellm.caching.caching import DualCache +from unittest.mock import patch, AsyncMock proxy_logging_obj = ProxyLogging(user_api_key_cache=DualCache()) import json