@kweaver-ai/chatkit 0.1.11 → 0.1.13

package/dist/components/base/ChatKitBase.d.ts

@@ -1,5 +1,5 @@
 import { Component } from 'react';
-import { ChatMessage, ApplicationContext, ChatKitInterface, OnboardingInfo, WebSearchQuery, ExecuteCodeResult, Text2SqlResult, Text2MetricResult, AfSailorResult, DatasourceFilterResult, DatasourceRerankResult, ChartDataSchema, DefaultToolResult } from '../../types';
+import { ChatMessage, ApplicationContext, ChatKitInterface, OnboardingInfo, WebSearchQuery, ExecuteCodeResult, Text2SqlResult, Text2MetricResult, AfSailorResult, DatasourceFilterResult, DatasourceRerankResult, ChartDataSchema, DefaultToolResult, MessageContext } from '../../types';
 /**
  * ChatKitBase 组件的属性接口
  */
@@ -56,6 +56,11 @@ export declare abstract class ChatKitBase<P extends ChatKitBaseProps = ChatKitBa
      */
     private isInitializing;
     private hasInitialized;
+    /**
+     * 流式响应时是否处于「按 chunk 批处理」中，避免同一 chunk 内多行事件触发多次 setState 导致 Maximum update depth exceeded
+     */
+    private _streamingBatch;
+    private _pendingStreamingUpdates;
     /**
      * 调用接口时携带的令牌
      */
@@ -255,6 +260,26 @@ export declare abstract class ChatKitBase<P extends ChatKitBaseProps = ChatKitBa
      * @param consumeTime 耗时（毫秒），可选
      */
     protected appendDefaultToolBlock(messageId: string, toolName: string, result: DefaultToolResult, consumeTime?: number): void;
+    /**
+     * 更新指定消息的 messageContext（合并更新）
+     * 用于在流式过程中或流结束时写入相关问题、耗时、Token 等辅助信息
+     */
+    protected updateMessageContext(messageId: string, patch: Partial<MessageContext>): void;
+    /**
+     * 流式期间将状态更新入队，在 handleStreamResponse 每 chunk 结束时统一 flush，避免同一 chunk 内多行事件导致多次 setState 触发 Maximum update depth exceeded
+     */
+    protected applyStreamingUpdate(updater: (prev: ChatKitBaseState) => Partial<ChatKitBaseState>): void;
+    /**
+     * 将当前批内的所有流式更新合并为一次 setState 执行
+     */
+    protected flushStreamingUpdates(): void;
+    /**
+     * 更新指定工具名称对应的最后一个工具块的结果（用于流式工具如 contextloader_data_enhanced 的组装）
+     * @param messageId 消息 ID
+     * @param toolName 工具名称
+     * @param result 更新后的默认工具结果
+     */
+    protected updateDefaultToolBlockResult(messageId: string, toolName: string, result: DefaultToolResult): void;
     /**
      * 创建新的会话
      * 内部会调用子类实现的 generateConversation() 和 getOnboardingInfo() 方法

package/dist/components/base/DrawerPortalContext.d.ts

@@ -0,0 +1,16 @@
+import React from 'react';
+export interface DrawerPortalProviderProps {
+    /** Portal 挂载的 DOM 节点；不传或 null 表示使用默认（root 优先，再 body） */
+    value?: HTMLElement | null;
+    children: React.ReactNode;
+}
+export declare function DrawerPortalProvider({ value, children }: DrawerPortalProviderProps): import("react/jsx-runtime").JSX.Element;
+/**
+ * 获取抽屉 Portal 挂载容器（来自 Context）。
+ * 返回 null 表示使用组件内部默认：document.getElementById('root') ?? document.body
+ */
+export declare function useDrawerPortalContainer(): HTMLElement | null;
+/**
+ * 解析最终挂载节点：prop/context 优先，否则 root 再 body（默认不挂到 body，而是 root）
+ */
+export declare function resolveDrawerPortalContainer(propContainer: HTMLElement | null | undefined, contextContainer: HTMLElement | null): HTMLElement;

package/dist/components/base/ToolBlockContext.d.ts

@@ -0,0 +1,38 @@
+import React from 'react';
+/**
+ * 内置工具抽屉展示所需的数据（与 ToolDrawer 展示内容对应）
+ * 状态与数据存放在 Context，避免 ToolBlock 因虚拟滚动/消息更新重渲染导致抽屉状态重置
+ */
+export interface ToolDrawerPayload {
+    toolName: string;
+    toolTitle: string;
+    toolIcon?: string | React.ReactNode;
+    input?: any;
+    output?: any;
+}
+/**
+ * 工具块上下文：管理「注册工具 onClick 返回的关闭函数」及「内置工具抽屉的显示状态与数据」。
+ * 点击任意工具块时先调用上次存储的关闭函数（若有），再执行当前点击逻辑；
+ * 若当前是注册工具且 onClick 返回了函数，则存储起来，供下次切换时调用。
+ * 内置抽屉的 open/close 与 payload 由 Context 持有，ToolDrawer 在 AssistantBase 单例渲染，ToolBlock 只负责调用 openToolDrawer/closeToolDrawer。
+ */
+export interface ToolBlockContextValue {
+    /** 执行上次注册的关闭函数并清空；切换工具块时由 ToolBlock 先调用 */
+    closeRegistered: () => void;
+    /** 存储本次打开的关闭函数（onClick 的返回值）；不返回或非函数则存空 */
+    registerClose: (close: (() => void) | undefined) => void;
+    /** 是否有内置工具抽屉处于打开状态（用于控制消息列表滚动行为等） */
+    hasOpenDrawer: boolean;
+    /** 更新内置工具抽屉的打开状态（内部使用，也可供外部同步） */
+    setHasOpenDrawer: (open: boolean) => void;
+    /** 当前要展示的抽屉数据，为 null 表示抽屉关闭 */
+    drawerPayload: ToolDrawerPayload | null;
+    /** 打开内置工具抽屉并设置展示数据 */
+    openToolDrawer: (payload: ToolDrawerPayload) => void;
+    /** 关闭内置工具抽屉并清空数据 */
+    closeToolDrawer: () => void;
+}
+export declare function ToolBlockProvider({ children }: {
+    children: React.ReactNode;
+}): import("react/jsx-runtime").JSX.Element;
+export declare function useToolBlockContext(): ToolBlockContextValue;

package/dist/components/base/assistant/AssistantBase.d.ts

@@ -7,6 +7,11 @@ import { ChatMessage } from '../../../types';
 export interface AssistantBaseProps extends ChatKitBaseProps {
     /** 是否启用历史对话功能，默认为 true */
     enableHistory?: boolean;
+    /**
+     * ToolDrawer Portal 挂载的 DOM 节点（可选）。
+     * 不传时抽屉挂到主内容区域 div（flex h-full w-full bg-white）下，便于在乾坤等微前端中正确挂载
+     */
+    drawerContainer?: HTMLElement | null;
 }
 /**
  * AssistantBase 组件的状态接口
@@ -14,6 +19,8 @@ export interface AssistantBaseProps extends ChatKitBaseProps {
  */
 export interface AssistantBaseState extends ChatKitBaseState {
     showHistory: boolean;
+    /** 主内容区域 DOM 引用，用于未传 drawerContainer 时作为抽屉挂载点 */
+    drawerContainerEl: HTMLElement | null;
 }
 /**
  * AssistantBase 基础组件
@@ -64,6 +71,8 @@ export declare abstract class AssistantBase<P extends AssistantBaseProps = Assis
      * @param previousUserMessage 上一条用户消息
      */
     handleRegenerate: (messageId: string, previousUserMessage: ChatMessage) => Promise<void>;
+    /** 点击相关问题：作为新问题发送 */
+    handleSelectQuestion: (question: string) => void;
     /**
      * 实现 React.Component.render() 方法
      * 渲染 Assistant 模式的界面

package/dist/components/base/assistant/MessageItem.d.ts

@@ -12,6 +12,10 @@ interface MessageItemProps {
     agentAvatar?: string;
     /** 重新生成回调函数 */
     onRegenerate?: (messageId: string) => Promise<void>;
+    /** 是否为最后一条助手消息（用于展示相关问题） */
+    isLastAssistantMessage?: boolean;
+    /** 点击相关问题时的回调（用于继续提问） */
+    onSelectQuestion?: (question: string) => void;
 }
 /**
  * MessageItem 组件

package/dist/components/base/assistant/MessageList.d.ts

@@ -12,6 +12,8 @@ interface MessageListProps {
     agentAvatar?: string;
     /** 重新生成回调函数 */
     onRegenerate?: (messageId: string, previousUserMessage: ChatMessage) => Promise<void>;
+    /** 点击相关问题时的回调（用于继续提问） */
+    onSelectQuestion?: (question: string) => void;
 }
 declare const MessageList: React.FC<MessageListProps>;
 export default MessageList;

package/dist/components/base/assistant/MessageStatsBar.d.ts

@@ -0,0 +1,15 @@
+import React from 'react';
+export interface MessageStatsBarProps {
+    /** 本次回答耗时（秒） */
+    elapsedSeconds?: number;
+    /** 本次回答 Token 总数 */
+    totalTokens?: number;
+    /** 自定义样式类名 */
+    className?: string;
+}
+/**
+ * 消息统计信息组件
+ * 在消息底部展示耗时和 Token 总数
+ */
+declare const MessageStatsBar: React.FC<MessageStatsBarProps>;
+export default MessageStatsBar;

package/dist/components/base/assistant/RelatedQuestions.d.ts

@@ -0,0 +1,15 @@
+import React from 'react';
+export interface RelatedQuestionsProps {
+    /** 相关推荐问题列表 */
+    questions: string[];
+    /** 点击某条问题时的回调（用于继续提问） */
+    onSelectQuestion?: (question: string) => void;
+    /** 自定义样式类名 */
+    className?: string;
+}
+/**
+ * 相关问题列表组件
+ * 展示在最后一条助手消息下方，支持点击继续提问
+ */
+declare const RelatedQuestions: React.FC<RelatedQuestionsProps>;
+export default RelatedQuestions;

package/dist/components/base/assistant/TaskInProgress.d.ts

@@ -8,7 +8,7 @@ interface TaskInProgressProps {
 }
 /**
  * TaskInProgress 组件
- * 显示任务进行中的加载动画
+ * 显示任务进行中的加载动画。使用 React.memo 避免流式更新消息时随父组件重复渲染导致卡顿。
  */
 declare const TaskInProgress: React.FC<TaskInProgressProps>;
 export default TaskInProgress;

package/dist/components/base/assistant/blocks/ToolBlock.d.ts

@@ -11,6 +11,7 @@ export interface ToolBlockProps {
  * ToolBlock 组件
  * 用于在消息流中渲染工具调用块的通用组件
  * 支持传入自定义图标和标题，并可点击右侧箭头查看详情
+ * 抽屉的显示状态与数据由 ToolBlockContext 持有，ToolDrawer 在 AssistantBase 单例渲染，本组件只负责调用 openToolDrawer/closeToolDrawer，避免虚拟滚动或消息更新导致抽屉状态重置。
  *
  * 根据设计文档（ChatKit for DIP.pdf）的 Block工具处理流程：
  * 1. 获取已注册工具名称

package/dist/components/base/assistant/blocks/ToolDrawer.d.ts

@@ -17,6 +17,11 @@ export interface ToolDrawerProps {
     input?: any;
     /** 工具输出结果 */
     output?: any;
+    /**
+     * Portal 挂载的 DOM 节点（可选）。
+     * 不传时优先使用 DrawerPortalContext，再默认挂到 document.getElementById('root') ?? document.body
+     */
+    container?: HTMLElement | null;
 }
 /**
  * ToolDrawer 组件

package/dist/components/base/copilot/CopilotBase.d.ts

@@ -1,5 +1,16 @@
 import { ChatKitBase, ChatKitBaseProps, ChatKitBaseState } from '../ChatKitBase';
 import { ChatMessage } from '../../../types';
+/**
+ * CopilotBase 组件的属性接口
+ * 扩展 ChatKitBaseProps，支持抽屉 Portal 挂载节点配置
+ */
+export interface CopilotBaseProps extends ChatKitBaseProps {
+    /**
+     * ToolDrawer Portal 挂载的 DOM 节点（可选）。
+     * 不传时抽屉挂到 document.getElementById('root') ?? document.body，避免虚拟滚动影响定位
+     */
+    drawerContainer?: HTMLElement | null;
+}
 /**
  * CopilotBase 组件的状态接口
  * 扩展 ChatKitBaseState（当前不需要额外状态，保留接口以便未来扩展）
@@ -13,7 +24,7 @@ export interface CopilotBaseState extends ChatKitBaseState {
  * 该类继承自 ChatKitBase，实现了 Copilot 模式的交互界面和交互逻辑
  * 区别于 AssistantBase，CopilotBase 在 render() 函数中渲染 Copilot 模式的界面
  */
-export declare abstract class CopilotBase<P extends ChatKitBaseProps = ChatKitBaseProps> extends ChatKitBase<P> {
+export declare abstract class CopilotBase<P extends CopilotBaseProps = CopilotBaseProps> extends ChatKitBase<P> {
     /**
      * 组件状态，包含历史对话显示状态
      */
@@ -42,6 +53,8 @@ export declare abstract class CopilotBase<P extends ChatKitBaseProps = ChatKitBa
      * @param previousUserMessage 上一条用户消息
      */
     handleRegenerate: (messageId: string, previousUserMessage: ChatMessage) => Promise<void>;
+    /** 点击相关问题：作为新问题发送 */
+    handleSelectQuestion: (question: string) => void;
     /**
      * 实现 React.Component.render() 方法
      * 渲染 Copilot 模式的界面

package/dist/components/base/copilot/MessageItem.d.ts

@@ -10,6 +10,10 @@ interface MessageItemProps {
     isStreaming?: boolean;
     /** 重新生成回调函数 */
     onRegenerate?: (messageId: string) => Promise<void>;
+    /** 是否为最后一条助手消息（用于展示相关问题） */
+    isLastAssistantMessage?: boolean;
+    /** 点击相关问题时的回调（用于继续提问） */
+    onSelectQuestion?: (question: string) => void;
 }
 /**
  * MessageItem 组件

package/dist/components/base/copilot/MessageList.d.ts

@@ -10,6 +10,8 @@ interface MessageListProps {
     streamingMessageId?: string | null;
     /** 重新生成回调函数 */
     onRegenerate?: (messageId: string, previousUserMessage: ChatMessage) => Promise<void>;
+    /** 点击相关问题时的回调（用于继续提问） */
+    onSelectQuestion?: (question: string) => void;
 }
 declare const MessageList: React.FC<MessageListProps>;
 export default MessageList;

package/dist/components/base/copilot/MessageStatsBar.d.ts

@@ -0,0 +1,15 @@
+import React from 'react';
+export interface MessageStatsBarProps {
+    /** 本次回答耗时（秒） */
+    elapsedSeconds?: number;
+    /** 本次回答 Token 总数 */
+    totalTokens?: number;
+    /** 自定义样式类名 */
+    className?: string;
+}
+/**
+ * Copilot 模式下的消息统计信息组件
+ * 在消息底部展示耗时和 Token 总数
+ */
+declare const MessageStatsBar: React.FC<MessageStatsBarProps>;
+export default MessageStatsBar;

package/dist/components/base/copilot/RelatedQuestions.d.ts

@@ -0,0 +1,15 @@
+import React from 'react';
+export interface RelatedQuestionsProps {
+    /** 相关推荐问题列表 */
+    questions: string[];
+    /** 点击某条问题时的回调（用于继续提问） */
+    onSelectQuestion?: (question: string) => void;
+    /** 自定义样式类名 */
+    className?: string;
+}
+/**
+ * Copilot 模式下的相关问题列表组件
+ * 展示在最后一条助手消息下方，支持点击继续提问
+ */
+declare const RelatedQuestions: React.FC<RelatedQuestionsProps>;
+export default RelatedQuestions;

package/dist/components/base/copilot/blocks/ToolBlock.d.ts

@@ -11,6 +11,7 @@ export interface ToolBlockProps {
  * ToolBlock 组件
  * 用于在消息流中渲染工具调用块的通用组件
  * 支持传入自定义图标和标题，并可点击右侧箭头查看详情
+ * 抽屉的显示状态与数据由 ToolBlockContext 持有，ToolDrawer 在 CopilotBase 单例渲染，本组件只负责调用 openToolDrawer/closeToolDrawer，避免虚拟滚动或消息更新导致抽屉状态重置。
  *
  * 根据设计文档（ChatKit for DIP.pdf）的 Block工具处理流程：
  * 1. 获取已注册工具名称

package/dist/components/base/copilot/blocks/ToolDrawer.d.ts

@@ -17,6 +17,11 @@ export interface ToolDrawerProps {
     input?: any;
     /** 工具输出结果 */
     output?: any;
+    /**
+     * Portal 挂载的 DOM 节点（可选）。
+     * 不传时优先使用 DrawerPortalContext，再默认挂到 document.getElementById('root') ?? document.body
+     */
+    container?: HTMLElement | null;
 }
 /**
  * ToolDrawer 组件

package/dist/components/dip/DIPBase.d.ts

@@ -117,6 +117,12 @@ export declare function DIPBaseMixin<TBase extends Constructor>(Base: TBase): {
         dipToken: string;
         /** DIP 刷新 token 的方法 */
         dipRefreshToken?: () => Promise<string>;
+        /** LeftHeaderTool 使用的 API 方法（构造时缓存，避免流式消息时重复创建） */
+        _leftHeaderApiMethods?: {
+            getKnowledgeNetworksDetail: (id: string) => Promise<any>;
+            getKnowledgeNetworkObjectTypes: (id: string, offset?: number, limit?: number) => Promise<any>;
+            getMetricInfoByIds: (ids: string[]) => Promise<any[]>;
+        };
         /**
          * 获取开场白和预置问题
          * 调用 AISHU DIP 的 agent-factory API 获取智能体配置信息，提取开场白和预置问题
@@ -398,6 +404,18 @@ export declare function DIPBaseMixin<TBase extends Constructor>(Base: TBase): {
          * 设置嵌套属性
          */
         setNestedProperty(obj: any, key: Array<string | number>, value: any): void;
+        /**
+         * 按路径不可变设置：只克隆从根到 key 路径上的节点，其余引用复用，避免全量深拷贝大对象
+         */
+        immutableSetNested(obj: any, key: Array<string | number>, value: any): any;
+        /**
+         * 不可变 upsert：按路径更新，不深拷贝整棵树
+         */
+        immutableApplyUpsert(obj: AssistantMessage, key: Array<string | number>, content: any): AssistantMessage;
+        /**
+         * 不可变 append：按路径追加后按路径更新，不深拷贝整棵树
+         */
+        immutableApplyAppend(obj: AssistantMessage, key: Array<string | number>, content: any): AssistantMessage;
         /**
          * 检查是否需要刷新 token
          * AISHU DIP 平台返回 401 状态码时表示 token 失效

package/dist/types/index.d.ts

@@ -97,14 +97,26 @@ export interface ToolBlockRegistration {
     Icon?: React.ReactNode;
     /** 工具点击事件
      * @param block 工具块数据，类型为 Record<string, any>
+     * @returns 可选返回关闭函数；切换工具块时 SDK 会先调用该函数，用于关闭本次打开的外层内容
      */
-    onClick?: (block: Record<string, any>) => void;
+    onClick?: (block: Record<string, any>) => void | (() => void);
 }
 /**
  * JSON2Plot 图表类型的消息块
  */
 export interface Json2PlotBlock extends ContentBlock<BlockType.JSON2PLOT, ChartDataSchema> {
 }
+/**
+ * 单条消息的扩展上下文（不参与流式内容渲染，用于相关问题、耗时、Token 等）
+ */
+export interface MessageContext {
+    /** 相关推荐问题列表，来自 message.ext.related_queries */
+    relatedQuestions?: string[];
+    /** 本次回答耗时（秒），用于展示「耗时: xx.xxs」 */
+    elapsedSeconds?: number;
+    /** Token 总数，用于展示「Token: x,xxx」 */
+    totalTokens?: number;
+}
 /**
  * 消息接口
  * 展示在消息区消息列表中的一条消息
@@ -120,6 +132,8 @@ export interface ChatMessage {
     content: Array<TextBlock | MarkdownBlock | WebSearchBlock | ToolBlock | Json2PlotBlock>;
     /** 与该消息关联的应用上下文（可选），仅用户消息可能包含此字段 */
     applicationContext?: ApplicationContext;
+    /** 消息扩展上下文：相关问题、耗时、Token 等，仅助手消息在流结束后可能包含 */
+    messageContext?: MessageContext;
 }
 /**
  * 应用上下文接口

package/dist/utils/BlockRegistry.d.ts

@@ -10,8 +10,9 @@ export interface ToolBlockRegistration {
     Icon?: React.ReactNode;
     /** 工具点击事件
      * @param block 工具块数据，类型为 Record<string, any>
+     * @returns 可选返回一个关闭函数；切换工具块时 SDK 会先调用该函数，用于关闭本次打开的外层内容（如抽屉/弹窗）
      */
-    onClick?: (block: Record<string, any>) => void;
+    onClick?: (block: Record<string, any>) => void | (() => void);
 }
 /**
  * Block 注册表

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@kweaver-ai/chatkit",
-  "version": "0.1.11",
+  "version": "0.1.13",
   "publishConfig": {
     "access": "public"
   },