From 0e11fdda42bc8acd929844e3fee110d6537902b6 Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Fri, 29 Sep 2023 15:31:53 -0700 Subject: [PATCH] test fixes for embedding --- litellm/utils.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/litellm/utils.py b/litellm/utils.py index 897ca0be19..c8bf7f6ceb 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()