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

fix(router.py): handle id being passed in as int

Browse source
This commit is contained in:
Krrish Dholakia 2024-04-04 14:22:50 -07:00
parent 9638e244f8
commit 48a5948081
7 changed files with 29 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

4
litellm/router.py
View file

@ -37,9 +37,11 @@ class ModelInfo(BaseModel):
str
] # Allow id to be optional on input, but it will always be present as a str in the model instance
def __init__(self, id: Optional[str] = None, **params):
def __init__(self, id: Optional[Union[str, int]] = None, **params):
if id is None:
id = str(uuid.uuid4()) # Generate a UUID if id is None or not provided
elif isinstance(id, int):
id = str(id)
super().__init__(id=id, **params)
class Config: