From 659793fceef28370438f0a77eaf871bc9341f447 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Sat, 22 Jun 2024 21:13:44 -0300 Subject: [PATCH] fix: correct argument for cohere_api_key --- .../base/langflow/components/retrievers/CohereRerank.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/src/backend/base/langflow/components/retrievers/CohereRerank.py b/src/backend/base/langflow/components/retrievers/CohereRerank.py index 6517fd97e..dd560d588 100644 --- a/src/backend/base/langflow/components/retrievers/CohereRerank.py +++ b/src/backend/base/langflow/components/retrievers/CohereRerank.py @@ -11,7 +11,7 @@ from langflow.schema import Data class CohereRerankComponent(LCVectorStoreComponent): display_name = "Cohere Rerank" - description = "Rerank documents using the Cohere API." + description = "Rerank documents using the Cohere API and a retriever." icon = "Cohere" inputs = [ @@ -38,13 +38,13 @@ class CohereRerankComponent(LCVectorStoreComponent): def build_base_retriever(self) -> Retriever: cohere_reranker = CohereRerank( - api_key=self.api_key, model=self.model, top_n=self.top_n, user_agent=self.user_agent + cohere_api_key=self.api_key, model=self.model, top_n=self.top_n, user_agent=self.user_agent ) retriever = ContextualCompressionRetriever(base_compressor=cohere_reranker, base_retriever=self.retriever) return retriever async def search_documents(self) -> List[Data]: - retriever = self.build_base_retriever() + retriever: ContextualCompressionRetriever = self.build_base_retriever() documents = await retriever.ainvoke(self.search_query) data = self.to_data(documents) self.status = data