wtfai 1.5.6 → 1.5.7

package/dist/iframe-bridge.js

@@ -55,6 +55,24 @@ class IframeBridgeHost {
             } else if ('uploadFile' === method) {
                 const [options] = params;
                 result = await this.session.uploadFileFromIframe(options);
+            } else if ('createRecord' === method) {
+                const [collection, data] = params;
+                result = await this.session.createRecord(collection, data);
+            } else if ('getRecord' === method) {
+                const [collection, id] = params;
+                result = await this.session.getRecord(collection, id);
+            } else if ('updateRecord' === method) {
+                const [collection, id, data] = params;
+                result = await this.session.updateRecord(collection, id, data);
+            } else if ('deleteRecord' === method) {
+                const [collection, id] = params;
+                result = await this.session.deleteRecord(collection, id);
+            } else if ('listRecords' === method) {
+                const [collection, options] = params;
+                result = await this.session.listRecords(collection, options);
+            } else if ('clearCollection' === method) {
+                const [collection] = params;
+                result = await this.session.clearCollection(collection);
             } else {
                 const userMethods = this.session.getIframeMethods();
                 const fn = userMethods[method];
@@ -78,7 +96,13 @@ class IframeBridgeHost {
             'hasMethod',
             'executeWorkflow',
             'uploadFile',
-            'updateSessionTitle'
+            'updateSessionTitle',
+            'createRecord',
+            'getRecord',
+            'updateRecord',
+            'deleteRecord',
+            'listRecords',
+            'clearCollection'
         ];
         if (builtIns.includes(methodName)) return true;
         const userMethods = this.session.getIframeMethods();

package/dist/session.d.ts

@@ -71,6 +71,54 @@ export declare class WorkflowSession {
      * @param title 新的标题
      */
     updateSessionTitle(title: string): Promise<void>;
+    /**
+     * 创建数据记录
+     * @param collection 集合名称
+     * @param data 数据内容
+     * @returns 新创建记录的 ID
+     */
+    createRecord(collection: string, data: Record<string, unknown>): Promise<string>;
+    /**
+     * 获取数据记录
+     * @param collection 集合名称
+     * @param id 记录 ID
+     * @returns 记录数据，不存在时返回 null
+     */
+    getRecord(collection: string, id: string): Promise<Record<string, unknown> | null>;
+    /**
+     * 更新数据记录
+     * @param collection 集合名称
+     * @param id 记录 ID
+     * @param data 新的数据内容
+     */
+    updateRecord(collection: string, id: string, data: Record<string, unknown>): Promise<void>;
+    /**
+     * 删除数据记录
+     * @param collection 集合名称
+     * @param id 记录 ID
+     */
+    deleteRecord(collection: string, id: string): Promise<void>;
+    /**
+     * 列表查询数据记录
+     * @param collection 集合名称
+     * @param options 分页选项
+     */
+    listRecords(collection: string, options?: {
+        page?: number;
+        pageSize?: number;
+    }): Promise<{
+        records: Array<{
+            id: string;
+            data: Record<string, unknown>;
+        }>;
+        total: number;
+    }>;
+    /**
+     * 清空集合中的所有数据
+     * @param collection 集合名称
+     * @returns 被删除的记录数量
+     */
+    clearCollection(collection: string): Promise<number>;
     /**
      * 执行指定的工作流（临时/嵌套执行），工作流需要使用消息生成组件来生成消息供收集
      * 不会影响当前会话的状态（messages 等）

package/dist/session.js

@@ -158,6 +158,90 @@ class WorkflowSession {
             }, this.headers).catch(reject);
         });
     }
+    async createRecord(collection, data) {
+        this.assertNotDisposed();
+        if (!this.state.threadId) throw new Error('No active session (missing threadId)');
+        const response = await fetch(`${this.baseUrl}/workflows/${this.workflowId}/data/${encodeURIComponent(collection)}`, {
+            method: 'POST',
+            headers: {
+                'Content-Type': 'application/json',
+                ...this.headers
+            },
+            body: JSON.stringify({
+                threadId: this.state.threadId,
+                data
+            })
+        });
+        if (!response.ok) throw new Error('Failed to create record');
+        const result = await response.json();
+        return result.id;
+    }
+    async getRecord(collection, id) {
+        this.assertNotDisposed();
+        if (!this.state.threadId) throw new Error('No active session (missing threadId)');
+        const url = new URL(`${this.baseUrl}/workflows/${this.workflowId}/data/${encodeURIComponent(collection)}/${encodeURIComponent(id)}`);
+        url.searchParams.set('threadId', this.state.threadId);
+        const response = await fetch(url.toString(), {
+            headers: this.headers
+        });
+        if (404 === response.status) return null;
+        if (!response.ok) throw new Error('Failed to get record');
+        const result = await response.json();
+        return result.data;
+    }
+    async updateRecord(collection, id, data) {
+        this.assertNotDisposed();
+        if (!this.state.threadId) throw new Error('No active session (missing threadId)');
+        const response = await fetch(`${this.baseUrl}/workflows/${this.workflowId}/data/${encodeURIComponent(collection)}/${encodeURIComponent(id)}`, {
+            method: 'PUT',
+            headers: {
+                'Content-Type': 'application/json',
+                ...this.headers
+            },
+            body: JSON.stringify({
+                threadId: this.state.threadId,
+                data
+            })
+        });
+        if (!response.ok) throw new Error('Failed to update record');
+    }
+    async deleteRecord(collection, id) {
+        this.assertNotDisposed();
+        if (!this.state.threadId) throw new Error('No active session (missing threadId)');
+        const url = new URL(`${this.baseUrl}/workflows/${this.workflowId}/data/${encodeURIComponent(collection)}/${encodeURIComponent(id)}`);
+        url.searchParams.set('threadId', this.state.threadId);
+        const response = await fetch(url.toString(), {
+            method: 'DELETE',
+            headers: this.headers
+        });
+        if (!response.ok) throw new Error('Failed to delete record');
+    }
+    async listRecords(collection, options) {
+        this.assertNotDisposed();
+        if (!this.state.threadId) throw new Error('No active session (missing threadId)');
+        const url = new URL(`${this.baseUrl}/workflows/${this.workflowId}/data/${encodeURIComponent(collection)}`);
+        url.searchParams.set('threadId', this.state.threadId);
+        if (null == options ? void 0 : options.page) url.searchParams.set('page', String(options.page));
+        if (null == options ? void 0 : options.pageSize) url.searchParams.set('pageSize', String(options.pageSize));
+        const response = await fetch(url.toString(), {
+            headers: this.headers
+        });
+        if (!response.ok) throw new Error('Failed to list records');
+        return response.json();
+    }
+    async clearCollection(collection) {
+        this.assertNotDisposed();
+        if (!this.state.threadId) throw new Error('No active session (missing threadId)');
+        const url = new URL(`${this.baseUrl}/workflows/${this.workflowId}/data/${encodeURIComponent(collection)}`);
+        url.searchParams.set('threadId', this.state.threadId);
+        const response = await fetch(url.toString(), {
+            method: 'DELETE',
+            headers: this.headers
+        });
+        if (!response.ok) throw new Error('Failed to clear collection');
+        const result = await response.json();
+        return result.deleted;
+    }
     async executeWorkflow(workflowId, input) {
         this.assertNotDisposed();
         const { parts: inputParts, ...rest } = input;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "wtfai",
-  "version": "1.5.6",
+  "version": "1.5.7",
   "type": "module",
   "exports": {
     ".": {
@@ -31,8 +31,8 @@
     "react-dom": ">=16.9.0"
   },
   "dependencies": {
-    "@ant-design/x": "^2.2.1",
-    "@ant-design/x-markdown": "^2.2.1",
+    "@ant-design/x": "^2.2.2",
+    "@ant-design/x-markdown": "^2.2.2",
     "@microsoft/fetch-event-source": "^2.0.1",
     "clsx": "^2.1.1",
     "compressorjs": "^1.2.1",