Fix cohere embedder (#1353)

This commit is contained in:
LeonieFreisinger
2024-05-22 07:55:10 +02:00
committed by GitHub
parent ebc5e25f98
commit 9ba445e623
4 changed files with 290 additions and 196 deletions

View File

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

View File

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