@kweaver-ai/chatkit 0.1.12 → 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,19 @@ 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

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

@@ -1,8 +1,20 @@
 import React from 'react';
 /**
- * 工具块上下文：管理「注册工具 onClick 返回的关闭函数」。
+ * 内置工具抽屉展示所需的数据（与 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 先调用 */
@@ -11,8 +23,14 @@ export interface ToolBlockContextValue {
     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;

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

@@ -71,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/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/copilot/CopilotBase.d.ts

@@ -53,6 +53,8 @@ export declare abstract class CopilotBase<P extends CopilotBaseProps = CopilotBa
      * @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/types/index.d.ts

@@ -106,6 +106,17 @@ export interface ToolBlockRegistration {
  */
 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;
+}
 /**
  * 消息接口
  * 展示在消息区消息列表中的一条消息
@@ -121,6 +132,8 @@ export interface ChatMessage {
     content: Array<TextBlock | MarkdownBlock | WebSearchBlock | ToolBlock | Json2PlotBlock>;
     /** 与该消息关联的应用上下文（可选），仅用户消息可能包含此字段 */
     applicationContext?: ApplicationContext;
+    /** 消息扩展上下文：相关问题、耗时、Token 等，仅助手消息在流结束后可能包含 */
+    messageContext?: MessageContext;
 }
 /**
  * 应用上下文接口

package/package.json

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