scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-11 22:51:28 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 123

test(test_config.py): fix test

Browse source
This commit is contained in:
Krrish Dholakia 2024-05-07 21:31:03 -07:00
parent 8d04f4686f
commit f45954d154
1 changed files with 3 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
litellm/tests/test_config.py
View file

@ -140,6 +140,8 @@ async def test_add_existing_deployment():
deployment_2.to_json(exclude_none=True),
]
)
init_len_list = len(llm_router.model_list)
print(f"llm_router: {llm_router}")
master_key = "sk-1234"
setattr(litellm.proxy.proxy_server, "llm_router", llm_router)
@ -164,7 +166,7 @@ async def test_add_existing_deployment():
db_models = [db_model]
num_added = pc._add_deployment(db_models=db_models)
assert num_added == 0
assert init_len_list == len(llm_router.model_list)
litellm_params = LiteLLM_Params(