fix co embeddings

This commit is contained in:
ishaan-jaff 2023-09-29 16:00:49 -07:00
parent 76956cce52
commit 762d590500

View file

@ -1240,7 +1240,7 @@ def get_llm_provider(model: str, custom_llm_provider: Optional[str] = None):
elif model in litellm.open_ai_embedding_models: elif model in litellm.open_ai_embedding_models:
custom_llm_provider = "openai" custom_llm_provider = "openai"
elif model in litellm.cohere_embedding_models: elif model in litellm.cohere_embedding_models:
custom_llm_provider == "cohere" custom_llm_provider = "cohere"
if custom_llm_provider is None or custom_llm_provider=="": if custom_llm_provider is None or custom_llm_provider=="":
print() print()