From 7f37d7e44f2fd63fa84de76fcdc7c400bc8f2e3e Mon Sep 17 00:00:00 2001
From: ishaan-jaff <ishaanjaffer0324@gmail.com>
Date: Fri, 12 Jan 2024 17:35:08 -0800
Subject: [PATCH] (feat) set custom_llm_provider for embedding hidden params

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

diff --git a/litellm/main.py b/litellm/main.py
index df570bb36..1655846f6 100644
--- a/litellm/main.py
+++ b/litellm/main.py
@@ -2512,6 +2512,8 @@ def embedding(
         else:
             args = locals()
             raise ValueError(f"No valid embedding model args passed in - {args}")
+        if response is not None:
+            response._hidden_params["custom_llm_provider"] = custom_llm_provider
         return response
     except Exception as e:
         ## LOGGING