From 9825f93fd2619b70dc70e8f63db98c4d536d2c80 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Wed, 19 Jun 2024 20:01:49 -0300 Subject: [PATCH] refactor: Update ChromaVectorStoreComponent search_query handling --- .../langflow/components/vectorstores/Chroma.py | 15 +++++---------- 1 file changed, 5 insertions(+), 10 deletions(-) diff --git a/src/backend/base/langflow/components/vectorstores/Chroma.py b/src/backend/base/langflow/components/vectorstores/Chroma.py index ef709b593..d96be5269 100644 --- a/src/backend/base/langflow/components/vectorstores/Chroma.py +++ b/src/backend/base/langflow/components/vectorstores/Chroma.py @@ -172,24 +172,19 @@ class ChromaVectorStoreComponent(LCVectorStoreComponent): """ 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 = "" - return + return [] 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"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( - 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 return search_results