From 7969849c1dfe5ec42b927e01b04705116cf859eb Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Wed, 28 Feb 2024 22:42:17 -0300 Subject: [PATCH] Refactor LCVectorStoreComponent search method --- .../langflow/components/vectorstores/base/model.py | 14 ++++++++++---- 1 file changed, 10 insertions(+), 4 deletions(-) diff --git a/src/backend/langflow/components/vectorstores/base/model.py b/src/backend/langflow/components/vectorstores/base/model.py index d9ea3875f..2bc766b8b 100644 --- a/src/backend/langflow/components/vectorstores/base/model.py +++ b/src/backend/langflow/components/vectorstores/base/model.py @@ -5,8 +5,8 @@ from langchain_core.retrievers import BaseRetriever from langchain_core.vectorstores import VectorStore from langflow import CustomComponent -from langflow.field_typing import Text, docs_to_records -from langflow.schema import Record +from langflow.field_typing import Text +from langflow.schema import Record, docs_to_records class LCVectorStoreComponent(CustomComponent): @@ -36,8 +36,14 @@ class LCVectorStoreComponent(CustomComponent): """ docs: List[Document] = [] - if input_value and isinstance(input_value, str) and hasattr(vector_store, "search"): - docs = vector_store.search(query=input_value, search_type=search_type.lower()) + if ( + input_value + and isinstance(input_value, str) + and hasattr(vector_store, "search") + ): + docs = vector_store.search( + query=input_value, search_type=search_type.lower() + ) else: raise ValueError("Invalid inputs provided.") return docs_to_records(docs)