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

test_embedding

Browse source
This commit is contained in:
Ishaan Jaffer 2025-09-27 16:57:27 -07:00
parent 10a801ce83
commit 0ec7dace79
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/proxy_unit_tests/test_proxy_custom_logger.py
View file

@ -80,7 +80,7 @@ def test_embedding(client):
my_custom_logger.async_success_embedding is True
) # checks if the status of async_success is True, only the async_log_success_event can set this to true
assert (
my_custom_logger.async_embedding_kwargs["model"] == "azure-embedding-model"
my_custom_logger.async_embedding_kwargs["model"] == "text-embedding-ada-002"
) # checks if kwargs passed to async_log_success_event are correct
kwargs = my_custom_logger.async_embedding_kwargs
litellm_params = kwargs.get("litellm_params")