From ccf63c98ff1e886d66890e276a4d6054cef32511 Mon Sep 17 00:00:00 2001 From: ogabrielluiz Date: Tue, 18 Jun 2024 12:58:45 -0300 Subject: [PATCH] =?UTF-8?q?=F0=9F=90=9B=20(Chroma.py):=20Fix=20passing=20l?= =?UTF-8?q?imit=20parameter=20to=20chroma=5Fcollection=5Fto=5Fdata=20metho?= =?UTF-8?q?d=20to=20correctly=20retrieve=20data=20from=20Chroma=20vector?= =?UTF-8?q?=20store?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/backend/base/langflow/components/vectorstores/Chroma.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/backend/base/langflow/components/vectorstores/Chroma.py b/src/backend/base/langflow/components/vectorstores/Chroma.py index ae9b08ede..1d7d89cef 100644 --- a/src/backend/base/langflow/components/vectorstores/Chroma.py +++ b/src/backend/base/langflow/components/vectorstores/Chroma.py @@ -7,7 +7,7 @@ from loguru import logger from langflow.base.vectorstores.model import LCVectorStoreComponent from langflow.base.vectorstores.utils import chroma_collection_to_data -from langflow.inputs import BoolInput, DataInput, DropdownInput, HandleInput, IntInput, MessageInput, StrInput, TextInput +from langflow.inputs import BoolInput, DataInput, DropdownInput, HandleInput, IntInput, StrInput, TextInput from langflow.schema import Data if TYPE_CHECKING: @@ -140,7 +140,7 @@ class ChromaVectorStoreComponent(LCVectorStoreComponent): ) self._add_documents_to_vector_store(chroma) - self.status = chroma_collection_to_data(chroma.get(self.limit)) + self.status = chroma_collection_to_data(chroma.get(limit=self.limit)) return chroma def _add_documents_to_vector_store(self, vector_store: "Chroma") -> None: