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__( def __init__(
self, self,
model_list: Optional[List[DeploymentTypedDict]] = None, model_list: Optional[List[Union[DeploymentTypedDict, Dict]]] = None,
## CACHING ## ## CACHING ##
redis_url: Optional[str] = None, redis_url: Optional[str] = None,
redis_host: 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(): async def _openai_completion():