From e3b24ec79754e0a69f1b5bc5bc57ae689a746b5d Mon Sep 17 00:00:00 2001
From: ishaan-jaff <ishaanjaffer0324@gmail.com>
Date: Wed, 6 Dec 2023 19:03:41 -0800
Subject: [PATCH] (feat) aembedding - add custom logging support

---
 litellm/main.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/litellm/main.py b/litellm/main.py
index d78c8ceb9c..a8b499af44 100644
--- a/litellm/main.py
+++ b/litellm/main.py
@@ -1677,7 +1677,7 @@ def batch_completion_models_all_responses(*args, **kwargs):
     return responses
 
 ### EMBEDDING ENDPOINTS ####################
-
+@client
 async def aembedding(*args, **kwargs):
     """
     Asynchronously calls the `embedding` function with the given arguments and keyword arguments.