phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

Litellm fix router testing (#5748)

Browse source 
* test: fix testing - azure changed content policy error logic

* test: fix tests to use mock responses

* test(test_image_generation.py): handle api instability

* test(test_image_generation.py): handle azure api instability

* fix(utils.py): fix unbounded variable error

* fix(utils.py): fix unbounded variable error

* test: refactor test to use mock response

* test: mark flaky azure tests
This commit is contained in:
Krish Dholakia 2024-09-17 18:02:23 -07:00 • committed by GitHub
parent 8d4339c702
commit dd602753c0
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
8 changed files with 36 additions and 11 deletions
Download patch file Download diff file Expand all files Collapse all files

1
litellm/tests/test_custom_callback_router.py
View file

@ -381,6 +381,7 @@ class CompletionCustomHandler(
# Simple Azure OpenAI call
## COMPLETION
@pytest.mark.flaky(retries=5, delay=1)
@pytest.mark.asyncio
async def test_async_chat_azure():
try: