@ant-design/agentic-ui 2.28.2 → 2.28.5

package/dist/Plugins/code/components/CodeRenderer.js

@@ -214,9 +214,7 @@ import { AceEditor, AceEditorContainer, CodeContainer, CodeToolbar, HtmlPreview,
         }
         // 只读模式下的思考块特殊渲染
         if (shouldRenderAsThinkBlock) {
-            return /*#__PURE__*/ React.createElement(ThinkBlock, {
-                element: props.element
-            });
+            return /*#__PURE__*/ React.createElement(ThinkBlock, props);
         }
         // 主要的代码编辑器渲染
         if (shouldRenderAsCodeEditor) {

package/dist/Plugins/code/components/ThinkBlock.d.ts

@@ -3,9 +3,35 @@
  * 只读模式下渲染思考类型的代码块
  */
 import React from 'react';
-import { CodeNode } from '../../../MarkdownEditor/el';
-interface ThinkBlockProps {
-    element: CodeNode;
+import { CodeNode, ElementProps } from '../../../MarkdownEditor/el';
+type ThinkBlockProps = ElementProps<CodeNode>;
+/**
+ * ThinkBlock Context 类型定义
+ */
+export interface ThinkBlockContextType {
+    /** 受控的展开状态 */
+    expanded?: boolean;
+    /** 展开状态变更回调 */
+    onExpandedChange?: (expanded: boolean) => void;
 }
-export declare function ThinkBlock({ element }: ThinkBlockProps): React.JSX.Element;
+/**
+ * ThinkBlock Context
+ */
+export declare const ThinkBlockContext: React.Context<ThinkBlockContextType | null>;
+/**
+ * ThinkBlock Provider 组件
+ *
+ * 用于全局控制所有 ThinkBlock 实例的展开/收起状态
+ *
+ * @example
+ * ```tsx
+ * <ThinkBlockProvider expanded={isExpanded} onExpandedChange={setIsExpanded}>
+ *   <MarkdownEditor ... />
+ * </ThinkBlockProvider>
+ * ```
+ */
+export declare const ThinkBlockProvider: React.FC<ThinkBlockContextType & {
+    children: React.ReactNode;
+}>;
+export declare function ThinkBlock(props: ThinkBlockProps): React.JSX.Element;
 export {};

package/dist/Plugins/code/components/ThinkBlock.js

@@ -48,11 +48,35 @@ function _unsupported_iterable_to_array(o, minLen) {
     if (n === "Arguments" || /^(?:Ui|I)nt(?:8|16|32)(?:Clamped)?Array$/.test(n)) return _array_like_to_array(o, minLen);
 }
 import { useMergedState } from "rc-util";
-import React, { useContext, useEffect } from "react";
+import React, { createContext, useContext, useEffect, useMemo, useRef } from "react";
 import { MessagesContext } from "../../../Bubble/MessagesContent/BubbleContext";
 import { I18nContext } from "../../../I18n";
-import { EditorStoreContext } from "../../../MarkdownEditor/editor/store";
+import { EditorStoreContext, useEditorStore } from "../../../MarkdownEditor/editor/store";
+import { EditorUtils } from "../../../MarkdownEditor/editor/utils/editorUtils";
 import { ToolUseBarThink } from "../../../ToolUseBarThink";
+/**
+ * ThinkBlock Context
+ */ export var ThinkBlockContext = /*#__PURE__*/ createContext(null);
+/**
+ * ThinkBlock Provider 组件
+ *
+ * 用于全局控制所有 ThinkBlock 实例的展开/收起状态
+ *
+ * @example
+ * ```tsx
+ * <ThinkBlockProvider expanded={isExpanded} onExpandedChange={setIsExpanded}>
+ *   <MarkdownEditor ... />
+ * </ThinkBlockProvider>
+ * ```
+ */ export var ThinkBlockProvider = function(param) {
+    var expanded = param.expanded, onExpandedChange = param.onExpandedChange, children = param.children;
+    return /*#__PURE__*/ React.createElement(ThinkBlockContext.Provider, {
+        value: {
+            expanded: expanded,
+            onExpandedChange: onExpandedChange
+        }
+    }, children);
+};
 /**
  * 将特殊标记恢复为代码块格式
  * @param content - 包含特殊标记的内容
@@ -65,27 +89,87 @@ import { ToolUseBarThink } from "../../../ToolUseBarThink";
         return "```".concat(lang, "\n").concat(code, "\n```");
     });
 };
-export function ThinkBlock(param) {
-    var element = param.element;
+export function ThinkBlock(props) {
     var _editorProps_codeProps, _editorProps_codeProps1;
+    var element = props.element;
     var locale = useContext(I18nContext).locale;
     var editorProps = (useContext(EditorStoreContext) || {}).editorProps;
     var message = useContext(MessagesContext).message;
+    var markdownEditorRef = useEditorStore().markdownEditorRef;
+    var thinkBlockContext = useContext(ThinkBlockContext);
     // 获取当前 Bubble 的 isFinished 状态
     var bubbleIsFinished = message === null || message === void 0 ? void 0 : message.isFinished;
-    var _useMergedState = _sliced_to_array(useMergedState(function() {
-        return bubbleIsFinished ? undefined : false;
-    }, {
-        value: editorProps === null || editorProps === void 0 ? void 0 : (_editorProps_codeProps = editorProps.codeProps) === null || _editorProps_codeProps === void 0 ? void 0 : _editorProps_codeProps.alwaysExpandedDeepThink,
-        defaultValue: editorProps === null || editorProps === void 0 ? void 0 : (_editorProps_codeProps1 = editorProps.codeProps) === null || _editorProps_codeProps1 === void 0 ? void 0 : _editorProps_codeProps1.alwaysExpandedDeepThink
+    // 获取当前元素的 path
+    var elementPath = useMemo(function() {
+        if (!markdownEditorRef.current) {
+            return null;
+        }
+        try {
+            return EditorUtils.findPath(markdownEditorRef.current, element);
+        } catch (error) {
+            console.error('Error finding element path:', error);
+            return null;
+        }
+    }, [
+        markdownEditorRef.current,
+        element
+    ]);
+    // 判断是否是最后一个节点
+    var isLastNode = useMemo(function() {
+        if (!markdownEditorRef.current || !elementPath) {
+            return false;
+        }
+        try {
+            return EditorUtils.checkSelEnd(markdownEditorRef.current, elementPath);
+        } catch (error) {
+            console.error('Error checking if last node:', error);
+            return false;
+        }
+    }, [
+        markdownEditorRef.current,
+        elementPath
+    ]);
+    var _editorProps_codeProps_alwaysExpandedDeepThink, _thinkBlockContext_expanded;
+    // 状态优先级（从高到低）：
+    // 1. Context 提供的 expanded（受控模式）
+    // 2. editorProps?.codeProps?.alwaysExpandedDeepThink
+    // 3. 默认值（false）
+    var _useMergedState = _sliced_to_array(useMergedState((_editorProps_codeProps_alwaysExpandedDeepThink = editorProps === null || editorProps === void 0 ? void 0 : (_editorProps_codeProps = editorProps.codeProps) === null || _editorProps_codeProps === void 0 ? void 0 : _editorProps_codeProps.alwaysExpandedDeepThink) !== null && _editorProps_codeProps_alwaysExpandedDeepThink !== void 0 ? _editorProps_codeProps_alwaysExpandedDeepThink : false, {
+        value: (editorProps === null || editorProps === void 0 ? void 0 : (_editorProps_codeProps1 = editorProps.codeProps) === null || _editorProps_codeProps1 === void 0 ? void 0 : _editorProps_codeProps1.alwaysExpandedDeepThink) ? true : (_thinkBlockContext_expanded = thinkBlockContext === null || thinkBlockContext === void 0 ? void 0 : thinkBlockContext.expanded) !== null && _thinkBlockContext_expanded !== void 0 ? _thinkBlockContext_expanded : undefined,
+        defaultValue: true,
+        onChange: thinkBlockContext === null || thinkBlockContext === void 0 ? void 0 : thinkBlockContext.onExpandedChange
     }), 2), expanded = _useMergedState[0], setExpanded = _useMergedState[1];
+    // 当 bubble 完成时，自动展开
     useEffect(function() {
         if (bubbleIsFinished) {
-            setExpanded(true);
+            setExpanded(false);
         }
     }, [
         bubbleIsFinished
     ]);
+    // 如果不是最后一个节点，自动收起
+    useEffect(function() {
+        if (!isLastNode) {
+            setExpanded(false);
+        }
+    }, [
+        isLastNode
+    ]);
+    // 跟踪 thinkBlockContext?.expanded 的之前值
+    var prevExpandedRef = useRef(thinkBlockContext === null || thinkBlockContext === void 0 ? void 0 : thinkBlockContext.expanded);
+    // 当 thinkBlockContext?.expanded 从其他值变成 undefined 时，自动收起
+    useEffect(function() {
+        var prevExpanded = prevExpandedRef.current;
+        var currentExpanded = thinkBlockContext === null || thinkBlockContext === void 0 ? void 0 : thinkBlockContext.expanded;
+        // 如果之前有值（true 或 false），现在变成 undefined，则收起
+        if (prevExpanded !== undefined && currentExpanded === undefined) {
+            setExpanded(false);
+        }
+        // 更新之前的值
+        prevExpandedRef.current = currentExpanded;
+    }, [
+        thinkBlockContext === null || thinkBlockContext === void 0 ? void 0 : thinkBlockContext.expanded
+    ]);
     var rawContent = (element === null || element === void 0 ? void 0 : element.value) !== null && (element === null || element === void 0 ? void 0 : element.value) !== undefined ? String(element.value).trim() : '';
     // 恢复内容中被转义的代码块
     var content = restoreCodeBlocks(rawContent);

package/dist/Plugins/code/components/index.d.ts

@@ -11,4 +11,4 @@ export { CodeToolbar, type CodeToolbarProps } from './CodeToolbar';
 export { HtmlPreview } from './HtmlPreview';
 export { LanguageSelector, type LanguageSelectorProps, } from './LanguageSelector';
 export { LoadImage } from './LoadImage';
-export { ThinkBlock } from './ThinkBlock';
+export { ThinkBlock, ThinkBlockProvider, type ThinkBlockContextType, } from './ThinkBlock';

package/dist/Plugins/code/components/index.js

@@ -10,4 +10,4 @@ export { CodeToolbar } from "./CodeToolbar";
 export { HtmlPreview } from "./HtmlPreview";
 export { LanguageSelector } from "./LanguageSelector";
 export { LoadImage } from "./LoadImage";
-export { ThinkBlock } from "./ThinkBlock";
+export { ThinkBlock, ThinkBlockProvider } from "./ThinkBlock";

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@ant-design/agentic-ui",
-  "version": "2.28.2",
+  "version": "2.28.5",
   "description": "面向智能体的 UI 组件库，提供多步推理可视化、工具调用展示、任务执行协同等 Agentic UI 能力",
   "repository": "git@github.com:ant-design/agentic-ui.git",
   "license": "MIT",