From e28f4a5f391bc8c90079c26d3ba92256e1d589a9 Mon Sep 17 00:00:00 2001 From: rajib Date: Tue, 16 May 2023 16:27:23 -0700 Subject: [PATCH] changed cohere.py to update the default model of embedding (#4709) # The cohere embedding model do not use large, small. It is deprecated. Changed the modules default model Fixes #4694 Co-authored-by: rajib76 Co-authored-by: Dev 2049 --- langchain/embeddings/cohere.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/langchain/embeddings/cohere.py b/langchain/embeddings/cohere.py index 527bb2c5..e4fe18b2 100644 --- a/langchain/embeddings/cohere.py +++ b/langchain/embeddings/cohere.py @@ -18,11 +18,13 @@ class CohereEmbeddings(BaseModel, Embeddings): .. code-block:: python from langchain.embeddings import CohereEmbeddings - cohere = CohereEmbeddings(model="medium", cohere_api_key="my-api-key") + cohere = CohereEmbeddings( + model="embed-english-light-v2.0", cohere_api_key="my-api-key" + ) """ client: Any #: :meta private: - model: str = "large" + model: str = "embed-english-v2.0" """Model name to use.""" truncate: Optional[str] = None