From 331530833107431f5f1582032ebf9a34627aee08 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Tue, 2 Apr 2024 19:14:26 -0300 Subject: [PATCH] Add exception handling for missing 'content' field in search_with_vector_store method --- .../langflow/components/vectorsearch/AstraDBSearch.py | 10 +++++++++- 1 file changed, 9 insertions(+), 1 deletion(-) diff --git a/src/backend/base/langflow/components/vectorsearch/AstraDBSearch.py b/src/backend/base/langflow/components/vectorsearch/AstraDBSearch.py index f11c04748..e2d8e61e9 100644 --- a/src/backend/base/langflow/components/vectorsearch/AstraDBSearch.py +++ b/src/backend/base/langflow/components/vectorsearch/AstraDBSearch.py @@ -137,4 +137,12 @@ class AstraDBSearchComponent(LCVectorStoreComponent): metadata_indexing_exclude=metadata_indexing_exclude, collection_indexing_policy=collection_indexing_policy, ) - return self.search_with_vector_store(input_value, search_type, vector_store, k=number_of_results) + try: + return self.search_with_vector_store(input_value, search_type, vector_store, k=number_of_results) + except KeyError as e: + if "content" in str(e): + raise ValueError( + "You should ingest data through Langflow (or LangChain) to query it in Langflow. Your collection does not contain a field name 'content'." + ) + else: + raise e