diff --git a/litellm-proxy-extras/litellm_proxy_extras/utils.py b/litellm-proxy-extras/litellm_proxy_extras/utils.py index 793e2fefa8e..92cdf476be6 100644 --- a/litellm-proxy-extras/litellm_proxy_extras/utils.py +++ b/litellm-proxy-extras/litellm_proxy_extras/utils.py @@ -521,7 +521,7 @@ class ProxyExtrasDBManager: ProxyExtrasDBManager._roll_back_migration( migration_name ) - except subprocess.CalledProcessError as rollback_err: + except (subprocess.CalledProcessError, subprocess.TimeoutExpired) as rollback_err: logger.warning( f"Failed to roll back migration {migration_name}: {rollback_err}. " f"It may already be in a rolled-back state."