wtfai 1.5.5 → 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

@@ -16,6 +16,7 @@ export declare class WorkflowSession {
     private disposed;
     constructor(workflowId: string, baseUrl: string, uploadService: UploadService, headers?: Record<string, string>, options?: SessionOptions);
     private iframeMethods;
+    getWorkflowId(): string;
     /**
      * 注册可供 iframe 调用的自定义方法
      */
@@ -71,7 +72,55 @@ export declare class WorkflowSession {
      */
     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 等）
      * @param workflowId 目标工作流 ID
      * @param input 输入数据（与 send 方法一致）

package/dist/session.js

@@ -11,6 +11,9 @@ function _define_property(obj, key, value) {
     return obj;
 }
 class WorkflowSession {
+    getWorkflowId() {
+        return this.workflowId;
+    }
     registerIframeMethods(methods) {
         this.assertNotDisposed();
         this.iframeMethods = {
@@ -155,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;
@@ -218,6 +305,7 @@ class WorkflowSession {
             executeWorkflowSSE(`${this.baseUrl}/workflows/${workflowId}/execute`, {
                 input: executionInput,
                 threadId: tempThreadId,
+                hostWorkflowId: this.workflowId,
                 ...rest
             }, {
                 onNodeEnd: (data)=>{

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "wtfai",
-  "version": "1.5.5",
+  "version": "1.5.7",
   "type": "module",
   "exports": {
     ".": {
@@ -14,13 +14,13 @@
   ],
   "devDependencies": {
     "@eslint/js": "^9.39.2",
-    "@rsbuild/plugin-react": "^1.4.3",
-    "@rslib/core": "^0.19.3",
-    "@types/react": "^19.2.10",
+    "@rsbuild/plugin-react": "^1.4.5",
+    "@rslib/core": "^0.19.5",
+    "@types/react": "^19.2.13",
     "eslint": "^9.39.2",
     "eslint-config-prettier": "^10.1.8",
     "eslint-plugin-prettier": "^5.5.5",
-    "globals": "^17.2.0",
+    "globals": "^17.3.0",
     "prettier": "^3.8.1",
     "react": "^19.2.4",
     "typescript": "^5.9.3",
@@ -31,8 +31,8 @@
     "react-dom": ">=16.9.0"
   },
   "dependencies": {
-    "@ant-design/x": "^2.2.0",
-    "@ant-design/x-markdown": "^2.2.0",
+    "@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",