Merge pull request #393 from seatable/fix-embedding-collection-api

fix: embedding collection api

Author: freeplant

package-lock.json

@@ -1,6 +1,6 @@
 {
   "name": "dtable-web-api",
-  "version": "3.5.7",
+  "version": "3.5.8",
   "description": "an interface for dtable-web",
   "main": "lib/dtable-web-api.js",
   "scripts": {

package.json

@@ -2649,41 +2649,51 @@ class DTableWebAPI {
     return this.req.delete(url);
   }
 
+  // embedding collections
   listEmbeddingCollections(dtableUuid) {
     const url = this.server + '/api/v2.1/embedding-collections/?dtable_uuid=' + dtableUuid;
     return this.req.get(url);
   }
 
-  createEmbeddingCollection(dtableUuid, tableId) {
+  createEmbeddingCollection(dtableUuid, tableIds) {
     let url = this.server + '/api/v2.1/embedding-collections/';
     let data = {
       'dtable_uuid': dtableUuid,
-      'table_id': tableId,
+      'table_ids': tableIds,
     };
-    return this._sendPostRequest(url, data, {headers: {'Content-Type': 'application/json'}});
+    return this._sendPostRequest(url, data, { headers: {'Content-Type': 'application/json' }});
+  }
+
+  deleteEmbeddingCollection(collectionId) {
+    const url = this.server + '/api/v2.1/embedding-collections/' + collectionId + '/';
+    return this.req.delete(url);
   }
 
-  semanticSearch(dtableUuid, tableId, query) {
+  updateEmbeddingCollection(collectionId) {
+    const url = this.server + '/api/v2.1/embedding-collections/' + collectionId + '/';
+    return this.req.put(url);
+  }
+
+  queryEmbeddingCollectionStatus(taskId) {
+    const url = this.server + '/api/v2.1/embedding-collections/task-status/?task_id=' + taskId;
+    return this.req.get(url);
+  }
+
+  semanticSearch(dtableUuid, query, { tableIds, count } = {}) {
     let url = this.server + '/api/v2.1/similarity-search/';
     let params = {
       'dtable_uuid': dtableUuid,
-      'table_id': tableId,
-      'query': query
+      'query': query,
     };
+    if (tableIds) {
+      params['table_ids'] = tableIds;
+    }
+    if (count) {
+      params['count'] = count;
+    }
     return this._sendPostRequest(url, params, {headers: {'Content-Type': 'application/json'}});
   }
 
-  deleteEmbeddingCollection(dtableUuid, collectionId) {
-    const url = this.server + '/api/v2.1/embedding-collections/' + collectionId + '/?dtable_uuid=' + dtableUuid;
-    return this.req.delete(url);
-  }
-
-  updateEmbeddingCollection(dtableUuid, collectionId) {
-    const url = this.server + '/api/v2.1/embedding-collections/' + collectionId + '/';
-    let params = { dtable_uuid: dtableUuid };
-    return this.req.put(url, params);
-  }
-
   // org admin api
   orgAdminUpdateOrgInfo(newOrgName) {
     let url = this.server + '/api/v2.1/org/admin/info/';