diff --git a/litellm/utils.py b/litellm/utils.py index 897ca0be1..c8bf7f6ce 100644 --- a/litellm/utils.py +++ b/litellm/utils.py @@ -1223,6 +1223,10 @@ def get_llm_provider(model: str, custom_llm_provider: Optional[str] = None): ## petals elif model in litellm.petals_models: custom_llm_provider = "petals" + elif model in litellm.open_ai_embedding_models: + custom_llm_provider = "openai" + elif model in litellm.cohere_embedding_models: + custom_llm_provider == "cohere" if custom_llm_provider is None or custom_llm_provider=="": print()