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

test(test_custom_callback_router.py): add async azure testing for router

Browse source
This commit is contained in:
Krrish Dholakia 2023-12-11 16:40:23 -08:00
parent 5c1322e574
commit 3b6099633c
4 changed files with 404 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

5
litellm/tests/test_custom_callback_input.py
View file

@ -22,8 +22,7 @@ from litellm.integrations.custom_logger import CustomLogger
# Test interfaces
## 1. litellm.completion() + litellm.embeddings()
## 2. router.completion() + router.embeddings()
## 3. proxy.completions + proxy.embeddings
## refer to test_custom_callback_input_router.py for the router + proxy tests
class CompletionCustomHandler(CustomLogger): # https://docs.litellm.ai/docs/observability/custom_callback#callback-class
"""
@ -577,4 +576,4 @@ async def test_async_embedding_bedrock():
except Exception as e:
pytest.fail(f"An exception occurred: {str(e)}")
asyncio.run(test_async_embedding_bedrock())
# asyncio.run(test_async_embedding_bedrock())