fix(router.py): fix typing

This commit is contained in:
Krrish Dholakia 2024-05-13 18:06:10 -07:00
parent 5488bf4921
commit 38988f030a
2 changed files with 2 additions and 2 deletions

View file

@ -63,7 +63,7 @@ class Router:
def __init__(
self,
model_list: Optional[List[DeploymentTypedDict]] = None,
model_list: Optional[List[Union[DeploymentTypedDict, Dict]]] = None,
## CACHING ##
redis_url: Optional[str] = None,
redis_host: Optional[str] = None,

View file

@ -26,7 +26,7 @@ model_list = [
}
]
router = litellm.Router(model_list=model_list)
router = litellm.Router(model_list=model_list) # type: ignore
async def _openai_completion():