From 750757f284aa49cb2473d4652ccc0bdd683a7428 Mon Sep 17 00:00:00 2001 From: jayden Date: Tue, 31 Mar 2026 01:55:01 +0000 Subject: [PATCH] chore: fix test --- tests/proxy_unit_tests/test_proxy_server.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/tests/proxy_unit_tests/test_proxy_server.py b/tests/proxy_unit_tests/test_proxy_server.py index d864a0d986a..23826a9f910 100644 --- a/tests/proxy_unit_tests/test_proxy_server.py +++ b/tests/proxy_unit_tests/test_proxy_server.py @@ -2673,7 +2673,9 @@ async def test_update_config_success_callback_normalization(): # Update config with mixed-case callbacks - expect normalization to lowercase config_update = ConfigYAML(litellm_settings={"success_callback": ["SQS", "sQs"]}) - await proxy_server.update_config(config_update) + from litellm.proxy._types import LitellmUserRoles, UserAPIKeyAuth + admin_user = UserAPIKeyAuth(user_role=LitellmUserRoles.PROXY_ADMIN, api_key="sk-test") + await proxy_server.update_config(config_update, user_api_key_dict=admin_user) saved = mock_proxy_config.saved_config assert saved is not None, "save_config was not called"