(test) custom logger embedding

2026-08-07 00:24:03 +00:00 · 2023-12-11 10:24:37 -08:00
parent 485c40b9ae
commit 6aa2f4bc1d
1 changed files with 1 additions and 1 deletions
@@ -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