diff --git a/tests/llm_translation/base_llm_unit_tests.py b/tests/llm_translation/base_llm_unit_tests.py index 230781c636..6a2cacd20a 100644 --- a/tests/llm_translation/base_llm_unit_tests.py +++ b/tests/llm_translation/base_llm_unit_tests.py @@ -947,6 +947,8 @@ class BaseLLMChatTest(ABC): second_response.choices[0].message.content is not None or second_response.choices[0].message.tool_calls is not None ) + except litellm.ServiceUnavailableError: + pytest.skip("Model is overloaded") except litellm.InternalServerError: pytest.skip("Model is overloaded") except litellm.RateLimitError: diff --git a/ui/litellm-dashboard/src/app/page.tsx b/ui/litellm-dashboard/src/app/page.tsx index 1fea83d054..2a3b7422ce 100644 --- a/ui/litellm-dashboard/src/app/page.tsx +++ b/ui/litellm-dashboard/src/app/page.tsx @@ -151,7 +151,7 @@ export default function CreateKeyPage() { if (redirectToLogin) { window.location.href = (proxyBaseUrl || "") + "/sso/key/generate" } - }, [token, authLoading]) + }, [redirectToLogin]) useEffect(() => { if (!token) { @@ -223,7 +223,7 @@ export default function CreateKeyPage() { } }, [accessToken, userID, userRole]); - if (authLoading) { + if (authLoading || redirectToLogin) { return }