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

fix(router.py): fix typing

Browse source
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
Download patch file Download diff file Expand all files Collapse all files

2
litellm/router.py
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,

2
litellm/tests/test_azure_perf.py
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():