Added type hints for model_list parameter in RouterConfig

This commit is contained in:
Ali 1. Hassan (Nokia) 2024-06-08 14:50:53 +05:30
parent af39e2d2dd
commit 5ef10e3c19

View file

@ -22,7 +22,7 @@ class ModelConfig(BaseModel):
class RouterConfig(BaseModel):
model_list: List[ModelConfig]
model_list: Optional[List[ModelConfig]]
redis_url: Optional[str] = None
redis_host: Optional[str] = None
@ -270,7 +270,7 @@ class LiteLLM_Params(GenericLiteLLMParams):
class updateLiteLLMParams(GenericLiteLLMParams):
# This class is used to update the LiteLLM_Params
# only differece is model is optional
# only difference is model is optional
model: Optional[str] = None