github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-06-02 04:58:46 +00:00
Code Issues Packages Projects Releases Wiki Activity

changed cohere.py to update the default model of embedding (#4709)

Browse Source 
# The cohere embedding model do not use large, small. It is deprecated.
Changed the modules default model

Fixes #4694


Co-authored-by: rajib76 <rajib76@yahoo.com>
Co-authored-by: Dev 2049 <dev.dev2049@gmail.com>
This commit is contained in:
rajib 2023-05-16 16:27:23 -07:00 committed by GitHub
parent 75fe9d3555
commit e28f4a5f39
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
langchain/embeddings/cohere.py
View File

@ -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