File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change 7
7
from langchain_core .runnables import RunnableLambda
8
8
from langchain_core .prompts import ChatPromptTemplate
9
9
from langchain_core .output_parsers .string import StrOutputParser
10
- from langchain_community . llms import Cohere
10
+ from langchain_cohere import ChatCohere
11
11
from langchain_openai import ChatOpenAI , OpenAI
12
12
from langchain_google_genai import ChatGoogleGenerativeAI
13
13
@@ -208,7 +208,7 @@ def load_language_embedder(self) -> embedders.EmbedderSettings:
208
208
# For Azure avoid automatic embedder selection
209
209
210
210
# Cohere
211
- elif type (self ._llm ) in [Cohere ]:
211
+ elif type (self ._llm ) in [ChatCohere ]:
212
212
embedder = embedders .EmbedderCohereConfig .get_embedder_from_config (
213
213
{
214
214
"cohere_api_key" : self ._llm .cohere_api_key ,
You can’t perform that action at this time.
0 commit comments