From 116e555a03fd98ae2e345816c6143ad0553a1c7f Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Mon, 6 May 2024 17:43:43 -0700 Subject: [PATCH] fix - setting model_group_retry_policy in db --- litellm/types/router.py | 1 + 1 file changed, 1 insertion(+) diff --git a/litellm/types/router.py b/litellm/types/router.py index dbfc549cd8b..4b54d8f41aa 100644 --- a/litellm/types/router.py +++ b/litellm/types/router.py @@ -55,6 +55,7 @@ class UpdateRouterConfig(BaseModel): routing_strategy_args: Optional[dict] = None routing_strategy: Optional[str] = None + model_group_retry_policy: Optional[dict] = None allowed_fails: Optional[int] = None cooldown_time: Optional[float] = None num_retries: Optional[int] = None