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

test(test_router_fallbacks.py): fix testing

Browse source
This commit is contained in:
Krrish Dholakia 2024-04-25 17:43:40 -07:00
parent 0dd918ef73
commit ee518b7728
2 changed files with 3 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

3
litellm/tests/test_router_fallbacks.py
View file

@ -258,6 +258,7 @@ def test_sync_fallbacks_embeddings():
model_list=model_list,
fallbacks=[{"bad-azure-embedding-model": ["good-azure-embedding-model"]}],
set_verbose=False,
num_retries=0,
)
customHandler = MyCustomHandler()
litellm.callbacks = [customHandler]
@ -393,7 +394,7 @@ def test_dynamic_fallbacks_sync():
},
]
router = Router(model_list=model_list, set_verbose=True)
router = Router(model_list=model_list, set_verbose=True, num_retries=0)
kwargs = {}
kwargs["model"] = "azure/gpt-3.5-turbo"
kwargs["messages"] = [{"role": "user", "content": "Hey, how's it going?"}]