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

refactor: Update ChromaVectorStoreComponent search_query handling

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-06-19 20:01:49 -03:00
commit 9825f93fd2
1 changed files with 5 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -172,24 +172,19 @@ class ChromaVectorStoreComponent(LCVectorStoreComponent):
""" """
Search for documents in the Chroma vector store. Search for documents in the Chroma vector store.
""" """
if not self.search_query: search_query: str = self.search_query
if not search_query:
self.status = "" self.status = ""
return return []
vector_store = self.build_vector_store() vector_store = self.build_vector_store()
logger.debug(f"Search input: {self.search_query}") logger.debug(f"Search input: {search_query}")
logger.debug(f"Search type: {self.search_type}") logger.debug(f"Search type: {self.search_type}")
logger.debug(f"Number of results: {self.number_of_results}") logger.debug(f"Number of results: {self.number_of_results}")
if isinstance(self.search_query, list):
if len(self.search_query) > 1:
raise ValueError("Input value must be a single-item list.")
else:
self.search_query = self.search_query[0]
search_results = self.search_with_vector_store( search_results = self.search_with_vector_store(
self.search_query, self.search_type, vector_store, k=self.number_of_results search_query, self.search_type, vector_store, k=self.number_of_results
) )
self.status = search_results self.status = search_results
return search_results return search_results