scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-09 03:18:44 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 108

fix: router management endpoints

Browse source
This commit is contained in:
Ishaan Jaff 2025-05-19 20:59:22 -07:00
parent 5fb5b2521d
commit 3c7210b148
1 changed files with 11 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
litellm/proxy/management_endpoints/router_management_endpoints.py
View file

@ -1,5 +1,6 @@
import asyncio
import copy
from typing import Dict, Union
from fastapi import APIRouter, Depends, HTTPException, status
@ -64,12 +65,16 @@ async def update_router_settings(
detail={"error": "Only proxy admin users can update router settings"},
)
existing_router_settings = (
await prisma_client.db.litellm_config.find_unique(
where={"param_name": "router_settings"}
)
or {}
router_settings_row = await prisma_client.db.litellm_config.find_unique(
where={"param_name": "router_settings"}
)
existing_router_settings: dict = {}
if router_settings_row:
param_value: Union[Dict, str] = router_settings_row.param_value
if isinstance(param_value, str):
existing_router_settings = safe_json_loads(data=param_value)
elif isinstance(param_value, dict):
existing_router_settings = param_value
# new router settings
new_router_settings: dict = copy.deepcopy(dict(existing_router_settings))
@ -86,6 +91,7 @@ async def update_router_settings(
"param_value": safe_dumps(new_router_settings),
},
"update": {
"param_name": "router_settings",
"param_value": safe_dumps(new_router_settings),
},
},