mirror of
https://github.com/BerriAI/litellm.git
synced 2025-04-26 19:24:27 +00:00
test: add direct test - fix code qa check
This commit is contained in:
parent
409175286a
commit
615c174be7
1 changed files with 18 additions and 0 deletions
|
@ -418,3 +418,21 @@ def test_router_handle_clientside_credential():
|
|||
|
||||
assert new_deployment.litellm_params.api_key == "123"
|
||||
assert len(router.get_model_list()) == 2
|
||||
|
||||
|
||||
def test_router_get_async_openai_model_client():
|
||||
router = Router(
|
||||
model_list=[
|
||||
{
|
||||
"model_name": "gemini/*",
|
||||
"litellm_params": {
|
||||
"model": "gemini/*",
|
||||
"api_base": "https://api.gemini.com",
|
||||
},
|
||||
}
|
||||
]
|
||||
)
|
||||
model_client = router._get_async_openai_model_client(
|
||||
deployment=MagicMock(), kwargs={}
|
||||
)
|
||||
assert model_client is None
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue