Skip to content

Commit db6eb81

Browse files
committed
Use official langchain cohere
1 parent 8c6238e commit db6eb81

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

core/cat/looking_glass/cheshire_cat.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@
77
from langchain_core.runnables import RunnableLambda
88
from langchain_core.prompts import ChatPromptTemplate
99
from langchain_core.output_parsers.string import StrOutputParser
10-
from langchain_community.llms import Cohere
10+
from langchain_cohere import ChatCohere
1111
from langchain_openai import ChatOpenAI, OpenAI
1212
from langchain_google_genai import ChatGoogleGenerativeAI
1313

@@ -208,7 +208,7 @@ def load_language_embedder(self) -> embedders.EmbedderSettings:
208208
# For Azure avoid automatic embedder selection
209209

210210
# Cohere
211-
elif type(self._llm) in [Cohere]:
211+
elif type(self._llm) in [ChatCohere]:
212212
embedder = embedders.EmbedderCohereConfig.get_embedder_from_config(
213213
{
214214
"cohere_api_key": self._llm.cohere_api_key,

0 commit comments

Comments
 (0)