wtfai 1.6.4 → 1.6.6

package/README.md

@@ -506,28 +506,30 @@ const url = await IframeBridge.uploadFile(file, {
 > - `saveData/loadData` 存储在工作流 state 中（适合简单 key-value）
 > - Session Data API 存储在独立数据表中（适合结构化数据、列表数据）
 
-#### `IframeBridge.createRecord(collection, data)`
+#### `IframeBridge.createRecord(collection, data, options?)`
 
-在指定集合中创建一条记录。
+在指定集合中创建一条记录。可以通过 `options.entityId` 字段进行二级分类归属（如老师 ID、学生 ID），后续查询时利用此字段能大幅提升检索性能。
 
 ```javascript
+// options = { entityId?: string }
 const id = await IframeBridge.createRecord('notes', {
   title: 'My Note',
   content: 'Hello World',
   createdAt: Date.now()
-});
+}, { entityId: 'user_123' }); // 可选的 entityId
 console.log('创建成功, ID:', id);
 ```
 
-#### `IframeBridge.createRecords(collection, dataList)`
+#### `IframeBridge.createRecords(collection, dataList, options?)`
 
-批量在指定集合中创建多条记录。
+批量在指定集合中创建多条记录。支持传入可选的配置项如 `entityId`。
 
 ```javascript
+// options = { entityId?: string }
 const ids = await IframeBridge.createRecords('logs', [
   { level: 'info', message: 'Task started' },
   { level: 'success', message: 'Task finished' }
-]);
+], { entityId: 'user_123' });
 console.log('创建成功, IDs:', ids);
 ```
 
@@ -566,13 +568,15 @@ await IframeBridge.deleteRecord('notes', id);
 
 #### `IframeBridge.listRecords(collection, options?)`
 
-分页查询集合中的记录。
+分页查询集合中的记录。可以通过传入 `entityId` 来极速筛选归属于特定实体的数据。
 
 ```javascript
+// options = { page?: number, pageSize?: number, order?: 'ASC' | 'DESC', entityId?: string }
 const result = await IframeBridge.listRecords('notes', {
   page: 1,
   pageSize: 10,
-  order: 'DESC'
+  order: 'DESC',
+  entityId: 'user_123' // 可选，利用组合索引极速过滤
 });
 console.log(`共 ${result.total} 条记录`);
 result.records.forEach(r => {

package/dist/iframe-bridge.js

@@ -56,11 +56,11 @@ class IframeBridgeHost {
                 const [options] = params;
                 result = await this.session.uploadFileFromIframe(options);
             } else if ('createRecord' === method) {
-                const [collection, data] = params;
-                result = await this.session.createRecord(collection, data);
+                const [collection, data, options] = params;
+                result = await this.session.createRecord(collection, data, options);
             } else if ('createRecords' === method) {
-                const [collection, dataList] = params;
-                result = await this.session.createRecords(collection, dataList);
+                const [collection, dataList, options] = params;
+                result = await this.session.createRecords(collection, dataList, options);
             } else if ('getRecord' === method) {
                 const [collection, id] = params;
                 result = await this.session.getRecord(collection, id);

package/dist/index.js

@@ -1,6 +1,6 @@
 import { Wtfai } from "./client.js";
-import { WorkflowSession } from "./session.js";
-import { UploadService } from "./upload.js";
-import { Markdown, WorkflowRegistry } from "./ui/markdown.js";
 const src = Wtfai;
-export { Markdown, UploadService, WorkflowRegistry, WorkflowSession, src as default };
+export { WorkflowSession } from "./session.js";
+export { UploadService } from "./upload.js";
+export { Markdown, WorkflowRegistry } from "./ui/markdown.js";
+export default src;

package/dist/session.d.ts

@@ -1,5 +1,5 @@
 import type { SimpleMessage } from '@our-llm/shared/types';
-import type { SessionOptions, SessionState, SessionEventListeners, SendInput } from './types';
+import type { SessionOptions, SessionState, SessionEventListeners, SendInput, ListRecordsOptions, ListRecordsResult } from './types';
 import { UploadService } from './upload';
 /**
  * 工作流会话
@@ -82,14 +82,18 @@ export declare class WorkflowSession {
      * @param data 数据内容
      * @returns 新创建记录的 ID
      */
-    createRecord(collection: string, data: Record<string, unknown>): Promise<string>;
+    createRecord(collection: string, data: Record<string, unknown>, options?: {
+        entityId?: string;
+    }): Promise<string>;
     /**
      * 批量创建数据记录
      * @param collection 集合名称
      * @param dataList 数据列表
      * @returns 新创建记录的 ID 列表
      */
-    createRecords(collection: string, dataList: Array<Record<string, unknown>>): Promise<string[]>;
+    createRecords(collection: string, dataList: Array<Record<string, unknown>>, options?: {
+        entityId?: string;
+    }): Promise<string[]>;
     /**
      * 获取数据记录
      * @param collection 集合名称
@@ -110,22 +114,7 @@ export declare class WorkflowSession {
      * @param id 记录 ID
      */
     deleteRecord(collection: string, id: string): Promise<void>;
-    /**
-     * 列表查询数据记录
-     * @param collection 集合名称
-     * @param options 分页选项
-     */
-    listRecords(collection: string, options?: {
-        page?: number;
-        pageSize?: number;
-        order?: 'ASC' | 'DESC';
-    }): Promise<{
-        records: Array<{
-            id: string;
-            data: Record<string, unknown>;
-        }>;
-        total: number;
-    }>;
+    listRecords(collection: string, options?: ListRecordsOptions): Promise<ListRecordsResult>;
     /**
      * 清空集合中的所有数据
      * @param collection 集合名称

package/dist/session.js

@@ -161,7 +161,7 @@ class WorkflowSession {
             }, this.headers).catch(reject);
         });
     }
-    async createRecord(collection, data) {
+    async createRecord(collection, data, options = {}) {
         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)}`, {
@@ -172,14 +172,15 @@ class WorkflowSession {
             },
             body: JSON.stringify({
                 threadId: this.state.threadId,
-                data
+                data,
+                entityId: options.entityId
             })
         });
         if (!response.ok) throw new Error('Failed to create record');
         const result = await response.json();
         return result.id;
     }
-    async createRecords(collection, dataList) {
+    async createRecords(collection, dataList, options = {}) {
         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)}/batch`, {
@@ -190,7 +191,8 @@ class WorkflowSession {
             },
             body: JSON.stringify({
                 threadId: this.state.threadId,
-                dataList
+                dataList,
+                entityId: options.entityId
             })
         });
         if (!response.ok) throw new Error('Failed to create records');
@@ -242,9 +244,9 @@ class WorkflowSession {
         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));
-        if (null == options ? void 0 : options.order) url.searchParams.set('order', options.order);
+        Object.entries(options || {}).forEach(([key, value])=>{
+            if (void 0 !== value) url.searchParams.set(key, String(value));
+        });
         const response = await fetch(url.toString(), {
             headers: this.headers
         });

package/dist/types.d.ts

@@ -193,3 +193,26 @@ export interface WorkflowInfo {
     name: string;
     description: string;
 }
+/**
+ * 分页与过滤选项
+ */
+export interface ListRecordsOptions {
+    page?: number;
+    pageSize?: number;
+    order?: 'ASC' | 'DESC';
+    entityId?: string;
+}
+/**
+ * 数据记录基类
+ */
+export interface BaseRecord {
+    id: string;
+    data: Record<string, unknown>;
+}
+/**
+ * 列表查询返回结构
+ */
+export interface ListRecordsResult {
+    records: BaseRecord[];
+    total: number;
+}

package/dist/ui/code.css

@@ -5,3 +5,37 @@
   display: block;
 }
 
+.html-preview-container {
+  background: #fff;
+  border: 1px solid #e5e7eb;
+  border-radius: 8px;
+  margin: 16px 0;
+  overflow: hidden;
+  box-shadow: 0 4px 6px -1px #0000001a, 0 2px 4px -2px #0000001a;
+}
+
+.html-preview-header {
+  color: #4b5563;
+  background: #f9fafb;
+  border-bottom: 1px solid #e5e7eb;
+  justify-content: space-between;
+  align-items: center;
+  padding: 8px 12px;
+  font-size: 13px;
+  font-weight: 500;
+  display: flex;
+}
+
+.html-preview-header .preview-label {
+  align-items: center;
+  gap: 8px;
+  display: flex;
+}
+
+.html-preview-iframe {
+  border: none;
+  width: 100%;
+  min-height: 400px;
+  display: block;
+}
+

package/dist/ui/code.js

@@ -1,14 +1,16 @@
-import { jsx } from "react/jsx-runtime";
+import { jsx, jsxs } from "react/jsx-runtime";
 import { useEffect, useRef } from "react";
 import { CodeHighlighter, Mermaid } from "@ant-design/x";
+import { EyeOutlined, LoadingOutlined } from "@ant-design/icons";
 import { jsonrepair } from "jsonrepair";
 import { useWorkflowSession } from "./context.js";
 import { IframeBridgeHost } from "../iframe-bridge.js";
 import "./code.css";
 const Code = (props)=>{
     var _className_match;
-    const { className, children } = props;
-    const lang = (null == className ? void 0 : null == (_className_match = className.match(/language-(\w+)/)) ? void 0 : _className_match[1]) || '';
+    const { className, children, streamStatus, lang: infoString } = props;
+    const fullLang = infoString || (null == className ? void 0 : null == (_className_match = className.match(/language-([\w:-]+)/)) ? void 0 : _className_match[1]) || '';
+    const lang = fullLang.split(':')[0];
     const session = useWorkflowSession();
     const iframeRef = useRef(null);
     const bridgeRef = useRef(null);
@@ -38,6 +40,41 @@ const Code = (props)=>{
             allow: "clipboard-read; clipboard-write; camera; microphone"
         });
     } catch  {}
+    if ('html:run' === fullLang) {
+        const isLoading = 'loading' === streamStatus;
+        return /*#__PURE__*/ jsxs("div", {
+            className: "html-preview-container",
+            children: [
+                /*#__PURE__*/ jsxs("div", {
+                    className: "html-preview-header",
+                    children: [
+                        /*#__PURE__*/ jsxs("div", {
+                            className: "preview-label",
+                            children: [
+                                /*#__PURE__*/ jsx(EyeOutlined, {}),
+                                /*#__PURE__*/ jsx("span", {
+                                    children: "预览"
+                                })
+                            ]
+                        }),
+                        isLoading && /*#__PURE__*/ jsx(LoadingOutlined, {
+                            spin: true,
+                            className: "loading-icon",
+                            style: {
+                                color: '#3b82f6'
+                            }
+                        })
+                    ]
+                }),
+                /*#__PURE__*/ jsx("iframe", {
+                    srcDoc: children,
+                    className: "html-preview-iframe",
+                    title: "HTML Preview",
+                    sandbox: "allow-scripts allow-forms allow-popups"
+                })
+            ]
+        });
+    }
     return /*#__PURE__*/ jsx(CodeHighlighter, {
         lang: lang,
         children: children

package/dist/ui/context.d.ts

@@ -7,5 +7,11 @@ export interface WorkflowInfoMapping {
 }
 export declare const SessionContext: import("react").Context<WorkflowSession | null>;
 export declare const WorkflowRegistryContext: import("react").Context<Record<string, WorkflowInfoMapping>>;
+export declare const MarkdownContext: import("react").Context<{
+    isStreaming?: boolean;
+}>;
 export declare const useWorkflowSession: () => WorkflowSession | null;
 export declare const useWorkflowRegistry: () => Record<string, WorkflowInfoMapping>;
+export declare const useMarkdownContext: () => {
+    isStreaming?: boolean;
+};

package/dist/ui/context.js

@@ -1,6 +1,10 @@
 import { createContext, useContext } from "react";
 const SessionContext = createContext(null);
 const WorkflowRegistryContext = createContext({});
+const MarkdownContext = createContext({
+    isStreaming: false
+});
 const useWorkflowSession = ()=>useContext(SessionContext);
 const useWorkflowRegistry = ()=>useContext(WorkflowRegistryContext);
-export { SessionContext, WorkflowRegistryContext, useWorkflowRegistry, useWorkflowSession };
+const useMarkdownContext = ()=>useContext(MarkdownContext);
+export { MarkdownContext, SessionContext, WorkflowRegistryContext, useMarkdownContext, useWorkflowRegistry, useWorkflowSession };

package/dist/ui/markdown.js

@@ -8,7 +8,7 @@ import clsx from "clsx";
 import { Code } from "./code.js";
 import { XProvider } from "@ant-design/x";
 import "./markdown.css";
-import { SessionContext, WorkflowRegistryContext, useWorkflowRegistry } from "./context.js";
+import { MarkdownContext, SessionContext, WorkflowRegistryContext, useWorkflowRegistry } from "./context.js";
 const WorkflowRegistry = ({ children, mapping })=>{
     const parentRegistry = useWorkflowRegistry();
     const mergedRegistry = {
@@ -29,47 +29,52 @@ const Markdown = ({ className, streaming, config, components, session, ...props
         locale: zh_CN,
         children: /*#__PURE__*/ jsx(SessionContext.Provider, {
             value: session || null,
-            children: /*#__PURE__*/ jsx(XMarkdown, {
-                className: clsx(className, 'x-markdown-light'),
-                streaming: {
-                    ...streaming,
-                    enableAnimation: true
+            children: /*#__PURE__*/ jsx(MarkdownContext.Provider, {
+                value: {
+                    isStreaming: !!(null == streaming ? void 0 : streaming.hasNextChunk)
                 },
-                config: {
-                    ...config,
-                    extensions: Latex()
-                },
-                components: {
-                    ...components,
-                    code: Code,
-                    workflow: (props)=>{
-                        const { id, children } = props;
-                        const info = id ? globalRegistry[id] : void 0;
-                        const displayContent = (null == info ? void 0 : info.name) || (Children.count(children) > 0 ? children : id);
-                        return /*#__PURE__*/ jsxs("a", {
-                            onClick: (e)=>{
-                                e.preventDefault();
-                                e.stopPropagation();
-                                null == session || session.emitContentAction({
-                                    type: 'workflow',
-                                    payload: {
-                                        workflowId: id,
-                                        extra: null == info ? void 0 : info.extra
-                                    }
-                                });
-                            },
-                            className: "workflow-link",
-                            children: [
-                                (null == info ? void 0 : info.icon) && /*#__PURE__*/ jsx("span", {
-                                    className: "workflow-link-icon",
-                                    children: info.icon
-                                }),
-                                displayContent
-                            ]
-                        });
-                    }
-                },
-                ...props
+                children: /*#__PURE__*/ jsx(XMarkdown, {
+                    className: clsx(className, 'x-markdown-light'),
+                    streaming: {
+                        ...streaming,
+                        enableAnimation: true
+                    },
+                    config: {
+                        ...config,
+                        extensions: Latex()
+                    },
+                    components: {
+                        ...components,
+                        code: Code,
+                        workflow: (props)=>{
+                            const { id, children } = props;
+                            const info = id ? globalRegistry[id] : void 0;
+                            const displayContent = (null == info ? void 0 : info.name) || (Children.count(children) > 0 ? children : id);
+                            return /*#__PURE__*/ jsxs("a", {
+                                onClick: (e)=>{
+                                    e.preventDefault();
+                                    e.stopPropagation();
+                                    null == session || session.emitContentAction({
+                                        type: 'workflow',
+                                        payload: {
+                                            workflowId: id,
+                                            extra: null == info ? void 0 : info.extra
+                                        }
+                                    });
+                                },
+                                className: "workflow-link",
+                                children: [
+                                    (null == info ? void 0 : info.icon) && /*#__PURE__*/ jsx("span", {
+                                        className: "workflow-link-icon",
+                                        children: info.icon
+                                    }),
+                                    displayContent
+                                ]
+                            });
+                        }
+                    },
+                    ...props
+                })
             })
         })
     });

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "wtfai",
-  "version": "1.6.4",
+  "version": "1.6.6",
   "type": "module",
   "exports": {
     ".": {
@@ -13,36 +13,38 @@
     "dist"
   ],
   "devDependencies": {
-    "@eslint/js": "^9.39.3",
-    "@rsbuild/plugin-react": "^1.4.5",
-    "@rslib/core": "^0.19.6",
+    "@eslint/js": "^9.39.4",
+    "@rsbuild/plugin-react": "^1.4.6",
+    "@rslib/core": "^0.20.0",
     "@types/react": "^19.2.14",
-    "eslint": "^9.39.3",
+    "eslint": "^9.39.4",
     "eslint-config-prettier": "^10.1.8",
     "eslint-plugin-prettier": "^5.5.5",
     "globals": "^17.4.0",
     "prettier": "^3.8.1",
     "react": "^19.2.4",
     "typescript": "^5.9.3",
-    "typescript-eslint": "^8.56.1"
+    "typescript-eslint": "^8.57.0"
   },
   "peerDependencies": {
     "react": ">=16.9.0",
     "react-dom": ">=16.9.0"
   },
   "dependencies": {
+    "@ant-design/icons": "^6.1.0",
     "@ant-design/x": "^2.3.0",
     "@ant-design/x-markdown": "^2.3.0",
     "@microsoft/fetch-event-source": "^2.0.1",
     "clsx": "^2.1.1",
     "compressorjs": "^1.2.1",
     "cos-js-sdk-v5": "^1.10.1",
-    "jsonrepair": "^3.13.2",
+    "jsonrepair": "^3.13.3",
     "uuid": "^13.0.0",
     "@our-llm/shared": "3.0.1"
   },
   "scripts": {
     "build": "rslib build",
+    "build:sdk": "rslib build -c rslib.config.sdk.ts",
     "dev": "rslib build --watch",
     "format": "prettier --write .",
     "lint": "eslint ."