From 390a6021f9a3d59e4f923993c6a63da36bc8cc28 Mon Sep 17 00:00:00 2001
From: Alex Sze <alexyysze@gmail.com>
Date: Mon, 1 Jun 2026 15:56:14 +0800
Subject: [PATCH] fix test

---
 tests/test_litellm/proxy/management_endpoints/test_ui_sso.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/tests/test_litellm/proxy/management_endpoints/test_ui_sso.py b/tests/test_litellm/proxy/management_endpoints/test_ui_sso.py
index 3aaaebe0fb6..f7bf507c6da 100644
--- a/tests/test_litellm/proxy/management_endpoints/test_ui_sso.py
+++ b/tests/test_litellm/proxy/management_endpoints/test_ui_sso.py
@@ -2115,6 +2115,7 @@ class TestCLIKeyRegenerationFlow:
 
         # Mock request (no query params needed - existing_key is in state)
         mock_request = MagicMock(spec=Request)
+        mock_request.query_params = {}
 
         # CLI state with existing_key embedded: {PREFIX}:{key}:{existing_key}
         cli_state = f"{LITELLM_CLI_SESSION_TOKEN_PREFIX}:sk-new-session-key-456:sk-existing-cli-key-123"