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

test_async_custom_handler_embedding_optional_param

Browse source
This commit is contained in:
Ishaan Jaffer 2025-09-27 12:18:17 -07:00
parent 1bb7ad4f69
commit f7213bacfe
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
tests/local_testing/test_custom_logger.py
View file

@ -392,7 +392,7 @@ async def test_async_custom_handler_embedding_optional_param():
customHandler_optional_params = MyCustomHandler()
litellm.callbacks = [customHandler_optional_params]
response = await litellm.aembedding(
model="azure/azure-embedding-model", input=["hello world"], user="John"
model="text-embedding-ada-002", input=["hello world"], user="John"
)
await asyncio.sleep(1) # success callback is async
assert customHandler_optional_params.user == "John"