jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

🐛 (Chroma.py): Fix passing limit parameter to chroma_collection_to_data method to correctly retrieve data from Chroma vector store

Browse source
This commit is contained in:
ogabrielluiz 2024-06-18 12:58:45 -03:00
commit ccf63c98ff
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
src/backend/base/langflow/components/vectorstores/Chroma.py
View file

@ -7,7 +7,7 @@ from loguru import logger
from langflow.base.vectorstores.model import LCVectorStoreComponent from langflow.base.vectorstores.model import LCVectorStoreComponent
from langflow.base.vectorstores.utils import chroma_collection_to_data 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 from langflow.schema import Data
if TYPE_CHECKING: if TYPE_CHECKING:
@ -140,7 +140,7 @@ class ChromaVectorStoreComponent(LCVectorStoreComponent):
) )
self._add_documents_to_vector_store(chroma) 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 return chroma
def _add_documents_to_vector_store(self, vector_store: "Chroma") -> None: def _add_documents_to_vector_store(self, vector_store: "Chroma") -> None: