fix - save router settings in db

This commit is contained in:
Ishaan Jaff 2024-04-16 15:11:28 -07:00
parent f64e261794
commit a710480629
2 changed files with 14 additions and 0 deletions

View file

@ -719,6 +719,10 @@ class ConfigYAML(LiteLLMBase):
description="litellm Module settings. See __init__.py for all, example litellm.drop_params=True, litellm.set_verbose=True, litellm.api_base, litellm.cache",
)
general_settings: Optional[ConfigGeneralSettings] = None
router_settings: Optional[dict] = Field(
None,
description="litellm router object settings. See router.py __init__ for all, example router.num_retries=5, router.timeout=5, router.max_retries=5, router.retry_after=5",
)
class Config:
protected_namespaces = ()

View file

@ -8190,6 +8190,16 @@ async def update_config(config_info: ConfigYAML):
"success_callback"
] = combined_success_callback
# router settings
if config_info.router_settings is not None:
config.setdefault("router_settings", {})
_updated_router_settings = config_info.router_settings
config["router_settings"] = {
**_updated_router_settings,
**config["router_settings"],
}
# Save the updated config
await proxy_config.save_config(new_config=config)