Quellcode durchsuchen

Fix cohere embedder (#1353)

LeonieFreisinger vor 1 Jahr
Ursprung
Commit
9ba445e623
4 geänderte Dateien mit 86 neuen und 57 gelöschten Zeilen
  1. 1 1
      embedchain/embedder/cohere.py
  2. 1 0
      embedchain/utils/misc.py
  3. 82 55
      poetry.lock
  4. 2 1
      pyproject.toml

+ 1 - 1
embedchain/embedder/cohere.py

@@ -1,6 +1,6 @@
 from typing import Optional
 
-from langchain_community.embeddings import CohereEmbeddings
+from langchain_cohere.embeddings import CohereEmbeddings
 
 from embedchain.config import BaseEmbedderConfig
 from embedchain.embedder.base import BaseEmbedder

+ 1 - 0
embedchain/utils/misc.py

@@ -451,6 +451,7 @@ def validate_config(config_data):
                     "mistralai",
                     "nvidia",
                     "ollama",
+                    "cohere",
                 ),
                 Optional("config"): {
                     Optional("model"): Optional(str),

Datei-Diff unterdrückt, da er zu groß ist
+ 82 - 55
poetry.lock


+ 2 - 1
pyproject.toml

@@ -119,7 +119,7 @@ twilio = { version = "^8.5.0", optional = true }
 fastapi-poe = { version = "0.0.16", optional = true }
 discord = { version = "^2.3.2", optional = true }
 slack-sdk = { version = "3.21.3", optional = true }
-cohere = { version = "^4.27", optional = true }
+cohere = { version = "^5.3", optional = true }
 together = { version = "^0.2.8", optional = true }
 weaviate-client = { version = "^3.24.1", optional = true }
 docx2txt = { version = "^0.8", optional = true }
@@ -154,6 +154,7 @@ langchain-openai = "^0.0.5"
 langchain-google-vertexai = { version = "^0.0.5", optional = true }
 sqlalchemy = "^2.0.27"
 alembic = "^1.13.1"
+langchain-cohere = "^0.1.4"
 
 [tool.poetry.group.dev.dependencies]
 black = "^23.3.0"

Einige Dateien werden nicht angezeigt, da zu viele Dateien in diesem Diff geändert wurden.