scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-05 08:07:05 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 137

(test) custom logger embedding

Browse source
This commit is contained in:
ishaan-jaff 2023-12-11 10:24:37 -08:00
parent 485c40b9ae
commit 6aa2f4bc1d
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
litellm/tests/test_configs/custom_callbacks.py
View file

@ -55,7 +55,7 @@ class MyCustomHandler(CustomLogger):
self.async_success = True
print("Value of async success: ", self.async_success)
print("\n kwargs: ", kwargs)
if kwargs.get("model") == "azure-embedding-model":
if kwargs.get("model") == "azure-embedding-model" or kwargs.get("model") == "ada":
self.async_success_embedding = True
self.async_embedding_kwargs = kwargs
self.async_embedding_response = response_obj