phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-25 10:44:24 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix(router.py): fix null model names error

Browse source 
https://github.com/BerriAI/litellm/issues/669#issuecomment-1775456245
This commit is contained in:
Krrish Dholakia 2023-10-23 09:29:50 -07:00
parent bbaa62af13
commit cfed9fff74
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
litellm/router.py
View file

@ -20,12 +20,12 @@ class Router:
router = Router(model_list=model_list) router = Router(model_list=model_list)
""" """
model_names = []
def __init__(self, def __init__(self,
model_list: Optional[list]=None, model_list: Optional[list]=None,
redis_host: Optional[str] = None, redis_host: Optional[str] = None,
redis_port: Optional[int] = None, redis_port: Optional[int] = None,
redis_password: Optional[str] = None) -> None: redis_password: Optional[str] = None) -> None:
self.model_names = []
if model_list: if model_list:
self.model_list = model_list self.model_list = model_list
self.model_names = [m["model_name"] for m in model_list] self.model_names = [m["model_name"] for m in model_list]