scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-20 00:11:50 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 75

fix: update model from db

Browse source
This commit is contained in:
Lucas 2026-04-03 01:41:26 +08:00
parent d1df4e838b
commit d784b788c8
2 changed files with 16 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
litellm/router.py
View file

@ -7318,6 +7318,7 @@ class Router:
if (
deployment.litellm_params == _deployment_on_router.litellm_params
and deployment.model_info == _deployment_on_router.model_info
and deployment.model_name == _deployment_on_router.model_name
):
# No need to update
return None

15
tests/router_unit_tests/test_router_helper_utils.py
View file

@ -828,6 +828,21 @@ def test_upsert_deployment(model_list):
assert len(router.model_list) == len(model_list)
def test_upsert_deployment_updates_model_name(model_list):
"""Test that changing only model_name still updates an existing deployment."""
router = Router(model_list=model_list)
deployment = router.get_deployment_by_model_group_name(
model_group_name="gpt-3.5-turbo"
)
deployment_id = deployment.model_info.id
deployment.model_name = "gpt-3.5-turbo-renamed"
updated_deployment = router.upsert_deployment(deployment=deployment)
assert updated_deployment is not None
assert router.get_deployment(model_id=deployment_id).model_name == deployment.model_name
def test_delete_deployment(model_list):
"""Test if the 'delete_deployment' function is working correctly"""
router = Router(model_list=model_list)