scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-11 03:38:38 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 84

fix: fix test

Browse source
This commit is contained in:
Krrish Dholakia 2025-09-27 14:14:40 -07:00
parent 7d8135f01e
commit 77670fa419
1 changed files with 2 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
tests/local_testing/test_custom_callback_input.py
View file

@ -775,6 +775,8 @@ async def test_async_embedding_openai():
litellm.callbacks = [customHandler_success]
response = await litellm.aembedding(
model="azure/text-embedding-ada-002",
input=["good morning from litellm"],
)
await asyncio.sleep(1)
print(f"customHandler_success.errors: {customHandler_success.errors}")
print(f"customHandler_success.states: {customHandler_success.states}")
@ -913,7 +915,6 @@ async def test_async_embedding_bedrock():
pytest.fail(f"An exception occurred: {str(e)}")
# Image Generation