@antv/dumi-theme-antv 0.8.0-beta.7 → 0.8.0-beta.9

package/dist/components/AI/HomeDialog/PromptTextarea/index.js

@@ -12,7 +12,7 @@ import { DatasourceCard } from "./DatasourceCard";
 import { useEventListener } from 'ahooks';
 import classnames from 'classnames';
 import _ from 'lodash';
-import React, { useState } from 'react';
+import React, { useRef, useState } from 'react';
 import styles from "./index.module.less";
 import { SendButton } from "./SendButton";
 import { ChooseLib } from "./ChooseLib";
@@ -41,6 +41,7 @@ export var PromptTextarea = /*#__PURE__*/React.memo(function PromptTextareaInner
   var authSnap = useSnapshot(authStore);
   var _useIntl = useIntl(),
     formatMessage = _useIntl.formatMessage;
+  var textareaRef = useRef(null);
 
   // 将fileMeta状态移到组件内部管理
   var _useState = useState(null),
@@ -92,14 +93,17 @@ export var PromptTextarea = /*#__PURE__*/React.memo(function PromptTextareaInner
   }
   var datasourceNode = renderDatasourceCard();
   var promptTextValid = Boolean(value);
+  var pureSend = function pureSend() {
+    if (promptTextValid) {
+      onConfirm === null || onConfirm === void 0 || onConfirm();
+    }
+  };
   var send = function send() {
     if (!authSnap.isAuthenticated) {
-      showLoginModal();
+      showLoginModal(pureSend);
       return;
     }
-    if (promptTextValid) {
-      onConfirm === null || onConfirm === void 0 || onConfirm();
-    }
+    pureSend();
   };
   useEventListener('keydown', function (event) {
     if (event.key === 'Enter') {
@@ -109,6 +113,8 @@ export var PromptTextarea = /*#__PURE__*/React.memo(function PromptTextareaInner
         send(); // 触发自定义事件
       }
     }
+  }, {
+    target: textareaRef
   });
   return /*#__PURE__*/React.createElement("div", {
     className: classnames(styles.container, _defineProperty(_defineProperty(_defineProperty({}, styles.active, focus), styles.compact, isCompact), styles.withDatasource, Boolean(datasourceNode))),
@@ -124,6 +130,7 @@ export var PromptTextarea = /*#__PURE__*/React.memo(function PromptTextareaInner
     },
     id: "prompt-textarea",
     className: classnames(styles.promptTextarea),
+    ref: textareaRef,
     placeholder:
     // (!isCompact && !themeConfig.isAntVSite && ic(themeConfig.metas.description)) ||
     !isCompact && !themeConfig.isAntVSite ? typedPlaceholder : formatMessage({

package/dist/components/AI/HomeDialog/RecommendCase/Card.js

@@ -7,19 +7,17 @@ import { FormattedMessage } from 'dumi';
 export var Card = function Card(_ref) {
   var _AIModeMeta$tag;
   var item = _ref.item,
-    index = _ref.index;
+    index = _ref.index,
+    onClick = _ref.onClick;
   var query = item.query,
     description = item.description,
     _item$imageUrls = item.imageUrls,
     imageUrls = _item$imageUrls === void 0 ? [] : _item$imageUrls,
-    link = item.link,
     tag = item.tag;
   var style = COLORS[index];
   var handleClick = function handleClick(e) {
     e.stopPropagation();
-    var urlObj = new URL(location.href);
-    urlObj.hash = "";
-    window.open(urlObj.toString(), '_blank');
+    onClick === null || onClick === void 0 || onClick();
   };
   var popoverContent = /*#__PURE__*/React.createElement("div", {
     className: styles.popoverContent

package/dist/components/AI/HomeDialog/RecommendCase/index.js

@@ -15,12 +15,15 @@ import styles from "./index.module.less";
 import { ReloadOutlined } from "@ant-design/icons";
 import RecommendJson from "./recommend.json";
 import classnames from "classnames";
-import { FormattedMessage } from 'dumi';
+import { FormattedMessage, useSiteData } from 'dumi';
+import { sampleSize } from "lodash-es";
 export var RecommendCase = function RecommendCase(props) {
   var _useState = useState(false),
     _useState2 = _slicedToArray(_useState, 2),
     loading = _useState2[0],
     setLoading = _useState2[1];
+  var _useSiteData = useSiteData(),
+    themeConfig = _useSiteData.themeConfig;
   var _useState3 = useState([]),
     _useState4 = _slicedToArray(_useState3, 2),
     list = _useState4[0],
@@ -30,20 +33,40 @@ export var RecommendCase = function RecommendCase(props) {
     return _regeneratorRuntime().wrap(function _callee$(_context) {
       while (1) switch (_context.prev = _context.next) {
         case 0:
-          try {
-            setLoading(true);
-            data = RecommendJson;
-            setList(data.slice(0, 4));
-          } catch (err) {
-            console.log(err);
-          } finally {
-            setLoading(false);
+          _context.prev = 0;
+          setLoading(true);
+          data = [];
+          if (!(themeConfig !== null && themeConfig !== void 0 && themeConfig.recommend)) {
+            _context.next = 9;
+            break;
           }
-        case 1:
+          _context.next = 6;
+          return fetch(themeConfig.recommend).then(function (res) {
+            return res.json();
+          });
+        case 6:
+          data = _context.sent;
+          _context.next = 10;
+          break;
+        case 9:
+          data = RecommendJson;
+        case 10:
+          setList(sampleSize(data, 4));
+          _context.next = 16;
+          break;
+        case 13:
+          _context.prev = 13;
+          _context.t0 = _context["catch"](0);
+          console.log(_context.t0);
+        case 16:
+          _context.prev = 16;
+          setLoading(false);
+          return _context.finish(16);
+        case 19:
         case "end":
           return _context.stop();
       }
-    }, _callee);
+    }, _callee, null, [[0, 13, 16, 19]]);
   })), [list]);
   useEffect(function () {
     fetchList();
@@ -75,7 +98,11 @@ export var RecommendCase = function RecommendCase(props) {
     return /*#__PURE__*/React.createElement(Card, {
       key: item.caseId,
       item: item,
-      index: index
+      index: index,
+      onClick: function onClick() {
+        var _props$onClick;
+        (_props$onClick = props.onClick) === null || _props$onClick === void 0 || _props$onClick.call(props, item);
+      }
     });
   }))));
 };

package/dist/components/AI/HomeDialog/RecommendCase/index.module.less

@@ -2,6 +2,7 @@
   margin-top: 32px;
   margin-left: auto;
   margin-right: auto;
+  z-index: 10;
 }
 
 .title {

package/dist/components/AI/HomeDialog/index.js

@@ -69,6 +69,11 @@ export function HomeDialog(props) {
     style: props.promptTextareaStyle,
     onDataSummaryChange: setFileSummary
   }), /*#__PURE__*/React.createElement(RecommendCase, {
-    className: props.recommendCaseClassName
+    className: props.recommendCaseClassName,
+    onClick: function onClick(val) {
+      if (val !== null && val !== void 0 && val.query) {
+        setPromptText(val.query);
+      }
+    }
   }));
 }

package/dist/components/Login/CheckCode/index.js

@@ -17,13 +17,11 @@ import { useRef, useState } from 'react';
 import { Form, Input, message } from 'antd';
 import CountDownButton from "../CountDownButton";
 import cls from 'classnames';
-import { UIA_UA_RE } from "../utils";
 import Captcha from "../Captcha";
 import { useIntl } from 'dumi';
 import "./index.less";
 import React from 'react';
 import { sendValidationCode } from "../../../model/auth";
-var isUIA = UIA_UA_RE.test(navigator.userAgent);
 function CheckCode(_ref) {
   var en = _ref.en,
     form = _ref.form,

package/dist/components/Login/LoginForm.js

@@ -15,7 +15,6 @@ function _iterableToArrayLimit(r, l) { var t = null == r ? null : "undefined" !=
 function _arrayWithHoles(arr) { if (Array.isArray(arr)) return arr; }
 import React, { useState } from 'react';
 import { Button, Checkbox, Form, Input, Modal } from 'antd';
-import { parse } from 'query-string';
 import { chinaMobilePhoneRE, getFingerprint, SignupRegion } from "./utils";
 import CheckCode from "./CheckCode";
 import classNames from 'classnames';
@@ -55,7 +54,7 @@ export default function LoginForm() {
   }
   function _handleLoginOrRegister() {
     _handleLoginOrRegister = _asyncToGenerator( /*#__PURE__*/_regeneratorRuntime().mark(function _callee(values) {
-      var platform, params, fingerprintRes, res, gotoFromServer, _parse, gotoFromUrl;
+      var platform, params, fingerprintRes;
       return _regeneratorRuntime().wrap(function _callee$(_context) {
         while (1) switch (_context.prev = _context.next) {
           case 0:
@@ -76,11 +75,6 @@ export default function LoginForm() {
             _context.next = 9;
             return loginOrRegister(params);
           case 9:
-            res = _context.sent;
-            gotoFromServer = res.goto;
-            _parse = parse(window.location.search), gotoFromUrl = _parse.goto;
-            window.location.href = gotoFromUrl || gotoFromServer || '/';
-          case 13:
           case "end":
             return _context.stop();
         }

package/dist/layouts/GlobalLayout/index.js

@@ -5,12 +5,14 @@ import React, { useEffect } from 'react';
 import { initializeAIChat } from "../../model/AIChat";
 import { initializeAuth } from "../../model/auth";
 var queryClient = new QueryClient();
-window.antd = antd;
 var GlobalLayout = function GlobalLayout() {
   var outlet = useOutlet();
   useEffect(function () {
     initializeAIChat();
     initializeAuth();
+    if (typeof window !== 'undefined') {
+      window.antd = antd;
+    }
   }, []);
   return outlet && /*#__PURE__*/React.createElement(QueryClientProvider, {
     client: queryClient

package/dist/model/AIChat.js

@@ -196,10 +196,16 @@ subscribeKey(AIChatStore, 'activeSessionId', function () {
   AIChatStore.codeBlock = null;
 });
 export var createPureNewSession = function createPureNewSession(title) {
+  var newConversationName = 'New Conversation';
+  if (AIChatStore.sessions.some(function (s) {
+    return s.title === newConversationName && s.messages.length === 0;
+  })) {
+    return;
+  }
   var newSessionId = crypto.randomUUID();
   AIChatStore.sessions.unshift({
     id: newSessionId,
-    title: title !== null && title !== void 0 ? title : 'New Conversation',
+    title: title !== null && title !== void 0 ? title : newConversationName,
     createdAt: Date.now(),
     messages: []
   });

package/dist/model/auth.js

@@ -10,6 +10,9 @@ import request from "../utils/request";
  * @returns {boolean} 如果包含则返回 true, 否则返回 false
  */
 function hasSkipLoginParam() {
+  if (typeof window === 'undefined') {
+    return false;
+  }
   // 1. 获取当前 URL 的查询字符串 (例如 "?foo=bar&skipLogin=1")
   var queryString = window.location.search;
 
@@ -25,19 +28,20 @@ function hasSkipLoginParam() {
 export var authStore = proxy({
   isModalOpen: false,
   isAuthenticated: false,
-  token: localStorage.getItem('authToken')
+  loginCallback: function loginCallback() {}
 });
 
 // 2. 定义 Actions (作为独立函数)
-export var showLoginModal = function showLoginModal() {
+export var showLoginModal = function showLoginModal(callback) {
   authStore.isModalOpen = true;
+  authStore.loginCallback = callback;
 };
 export var hideLoginModal = function hideLoginModal() {
   authStore.isModalOpen = false;
 };
 export var loginOrRegister = /*#__PURE__*/function () {
   var _ref = _asyncToGenerator( /*#__PURE__*/_regeneratorRuntime().mark(function _callee(params) {
-    var result;
+    var _authStore$loginCallb;
     return _regeneratorRuntime().wrap(function _callee$(_context) {
       while (1) switch (_context.prev = _context.next) {
         case 0:
@@ -45,24 +49,20 @@ export var loginOrRegister = /*#__PURE__*/function () {
           _context.next = 3;
           return request.post('/api/modules/user/api/accounts/login_or_register', params);
         case 3:
-          result = _context.sent;
-          authStore.token = result.token;
           authStore.isAuthenticated = true;
-          if (typeof window !== 'undefined') {
-            localStorage.setItem('authToken', result.token);
-          }
           hideLoginModal();
+          (_authStore$loginCallb = authStore.loginCallback) === null || _authStore$loginCallb === void 0 || _authStore$loginCallb.call(authStore);
           return _context.abrupt("return", true);
-        case 11:
-          _context.prev = 11;
+        case 9:
+          _context.prev = 9;
           _context.t0 = _context["catch"](0);
           console.error('Login failed in store:', _context.t0);
           return _context.abrupt("return", false);
-        case 15:
+        case 13:
         case "end":
           return _context.stop();
       }
-    }, _callee, null, [[0, 11]]);
+    }, _callee, null, [[0, 9]]);
   }));
   return function loginOrRegister(_x) {
     return _ref.apply(this, arguments);
@@ -85,15 +85,13 @@ export var logout = /*#__PURE__*/function () {
           console.error('Logout failed in store:', _context2.t0);
         case 8:
           _context2.prev = 8;
-          authStore.token = null;
           authStore.isAuthenticated = false;
-          localStorage.removeItem('authToken');
           return _context2.finish(8);
-        case 13:
+        case 11:
         case "end":
           return _context2.stop();
       }
-    }, _callee2, null, [[0, 5, 8, 13]]);
+    }, _callee2, null, [[0, 5, 8, 11]]);
   }));
   return function logout() {
     return _ref2.apply(this, arguments);

package/dist/pages/AIPlayground/components/ConversationsMenu/index.js

@@ -10,8 +10,7 @@ import { Dropdown, Input, Modal } from 'antd';
 import { Menu } from 'antd';
 import styles from "./index.module.less";
 import { useSnapshot } from 'valtio';
-import { AIChatStore, handleDeleteSession, handlePinSession, handleRenameSession } from "../../../../model/AIChat";
-import { history } from 'dumi';
+import { AIChatStore, createPureNewSession, handleDeleteSession, handlePinSession, handleRenameSession } from "../../../../model/AIChat";
 import { useSetState } from "ahooks";
 import { useIntl } from 'dumi';
 export var ConversationsMenu = function ConversationsMenu() {
@@ -54,7 +53,7 @@ export var ConversationsMenu = function ConversationsMenu() {
       id: 'ai.conversations.new'
     }),
     onClick: function onClick() {
-      return history.push('/');
+      return createPureNewSession();
     }
   }, {
     key: 'history',

package/dist/pages/AIPlayground/components/MarkdownComponent/MarkdownCodeBlock.js

@@ -19,7 +19,10 @@ import { useIntl } from 'dumi';
 export var MarkdownCodeBlock = function MarkdownCodeBlock(_ref) {
   var inline = _ref.inline,
     className = _ref.className,
-    children = _ref.children;
+    children = _ref.children,
+    _ref$showRunButton = _ref.showRunButton,
+    showRunButtonProp = _ref$showRunButton === void 0 ? true : _ref$showRunButton;
+  var intl = useIntl();
   var _useCopyToClipboard = useCopyToClipboard(),
     _useCopyToClipboard2 = _slicedToArray(_useCopyToClipboard, 2),
     copyState = _useCopyToClipboard2[0],
@@ -39,7 +42,7 @@ export var MarkdownCodeBlock = function MarkdownCodeBlock(_ref) {
   var codeString = String(children).replace(/\n$/, '');
 
   // 4. 判断是否显示“运行”按钮
-  var showRunButton = /\bimport\b/.test(codeString);
+  var showRunButton = showRunButtonProp && /\bimport\b/.test(codeString);
 
   // 5. 定义运行代码的逻辑
   var handleRunCode = function handleRunCode() {
@@ -48,7 +51,6 @@ export var MarkdownCodeBlock = function MarkdownCodeBlock(_ref) {
   };
 
   // 6. 返回最终的 JSX 结构
-  var intl = useIntl();
   return /*#__PURE__*/React.createElement("div", {
     style: {
       position: 'relative',

package/dist/pages/AIPlayground/components/MarkdownComponent/index.js

@@ -9,7 +9,8 @@ import remarkGfm from 'remark-gfm';
 import { MarkdownCodeBlock } from "./MarkdownCodeBlock"; // 引入我们创建的自定义组件
 
 export var MarkdownComponent = function MarkdownComponent(_ref) {
-  var content = _ref.content;
+  var content = _ref.content,
+    showRunButton = _ref.showRunButton;
   return /*#__PURE__*/React.createElement(Markdown, {
     remarkPlugins: [remarkGfm],
     rehypePlugins: [rehypeRaw],
@@ -27,7 +28,9 @@ export var MarkdownComponent = function MarkdownComponent(_ref) {
       },
       // 2. 【核心】重写 `code` 标签的渲染，使用我们自己的组件
       code: function code(props) {
-        return /*#__PURE__*/React.createElement(MarkdownCodeBlock, props);
+        return /*#__PURE__*/React.createElement(MarkdownCodeBlock, _extends({
+          showRunButton: showRunButton
+        }, props));
       },
       // 3. 重写非标准 `description` 标签的渲染
       // @ts-expect-error - 告知 TypeScript 我们知道这是一个自定义的、非标准的 HTML 标签

package/dist/pages/AIPlayground/components/MsgBox/index.js

@@ -9,7 +9,8 @@ import { Bubble } from '@ant-design/x';
 import { Button, Flex, Space, Tooltip } from 'antd';
 import { useChat } from '@ai-sdk/react';
 import { TextStreamChatTransport } from 'ai';
-import { history, useIntl, useSiteData } from 'dumi';
+import { useIntl, useSiteData } from 'dumi';
+import { findLast } from 'lodash-es';
 import React, { useEffect, useState, useMemo, useRef } from 'react';
 import { useCopyToClipboard } from 'react-use';
 import { useSnapshot } from 'valtio';
@@ -62,13 +63,11 @@ var getTextContent = function getTextContent(message) {
   }).join('');
 };
 function MsgBox(props) {
-  var _derivedSnap$activeSe, _derivedSnap$activeSe3, _derivedSnap$activeSe5, _derivedSnap$activeSe6, _messages;
+  var _derivedSnap$activeSe, _derivedSnap$activeSe2, _derivedSnap$activeSe4, _derivedSnap$activeSe7, _derivedSnap$activeSe8, _messages;
   var _props$messages = props.messages,
     initialMessages = _props$messages === void 0 ? [] : _props$messages,
     _props$simple = props.simple,
     simple = _props$simple === void 0 ? false : _props$simple,
-    _props$context = props.context,
-    context = _props$context === void 0 ? '' : _props$context,
     onCodegen = props.onCodegen,
     title = props.title;
   var _useSiteData = useSiteData(),
@@ -93,16 +92,17 @@ function MsgBox(props) {
     _useCopyToClipboard2 = _slicedToArray(_useCopyToClipboard, 2),
     copyState = _useCopyToClipboard2[0],
     copyToClipboard = _useCopyToClipboard2[1];
-
+  var latestUserMessage = findLast((_derivedSnap$activeSe = derivedSnap.activeSession) === null || _derivedSnap$activeSe === void 0 ? void 0 : _derivedSnap$activeSe.messages, function (msg) {
+    return msg.role === 'user';
+  });
   // 使用 ref 存储动态值，避免重新创建 transport
   var anonymousUserIdRef = useRef(snap.anonymousUserId);
-  var contextRef = useRef(props.context);
-  var activeSessionIdRef = useRef((_derivedSnap$activeSe = derivedSnap.activeSession) === null || _derivedSnap$activeSe === void 0 ? void 0 : _derivedSnap$activeSe.id);
+  var activeSessionIdRef = useRef((_derivedSnap$activeSe2 = derivedSnap.activeSession) === null || _derivedSnap$activeSe2 === void 0 ? void 0 : _derivedSnap$activeSe2.id);
   useEffect(function () {
-    var _derivedSnap$activeSe2;
+    var _derivedSnap$activeSe3;
     anonymousUserIdRef.current = snap.anonymousUserId;
-    activeSessionIdRef.current = (_derivedSnap$activeSe2 = derivedSnap.activeSession) === null || _derivedSnap$activeSe2 === void 0 ? void 0 : _derivedSnap$activeSe2.id;
-  }, [snap.anonymousUserId, (_derivedSnap$activeSe3 = derivedSnap.activeSession) === null || _derivedSnap$activeSe3 === void 0 ? void 0 : _derivedSnap$activeSe3.id]);
+    activeSessionIdRef.current = (_derivedSnap$activeSe3 = derivedSnap.activeSession) === null || _derivedSnap$activeSe3 === void 0 ? void 0 : _derivedSnap$activeSe3.id;
+  }, [snap.anonymousUserId, (_derivedSnap$activeSe4 = derivedSnap.activeSession) === null || _derivedSnap$activeSe4 === void 0 ? void 0 : _derivedSnap$activeSe4.id]);
 
   // 转换初始消息为新格式
   var convertedInitialMessages = useMemo(function () {
@@ -123,8 +123,9 @@ function MsgBox(props) {
             gptConversationId: activeSessionIdRef.current,
             anonymousUserId: anonymousUserIdRef.current,
             mountId: 'container',
-            antvContext: props.context,
-            library: lib
+            antvContext: (latestUserMessage === null || latestUserMessage === void 0 ? void 0 : latestUserMessage.context) || props.context,
+            library: (latestUserMessage === null || latestUserMessage === void 0 ? void 0 : latestUserMessage.lib) || lib,
+            mode: latestUserMessage === null || latestUserMessage === void 0 ? void 0 : latestUserMessage.mode
           };
         }
       }),
@@ -176,9 +177,12 @@ function MsgBox(props) {
 
   // 处理用户提交
   var handleSubmit = function handleSubmit() {
-    var _derivedState$activeS3;
-    if (!promptText.trim() || status === 'streaming' || status === 'submitted') return;
-
+    var _derivedSnap$activeSe5, _derivedState$activeS3;
+    var trimmedPrompt = promptText.trim();
+    if (!trimmedPrompt || status === 'streaming' || status === 'submitted') return;
+    if (!((_derivedSnap$activeSe5 = derivedSnap.activeSession.messages) !== null && _derivedSnap$activeSe5 !== void 0 && _derivedSnap$activeSe5.length)) {
+      derivedState.activeSession.title = trimmedPrompt;
+    }
     // 使用 sendMessage 函数发送新消息
     // 第二个参数传递每次请求特定的额外数据
     sendMessage({
@@ -208,17 +212,19 @@ function MsgBox(props) {
       return;
     }
     setTimeout(function () {
-      var _derivedSnap$activeSe4;
-      var sessionMessages = (_derivedSnap$activeSe4 = derivedSnap.activeSession) === null || _derivedSnap$activeSe4 === void 0 ? void 0 : _derivedSnap$activeSe4.messages;
+      var _derivedSnap$activeSe6;
+      var sessionMessages = (_derivedSnap$activeSe6 = derivedSnap.activeSession) === null || _derivedSnap$activeSe6 === void 0 ? void 0 : _derivedSnap$activeSe6.messages;
       if ((sessionMessages === null || sessionMessages === void 0 ? void 0 : sessionMessages.length) > 0) {
         var converted = convertToUIMessages(sessionMessages);
         // 避免无限循环，仅当消息数量或内容不同时更新
         if (messages.length !== converted.length || JSON.stringify(messages) !== JSON.stringify(converted)) {
           setMessages(converted);
         }
+      } else {
+        setMessages([]);
       }
     });
-  }, [(_derivedSnap$activeSe5 = derivedSnap.activeSession) === null || _derivedSnap$activeSe5 === void 0 ? void 0 : _derivedSnap$activeSe5.id, (_derivedSnap$activeSe6 = derivedSnap.activeSession) === null || _derivedSnap$activeSe6 === void 0 || (_derivedSnap$activeSe6 = _derivedSnap$activeSe6.messages) === null || _derivedSnap$activeSe6 === void 0 ? void 0 : _derivedSnap$activeSe6.length]);
+  }, [(_derivedSnap$activeSe7 = derivedSnap.activeSession) === null || _derivedSnap$activeSe7 === void 0 ? void 0 : _derivedSnap$activeSe7.id, (_derivedSnap$activeSe8 = derivedSnap.activeSession) === null || _derivedSnap$activeSe8 === void 0 || (_derivedSnap$activeSe8 = _derivedSnap$activeSe8.messages) === null || _derivedSnap$activeSe8 === void 0 ? void 0 : _derivedSnap$activeSe8.length]);
 
   // 处理从外部（如demo页）发起的对话
   useEffect(function () {
@@ -242,13 +248,13 @@ function MsgBox(props) {
   }, []);
   useEffect(function () {
     chatScrollIntoView();
+    stop();
   }, [snap.activeSessionId]);
 
   // 将 messages 数组作为依赖项。当它变化时，Hook 会运行。
   var _useAutoScroll = useAutoScroll(messages),
     containerRef = _useAutoScroll.containerRef,
-    anchorRef = _useAutoScroll.anchorRef,
-    showScrollDownButton = _useAutoScroll.showScrollDownButton;
+    anchorRef = _useAutoScroll.anchorRef;
   return /*#__PURE__*/React.createElement(React.Fragment, null, /*#__PURE__*/React.createElement(Flex, {
     gap: "middle",
     vertical: true,
@@ -259,7 +265,8 @@ function MsgBox(props) {
     return /*#__PURE__*/React.createElement(Bubble, {
       key: msg.id || index,
       content: /*#__PURE__*/React.createElement(MarkdownComponent, {
-        content: textContent
+        content: textContent,
+        showRunButton: !props.simple
       }),
       avatar: msg.role === 'assistant' ? avatar : null,
       footer: msg.role === 'assistant' && index > 0 && index === messages.length - 1 && status === 'ready' ? /*#__PURE__*/React.createElement(Space, {
@@ -303,7 +310,7 @@ function MsgBox(props) {
   }, /*#__PURE__*/React.createElement(Space, null, /*#__PURE__*/React.createElement("button", {
     type: "button",
     onClick: function onClick() {
-      return history.push('/');
+      return createPureNewSession();
     },
     className: styles.newButton
   }, /*#__PURE__*/React.createElement(Space, null, /*#__PURE__*/React.createElement(PlusSquareOutlined, null), formatMessage({

package/dist/slots/CodeEditor/index.js

@@ -15,7 +15,7 @@ function _arrayWithHoles(arr) { if (Array.isArray(arr)) return arr; }
 import MonacoEditor, { loader } from '@monaco-editor/react';
 import { Drawer, Switch } from 'antd';
 import { autoType as d3AutoType, dsvFormat } from 'd3-dsv';
-import { useLocale, useSiteData, useIntl } from 'dumi';
+import { useLocale, useSiteData, useIntl, useLocation } from 'dumi';
 import { debounce, noop } from 'lodash-es';
 import { format } from 'prettier';
 import parserBabel from 'prettier/parser-babel';
@@ -60,6 +60,7 @@ var CodeEditor = function CodeEditor(_ref) {
     onFullscreen = _ref$onFullscreen === void 0 ? noop : _ref$onFullscreen,
     _ref$showAI = _ref.showAI,
     showAI = _ref$showAI === void 0 ? true : _ref$showAI;
+  var umiLocation = useLocation();
   var locale = useLocale();
   var _useSiteData = useSiteData(),
     themeConfig = _useSiteData.themeConfig;
@@ -471,8 +472,10 @@ var CodeEditor = function CodeEditor(_ref) {
         return setShowAIDrawer(false);
       },
       rootClassName: styles.drawer,
-      width: '90%'
+      width: '80%',
+      key: "".concat(umiLocation.hash, "_").concat(umiLocation.key)
     }, /*#__PURE__*/React.createElement(MsgBox, {
+      key: "".concat(umiLocation.hash, "_").concat(umiLocation.key),
       simple: true,
       messages: [{
         id: crypto.randomUUID(),

package/dist/slots/Header/Search/SearchResult.js

@@ -24,6 +24,15 @@ export var SearchResult = function SearchResult(_ref) {
     themeConfig = _useSiteData.themeConfig;
   var intl = useIntl();
   var authSnap = useSnapshot(authStore);
+  function pureSearch() {
+    createNewSession({
+      promptText: keywords,
+      mode: 'solve',
+      lib: !themeConfig.isAntVSite ? themeConfig.title : undefined,
+      jump: true,
+      lang: intl.locale === 'zh' ? 'zh' : 'en'
+    });
+  }
   return /*#__PURE__*/React.createElement("div", {
     className: styles.searchResult
   }, /*#__PURE__*/React.createElement("div", {
@@ -39,16 +48,10 @@ export var SearchResult = function SearchResult(_ref) {
     className: styles.result,
     onClick: function onClick() {
       if (!authSnap.isAuthenticated) {
-        showLoginModal();
+        showLoginModal(pureSearch);
         return;
       }
-      createNewSession({
-        promptText: keywords,
-        mode: "solve",
-        lib: !themeConfig.isAntVSite ? themeConfig.title : undefined,
-        jump: true,
-        lang: intl.locale === 'zh' ? 'zh' : 'en'
-      });
+      pureSearch();
     }
   }, /*#__PURE__*/React.createElement("div", {
     className: classnames(styles.title, styles.highlighted)

package/dist/slots/Header/index.js

@@ -13,7 +13,7 @@ function _unsupportedIterableToArray(o, minLen) { if (!o) return; if (typeof o =
 function _arrayLikeToArray(arr, len) { if (len == null || len > arr.length) len = arr.length; for (var i = 0, arr2 = new Array(len); i < len; i++) arr2[i] = arr[i]; return arr2; }
 function _iterableToArrayLimit(r, l) { var t = null == r ? null : "undefined" != typeof Symbol && r[Symbol.iterator] || r["@@iterator"]; if (null != t) { var e, n, i, u, a = [], f = !0, o = !1; try { if (i = (t = t.call(r)).next, 0 === l) { if (Object(t) !== t) return; f = !1; } else for (; !(f = (e = i.call(t)).done) && (a.push(e.value), a.length !== l); f = !0); } catch (r) { o = !0, n = r; } finally { try { if (!f && null != t.return && (u = t.return(), Object(u) !== u)) return; } finally { if (o) throw n; } } return a; } }
 function _arrayWithHoles(arr) { if (Array.isArray(arr)) return arr; }
-import { CaretDownFilled, DownOutlined, GithubOutlined, LinkOutlined, LogoutOutlined, MenuOutlined, UserDeleteOutlined, UserOutlined, WechatOutlined } from '@ant-design/icons';
+import { CaretDownFilled, DownOutlined, GithubOutlined, LinkOutlined, LogoutOutlined, MenuOutlined, UserDeleteOutlined, WechatOutlined } from '@ant-design/icons';
 import { Alert, Button, Dropdown, Menu, Modal, Popover } from 'antd';
 import cx from 'classnames';
 import { FormattedMessage, Link, useLocale, useSiteData } from 'dumi';
@@ -26,6 +26,7 @@ import { Navs } from "./Navs";
 import { Products } from "./Products";
 import { Search } from "./Search";
 import { findVersion } from "./utils";
+import { ReactComponent as UserIcon } from "../../static/user.svg";
 import { Assistant } from '@petercatai/assistant';
 import '@petercatai/assistant/style';
 import { useLocation } from 'react-router-dom';
@@ -429,7 +430,9 @@ var HeaderComponent = function HeaderComponent(_ref) {
         icon: /*#__PURE__*/React.createElement(UserDeleteOutlined, null)
       }]
     }
-  }, /*#__PURE__*/React.createElement("a", null, /*#__PURE__*/React.createElement(UserOutlined, null)))));
+  }, /*#__PURE__*/React.createElement("a", null, /*#__PURE__*/React.createElement(UserIcon, {
+    className: styles.userIcon
+  })))));
   return /*#__PURE__*/React.createElement("header", {
     className: cx(styles.header, _defineProperty(_defineProperty(_defineProperty(_defineProperty(_defineProperty({}, styles.transparent, !!transparent && !productMenuVisible), styles.isHomePage, !!isHomePage && !isAntVHome), styles.lightTheme, !!isAntVHome && !productMenuVisible && isWide), styles.isAntVHome, !!isAntVHome), styles.fixed, popupMenuVisible))
   }, bannerVisible && announcementTitle && /*#__PURE__*/React.createElement(Alert, {

package/dist/slots/Header/index.module.less

@@ -460,3 +460,11 @@
     }
   }
 }
+
+.userIcon {
+  font-size: 16px;
+  width: 1em;
+  height: 1em;
+  vertical-align: -0.2em;
+  fill: currentColor;
+}

package/dist/static/user.svg

@@ -0,0 +1,3 @@
+<svg viewBox="0 0 1024 1024" xmlns="http://www.w3.org/2000/svg">
+  <path d="M512 74.666667C270.933333 74.666667 74.666667 270.933333 74.666667 512S270.933333 949.333333 512 949.333333 949.333333 753.066667 949.333333 512 753.066667 74.666667 512 74.666667z m0 160c70.4 0 128 57.6 128 128s-57.6 128-128 128-128-57.6-128-128 57.6-128 128-128z m236.8 507.733333c-23.466667 32-117.333333 100.266667-236.8 100.266667s-213.333333-68.266667-236.8-100.266667c-8.533333-10.666667-10.666667-21.333333-8.533333-32 29.866667-110.933333 130.133333-187.733333 245.333333-187.733333s215.466667 76.8 245.333333 187.733333c2.133333 10.666667 0 21.333333-8.533333 32z" fill="currentColor"></path>
+</svg>

package/dist/utils/env.js

@@ -11,10 +11,21 @@ export var safeWindow = function safeWindow(fn) {
 };
 
 /**
- * 根据当前窗口的域名动态获取 API 的 baseURL
+ * 根据当前环境动态获取 API 的 baseURL。
+ * 在浏览器中，它会根据域名判断；在服务端，它会返回一个固定的生产环境地址。
  * @returns {string} API 的 baseURL
  */
 export var getBaseURL = function getBaseURL() {
+  // 关键：检查是否在浏览器环境
+  if (typeof window === 'undefined') {
+    // === 服务端环境 (SSR/Pre-render) ===
+    // 在服务端渲染时，我们无法知道用户最终会通过哪个域名访问。
+    // 通常，我们默认返回生产环境的 API 地址。
+    // 这样预渲染出的页面如果需要请求数据，会直接请求线上API。
+    return 'https://www.weavefox.cn';
+  }
+
+  // === 浏览器环境 ===
   var hostname = window.location.hostname;
 
   // 生产环境
@@ -31,5 +42,7 @@ export var getBaseURL = function getBaseURL() {
   if (hostname.endsWith('.alipay.net')) {
     return 'https://weavefox.alipay.net:8443';
   }
+
+  // 默认返回生产环境地址，适用于其他未知域名（如 localhost）
   return 'https://www.weavefox.cn';
 };

package/dist/utils/request.js

@@ -6,24 +6,12 @@ function _toPropertyKey(t) { var i = _toPrimitive(t, "string"); return "symbol"
 function _toPrimitive(t, r) { if ("object" != _typeof(t) || !t) return t; var e = t[Symbol.toPrimitive]; if (void 0 !== e) { var i = e.call(t, r || "default"); if ("object" != _typeof(i)) return i; throw new TypeError("@@toPrimitive must return a primitive value."); } return ("string" === r ? String : Number)(t); }
 import axios from 'axios';
 import { getBaseURL } from "./env";
-import { getToken } from "./auth";
 var req = axios.create({
   baseURL: getBaseURL(),
   timeout: 60000,
   withCredentials: true
 });
 
-// 请求拦截器
-req.interceptors.request.use(function (config) {
-  var token = getToken();
-  if (token && config.headers) {
-    config.headers.Authorization = "Bearer ".concat(token);
-  }
-  return config;
-}, function (error) {
-  return Promise.reject(error);
-});
-
 // 响应拦截器
 req.interceptors.response.use(
 // @ts-ignore - Temporarily ignore type mismatch if it occurs with custom ApiResponse

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@antv/dumi-theme-antv",
-  "version": "0.8.0-beta.7",
+  "version": "0.8.0-beta.9",
   "description": "AntV website theme based on dumi2.",
   "keywords": [
     "dumi",

package/dist/utils/auth.js

@@ -1,10 +0,0 @@
-var TokenKey = 'My-App-Token';
-export var getToken = function getToken() {
-  return localStorage.getItem(TokenKey);
-};
-export var setToken = function setToken(token) {
-  return localStorage.setItem(TokenKey, token);
-};
-export var removeToken = function removeToken() {
-  return localStorage.removeItem(TokenKey);
-};