From 9cc6b2f243a518be1af9ca9466db8412753b8b68 Mon Sep 17 00:00:00 2001
From: joaoguilhermeS <j.guilherme.s.oliveira2@gmail.com>
Date: Sat, 22 Jun 2024 13:38:26 -0300
Subject: [PATCH] chore: Updating AstraDB parameters format

---
 .../components/vectorstores/AstraDB.py        | 35 ++++++++-----------
 1 file changed, 14 insertions(+), 21 deletions(-)

diff --git a/src/backend/base/langflow/components/vectorstores/AstraDB.py b/src/backend/base/langflow/components/vectorstores/AstraDB.py
index 1ce2819cc..6b8e3a8f6 100644
--- a/src/backend/base/langflow/components/vectorstores/AstraDB.py
+++ b/src/backend/base/langflow/components/vectorstores/AstraDB.py
@@ -38,15 +38,14 @@ class AstraVectorStoreComponent(LCVectorStoreComponent):
             info="API endpoint URL for the Astra DB service.",
             value="ASTRA_DB_API_ENDPOINT",
         ),
-        DataInput(
-            name="vector_store_inputs",
-            display_name="Vector Store Inputs",
-            is_list=True,
+        MultilineInput(
+            name="search_input",
+            display_name="Search Input",
         ),
-        HandleInput(
-            name="embedding",
-            display_name="Embedding",
-            input_types=["Embeddings"],
+        DataInput(
+            name="ingest_data",
+            display_name="Ingest Data",
+            is_list=True,
         ),
         StrInput(
             name="namespace",
@@ -105,6 +104,11 @@ class AstraVectorStoreComponent(LCVectorStoreComponent):
             info="Optional list of metadata fields to include in the indexing.",
             advanced=True,
         ),
+        HandleInput(
+            name="embedding",
+            display_name="Embedding",
+            input_types=["Embeddings"],
+        ),
         StrInput(
             name="metadata_indexing_exclude",
             display_name="Metadata Indexing Exclude",
@@ -117,20 +121,12 @@ class AstraVectorStoreComponent(LCVectorStoreComponent):
             info="Optional dictionary defining the indexing policy for the collection.",
             advanced=True,
         ),
-        BoolInput(
-            name="add_to_vector_store",
-            display_name="Add to Vector Store",
-            info="If true, the Vector Store Inputs will be added to the Vector Store.",
-        ),
-        MultilineInput(
-            name="search_input",
-            display_name="Search Input",
-        ),
         DropdownInput(
             name="search_type",
             display_name="Search Type",
             options=["Similarity", "MMR"],
             value="Similarity",
+            advanced=True,
         ),
         IntInput(
             name="number_of_results",
@@ -186,15 +182,12 @@ class AstraVectorStoreComponent(LCVectorStoreComponent):
         except Exception as e:
             raise ValueError(f"Error initializing AstraDBVectorStore: {str(e)}") from e
 
-        if self.add_to_vector_store:
-            self._add_documents_to_vector_store(vector_store)
-
         self.status = self._astradb_collection_to_data(vector_store.collection)
         return vector_store
 
     def _add_documents_to_vector_store(self, vector_store):
         documents = []
-        for _input in self.vector_store_inputs or []:
+        for _input in self.ingest_data or []:
             if isinstance(_input, Data):
                 documents.append(_input.to_lc_document())
             else: