updated

Author: Lokeshchand33

components/databricks/actions/delete-vector-search-index-data/delete-vector-search-index-data.mjs

@@ -1,4 +1,5 @@
 import databricks from "../../databricks.app.mjs";
+import utils from "../../common/utils.mjs";
 
 export default {
   key: "databricks-delete-vector-search-index-data",
@@ -23,7 +24,13 @@ export default {
     },
   },
   async run({ $ }) {
-    const keys = (this.primaryKeys || [])
+    const parsedKeys = utils.parseObject(this.primaryKeys);
+
+    const keys = (Array.isArray(parsedKeys)
+      ? parsedKeys
+      : [
+        parsedKeys,
+      ])
       .map((s) => String(s).trim())
       .filter(Boolean);
 
@@ -33,7 +40,7 @@ export default {
 
     const response = await this.databricks.deleteVectorSearchData({
       indexName: this.indexName,
-      data: {
+      params: {
         primary_keys: keys,
       },
       $,

components/databricks/actions/list-vector-search-indexes/list-vector-search-indexes.mjs

@@ -3,7 +3,7 @@ import databricks from "../../databricks.app.mjs";
 export default {
   key: "databricks-list-vector-search-indexes",
   name: "List Vector Search Indexes",
-  description: "Lists all vector search indexes in the workspace. [See the documentation](https://docs.databricks.com/api/workspace/vectorsearchindexes/listindexes)",
+  description: "Lists all vector search indexes for a given endpoint. [See the documentation](https://docs.databricks.com/api/workspace/vectorsearchindexes/listindexes)",
   version: "0.0.1",
   type: "action",
   props: {
@@ -17,38 +17,20 @@ export default {
   },
 
   async run({ $ }) {
-    const allIndexes = [];
-    let pageToken;
-
-    do {
-      const {
-        vector_indexes, next_page_token,
-      } = await this.databricks.listVectorSearchIndexes({
-        params: {
-          endpoint_name: this.endpointName,
-          ...(pageToken
-            ? {
-              page_token: pageToken,
-            }
-            : {}),
-        },
-        $,
-      });
-
-      if (vector_indexes?.length) {
-        allIndexes.push(...vector_indexes);
-      }
-
-      pageToken = next_page_token;
-    } while (pageToken);
+    const { vector_indexes = [] } = await this.databricks.listVectorSearchIndexes({
+      params: {
+        endpoint_name: this.endpointName,
+      },
+      $,
+    });
 
     $.export(
       "$summary",
-      `Successfully retrieved ${allIndexes.length} index${allIndexes.length === 1
+      `Successfully retrieved ${vector_indexes.length} index${vector_indexes.length === 1
         ? ""
         : "es"}.`,
     );
 
-    return allIndexes;
+    return vector_indexes;
   },
 };

components/databricks/databricks.app.mjs

@@ -281,9 +281,11 @@ export default {
       });
     },
 
-    listVectorSearchIndexes(args = {}) {
+    listVectorSearchIndexes({
+      params, ...args
+    }) {
       return this._makeRequest({
-        path: "/vector-search/indexes",
+        path: `/vector-search/indexes?endpoint_name=${params.endpoint_name}`,
         method: "GET",
         ...args,
       });
@@ -320,10 +322,11 @@ export default {
     },
 
     deleteVectorSearchData({
-      indexName, ...args
-    }) {
+      indexName, params, ...args
+    })
+    {
       return this._makeRequest({
-        path: `/vector-search/indexes/${indexName}/delete-data`,
+        path: `/vector-search/indexes/${indexName}/delete-data?primary_keys=${params.primary_keys}`,
         method: "DELETE",
         ...args,
       });