fix(router.py): fix check for context window fallbacks

fallback if list is not none
This commit is contained in:
Krrish Dholakia 2024-04-01 10:41:12 -07:00
parent c9e6b05cfb
commit f46a9d09a5

View file

@ -1241,7 +1241,7 @@ class Router:
### CHECK IF RATE LIMIT / CONTEXT WINDOW ERROR w/ fallbacks available / Bad Request Error
if (
isinstance(original_exception, litellm.ContextWindowExceededError)
and context_window_fallbacks is None
and context_window_fallbacks is not None
) or (
isinstance(original_exception, openai.RateLimitError)
and fallbacks is not None
@ -1442,7 +1442,7 @@ class Router:
### CHECK IF RATE LIMIT / CONTEXT WINDOW ERROR
if (
isinstance(original_exception, litellm.ContextWindowExceededError)
and context_window_fallbacks is None
and context_window_fallbacks is not None
) or (
isinstance(original_exception, openai.RateLimitError)
and fallbacks is not None