From 744e69f01f9b91f3b189737c8a77fdd850f89f5f Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Thu, 2 Nov 2023 10:05:22 -0700 Subject: [PATCH] (feat) add embed-english-v3.0 --- litellm/__init__.py | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/litellm/__init__.py b/litellm/__init__.py index 7bc023bc23..b2627ecbb6 100644 --- a/litellm/__init__.py +++ b/litellm/__init__.py @@ -316,7 +316,14 @@ longer_context_model_fallback_dict: dict = { ####### EMBEDDING MODELS ################### open_ai_embedding_models: List = ["text-embedding-ada-002"] -cohere_embedding_models: List = ["embed-english-v2.0", "embed-english-light-v2.0", "embed-multilingual-v2.0"] +cohere_embedding_models: List = [ + "embed-english-v3.0", + "embed-english-light-v3.0", + "embed-multilingual-v3.0", + "embed-english-v2.0", + "embed-english-light-v2.0", + "embed-multilingual-v2.0", +] bedrock_embedding_models: List = ["amazon.titan-embed-text-v1"] from .timeout import timeout