ai-workspace-services/litellm
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

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