@antv/dumi-theme-antv 0.8.0-beta.12 → 0.8.0-beta.14

package/dist/components/AI/HomeDialog/ModeSelector/ModeSelectorDropdown.js

@@ -0,0 +1,42 @@
+function _slicedToArray(arr, i) { return _arrayWithHoles(arr) || _iterableToArrayLimit(arr, i) || _unsupportedIterableToArray(arr, i) || _nonIterableRest(); }
+function _nonIterableRest() { throw new TypeError("Invalid attempt to destructure non-iterable instance.\nIn order to be iterable, non-array objects must have a [Symbol.iterator]() method."); }
+function _unsupportedIterableToArray(o, minLen) { if (!o) return; if (typeof o === "string") return _arrayLikeToArray(o, minLen); var n = Object.prototype.toString.call(o).slice(8, -1); if (n === "Object" && o.constructor) n = o.constructor.name; if (n === "Map" || n === "Set") return Array.from(o); if (n === "Arguments" || /^(?:Ui|I)nt(?:8|16|32)(?:Clamped)?Array$/.test(n)) return _arrayLikeToArray(o, minLen); }
+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 { Dropdown, Space } from 'antd';
+import { useIntl } from 'dumi';
+import React from 'react';
+import { useSnapshot } from 'valtio';
+import { AIChatStore } from "../../../../model/AIChat";
+import { AIModeMeta } from "../../constant";
+export function ModeSelectorDropdown() {
+  var _useIntl = useIntl(),
+    formatMessage = _useIntl.formatMessage;
+  var snap = useSnapshot(AIChatStore);
+  var items = Object.entries(AIModeMeta).map(function (_ref) {
+    var _ref2 = _slicedToArray(_ref, 2),
+      key = _ref2[0],
+      value = _ref2[1];
+    return {
+      key: key,
+      label: formatMessage({
+        id: value.shortName
+      }),
+      icon: value.icon,
+      // @ts-ignore
+      onClick: function onClick() {
+        return AIChatStore.mode = key;
+      }
+    };
+  });
+  return /*#__PURE__*/React.createElement(Dropdown, {
+    menu: {
+      items: items
+    }
+  }, /*#__PURE__*/React.createElement("button", {
+    type: "button"
+  }, /*#__PURE__*/React.createElement("a", null, /*#__PURE__*/React.createElement(Space, null, AIModeMeta[snap.mode].icon, formatMessage({
+    id: AIModeMeta[snap.mode].shortName
+  })))));
+}

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

@@ -1,6 +1,6 @@
 import React from 'react';
 import { Dropdown } from "antd";
-import { useProducts } from "../../../../../hooks/useProducts";
+import { useLibrary, useProducts } from "../../../../../hooks/useProducts";
 import { useLocale, FormattedMessage } from "dumi";
 import styles from "./index.module.less";
 export function ChooseLib(props) {
@@ -15,11 +15,16 @@ export function ChooseLib(props) {
   var _useProducts = useProducts(),
     _useProducts$data = _useProducts.data,
     data = _useProducts$data === void 0 ? [] : _useProducts$data;
+  var _useLibrary = useLibrary(),
+    _useLibrary$data = _useLibrary.data,
+    library = _useLibrary$data === void 0 ? [] : _useLibrary$data;
   var onSelect = function onSelect(key) {
     onChange === null || onChange === void 0 || onChange(key);
   };
   var items = data.filter(function (item) {
-    return item.lang === lang && ["G2", "F2", "S2", "G6", "X6", "AVA"].includes(item.title);
+    return item.lang === lang && library.map(function (l) {
+      return l.toUpperCase();
+    }).includes(item.title);
   }).map(function (item) {
     return {
       key: item.title,

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

@@ -22,6 +22,7 @@ import { useTypewriter } from "../../../../hooks/useTypewriter";
 import { authStore, showLoginModal } from "../../../../model/auth";
 import { useSnapshot } from "valtio";
 import { AIChatStore } from "../../../../model/AIChat";
+import { ModeSelectorDropdown } from "../ModeSelector/ModeSelectorDropdown";
 var PLACEHOLDER = {
   implement: 'ai.placeholder.implement',
   solve: 'ai.placeholder.solve'
@@ -34,9 +35,10 @@ export var PromptTextarea = /*#__PURE__*/React.memo(function PromptTextareaInner
     onConfirm = props.onConfirm,
     onCancel = props.onCancel,
     loading = props.loading,
-    mode = props.mode,
     _props$showAction = props.showAction,
-    showAction = _props$showAction === void 0 ? true : _props$showAction;
+    showAction = _props$showAction === void 0 ? true : _props$showAction,
+    _props$showModeSelect = props.showModeSelector,
+    showModeSelector = _props$showModeSelect === void 0 ? false : _props$showModeSelect;
   var snap = useSnapshot(AIChatStore);
   var authSnap = useSnapshot(authStore);
   var _useIntl = useIntl(),
@@ -138,7 +140,7 @@ export var PromptTextarea = /*#__PURE__*/React.memo(function PromptTextareaInner
     placeholder:
     // (!isCompact && !themeConfig.isAntVSite && ic(themeConfig.metas.description)) ||
     !isCompact && !themeConfig.isAntVSite ? typedPlaceholder : formatMessage({
-      id: _.get(PLACEHOLDER, mode, 'ai.placeholder.implement')
+      id: _.get(PLACEHOLDER, snap.mode, 'ai.placeholder.implement')
     }),
     value: value,
     onChange: function onChange(evt) {
@@ -154,7 +156,7 @@ export var PromptTextarea = /*#__PURE__*/React.memo(function PromptTextareaInner
     onChange: function onChange(s) {
       return AIChatStore.lib = s;
     }
-  }))), /*#__PURE__*/React.createElement("div", {
+  })), showModeSelector && /*#__PURE__*/React.createElement(React.Fragment, null, /*#__PURE__*/React.createElement(ModeSelectorDropdown, null))), /*#__PURE__*/React.createElement("div", {
     className: styles.actions
   }, loading ? /*#__PURE__*/React.createElement("img", {
     className: styles.actionBtn,

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

@@ -1,20 +1,25 @@
+function _typeof(o) { "@babel/helpers - typeof"; return _typeof = "function" == typeof Symbol && "symbol" == typeof Symbol.iterator ? function (o) { return typeof o; } : function (o) { return o && "function" == typeof Symbol && o.constructor === Symbol && o !== Symbol.prototype ? "symbol" : typeof o; }, _typeof(o); }
 import { Popover } from 'antd';
 import React from 'react';
 import styles from "./card.module.less";
 import { BarChartOutlined, QuestionCircleOutlined } from "@ant-design/icons";
 import { AIMode, AIModeMeta, COLORS } from "../../constant";
-import { FormattedMessage } from 'dumi';
+import { FormattedMessage, useLocale } from 'dumi';
 export var Card = function Card(_ref) {
   var _AIModeMeta$tag;
   var item = _ref.item,
     index = _ref.index,
     onClick = _ref.onClick;
-  var query = item.query,
-    description = item.description,
+  var _item$query = item.query,
+    query = _item$query === void 0 ? {} : _item$query,
+    _item$description = item.description,
+    description = _item$description === void 0 ? {} : _item$description,
     _item$imageUrls = item.imageUrls,
     imageUrls = _item$imageUrls === void 0 ? [] : _item$imageUrls,
     tag = item.tag;
   var style = COLORS[index];
+  var locale = useLocale();
+  var lang = locale.id === 'zh' ? 'zh' : 'en';
   var handleClick = function handleClick(e) {
     e.stopPropagation();
     onClick === null || onClick === void 0 || onClick();
@@ -29,7 +34,7 @@ export var Card = function Card(_ref) {
     id: "ai.recommend.card.caseName"
   })), /*#__PURE__*/React.createElement("div", {
     className: styles.popoverValue
-  }, query)), /*#__PURE__*/React.createElement("div", {
+  }, _typeof(query) === 'object' ? query[lang] : query)), /*#__PURE__*/React.createElement("div", {
     className: styles.popoverItem
   }, /*#__PURE__*/React.createElement("div", {
     className: styles.popoverLabel
@@ -37,7 +42,7 @@ export var Card = function Card(_ref) {
     id: "ai.recommend.card.description"
   })), /*#__PURE__*/React.createElement("div", {
     className: styles.popoverValue
-  }, description)));
+  }, _typeof(description) === 'object' ? description[lang] : description)));
   return /*#__PURE__*/React.createElement(Popover, {
     content: popoverContent,
     placement: "top",
@@ -61,7 +66,7 @@ export var Card = function Card(_ref) {
     id: ((_AIModeMeta$tag = AIModeMeta[tag]) === null || _AIModeMeta$tag === void 0 ? void 0 : _AIModeMeta$tag.name) || tag
   }))), /*#__PURE__*/React.createElement("div", {
     className: styles.title
-  }, query), /*#__PURE__*/React.createElement("div", {
+  }, _typeof(query) === 'object' ? query[lang] : query), /*#__PURE__*/React.createElement("div", {
     className: styles.imageContainer
   }, imageUrls.slice(0, 2).map(function (item, idx) {
     return /*#__PURE__*/React.createElement("img", {

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

@@ -100,7 +100,7 @@
   }
 
   .popoverContent {
-    max-width: 372px;
+    max-width: 800px;
 
     .popoverItem {
       display: flex;
@@ -112,7 +112,7 @@
         color: #1d2129e6;
         line-height: 22px;
         font-weight: 500;
-        min-width: 56px;
+        min-width: 80px;
       }
 
       .popoverValue {

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

@@ -9,14 +9,17 @@ function _arrayLikeToArray(arr, len) { if (len == null || len > arr.length) len
 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 { Spin } from 'antd';
-import React, { useCallback, useEffect, useState } from 'react';
+import React, { useEffect, useState } from 'react';
 import { Card } from "./Card";
 import styles from "./index.module.less";
 import { ReloadOutlined } from "@ant-design/icons";
 import RecommendJson from "./recommend.json";
 import classnames from "classnames";
 import { FormattedMessage, useSiteData } from 'dumi';
-import { sampleSize } from "lodash-es";
+import { sample, sampleSize } from "lodash-es";
+import { AIChatStore } from "../../../../model/AIChat";
+import { useLibrary } from "../../../../hooks/useProducts";
+import { getBaseSiteDataUrl } from "../../../../utils/env";
 export var RecommendCase = function RecommendCase(props) {
   var _useState = useState(false),
     _useState2 = _slicedToArray(_useState, 2),
@@ -28,46 +31,56 @@ export var RecommendCase = function RecommendCase(props) {
     _useState4 = _slicedToArray(_useState3, 2),
     list = _useState4[0],
     setList = _useState4[1];
-  var fetchList = useCallback( /*#__PURE__*/_asyncToGenerator( /*#__PURE__*/_regeneratorRuntime().mark(function _callee() {
-    var _themeConfig$ai, data;
-    return _regeneratorRuntime().wrap(function _callee$(_context) {
-      while (1) switch (_context.prev = _context.next) {
-        case 0:
-          _context.prev = 0;
-          setLoading(true);
-          data = [];
-          if (!(themeConfig !== null && themeConfig !== void 0 && (_themeConfig$ai = themeConfig.ai) !== null && _themeConfig$ai !== void 0 && _themeConfig$ai.recommend)) {
-            _context.next = 9;
+  var _useLibrary = useLibrary(),
+    _useLibrary$data = _useLibrary.data,
+    library = _useLibrary$data === void 0 ? [] : _useLibrary$data;
+  var fetchList = /*#__PURE__*/function () {
+    var _ref = _asyncToGenerator( /*#__PURE__*/_regeneratorRuntime().mark(function _callee() {
+      var _themeConfig$ai, data, url;
+      return _regeneratorRuntime().wrap(function _callee$(_context) {
+        while (1) switch (_context.prev = _context.next) {
+          case 0:
+            _context.prev = 0;
+            setLoading(true);
+            data = [];
+            url = themeConfig.isAntVSite && library.length ? "".concat(getBaseSiteDataUrl(), "/").concat(sample(library).toLowerCase(), "/recommend.json") : (themeConfig === null || themeConfig === void 0 || (_themeConfig$ai = themeConfig.ai) === null || _themeConfig$ai === void 0 ? void 0 : _themeConfig$ai.recommend) || "".concat(getBaseSiteDataUrl(), "/").concat(themeConfig.title, "/recommend.json");
+            if (!url) {
+              _context.next = 10;
+              break;
+            }
+            _context.next = 7;
+            return fetch(url).then(function (res) {
+              return res.json();
+            });
+          case 7:
+            data = _context.sent;
+            _context.next = 11;
             break;
-          }
-          _context.next = 6;
-          return fetch(themeConfig.ai.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, null, [[0, 13, 16, 19]]);
-  })), [list]);
+          case 10:
+            data = RecommendJson;
+          case 11:
+            setList(sampleSize(data, 4));
+            _context.next = 18;
+            break;
+          case 14:
+            _context.prev = 14;
+            _context.t0 = _context["catch"](0);
+            setList(RecommendJson);
+            console.log(_context.t0);
+          case 18:
+            _context.prev = 18;
+            setLoading(false);
+            return _context.finish(18);
+          case 21:
+          case "end":
+            return _context.stop();
+        }
+      }, _callee, null, [[0, 14, 18, 21]]);
+    }));
+    return function fetchList() {
+      return _ref.apply(this, arguments);
+    };
+  }();
   useEffect(function () {
     fetchList();
   }, []);
@@ -82,14 +95,14 @@ export var RecommendCase = function RecommendCase(props) {
     className: styles.quickStart
   }, /*#__PURE__*/React.createElement(FormattedMessage, {
     id: "ai.recommend.title"
-  })), /*#__PURE__*/React.createElement("span", {
+  })), (list === null || list === void 0 ? void 0 : list.length) > 4 ? /*#__PURE__*/React.createElement("span", {
     className: styles.refresh,
     onClick: function onClick() {
       return fetchList();
     }
   }, /*#__PURE__*/React.createElement(ReloadOutlined, null), /*#__PURE__*/React.createElement(FormattedMessage, {
     id: "ai.recommend.refresh"
-  }))), /*#__PURE__*/React.createElement(Spin, {
+  })) : /*#__PURE__*/React.createElement(React.Fragment, null)), /*#__PURE__*/React.createElement(Spin, {
     spinning: loading,
     wrapperClassName: classnames(styles.listContainer, props.className)
   }, /*#__PURE__*/React.createElement("div", {
@@ -102,6 +115,9 @@ export var RecommendCase = function RecommendCase(props) {
       onClick: function onClick() {
         var _props$onClick;
         (_props$onClick = props.onClick) === null || _props$onClick === void 0 || _props$onClick.call(props, item);
+        if (item.tag) {
+          AIChatStore.mode = item.tag;
+        }
       }
     });
   }))));

package/dist/components/AI/HomeDialog/RecommendCase/recommend.json

@@ -2,41 +2,65 @@
   {
     "caseId": "1",
     "tag": "implement",
-    "query": "一个城市过去五年来的年度降雨量分别是：2016年800毫米，2017年900毫米，2018年700毫米，2019年1000毫米，2020年1100毫米。使用折线图描绘这些数据。",
-    "description": "自然语言描述需求，生成可视化代码",
+    "query": {
+      "zh": "你有一份公司多产品线销售数据：data = [ {month: '1月', productA: 12000, productB: 8500, productC: 15200}, {month: '2月', productA: 13500, productB: 9200, productC: 14800}, {month: '3月', productA: 15200, productB: 10800, productC: 16500}, {month: '4月', productA: 14800, productB: 11500, productC: 17200}, {month: '5月', productA: 16500, productB: 12200, productC: 18800}, {month: '6月', productA: 18200, productB: 13800, productC: 19500}, {month: '7月', productA: 19800, productB: 14500, productC: 20200}, {month: '8月', productA: 21500, productB: 15200, productC: 21800}, {month: '9月', productA: 20200, productB: 16800, productC: 22500}, {month: '10月', productA: 22800, productB: 17500, productC: 23200}, {month: '11月', productA: 24500, productB: 18200, productC: 24800}, {month: '12月', productA: 26200, productB: 19800, productC: 25500} ]， 请绘制一个多折线图展示三条产品线的销售趋势变化，通过不同线型和颜色清晰区分各产品表现。 同时在最高点进行数据标注。",
+      "en": "You have multi-product line sales data for a company: data = [ {month: 'January', productA: 12000, productB: 8500, productC: 15200}, {month: 'February', productA: 13500, productB: 9200, productC: 14800}, {month: 'March', productA: 15200, productB: 10800, productC: 16500}, {month: 'April', productA: 14800, productB: 11500, productC: 17200}, {month: 'May', productA: 16500, productB: 12200, productC: 18800}, {month: 'June', productA: 18200, productB: 13800, productC: 19500}, {month: 'July', productA: 19800, productB: 14500, productC: 20200}, {month: 'August', productA: 21500, productB: 15200, productC: 21800}, {month: 'September', productA: 20200, productB: 16800, productC: 22500}, {month: 'October', productA: 22800, productB: 17500, productC: 23200}, {month: 'November', productA: 24500, productB: 18200, productC: 24800}, {month: 'December', productA: 26200, productB: 19800, productC: 25500} ]. Please draw a multi-line chart to show the sales trend changes of the three product lines, clearly distinguishing each product's performance through different line styles and colors. Also add data annotations at the highest points."
+    },
+    "description": {
+      "zh": "自然语言描述需求，生成可视化代码",
+      "en": "Natural language description of requirements, generate visualization code"
+    },
     "imageUrls": [
-      "https://mdn.alipayobjects.com/huamei_qa8qxu/afts/img/A*K5iUTrYme5QAAAAAQpAAAAgAemJ7AQ/original",
-      "https://mdn.alipayobjects.com/huamei_qa8qxu/afts/img/A*K5iUTrYme5QAAAAAQpAAAAgAemJ7AQ/original"
+      "https://mdn.alipayobjects.com/huamei_qa8qxu/afts/img/A*zUKtR4U75fcAAAAAROAAAAgAemJ7AQ/original",
+      "https://mdn.alipayobjects.com/huamei_qa8qxu/afts/img/A*zUKtR4U75fcAAAAAROAAAAgAemJ7AQ/original"
     ]
   },
   {
     "caseId": "2",
     "tag": "implement",
-    "query": "使用文件中的数据，生成柱状图",
-    "description": "多种数据源引入，专业图表生成，有数据就能生成图表。",
+    "query": {
+      "zh": "一家咖啡店的饮品销售比例为：咖啡 60%，茶 25%，果汁 15%。请用环图可视化这些饮品销售数据，并将色板调整为橙色系。",
+      "en": "A coffee shop's beverage sales ratio is: Coffee 60%, Tea 25%, Juice 15%. Please visualize these beverage sales data with a donut chart and adjust the color palette to orange tones."
+    },
+    "description": {
+      "zh": "自然语言描述需求，生成可视化代码",
+      "en": "Natural language description of requirements, generate visualization code"
+    },
     "imageUrls": [
-      "https://mdn.alipayobjects.com/huamei_qa8qxu/afts/img/A*K5iUTrYme5QAAAAAQpAAAAgAemJ7AQ/original",
-      "https://mdn.alipayobjects.com/huamei_qa8qxu/afts/img/A*K5iUTrYme5QAAAAAQpAAAAgAemJ7AQ/original"
+      "https://mdn.alipayobjects.com/huamei_qa8qxu/afts/img/A*oMezTK-Yf2EAAAAAQmAAAAgAemJ7AQ/original",
+      "https://mdn.alipayobjects.com/huamei_qa8qxu/afts/img/A*oMezTK-Yf2EAAAAAQmAAAAgAemJ7AQ/original"
     ]
   },
   {
     "caseId": "3",
-    "tag": "implement",
-    "query": "基于设计稿生成图表代码，还原视觉细节",
-    "description": "基于设计稿生成图表代码，还原视觉细节",
+    "tag": "solve",
+    "query": {
+      "zh": "为什么配置了 state 没有生效？",
+      "en": "Why is the configured state not taking effect?"
+    },
+    "description": {
+      "zh": "精准定位并提供针对渲染异常、性能瓶颈及数据不匹配等问题的解决方案。",
+      "en": "Precisely locate and provide solutions for rendering anomalies, performance bottlenecks, and data mismatches."
+    },
     "imageUrls": [
-      "https://mdn.alipayobjects.com/huamei_qa8qxu/afts/img/A*K5iUTrYme5QAAAAAQpAAAAgAemJ7AQ/original",
-      "https://mdn.alipayobjects.com/huamei_qa8qxu/afts/img/A*K5iUTrYme5QAAAAAQpAAAAgAemJ7AQ/original"
+      "https://mdn.alipayobjects.com/huamei_qa8qxu/afts/img/A*15mgQ4lA5CUAAAAASzAAAAgAemJ7AQ/original",
+      "https://mdn.alipayobjects.com/huamei_qa8qxu/afts/img/A*15mgQ4lA5CUAAAAASzAAAAgAemJ7AQ/original"
     ]
   },
   {
     "caseId": "4",
     "tag": "solve",
-    "query": "怎么配置堆叠面积图的描边为不同的颜色",
-    "description": "精准定位并提供针对渲染异常、性能瓶颈及数据不匹配等问题的解决方案。",
+    "query": {
+      "zh": "数据标签互相遮挡了怎么办？",
+      "en": "What to do when data labels overlap each other?"
+    },
+    "description": {
+      "zh": "精准定位并提供针对渲染异常、性能瓶颈及数据不匹配等问题的解决方案。",
+      "en": "Precisely locate and provide solutions for rendering anomalies, performance bottlenecks, and data mismatches."
+    },
     "imageUrls": [
-      "https://mdn.alipayobjects.com/huamei_qa8qxu/afts/img/A*K5iUTrYme5QAAAAAQpAAAAgAemJ7AQ/original",
-      "https://mdn.alipayobjects.com/huamei_qa8qxu/afts/img/A*K5iUTrYme5QAAAAAQpAAAAgAemJ7AQ/original"
+      "https://mdn.alipayobjects.com/huamei_qa8qxu/afts/img/A*thpQQIB3LG4AAAAASMAAAAgAemJ7AQ/original",
+      "https://mdn.alipayobjects.com/huamei_qa8qxu/afts/img/A*thpQQIB3LG4AAAAASMAAAAgAemJ7AQ/original"
     ]
   }
 ]

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

@@ -10,19 +10,14 @@ import { AntVBanner } from "./AntVBanner";
 import { PromptTextarea } from "./PromptTextarea";
 import { RecommendCase } from "./RecommendCase";
 import { ModeSelector } from "./ModeSelector";
-import { useLocalStorageState } from 'ahooks';
-import { AIMode } from "../constant";
 import classnames from 'classnames';
 import { useLocale } from 'dumi';
-import { createNewSession } from "../../../model/AIChat";
+import { AIChatStore, createNewSession } from "../../../model/AIChat";
+import { useSnapshot } from "valtio";
 export function HomeDialog(props) {
   var locale = useLocale();
-  var _useLocalStorageState = useLocalStorageState('use-local-storage-ai-mode-type', {
-      defaultValue: AIMode.implement
-    }),
-    _useLocalStorageState2 = _slicedToArray(_useLocalStorageState, 2),
-    mode = _useLocalStorageState2[0],
-    setMode = _useLocalStorageState2[1];
+  var lang = locale.id === 'zh' ? 'zh' : 'en';
+  var snap = useSnapshot(AIChatStore);
   var _useState = useState(''),
     _useState2 = _slicedToArray(_useState, 2),
     promptText = _useState2[0],
@@ -36,12 +31,10 @@ export function HomeDialog(props) {
     style: props.style
   }, /*#__PURE__*/React.createElement(AntVBanner, null), /*#__PURE__*/React.createElement(ModeSelector, {
     onChange: function onChange(v) {
-      setMode(v);
-      // todo  埋点
+      AIChatStore.mode = v;
     },
-    value: mode
+    value: snap.mode
   }), /*#__PURE__*/React.createElement(PromptTextarea, {
-    mode: mode,
     value: promptText,
     onChange: function onChange(val) {
       setPromptText(val);
@@ -49,7 +42,7 @@ export function HomeDialog(props) {
     onConfirm: function onConfirm() {
       createNewSession({
         promptText: promptText,
-        mode: mode,
+        mode: snap.mode,
         jump: true,
         context: fileSummary,
         lang: locale.id
@@ -61,7 +54,7 @@ export function HomeDialog(props) {
     className: props.recommendCaseClassName,
     onClick: function onClick(val) {
       if (val !== null && val !== void 0 && val.query) {
-        setPromptText(val.query);
+        setPromptText(val.query[lang]);
       }
     }
   }));

package/dist/components/AI/constant.js

@@ -7,11 +7,13 @@ export var AIMode = {
 export var AIModeMeta = {
   implement: {
     name: 'ai.mode.implement',
-    icon: /*#__PURE__*/React.createElement(BarChartOutlined, null)
+    icon: /*#__PURE__*/React.createElement(BarChartOutlined, null),
+    shortName: 'ai.mode.implement.short'
   },
   solve: {
     name: 'ai.mode.solve',
-    icon: /*#__PURE__*/React.createElement(QuestionCircleOutlined, null)
+    icon: /*#__PURE__*/React.createElement(QuestionCircleOutlined, null),
+    shortName: 'ai.mode.solve.short'
   }
 };
 

package/dist/hooks/useProducts.js

@@ -1,7 +1,8 @@
 import { useQuery } from '@tanstack/react-query';
+import { getBaseSiteDataUrl } from "../utils/env";
 export function getProducts() {
   // 如需要修改产品信息，请到 https://yuyan.antfin-inc.com/antv/site-data/sprints 修改区块内容
-  return fetch('https://assets.antv.antgroup.com/antv/products.json' // 生产环境
+  return fetch("".concat(getBaseSiteDataUrl(), "/antv/products.json") // 生产环境
   // 'https://site-data-pre.alipay.com/antv/products.json', // 预发测试
   ).then(function (res) {
     return res.json();
@@ -15,4 +16,21 @@ export function useProducts() {
     queryFn: getProducts,
     staleTime: 24 * 60 * 60 * 1000 // 一天内数据不会被认为是 "stale"，不会触发后台刷新
   });
+}
+export function getLibrary() {
+  // 如需要修改产品信息，请到 https://yuyan.antfin-inc.com/antv/site-data/sprints 修改区块内容
+  return fetch("".concat(getBaseSiteDataUrl(), "/antv/library.json") // 生产环境
+  // 'https://site-data-pre.alipay.com/antv/products.json', // 预发测试
+  ).then(function (res) {
+    return res.json();
+  });
+}
+
+// 封装了 queryKey 和 fetcher 的自定义 Hook
+export function useLibrary() {
+  return useQuery({
+    queryKey: ['antv-library'],
+    queryFn: getLibrary,
+    staleTime: 24 * 60 * 60 * 1000 // 一天内数据不会被认为是 "stale"，不会触发后台刷新
+  });
 }

package/dist/locales/en.json

@@ -182,7 +182,9 @@
   "ai.placeholder.implement": "What would you like to visualize today?",
   "ai.placeholder.solve": "What visualization problem would you like to solve today?",
   "ai.mode.implement": "Develop",
+  "ai.mode.implement.short": "Develop",
   "ai.mode.solve": "Q&A",
+  "ai.mode.solve.short": "Q&A",
   "ai.upload.tooltip": "Only supports csv,json,tsv,txt files. For performance and cost reasons, only the first few lines will be used as samples",
   "ai.upload.data": "Upload Data",
   "ai.upload.image": "Upload Image",
@@ -205,6 +207,7 @@
   "ai.msgbox.start.new.chat": "Start New Chat",
   "ai.msgbox.send.tip": "Please enter content before sending",
   "ai.msgbox.error.response": "Sorry, I couldn't successfully process your request, please try again later",
+  "ai.msgbox.continue.from.here": "Continue from here",
   "ai.recommend.title": "Featured Cases",
   "ai.recommend.refresh": "Refresh",
   "ai.recommend.card.caseName": "Case Name",
@@ -227,12 +230,14 @@
   "ai.conversations.pin": "Pin",
   "ai.conversations.delete": "Delete",
   "ai.conversations.edit.title": "Edit Conversation Name",
+  "ai.conversations.ok": "OK",
+  "ai.conversations.cancel": "Cancel",
+  "ai.msgbox.delete": "Delete",
   "ai.chooseLib.placeholder": "Choose Technology Stack",
   "ai.search.try": "Try",
   "ai.search.visualization": "Q&A",
   "ai.assistant.editor.intro": "I am AntV AI Assistant. You can ask me questions at any time to:\n1. Interpret the code configuration of the current chart.\n2. Generate new code based on the current case through natural language dialogue to customize the chart.",
   "ai.placeholder.whatis": "What is {title}?",
-  "header.ai.code": "AI Code Generation",
   "header.user.history": "History",
   "header.user.logout": "Logout",
   "header.user.deleteAccount": "Delete Account",

package/dist/locales/zh.json

@@ -181,7 +181,9 @@
   "ai.placeholder.implement": "今天，你想可视化什么？",
   "ai.placeholder.solve": "今天，你想解决什么可视化问题？",
   "ai.mode.implement": "可视化研发",
+  "ai.mode.implement.short": "研发",
   "ai.mode.solve": "可视化答疑",
+  "ai.mode.solve.short": "答疑",
   "ai.upload.tooltip": "仅支持csv,json,tsv,txt文件，为了性能和成本，只会使用文件的前几行作为样本",
   "ai.upload.data": "上传数据",
   "ai.upload.image": "上传图片",
@@ -220,18 +222,21 @@
   "ai.toolbar.clear.conversation": "清空对话",
   "ai.conversations.expand": "展开",
   "ai.conversations.collapse": "收起",
+  "ai.msgbox.continue.from.here": "从这里继续",
   "ai.conversations.new": "开始新对话",
   "ai.conversations.history": "历史对话",
   "ai.conversations.rename": "重命名",
   "ai.conversations.pin": "置顶",
   "ai.conversations.delete": "删除",
   "ai.conversations.edit.title": "编辑对话名称",
+  "ai.conversations.ok": "确定",
+  "ai.conversations.cancel": "取消",
+  "ai.msgbox.delete": "删除",
   "ai.chooseLib.placeholder": "选择技术栈",
   "ai.search.try": "试试",
   "ai.search.visualization": "可视化答疑",
   "ai.assistant.editor.intro": "我是AntV AI助手。您可以随时向我提问，让我为您：\n\n1、解读当前图表的代码配置。\n2、通过自然语言对话，基于当前案例生成新代码以定制图表。",
   "ai.placeholder.whatis": "{title}是什么？",
-  "header.ai.code": "AI生码",
   "header.user.history": "历史会话",
   "header.user.logout": "退出登录",
   "header.user.deleteAccount": "注销账号",

package/dist/model/AIChat.js

@@ -14,6 +14,7 @@ import localforage from 'localforage';
 import FingerprintJS from '@fingerprintjs/fingerprintjs';
 import { history } from "dumi";
 import { message } from "antd";
+import { AIMode } from "../components/AI/constant";
 
 // --- 配置 ---
 // 定义需要持久化的 state key
@@ -28,7 +29,8 @@ var initialState = {
   activeSessionId: null,
   tempMessage: null,
   codeBlock: null,
-  lib: null
+  lib: null,
+  mode: AIMode.implement
 };
 
 // --- valtio Store 创建 ---
@@ -203,9 +205,11 @@ subscribeKey(AIChatStore, 'activeSessionId', function () {
 });
 export var createPureNewSession = function createPureNewSession(title) {
   var newConversationName = 'New Conversation';
-  if (AIChatStore.sessions.some(function (s) {
+  var existNewSession = AIChatStore.sessions.find(function (s) {
     return s.title === newConversationName && s.messages.length === 0;
-  })) {
+  });
+  if (existNewSession) {
+    AIChatStore.activeSessionId = existNewSession.id;
     return;
   }
   var newSessionId = crypto.randomUUID();
@@ -285,4 +289,14 @@ export var clearAllChatData = /*#__PURE__*/function () {
   return function clearAllChatData() {
     return _ref2.apply(this, arguments);
   };
-}();
+}();
+export function deleteMessage(msgId) {
+  derivedState.activeSession.messages = derivedState.activeSession.messages.filter(function (m) {
+    return m.id !== msgId;
+  });
+}
+export function branchMessage(index) {
+  var messages = derivedState.activeSession.messages.slice(0, index + 1);
+  createPureNewSession(messages[0].content);
+  AIChatStore.sessions[0].messages = messages;
+}

package/dist/model/auth.js

@@ -4,6 +4,8 @@ function asyncGeneratorStep(gen, resolve, reject, _next, _throw, key, arg) { try
 function _asyncToGenerator(fn) { return function () { var self = this, args = arguments; return new Promise(function (resolve, reject) { var gen = fn.apply(self, args); function _next(value) { asyncGeneratorStep(gen, resolve, reject, _next, _throw, "next", value); } function _throw(err) { asyncGeneratorStep(gen, resolve, reject, _next, _throw, "throw", err); } _next(undefined); }); }; }
 import { proxy } from 'valtio';
 import request from "../utils/request";
+import { history } from 'dumi';
+import { message } from "antd";
 
 /**
  * 检查当前 URL 是否包含 'skipLogin=1' 参数
@@ -41,7 +43,7 @@ export var hideLoginModal = function hideLoginModal() {
 };
 export var loginOrRegister = /*#__PURE__*/function () {
   var _ref = _asyncToGenerator( /*#__PURE__*/_regeneratorRuntime().mark(function _callee(params) {
-    var _authStore$loginCallb;
+    var _authStore$loginCallb, _error$response, errorMessage;
     return _regeneratorRuntime().wrap(function _callee$(_context) {
       while (1) switch (_context.prev = _context.next) {
         case 0:
@@ -57,8 +59,12 @@ export var loginOrRegister = /*#__PURE__*/function () {
           _context.prev = 9;
           _context.t0 = _context["catch"](0);
           console.error('Login failed in store:', _context.t0);
+          errorMessage = _context.t0 === null || _context.t0 === void 0 || (_error$response = _context.t0.response) === null || _error$response === void 0 || (_error$response = _error$response.data) === null || _error$response === void 0 ? void 0 : _error$response.message;
+          if (errorMessage) {
+            message.error(errorMessage);
+          }
           return _context.abrupt("return", false);
-        case 13:
+        case 15:
         case "end":
           return _context.stop();
       }
@@ -86,12 +92,13 @@ export var logout = /*#__PURE__*/function () {
         case 8:
           _context2.prev = 8;
           authStore.isAuthenticated = false;
+          history.push('/');
           return _context2.finish(8);
-        case 11:
+        case 12:
         case "end":
           return _context2.stop();
       }
-    }, _callee2, null, [[0, 5, 8, 11]]);
+    }, _callee2, null, [[0, 5, 8, 12]]);
   }));
   return function logout() {
     return _ref2.apply(this, arguments);

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

@@ -85,7 +85,7 @@ export var ConversationsMenu = function ConversationsMenu() {
                 setState({
                   open: true,
                   session: session,
-                  rename: session.title
+                  rename: session.title.slice(0, 100)
                 });
               }
             }, {
@@ -156,10 +156,16 @@ export var ConversationsMenu = function ConversationsMenu() {
     },
     okButtonProps: {
       disabled: !state.rename
-    }
+    },
+    okText: formatMessage({
+      id: 'ai.conversations.ok'
+    }),
+    cancelText: formatMessage({
+      id: 'ai.conversations.cancel'
+    })
   }, /*#__PURE__*/React.createElement(Input, {
     showCount: true,
-    maxLength: 20,
+    maxLength: 100,
     onChange: function onChange(e) {
       return setState({
         rename: e.target.value

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

@@ -4,23 +4,24 @@ 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 { CheckOutlined, CopyOutlined, PlusSquareOutlined, SyncOutlined } from '@ant-design/icons';
+import { BranchesOutlined, CheckOutlined, CopyOutlined, DeleteOutlined, PlusSquareOutlined, SyncOutlined } from '@ant-design/icons';
 import { Bubble } from '@ant-design/x';
 import { Button, Flex, Space, Tooltip } from 'antd';
 import { useChat } from '@ai-sdk/react';
 import { TextStreamChatTransport } from 'ai';
-import { useIntl } 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';
 import { PromptTextarea } from "../../../../components/AI/HomeDialog/PromptTextarea";
-import { AIChatStore, clearEmptySession, createPureNewSession, derivedState } from "../../../../model/AIChat";
+import { AIChatStore, branchMessage, clearEmptySession, createPureNewSession, deleteMessage, derivedState } from "../../../../model/AIChat";
 import { getCodeFromMarkdown, isPreviewable } from "../../../../utils/code";
 import { MarkdownComponent } from "../MarkdownComponent";
 import styles from "./index.module.less";
 import { useAutoScroll } from "./useAutoScroll";
 import { getBaseURL } from "../../../../utils/env";
+import { AIMode } from "../../../../components/AI/constant";
 var avatar = {
   icon: /*#__PURE__*/React.createElement("img", {
     draggable: false,
@@ -92,6 +93,8 @@ function MsgBox(props) {
   // 使用 ref 存储动态值，避免重新创建 transport
   var anonymousUserIdRef = useRef(snap.anonymousUserId);
   var activeSessionIdRef = useRef((_derivedSnap$activeSe2 = derivedSnap.activeSession) === null || _derivedSnap$activeSe2 === void 0 ? void 0 : _derivedSnap$activeSe2.id);
+  var _useSiteData = useSiteData(),
+    themeConfig = _useSiteData.themeConfig;
   useEffect(function () {
     var _derivedSnap$activeSe3;
     anonymousUserIdRef.current = snap.anonymousUserId;
@@ -119,7 +122,7 @@ function MsgBox(props) {
             mountId: 'container',
             antvContext: (latestUserMessage === null || latestUserMessage === void 0 ? void 0 : latestUserMessage.context) || props.context,
             library: AIChatStore.lib,
-            mode: latestUserMessage === null || latestUserMessage === void 0 ? void 0 : latestUserMessage.mode
+            mode: AIChatStore.mode
           };
         }
       }),
@@ -187,7 +190,7 @@ function MsgBox(props) {
       body: {
         context: fileSummary,
         lib: snap.lib,
-        mode: 'implement'
+        mode: snap.mode
       }
     });
     setPromptText('');
@@ -237,6 +240,10 @@ function MsgBox(props) {
     chatScrollIntoView();
     if (simple) {
       createPureNewSession(title);
+      AIChatStore.mode = AIMode.implement;
+      if (!themeConfig.isAntVSite && themeConfig.title) {
+        AIChatStore.lib = themeConfig.title;
+      }
     }
     return function () {
       if (simple) {
@@ -267,9 +274,9 @@ function MsgBox(props) {
         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, {
+      footer: status === 'ready' && /*#__PURE__*/React.createElement(Space, {
         size: "small"
-      }, /*#__PURE__*/React.createElement(Tooltip, {
+      }, msg.role === 'assistant' && index === messages.length - 1 && /*#__PURE__*/React.createElement(Tooltip, {
         title: formatMessage({
           id: 'ai.msgbox.retry'
         })
@@ -294,7 +301,31 @@ function MsgBox(props) {
           return copyToClipboard(textContent);
         },
         icon: copyState.value === textContent ? /*#__PURE__*/React.createElement(CheckOutlined, null) : /*#__PURE__*/React.createElement(CopyOutlined, null)
-      }))) : null,
+      })), /*#__PURE__*/React.createElement(Tooltip, {
+        title: formatMessage({
+          id: 'ai.msgbox.continue.from.here'
+        })
+      }, /*#__PURE__*/React.createElement(Button, {
+        color: "default",
+        variant: "text",
+        size: "small",
+        onClick: function onClick() {
+          return branchMessage(index);
+        },
+        icon: /*#__PURE__*/React.createElement(BranchesOutlined, null)
+      })), msg.role === 'assistant' && /*#__PURE__*/React.createElement(Tooltip, {
+        title: formatMessage({
+          id: 'ai.msgbox.delete'
+        })
+      }, /*#__PURE__*/React.createElement(Button, {
+        color: "default",
+        variant: "text",
+        size: "small",
+        onClick: function onClick() {
+          return deleteMessage(msg.id);
+        },
+        icon: /*#__PURE__*/React.createElement(DeleteOutlined, null)
+      }))),
       placement: msg.role === 'user' ? 'end' : 'start'
     });
   }), (status === 'streaming' || status === 'submitted') && ((_messages = messages[messages.length - 1]) === null || _messages === void 0 ? void 0 : _messages.role) === 'user' && /*#__PURE__*/React.createElement(Bubble, {
@@ -316,7 +347,6 @@ function MsgBox(props) {
     id: 'ai.msgbox.start.new.chat'
   }))))), /*#__PURE__*/React.createElement(PromptTextarea, {
     size: "compact",
-    mode: "implement",
     value: promptText,
     onChange: setPromptText,
     loading: status === 'streaming' || status === 'submitted',
@@ -326,7 +356,8 @@ function MsgBox(props) {
       marginBottom: 0
     },
     onConfirm: handleSubmit,
-    onDataSummaryChange: setFileSummary
+    onDataSummaryChange: setFileSummary,
+    showModeSelector: true
   })));
 }
 export default MsgBox;

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

@@ -38,7 +38,8 @@ function SessionLayout(props) {
     primary: "second",
     style: {
       position: 'unset'
-    }
+    },
+    minSize: 100
   }, /*#__PURE__*/React.createElement("div", {
     className: classnames(styles.msgBox)
   }, children[0]), /*#__PURE__*/React.createElement("div", {

package/dist/pages/AIPlayground/components/SessionLayout/index.module.less

@@ -2,8 +2,9 @@
   display: flex;
   gap: 12px;
   background: #f6f7fb;
-  height: calc(100vh - 64px);
   padding: 0;
+  flex-grow: 1;
+  overflow-y: auto;
 }
 
 .msgBox, .taskBox{

package/dist/pages/AIPlayground/components/TaskBox/generateCode.js

@@ -79,11 +79,17 @@ export function wrap2VisionSnap() {
   var codeBlock = arguments.length > 0 && arguments[0] !== undefined ? arguments[0] : '';
   var dependencies = generateDependencies(codeBlock);
   var rootElementType = dependencies['@antv/f2'] ? 'canvas' : 'div';
+  var dependenciesJSON = {
+    "name": "AntV-adapted-project",
+    "version": "1.0.0",
+    "main": "/src/index.jsx",
+    "dependencies": dependencies
+  };
   return {
     modules: {
       '/package.json': {
         fpath: '/package.json',
-        code: "{\n  \"name\": \"AntV-adapted-project\",\n  \"version\": \"1.0.0\",\n  \"main\": \"/src/index.jsx\",\n  \"dependencies\": ".concat(JSON.stringify(dependencies, null, 2), "\n}")
+        code: JSON.stringify(dependenciesJSON, null, 2)
       },
       '/src/index.jsx': {
         fpath: '/src/index.jsx',

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

@@ -48,7 +48,8 @@ function TaskBox() {
   if (loading) {
     return /*#__PURE__*/React.createElement(Loading, null);
   }
-  if (themeConfig.isAntVSite || ((_themeConfig$ai = themeConfig.ai) === null || _themeConfig$ai === void 0 ? void 0 : _themeConfig$ai.codeRunner) === "VisionSnap" || !((_themeConfig$ai2 = themeConfig.ai) !== null && _themeConfig$ai2 !== void 0 && _themeConfig$ai2.codeRunner)) {
+  var wrappedVisionSnapCode = wrap2VisionSnap(snap.codeBlock);
+  if (themeConfig.isAntVSite || ((_themeConfig$ai = themeConfig.ai) === null || _themeConfig$ai === void 0 ? void 0 : _themeConfig$ai.codeRunner) === "VisionSnap" || !((_themeConfig$ai2 = themeConfig.ai) !== null && _themeConfig$ai2 !== void 0 && _themeConfig$ai2.codeRunner) || !wrappedVisionSnapCode.modules["/package.json"].code.includes("@antv/f2")) {
     return /*#__PURE__*/React.createElement(ErrorBoundary, {
       FallbackComponent: ErrorFallback
     }, /*#__PURE__*/React.createElement(sdk.VisionPreview, {
@@ -61,7 +62,7 @@ function TaskBox() {
       displayMode: "code-and-preview",
       initialView: "preview",
       theme: "light",
-      code: wrap2VisionSnap(snap.codeBlock),
+      code: wrappedVisionSnapCode,
       requestProxy: requestProxy,
       isStreaming: false,
       proxyOptions: {

package/dist/pages/AIPlayground/index.js

@@ -3,7 +3,10 @@ import { SessionLayout } from "./components/SessionLayout";
 import MsgBox from "./components/MsgBox";
 import TaskBox from "./components/TaskBox";
 import Header from "dumi/theme/slots/Header";
+import styles from "./index.module.less";
 function AIPlayground() {
-  return /*#__PURE__*/React.createElement(React.Fragment, null, /*#__PURE__*/React.createElement(Header, null), /*#__PURE__*/React.createElement(SessionLayout, null, /*#__PURE__*/React.createElement(MsgBox, null), /*#__PURE__*/React.createElement(TaskBox, null)));
+  return /*#__PURE__*/React.createElement("div", {
+    className: styles.aiPlayground
+  }, /*#__PURE__*/React.createElement(Header, null), /*#__PURE__*/React.createElement(SessionLayout, null, /*#__PURE__*/React.createElement(MsgBox, null), /*#__PURE__*/React.createElement(TaskBox, null)));
 }
 export default AIPlayground;

package/dist/pages/AIPlayground/index.module.less

@@ -0,0 +1,5 @@
+.aiPlayground {
+  display: flex;
+  flex-direction: column;
+  height: 100vh;
+}

package/dist/slots/Detail/index.js

@@ -23,6 +23,7 @@ import { News } from "./News";
 import { ic } from "../hooks";
 import styles from "./index.module.less";
 import { HomeDialog } from "../../components/AI/HomeDialog";
+import { getBaseSiteDataUrl } from "../../utils/env";
 /**
  * Index.技术栈的描述区域！
  * 各自配置
@@ -48,7 +49,7 @@ export var Detail = function Detail(_ref) {
     setRemoteNews = _useState2[1];
   var lang = useLocale().id;
   useEffect(function () {
-    fetch('https://assets.antv.antgroup.com/antv/banner-messages.json' // 生产环境
+    fetch("".concat(getBaseSiteDataUrl(), "/antv/banner-messages.json") // 生产环境
     // 'https://site-data-pre.alipay.com/antv/banner-messages.json', // 预发测试
     ).then(function (res) {
       return res.json();

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

@@ -2,7 +2,7 @@ import { useIntl, useSiteData } from 'dumi';
 import React from 'react';
 import styles from "./SearchResult.module.less";
 import classnames from "classnames";
-import { createNewSession } from "../../../model/AIChat";
+import { AIChatStore, createNewSession } from "../../../model/AIChat";
 import { authStore, showLoginModal } from "../../../model/auth";
 import { useSnapshot } from "valtio";
 var getHighlightInfo = function getHighlightInfo(textSegments) {
@@ -25,6 +25,7 @@ export var SearchResult = function SearchResult(_ref) {
   var intl = useIntl();
   var authSnap = useSnapshot(authStore);
   function pureSearch() {
+    AIChatStore.mode = 'solve';
     createNewSession({
       promptText: keywords,
       mode: 'solve',

package/dist/slots/Header/index.js

@@ -18,7 +18,7 @@ import { Alert, Button, Dropdown, Menu, Modal, Popover } from 'antd';
 import cx from 'classnames';
 import { FormattedMessage, Link, useLocale, useSiteData, useIntl, history } from 'dumi';
 import { get, map, size } from 'lodash-es';
-import React, { useEffect, useMemo, useState } from 'react';
+import React, { useEffect, useState } from 'react';
 import { useMedia } from 'react-use';
 import { getPurePathname } from "../../utils/location";
 import { ic, icWithLocale } from "../hooks";
@@ -57,9 +57,6 @@ var HeaderComponent = function HeaderComponent(_ref) {
     showAntVProductsCard = _ref$showAntVProducts === void 0 ? true : _ref$showAntVProducts,
     _ref$showLanguageSwit = _ref.showLanguageSwitcher,
     showLanguageSwitcher = _ref$showLanguageSwit === void 0 ? true : _ref$showLanguageSwit,
-    _ref$showWeavefox = _ref.showWeavefox,
-    showWeavefox = _ref$showWeavefox === void 0 ? {} : _ref$showWeavefox,
-    isInternalUser = _ref.isInternalUser,
     logo = _ref.logo,
     onLanguageChange = _ref.onLanguageChange,
     _ref$showWxQrcode = _ref.showWxQrcode,
@@ -154,8 +151,7 @@ var HeaderComponent = function HeaderComponent(_ref) {
       }),
       link: ''
     }, logo),
-    img = _img$link$logo.img,
-    link = _img$link$logo.link;
+    img = _img$link$logo.img;
   var _useLocation = useLocation(),
     pathname = _useLocation.pathname;
   useEffect(function () {
@@ -190,27 +186,13 @@ var HeaderComponent = function HeaderComponent(_ref) {
   var handleSwitchLanguage = function handleSwitchLanguage() {
     onLanguageChange === null || onLanguageChange === void 0 || onLanguageChange(lang);
   };
-  var weavefox = useMemo(function () {
-    var configKey = isInternalUser ? 'internal' : 'public';
-    var configValue = showWeavefox[configKey];
-    if (!configValue) return /*#__PURE__*/React.createElement(React.Fragment, null);
-    var defaultLink = isInternalUser ? 'https://weavefox.antgroup-inc.cn/agent/@huiyu.zjt/AntV' : 'https://weavefox.alipay.com/agent/@ufox-b8tydq-0758/202505AP7vfl00422922';
-    var href = typeof configValue === 'string' ? configValue : defaultLink;
-    return /*#__PURE__*/React.createElement("li", null, /*#__PURE__*/React.createElement("a", {
-      href: href,
-      target: "_blank",
-      rel: "noreferrer"
-    }, /*#__PURE__*/React.createElement(FormattedMessage, {
-      id: "header.ai.code"
-    })));
-  }, [isInternalUser, showWeavefox]);
   var menu = /*#__PURE__*/React.createElement("ul", {
     className: cx(styles.menu, _defineProperty(_defineProperty({}, styles.popup, !isWide), styles.popupHidden, !popupMenuVisible))
   }, /** 最左侧的菜单，一般是 教程、API、示例，或者其他自定义，有配置文件中的 `navs` 决定 */
   size(navs) ? /*#__PURE__*/React.createElement(Navs, {
     navs: navs,
     path: pathname
-  }) : null, weavefox, /** 生态产品 */
+  }) : null, /** 生态产品 */
   size(ecosystems) ? /*#__PURE__*/React.createElement("li", null, /*#__PURE__*/React.createElement(Dropdown, {
     className: styles.ecoSystems,
     overlay: /*#__PURE__*/React.createElement(Menu, null, map(ecosystems, function (_ref2) {

package/dist/utils/env.js

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

package/package.json

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