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