@nocobase/plugin-ai 2.1.0-beta.14 → 2.1.0-beta.16

package/dist/node_modules/zod/package.json

@@ -1 +1 @@
-{"name":"zod","version":"4.3.5","type":"module","license":"MIT","author":"Colin McDonnell <zod@colinhacks.com>","description":"TypeScript-first schema declaration and validation library with static type inference","homepage":"https://zod.dev","llms":"https://zod.dev/llms.txt","llmsFull":"https://zod.dev/llms-full.txt","mcpServer":"https://mcp.inkeep.com/zod/mcp","funding":"https://github.com/sponsors/colinhacks","sideEffects":false,"files":["src","**/*.js","**/*.mjs","**/*.cjs","**/*.d.ts","**/*.d.mts","**/*.d.cts","**/package.json"],"keywords":["typescript","schema","validation","type","inference"],"main":"./index.cjs","types":"./index.d.cts","module":"./index.js","zshy":{"exports":{"./package.json":"./package.json",".":"./src/index.ts","./mini":"./src/mini/index.ts","./locales":"./src/locales/index.ts","./v3":"./src/v3/index.ts","./v4":"./src/v4/index.ts","./v4-mini":"./src/v4-mini/index.ts","./v4/mini":"./src/v4/mini/index.ts","./v4/core":"./src/v4/core/index.ts","./v4/locales":"./src/v4/locales/index.ts","./v4/locales/*":"./src/v4/locales/*"},"conditions":{"@zod/source":"src"}},"exports":{"./package.json":"./package.json",".":{"@zod/source":"./src/index.ts","types":"./index.d.cts","import":"./index.js","require":"./index.cjs"},"./mini":{"@zod/source":"./src/mini/index.ts","types":"./mini/index.d.cts","import":"./mini/index.js","require":"./mini/index.cjs"},"./locales":{"@zod/source":"./src/locales/index.ts","types":"./locales/index.d.cts","import":"./locales/index.js","require":"./locales/index.cjs"},"./v3":{"@zod/source":"./src/v3/index.ts","types":"./v3/index.d.cts","import":"./v3/index.js","require":"./v3/index.cjs"},"./v4":{"@zod/source":"./src/v4/index.ts","types":"./v4/index.d.cts","import":"./v4/index.js","require":"./v4/index.cjs"},"./v4-mini":{"@zod/source":"./src/v4-mini/index.ts","types":"./v4-mini/index.d.cts","import":"./v4-mini/index.js","require":"./v4-mini/index.cjs"},"./v4/mini":{"@zod/source":"./src/v4/mini/index.ts","types":"./v4/mini/index.d.cts","import":"./v4/mini/index.js","require":"./v4/mini/index.cjs"},"./v4/core":{"@zod/source":"./src/v4/core/index.ts","types":"./v4/core/index.d.cts","import":"./v4/core/index.js","require":"./v4/core/index.cjs"},"./v4/locales":{"@zod/source":"./src/v4/locales/index.ts","types":"./v4/locales/index.d.cts","import":"./v4/locales/index.js","require":"./v4/locales/index.cjs"},"./v4/locales/*":{"@zod/source":"./src/v4/locales/*","types":"./v4/locales/*","import":"./v4/locales/*","require":"./v4/locales/*"}},"repository":{"type":"git","url":"git+https://github.com/colinhacks/zod.git"},"bugs":{"url":"https://github.com/colinhacks/zod/issues"},"support":{"backing":{"npm-funding":true}},"scripts":{"clean":"git clean -xdf . -e node_modules","build":"zshy --project tsconfig.build.json","postbuild":"tsx ../../scripts/write-stub-package-jsons.ts && pnpm biome check --write .","test:watch":"pnpm vitest","test":"pnpm vitest run","prepublishOnly":"tsx ../../scripts/check-versions.ts"},"_lastModified":"2026-04-20T09:20:58.182Z"}
+{"name":"zod","version":"4.3.5","type":"module","license":"MIT","author":"Colin McDonnell <zod@colinhacks.com>","description":"TypeScript-first schema declaration and validation library with static type inference","homepage":"https://zod.dev","llms":"https://zod.dev/llms.txt","llmsFull":"https://zod.dev/llms-full.txt","mcpServer":"https://mcp.inkeep.com/zod/mcp","funding":"https://github.com/sponsors/colinhacks","sideEffects":false,"files":["src","**/*.js","**/*.mjs","**/*.cjs","**/*.d.ts","**/*.d.mts","**/*.d.cts","**/package.json"],"keywords":["typescript","schema","validation","type","inference"],"main":"./index.cjs","types":"./index.d.cts","module":"./index.js","zshy":{"exports":{"./package.json":"./package.json",".":"./src/index.ts","./mini":"./src/mini/index.ts","./locales":"./src/locales/index.ts","./v3":"./src/v3/index.ts","./v4":"./src/v4/index.ts","./v4-mini":"./src/v4-mini/index.ts","./v4/mini":"./src/v4/mini/index.ts","./v4/core":"./src/v4/core/index.ts","./v4/locales":"./src/v4/locales/index.ts","./v4/locales/*":"./src/v4/locales/*"},"conditions":{"@zod/source":"src"}},"exports":{"./package.json":"./package.json",".":{"@zod/source":"./src/index.ts","types":"./index.d.cts","import":"./index.js","require":"./index.cjs"},"./mini":{"@zod/source":"./src/mini/index.ts","types":"./mini/index.d.cts","import":"./mini/index.js","require":"./mini/index.cjs"},"./locales":{"@zod/source":"./src/locales/index.ts","types":"./locales/index.d.cts","import":"./locales/index.js","require":"./locales/index.cjs"},"./v3":{"@zod/source":"./src/v3/index.ts","types":"./v3/index.d.cts","import":"./v3/index.js","require":"./v3/index.cjs"},"./v4":{"@zod/source":"./src/v4/index.ts","types":"./v4/index.d.cts","import":"./v4/index.js","require":"./v4/index.cjs"},"./v4-mini":{"@zod/source":"./src/v4-mini/index.ts","types":"./v4-mini/index.d.cts","import":"./v4-mini/index.js","require":"./v4-mini/index.cjs"},"./v4/mini":{"@zod/source":"./src/v4/mini/index.ts","types":"./v4/mini/index.d.cts","import":"./v4/mini/index.js","require":"./v4/mini/index.cjs"},"./v4/core":{"@zod/source":"./src/v4/core/index.ts","types":"./v4/core/index.d.cts","import":"./v4/core/index.js","require":"./v4/core/index.cjs"},"./v4/locales":{"@zod/source":"./src/v4/locales/index.ts","types":"./v4/locales/index.d.cts","import":"./v4/locales/index.js","require":"./v4/locales/index.cjs"},"./v4/locales/*":{"@zod/source":"./src/v4/locales/*","types":"./v4/locales/*","import":"./v4/locales/*","require":"./v4/locales/*"}},"repository":{"type":"git","url":"git+https://github.com/colinhacks/zod.git"},"bugs":{"url":"https://github.com/colinhacks/zod/issues"},"support":{"backing":{"npm-funding":true}},"scripts":{"clean":"git clean -xdf . -e node_modules","build":"zshy --project tsconfig.build.json","postbuild":"tsx ../../scripts/write-stub-package-jsons.ts && pnpm biome check --write .","test:watch":"pnpm vitest","test":"pnpm vitest run","prepublishOnly":"tsx ../../scripts/check-versions.ts"},"_lastModified":"2026-04-24T10:36:51.643Z"}

package/dist/server/ai-employees/ai-conversations.d.ts

@@ -0,0 +1,70 @@
+/**
+ * This file is part of the NocoBase (R) project.
+ * Copyright (c) 2020-2024 NocoBase Co., Ltd.
+ * Authors: NocoBase Team.
+ *
+ * This project is dual-licensed under AGPL-3.0 and NocoBase Commercial License.
+ * For more information, please refer to: https://www.nocobase.com/agreement.
+ */
+import { Model, Transaction } from '@nocobase/database';
+import PluginAIServer from '../plugin';
+import { AIMessage, SubAgentConversationMetadata, UserDecision } from '../types';
+export type AIConversationsOptions = {
+    systemMessage?: unknown;
+    skillSettings?: unknown;
+    conversationSettings?: unknown;
+    [key: string]: unknown;
+};
+export type AIConversationFilterParams = {
+    filter: {
+        userId: string;
+    };
+};
+export type CreateAIConversationParams = {
+    userId?: string;
+    aiEmployee: {
+        username: string;
+    };
+    title?: string;
+    options?: AIConversationsOptions;
+    from?: 'main-agent' | 'sub-agent';
+    transaction?: Transaction;
+    category?: 'chat' | 'task';
+};
+export type UpdateAIConversationParams = {
+    userId: string;
+    sessionId: string;
+    title?: string;
+    options?: AIConversationsOptions;
+};
+export type GetAIConversationMessagesParams = {
+    userId: string;
+    sessionId: string;
+    cursor?: string;
+    paginate?: boolean;
+};
+export type ParsedMessageRow = AIMessage & Model;
+export type GetAIConversationMessagesResult = {
+    rows: any[];
+    hasMore?: boolean;
+    cursor?: string | null;
+};
+export declare class AIConversationsManager {
+    protected plugin: PluginAIServer;
+    constructor(plugin: PluginAIServer);
+    create({ userId, aiEmployee, title, options, from, transaction, category, }: CreateAIConversationParams): Promise<any>;
+    update({ userId, sessionId, title, options: inputOptions }: UpdateAIConversationParams): Promise<any>;
+    getConversation({ sessionId, userId }: {
+        sessionId: string;
+        userId?: string;
+    }): Promise<any>;
+    getMessages({ userId, sessionId, cursor, paginate, }: GetAIConversationMessagesParams): Promise<GetAIConversationMessagesResult>;
+    getUserDecisions(messageId: string): Promise<{
+        interruptId?: string;
+        decisions: UserDecision[];
+    } | undefined>;
+    resolveSubAgentConversation(sessionId: string, toolCallId: string): Promise<SubAgentConversationMetadata>;
+    private get aiConversationsRepo();
+    private get aiMessagesRepo();
+    private get aiToolMessagesRepo();
+}

package/dist/server/ai-employees/ai-conversations.js

@@ -0,0 +1,325 @@
+/**
+ * This file is part of the NocoBase (R) project.
+ * Copyright (c) 2020-2024 NocoBase Co., Ltd.
+ * Authors: NocoBase Team.
+ *
+ * This project is dual-licensed under AGPL-3.0 and NocoBase Commercial License.
+ * For more information, please refer to: https://www.nocobase.com/agreement.
+ */
+
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __export = (target, all) => {
+  for (var name in all)
+    __defProp(target, name, { get: all[name], enumerable: true });
+};
+var __copyProps = (to, from, except, desc) => {
+  if (from && typeof from === "object" || typeof from === "function") {
+    for (let key of __getOwnPropNames(from))
+      if (!__hasOwnProp.call(to, key) && key !== except)
+        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
+  }
+  return to;
+};
+var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
+var ai_conversations_exports = {};
+__export(ai_conversations_exports, {
+  AIConversationsManager: () => AIConversationsManager
+});
+module.exports = __toCommonJS(ai_conversations_exports);
+var import_database = require("@nocobase/database");
+var import_utils = require("../utils");
+class AIConversationsManager {
+  constructor(plugin) {
+    this.plugin = plugin;
+  }
+  async create({
+    userId,
+    aiEmployee,
+    title,
+    options = {},
+    from = "main-agent",
+    transaction,
+    category = "chat"
+  }) {
+    return await this.aiConversationsRepo.create({
+      values: {
+        userId,
+        title,
+        aiEmployee,
+        options,
+        thread: 1,
+        from,
+        category
+      },
+      transaction
+    });
+  }
+  async update({ userId, sessionId, title, options: inputOptions }) {
+    const conversation = await this.getConversation({
+      sessionId,
+      userId
+    });
+    if (!conversation) {
+      throw new Error("invalid sessionId");
+    }
+    const { systemMessage, skillSettings, conversationSettings } = inputOptions ?? {};
+    const options = conversation.options ?? {};
+    if (systemMessage) {
+      options["systemMessage"] = systemMessage;
+    }
+    if (skillSettings) {
+      options["skillSettings"] = skillSettings;
+    }
+    if (conversationSettings) {
+      options["conversationSettings"] = conversationSettings;
+    }
+    const values = { options };
+    if (title) {
+      values.title = title;
+    }
+    return await this.aiConversationsRepo.update({
+      filter: {
+        userId,
+        sessionId
+      },
+      values
+    });
+  }
+  async getConversation({ sessionId, userId }) {
+    const conversation = await this.aiConversationsRepo.findOne({
+      filter: {
+        sessionId
+      }
+    });
+    if (!userId) {
+      return conversation;
+    }
+    if (!conversation) {
+      return null;
+    }
+    let ownershipCheck = await this.aiConversationsRepo.count({
+      filter: {
+        sessionId,
+        userId
+      }
+    });
+    if (ownershipCheck === 0) {
+      ownershipCheck = await this.plugin.db.getRepository("aiWorkflowTasks").count({
+        filter: {
+          sessionId,
+          "users.id": userId
+        }
+      });
+    }
+    if (ownershipCheck) {
+      return conversation;
+    } else {
+      return null;
+    }
+  }
+  async getMessages({
+    userId,
+    sessionId,
+    cursor,
+    paginate = true
+  }) {
+    const conversation = await this.getConversation({
+      sessionId,
+      userId
+    });
+    if (!conversation) {
+      throw new Error("invalid sessionId");
+    }
+    const pageSize = 10;
+    const maxLimit = 200;
+    const messageRepository = this.plugin.db.getRepository("aiConversations.messages", sessionId);
+    const filter = {
+      role: {
+        $notIn: ["tool"]
+      }
+    };
+    if (paginate && cursor) {
+      filter["messageId"] = {
+        $lt: cursor
+      };
+    }
+    const rows = await messageRepository.find({
+      sort: ["-messageId"],
+      limit: paginate ? pageSize + 1 : maxLimit,
+      filter
+    });
+    const hasMore = paginate && rows.length > pageSize;
+    const data = hasMore ? rows.slice(0, -1) : rows;
+    const newCursor = data.length ? data[data.length - 1].messageId : null;
+    const subAgentConversations = data.filter((row) => {
+      var _a, _b;
+      return ((_b = (_a = row.metadata) == null ? void 0 : _a.subAgentConversations) == null ? void 0 : _b.length) ?? 0 > 0;
+    }).flatMap((row) => row.metadata.subAgentConversations);
+    const subAgentConversationSessionIds = [...new Set(subAgentConversations.map((item) => item.sessionId))];
+    const subAgentConversationMessages = subAgentConversationSessionIds.length ? await this.aiMessagesRepo.find({
+      sort: ["messageId"],
+      filter: {
+        sessionId: {
+          $in: subAgentConversationSessionIds
+        },
+        role: {
+          $notIn: ["tool"]
+        }
+      }
+    }) : [];
+    const subAgentConversationMessageMap = /* @__PURE__ */ new Map();
+    const toolCallIds = [
+      ...data.filter((row) => {
+        var _a;
+        return ((_a = row == null ? void 0 : row.toolCalls) == null ? void 0 : _a.length) ?? 0 > 0;
+      }).flatMap((row) => row.toolCalls).map((toolCall) => toolCall.id),
+      ...subAgentConversationMessages.filter((row) => {
+        var _a;
+        return ((_a = row == null ? void 0 : row.toolCalls) == null ? void 0 : _a.length) ?? 0 > 0;
+      }).flatMap((row) => row.toolCalls).map((toolCall) => toolCall.id)
+    ];
+    const toolMessages = await this.aiToolMessagesRepo.find({
+      filter: {
+        toolCallId: {
+          $in: toolCallIds
+        }
+      }
+    });
+    const toolMessageKey = (messageId, toolCallId) => `${messageId}:${toolCallId}`;
+    const toolMessageMap = new Map(
+      toolMessages.map((toolMessage) => [
+        toolMessageKey(toolMessage.messageId, toolMessage.toolCallId),
+        toolMessage
+      ])
+    );
+    const toolsList = await this.plugin.aiManager.toolManager.listTools();
+    const toolsMap = new Map(
+      toolsList.map((group) => group.tools).flat().map((tool) => [tool.name, tool])
+    );
+    const parseMessageRow = (row) => {
+      var _a, _b;
+      if (((_a = row == null ? void 0 : row.toolCalls) == null ? void 0 : _a.length) ?? 0 > 0) {
+        for (const toolCall of row.toolCalls) {
+          const tool = toolsMap.get(toolCall.name);
+          const toolMessage = toolMessageMap.get(toolMessageKey(row.messageId, toolCall.id));
+          toolCall.invokeStatus = toolMessage == null ? void 0 : toolMessage.invokeStatus;
+          toolCall.invokeStartTime = toolMessage == null ? void 0 : toolMessage.invokeStartTime;
+          toolCall.invokeEndTime = toolMessage == null ? void 0 : toolMessage.invokeEndTime;
+          toolCall.auto = toolMessage == null ? void 0 : toolMessage.auto;
+          toolCall.status = toolMessage == null ? void 0 : toolMessage.status;
+          toolCall.content = toolMessage == null ? void 0 : toolMessage.content;
+          toolCall.execution = tool == null ? void 0 : tool.execution;
+          toolCall.willInterrupt = (tool == null ? void 0 : tool.execution) === "frontend" || (toolMessage == null ? void 0 : toolMessage.auto) === false;
+          toolCall.defaultPermission = tool == null ? void 0 : tool.defaultPermission;
+        }
+      }
+      const providerOptions = this.plugin.aiManager.llmProviders.get((_b = row.metadata) == null ? void 0 : _b.provider);
+      if (!providerOptions) {
+        return (0, import_utils.parseResponseMessage)(row);
+      }
+      const Provider = providerOptions.provider;
+      const provider = new Provider({
+        app: this.plugin.app
+      });
+      return provider.parseResponseMessage(row);
+    };
+    for (const row of subAgentConversationMessages) {
+      const sessionMessages = subAgentConversationMessageMap.get(row.sessionId) ?? [];
+      sessionMessages.push(parseMessageRow(row));
+      sessionMessages.forEach((it) => it.content.from = "sub-agent");
+      subAgentConversationMessageMap.set(row.sessionId, sessionMessages);
+    }
+    return {
+      rows: data.map((row) => {
+        var _a;
+        const parsedRow = parseMessageRow(row);
+        const subAgentConversationItems = ((_a = row.metadata) == null ? void 0 : _a.subAgentConversations) ?? [];
+        if (subAgentConversationItems.length) {
+          parsedRow.content.subAgentConversations = subAgentConversationItems.map((item) => ({
+            sessionId: item.sessionId,
+            toolCallId: item.toolCallId,
+            status: item.status,
+            messages: subAgentConversationMessageMap.get(item.sessionId) ?? []
+          }));
+        }
+        parsedRow.content.from = "main-agent";
+        return parsedRow;
+      }),
+      ...paginate && {
+        hasMore,
+        cursor: newCursor
+      }
+    };
+  }
+  async getUserDecisions(messageId) {
+    var _a;
+    const allInterruptedToolCall = await this.aiToolMessagesRepo.find({
+      filter: {
+        messageId,
+        interruptActionOrder: { [import_database.Op.not]: null }
+      },
+      order: [["interruptActionOrder", "ASC"]]
+    });
+    if (!allInterruptedToolCall.every((t) => t.invokeStatus === "waiting")) {
+      return;
+    }
+    const message = await this.aiMessagesRepo.findOne({
+      filter: {
+        messageId
+      }
+    });
+    const interruptId = (_a = message == null ? void 0 : message.get("metadata")) == null ? void 0 : _a.interruptId;
+    return {
+      interruptId,
+      decisions: allInterruptedToolCall.map((item) => item.userDecision)
+    };
+  }
+  async resolveSubAgentConversation(sessionId, toolCallId) {
+    var _a, _b;
+    if (!sessionId || !toolCallId) {
+      return null;
+    }
+    const toolMessage = await this.aiToolMessagesRepo.findOne({
+      filter: {
+        sessionId,
+        toolCallId
+      }
+    });
+    if (!toolMessage) {
+      return null;
+    }
+    const aiMessage = await this.aiMessagesRepo.findOne({
+      filter: {
+        sessionId,
+        messageId: toolMessage.messageId
+      }
+    });
+    if (!aiMessage) {
+      return null;
+    }
+    if (!((_b = (_a = aiMessage.metadata) == null ? void 0 : _a.subAgentConversations) == null ? void 0 : _b.length)) {
+      return null;
+    }
+    const subAgentConversation = aiMessage.metadata.subAgentConversations.find((it) => it.toolCallId == toolCallId);
+    if (!subAgentConversation) {
+      return null;
+    }
+    return subAgentConversation;
+  }
+  get aiConversationsRepo() {
+    return this.plugin.db.getRepository("aiConversations");
+  }
+  get aiMessagesRepo() {
+    return this.plugin.db.getRepository("aiMessages");
+  }
+  get aiToolMessagesRepo() {
+    return this.plugin.db.getRepository("aiToolMessages");
+  }
+}
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  AIConversationsManager
+});

package/dist/server/ai-employees/ai-employee.d.ts

@@ -7,43 +7,73 @@
  * For more information, please refer to: https://www.nocobase.com/agreement.
  */
 import { Model, Transaction } from '@nocobase/database';
-import { Context } from '@nocobase/actions';
 import { LLMProvider } from '../llm-providers/provider';
-import { AIChatContext, AIChatConversation, AIMessageInput, UserDecision } from '../types';
-import { DocumentSegmentedWithScore } from '../features';
-import { KnowledgeBaseGroup } from '../types';
+import { AIChatContext, AIChatConversation, AIMessageInput, AIToolCall, UserDecision } from '../types';
 import { ToolsEntry } from '@nocobase/ai';
 import { AIToolMessage } from '../types/ai-message.type';
+import { Context } from '@nocobase/actions';
 export interface ModelRef {
     llmService: string;
     model: string;
 }
+export interface AIEmployeeOptions {
+    ctx: Context;
+    employee: Model;
+    sessionId: string;
+    systemMessage?: string;
+    skillSettings?: Record<string, any>;
+    webSearch?: boolean;
+    model?: ModelRef;
+    legacy?: boolean;
+    from?: 'main-agent' | 'sub-agent';
+    tools?: {
+        name: string;
+    }[];
+}
 export declare class AIEmployee {
+    sessionId: string;
+    from: string;
     employee: Model;
     aiChatConversation: AIChatConversation;
     skillSettings?: Record<string, any>;
     private plugin;
     private db;
-    private sessionId;
     private ctx;
     private systemMessage;
+    private protocol;
     private webSearch?;
     private model?;
     private legacy?;
-    private protocol;
-    constructor(ctx: Context, employee: Model, sessionId: string, systemMessage?: string, skillSettings?: Record<string, any>, webSearch?: boolean, model?: ModelRef, legacy?: boolean);
+    private tools;
+    private inWorkflow?;
+    constructor({ ctx, employee, sessionId, systemMessage, skillSettings, webSearch, model, legacy, from, tools, }: AIEmployeeOptions);
+    getFormatMessages(userMessages: AIMessageInput[]): Promise<{
+        role: "user" | "assistant" | "tool" | "system";
+        content: any;
+        tool_call_id?: string;
+        tool_calls?: AIToolCall[];
+    }[]>;
+    isInWorkflow(): Promise<boolean>;
     private buildState;
     private initSession;
     private buildChatContext;
     stream({ messageId, userMessages, userDecisions, }: {
         messageId?: string;
         userMessages?: AIMessageInput[];
-        userDecisions?: UserDecision[];
+        userDecisions?: {
+            interruptId?: string;
+            decisions: UserDecision[];
+        };
     }): Promise<boolean>;
-    invoke({ messageId, userMessages, userDecisions, }: {
+    invoke({ messageId, userMessages, userDecisions, writer, context, }: {
         messageId?: string;
         userMessages?: AIMessageInput[];
-        userDecisions?: UserDecision[];
+        userDecisions?: {
+            interruptId?: string;
+            decisions: UserDecision[];
+        };
+        writer?: (chunk: any) => void;
+        context?: any;
     }): Promise<any>;
     createAgent({ provider, systemPrompt, tools, middleware, }: {
         provider: LLMProvider;
@@ -74,21 +104,15 @@ export declare class AIEmployee {
         allowEmpty?: boolean;
         responseMetadata: Map<string, any>;
     }): Promise<void>;
+    private handleInterruptedToolCalls;
     getEmployeeDataSourceContext(): string;
     getSystemPrompt(userMessages: AIMessageInput[]): Promise<string>;
-    retrieveKnowledgeBase(userMessage: AIMessageInput): Promise<DocumentSegmentedWithScore[]>;
-    isEnabledKnowledgeBase(): boolean;
-    getAIEmployeeKnowledgeBaseConfig(): {
-        topK: number;
-        score: string;
-    };
-    getKnowledgeBaseGroup(): Promise<KnowledgeBaseGroup[]>;
     initToolCall(transaction: Transaction, messageId: string, toolCalls: {
         id: string;
         name: string;
         args: any;
     }[]): Promise<Model<AIToolMessage>[]>;
-    updateToolCallInterrupted(messageId: string, toolCallId: string, interruptAction: {
+    updateToolCallInterrupted(sessionId: string, messageId: string, toolCallId: string, interruptId: string, interruptAction: {
         order: number;
         description?: string;
         allowed_decisions?: string[];
@@ -98,7 +122,6 @@ export declare class AIEmployee {
     confirmToolCall(transaction: Transaction, messageId: string, toolCallIds: string[]): Promise<number>;
     getToolCallResult(messageId: string, toolCallId: string): Promise<AIToolMessage>;
     getToolCallResultMap(messageId: string, toolCallIds: string[]): Promise<Map<string, AIToolMessage>>;
-    getUserDecisions(messageId: string): Promise<UserDecision[]>;
     cancelToolCall(): Promise<any>;
     get logger(): any;
     sendErrorResponse(errorMessage: string): void;
@@ -118,11 +141,17 @@ export declare class AIEmployee {
     private getToolCallMap;
     private toInterruptActions;
     private getAIEmployeeTools;
+    private getAvailableSkills;
+    private getAgentTools;
+    getLoadedSkillNames(): Promise<string[]>;
+    getActivatedSkillToolNames(): Promise<Set<string>>;
+    private getAvailableAIEmployees;
     private getMiddleware;
     private getCurrentThread;
     private forkCurrentThread;
     getToolsMap(): Promise<Map<string, import("@nocobase/ai").ToolsOptions>>;
     private listTools;
+    private withRunMetadata;
     private get toolsManager();
     private get aiConversationsRepo();
     private get aiMessagesRepo();
@@ -131,3 +160,55 @@ export declare class AIEmployee {
     private get aiToolMessagesModel();
     private get aiFilesModel();
 }
+export declare class ChatStreamProtocol {
+    private readonly streamConsumer;
+    private _statistics;
+    constructor(streamConsumer: StreamConsumer);
+    static fromContext(ctx: Context): ChatStreamProtocol;
+    with(conversation: {
+        sessionId: string;
+        from: string;
+        username: string;
+    }): {
+        startStream: () => void;
+        endStream: () => void;
+        subAgentCompleted: () => void;
+        newMessage: (content?: unknown) => void;
+        content: (content: string) => void;
+        webSearch: (content: {
+            type: string;
+            query: string;
+        }[]) => void;
+        reasoning: (content: {
+            status: string;
+            content: string;
+        }) => void;
+        stopReasoning: () => void;
+        toolCallChunks: (content: unknown) => void;
+        toolCalls: (content: unknown) => void;
+        toolCallStatus: ({ toolCall, invokeStatus, status, invokeStartTime, invokeEndTime, content, interruptAction, }: {
+            toolCall: {
+                messageId: string;
+                id: string;
+                name: string;
+                willInterrupt: boolean;
+            };
+            invokeStatus: string;
+            status?: string;
+            invokeStartTime?: string | Date | null;
+            invokeEndTime?: string | Date | null;
+            content?: unknown;
+            interruptAction?: {
+                order: number;
+                description: string;
+                allowedDecisions: string[];
+            };
+        }) => void;
+    };
+    get statistics(): {
+        sent: number;
+    };
+}
+export type StreamConsumer = {
+    write: (chunk: any) => void;
+};