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

(test) debug proxy logger test

Browse Source
This commit is contained in:
ishaan-jaff 2023-12-11 10:40:01 -08:00
parent 426c741b40
commit 38fe80266f
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
litellm/tests/test_configs/custom_callbacks.py
View File

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