dskcode 0.1.11 → 0.1.12

package/dist/index.js

@@ -4,6 +4,7 @@ import {
   ModelNotSupportedError
 } from "./chunk-FODNQIT4.js";
 import {
+  SUPPORTED_MODELS,
   calculateCost,
   estimateTokens,
   getModelMeta,
@@ -303,6 +304,34 @@ async function saveApiKey(apiKey) {
   await writeFile(configFile, JSON.stringify(configData, null, 2), "utf-8");
   return configFile;
 }
+async function saveModelConfig(model) {
+  const home = process.env.HOME ?? process.env.USERPROFILE ?? "~";
+  const configDir = join(home, ".dskcode");
+  const configFile = join(configDir, "settings.json");
+  await mkdir(configDir, { recursive: true });
+  let configData;
+  try {
+    const raw = await readFile(configFile, "utf-8");
+    configData = JSON.parse(raw);
+  } catch {
+    configData = structuredClone(defaultConfig);
+  }
+  const providers = configData.providers ?? [];
+  const defaultProviderName = configData.defaultProvider ?? "deepseek";
+  const existing = providers.find((p) => p.name === defaultProviderName);
+  if (existing) {
+    existing.model = model;
+  } else {
+    providers.push({
+      name: defaultProviderName,
+      baseUrl: "https://api.deepseek.com",
+      model
+    });
+  }
+  configData.providers = providers;
+  await writeFile(configFile, JSON.stringify(configData, null, 2), "utf-8");
+  return configFile;
+}
 async function saveStockConfig(symbols) {
   const home = process.env.HOME ?? process.env.USERPROFILE ?? "~";
   const configDir = join(home, ".dskcode");
@@ -1308,6 +1337,14 @@ var IDLE_GRADIENT_STOPS = [
   [189, 147, 249]
   // #bd93f9
 ];
+var CMD_TIP_GRADIENT_STOPS = [
+  [255, 245, 180],
+  // #fff5b4 浅柠黄
+  [255, 210, 80],
+  // #ffd250 暖黄
+  [255, 150, 50]
+  // #ff9632 橙色
+];
 var STREAMING_GRADIENT_STOPS = [
   [255, 191, 0],
   // #ffbf00 琥珀金
@@ -1324,7 +1361,11 @@ var GRADIENT_ANIMATION = {
   /** 流式占位符动画：每帧相位步进 */
   streamingPhaseStep: 0.05,
   /** 流式占位符动画：帧间隔（ms） */
-  streamingInterval: 40
+  streamingInterval: 40,
+  /** 命令提示条动画：每帧相位步进 */
+  cmdTipPhaseStep: 0.05,
+  /** 命令提示条动画：帧间隔（ms） */
+  cmdTipInterval: 40
 };
 var SKIP_CHARS = /* @__PURE__ */ new Set([" ", "~", ".", "\u3002", "\uFF01", "\u{1F447}"]);
 function lerpStopsToHex(t, stops) {
@@ -1351,7 +1392,7 @@ function getGradientColors(text, phase, stops) {
 }
 
 // src/ui/ChatSession.tsx
-import { jsx as jsx5, jsxs as jsxs5 } from "react/jsx-runtime";
+import { Fragment, jsx as jsx5, jsxs as jsxs5 } from "react/jsx-runtime";
 var commandRegistry = /* @__PURE__ */ new Map();
 function registerCommand(name, cmd) {
   commandRegistry.set(name, cmd);
@@ -1374,6 +1415,7 @@ registerCommand("/help", {
 });
 registerCommand("/clear", { desc: "\u6E05\u7A7A\u5BF9\u8BDD\u5386\u53F2", handler: () => ({ kind: "clear" }) });
 registerCommand("/version", { desc: "\u663E\u793A\u7248\u672C\u4FE1\u606F", handler: () => ({ kind: "text", content: "dskcode v0.1.10" }) });
+registerCommand("/model", { desc: "\u5207\u6362\u6A21\u578B", handler: () => ({ kind: "text", content: "\u8BF7\u76F4\u63A5\u8F93\u5165 /model \u8FDB\u5165\u9009\u62E9\u754C\u9762" }) });
 registerCommand("/game", { desc: "\u542F\u52A8\u6E38\u620F", handler: () => ({ kind: "navigate", target: "game" }) });
 registerCommand("/stock", { desc: "\u67E5\u770B\u80A1\u7968\u884C\u60C5", handler: () => ({ kind: "navigate", target: "stock" }) });
 var STREAMING_PLACEHOLDERS = [
@@ -1388,7 +1430,7 @@ var IDLE_PLACEHOLDERS = [
   "\u60F3\u5E72\u5565\uFF1F\u76F4\u63A5\u8BF4~",
   "\u6765\u5427\uFF0C\u5429\u5490\u70B9\u5565",
   "\u968F\u65F6\u5F85\u547D...",
-  "\u6233\u8FD9\u91CC\u5F00\u804A \u{1F447}",
+  "\u6233\u8FD9\u91CC\u5F00\u804A...",
   "\u7B49\u4F60\u5F00\u53E3...",
   "\u5C3D\u7BA1\u4F7F\u5524~"
 ];
@@ -1426,8 +1468,16 @@ function ChatSession({
   const [currentUsage, setCurrentUsage] = useState3(void 0);
   const [currentElapsed, setCurrentElapsed] = useState3(void 0);
   const [currentCost, setCurrentCost] = useState3(void 0);
-  const [currentModel, setCurrentModel] = useState3(void 0);
+  const [activeModel, setActiveModel] = useState3(model);
+  const [streamingModel, setStreamingModel] = useState3(void 0);
   const [streamError, setStreamError] = useState3(void 0);
+  const cmdTips = Array.from(getRegisteredCommands()).filter(([name]) => name !== "/exit" && name !== "/quit").map(([name, cmd]) => ({ name, desc: cmd.desc }));
+  const [cmdTipIndex, setCmdTipIndex] = useState3(0);
+  const [cmdTipGradientColors, setCmdTipGradientColors] = useState3([]);
+  const cmdTipPhaseRef = useRef2(0);
+  const [selectingModel, setSelectingModel] = useState3(false);
+  const [modelSelectIndex, setModelSelectIndex] = useState3(0);
+  const modelOptions = ["deepseek-v4-flash", "deepseek-v4-pro"];
   const sessionRef = useRef2(null);
   const abortRef = useRef2(null);
   const currentContentRef = useRef2("");
@@ -1443,6 +1493,34 @@ function ChatSession({
   useInput(
     useCallback2(
       (_input, key) => {
+        if (selectingModel) {
+          if (key.upArrow) {
+            setModelSelectIndex((prev) => (prev - 1 + modelOptions.length) % modelOptions.length);
+          } else if (key.downArrow) {
+            setModelSelectIndex((prev) => (prev + 1) % modelOptions.length);
+          } else if (key.return) {
+            const selected = modelOptions[modelSelectIndex];
+            if (selected === activeModel) {
+              setDisplayMessages((prev) => [
+                ...prev,
+                { role: "assistant", content: `\u5DF2\u7ECF\u5728\u4F7F\u7528 ${SUPPORTED_MODELS[selected].displayName}` }
+              ]);
+            } else {
+              setActiveModel(selected);
+              sessionRef.current?.reset();
+              saveModelConfig(selected).catch(() => {
+              });
+              setDisplayMessages((prev) => [
+                ...prev,
+                { role: "assistant", content: `\u6A21\u578B\u5DF2\u5207\u6362\u4E3A ${SUPPORTED_MODELS[selected].displayName}\uFF08${selected}\uFF09` }
+              ]);
+            }
+            setSelectingModel(false);
+          } else if (key.escape) {
+            setSelectingModel(false);
+          }
+          return;
+        }
         if (key.ctrl && _input === "c") {
           if (isStreaming) {
             abortRef.current?.abort();
@@ -1455,9 +1533,31 @@ function ChatSession({
           setInput(_input);
         }
       },
-      [isStreaming, handleCtrlC, input]
+      [selectingModel, modelSelectIndex, modelOptions, activeModel, isStreaming, handleCtrlC, input]
     )
   );
+  useEffect3(() => {
+    if (cmdTips.length <= 1) return;
+    const timer = setInterval(() => {
+      setCmdTipIndex((prev) => (prev + 1) % cmdTips.length);
+    }, 5e3);
+    return () => clearInterval(timer);
+  }, [cmdTips.length]);
+  useEffect3(() => {
+    const tip = cmdTips[cmdTipIndex % cmdTips.length];
+    if (!tip) {
+      setCmdTipGradientColors([]);
+      return;
+    }
+    const text = `${tip.name} ${tip.desc}`;
+    cmdTipPhaseRef.current = 0;
+    setCmdTipGradientColors(getGradientColors(text, 1, CMD_TIP_GRADIENT_STOPS));
+    const interval = setInterval(() => {
+      cmdTipPhaseRef.current = (cmdTipPhaseRef.current + GRADIENT_ANIMATION.cmdTipPhaseStep) % 1;
+      setCmdTipGradientColors(getGradientColors(text, 1 - cmdTipPhaseRef.current, CMD_TIP_GRADIENT_STOPS));
+    }, GRADIENT_ANIMATION.cmdTipInterval);
+    return () => clearInterval(interval);
+  }, [cmdTipIndex, cmdTips.length]);
   useEffect3(() => {
     const timer = setInterval(() => {
       setOffset((prev) => (prev + 1) % CYBER_PALETTE.length);
@@ -1470,7 +1570,7 @@ function ChatSession({
       name: "deepseek",
       apiKey,
       baseUrl,
-      model
+      model: activeModel
     });
     const tracker = externalCostTracker ?? new CostTracker();
     const session = new Session(provider, [], tracker, {
@@ -1480,7 +1580,7 @@ function ChatSession({
     return () => {
       sessionRef.current = null;
     };
-  }, [apiKey, baseUrl, model, externalCostTracker]);
+  }, [apiKey, baseUrl, activeModel, externalCostTracker]);
   useEffect3(() => {
     if (!apiKey || !baseUrl) return;
     let cancelled = false;
@@ -1530,10 +1630,10 @@ function ChatSession({
       return;
     }
     gradientPhaseRef.current = 0;
-    setGradientColors(getGradientColors(idlePlaceholder, 0, IDLE_GRADIENT_STOPS));
+    setGradientColors(getGradientColors(idlePlaceholder, 1, IDLE_GRADIENT_STOPS));
     const interval = setInterval(() => {
       gradientPhaseRef.current = (gradientPhaseRef.current + GRADIENT_ANIMATION.idlePhaseStep) % 1;
-      setGradientColors(getGradientColors(idlePlaceholder, gradientPhaseRef.current, IDLE_GRADIENT_STOPS));
+      setGradientColors(getGradientColors(idlePlaceholder, 1 - gradientPhaseRef.current, IDLE_GRADIENT_STOPS));
     }, GRADIENT_ANIMATION.idleInterval);
     return () => clearInterval(interval);
   }, [isStreaming, idlePlaceholder]);
@@ -1543,10 +1643,10 @@ function ChatSession({
       return;
     }
     streamingPhaseRef.current = 0;
-    setStreamingGradientColors(getGradientColors(streamingPlaceholder, 0, STREAMING_GRADIENT_STOPS));
+    setStreamingGradientColors(getGradientColors(streamingPlaceholder, 1, STREAMING_GRADIENT_STOPS));
     const interval = setInterval(() => {
       streamingPhaseRef.current = (streamingPhaseRef.current + GRADIENT_ANIMATION.streamingPhaseStep) % 1;
-      setStreamingGradientColors(getGradientColors(streamingPlaceholder, streamingPhaseRef.current, STREAMING_GRADIENT_STOPS));
+      setStreamingGradientColors(getGradientColors(streamingPlaceholder, 1 - streamingPhaseRef.current, STREAMING_GRADIENT_STOPS));
     }, GRADIENT_ANIMATION.streamingInterval);
     return () => clearInterval(interval);
   }, [isStreaming, streamingPlaceholder]);
@@ -1554,6 +1654,13 @@ function ChatSession({
     const trimmed = value.trim();
     if (!trimmed) return;
     if (trimmed.startsWith("/")) {
+      if (trimmed.toLowerCase() === "/model") {
+        const curIdx = modelOptions.indexOf(activeModel);
+        setModelSelectIndex(curIdx >= 0 ? curIdx : 0);
+        setSelectingModel(true);
+        setInput("");
+        return;
+      }
       const cmd = commandRegistry.get(trimmed.toLowerCase());
       if (cmd) {
         const result = cmd.handler();
@@ -1613,7 +1720,7 @@ function ChatSession({
     setCurrentUsage(void 0);
     setCurrentElapsed(void 0);
     setCurrentCost(void 0);
-    setCurrentModel(void 0);
+    setStreamingModel(void 0);
     setStreamError(void 0);
     currentContentRef.current = "";
     currentToolCallsRef.current = [];
@@ -1645,7 +1752,7 @@ function ChatSession({
             break;
           case "usage":
             setCurrentUsage(event.usage);
-            setCurrentModel(event.model);
+            setStreamingModel(event.model);
             currentUsageRef.current = event.usage;
             currentModelRef.current = event.model;
             break;
@@ -1696,14 +1803,14 @@ function ChatSession({
     }
   }, [isStreaming, externalCostTracker]);
   useEffect3(() => {
-    if (currentUsage && currentModel && !currentCost) {
+    if (currentUsage && streamingModel && !currentCost) {
       import("./models-NQGNKB4T.js").then(({ calculateCost: calculateCost2 }) => {
-        const cost = calculateCost2(currentUsage, currentModel);
+        const cost = calculateCost2(currentUsage, streamingModel);
         setCurrentCost(cost.totalCost);
         currentCostRef.current = cost.totalCost;
       });
     }
-  }, [currentUsage, currentModel, currentCost]);
+  }, [currentUsage, streamingModel, currentCost]);
   return /* @__PURE__ */ jsxs5(Box5, { flexDirection: "column", paddingLeft: 1, paddingRight: 1, children: [
     /* @__PURE__ */ jsxs5(Box5, { flexDirection: "row", marginBottom: 1, children: [
       /* @__PURE__ */ jsx5(Box5, { flexDirection: "column", marginRight: 4, children: LOGO_LINES.map((line, i) => {
@@ -1725,8 +1832,17 @@ function ChatSession({
         ] }),
         /* @__PURE__ */ jsxs5(Text6, { color: "#00ffff", children: [
           "  \u{1F527} \u6A21\u578B ",
-          model
+          SUPPORTED_MODELS[activeModel]?.displayName ?? activeModel
         ] }),
+        cmdTips.length > 0 && (() => {
+          const tip = cmdTips[cmdTipIndex % cmdTips.length];
+          if (!tip) return null;
+          const text = `${tip.name} ${tip.desc}`;
+          return /* @__PURE__ */ jsxs5(Text6, { children: [
+            /* @__PURE__ */ jsx5(Text6, { color: "#808080", children: "  \u{1F4A1} " }),
+            cmdTipGradientColors.length > 0 ? text.split("").map((ch, i) => /* @__PURE__ */ jsx5(Text6, { color: cmdTipGradientColors[i] || void 0, children: ch }, i)) : /* @__PURE__ */ jsx5(Text6, { color: "#808080", children: text })
+          ] });
+        })(),
         verbose ? /* @__PURE__ */ jsx5(Text6, { color: "#ff1493", children: "  \u26A1 Verbose" }) : null
       ] }),
       /* @__PURE__ */ jsxs5(Box5, { flexGrow: 1, flexDirection: "column", justifyContent: "center", alignItems: "flex-end", children: [
@@ -1783,20 +1899,54 @@ function ChatSession({
         streamError
       ] }) })
     ] }),
-    /* @__PURE__ */ jsx5(Box5, { marginTop: 1, children: /* @__PURE__ */ jsx5(Text6, { color: "#00ffff", dimColor: true, children: "\u2500".repeat(dividerWidth) }) }),
-    /* @__PURE__ */ jsxs5(Box5, { children: [
-      /* @__PURE__ */ jsx5(Box5, { width: 4, flexShrink: 0, children: /* @__PURE__ */ jsx5(Text6, { bold: true, color: "#00ff41", children: "\u26A1" }) }),
-      /* @__PURE__ */ jsx5(Box5, { flexGrow: 1, children: !input && !isStreaming && idlePlaceholder && gradientColors.length > 0 ? /* @__PURE__ */ jsx5(Text6, { children: idlePlaceholder.split("").map((ch, i) => /* @__PURE__ */ jsx5(Text6, { color: gradientColors[i] ?? void 0, children: ch }, i)) }) : !input && isStreaming && streamingPlaceholder && streamingGradientColors.length > 0 ? /* @__PURE__ */ jsx5(Text6, { children: streamingPlaceholder.split("").map((ch, i) => /* @__PURE__ */ jsx5(Text6, { color: streamingGradientColors[i] ?? void 0, children: ch }, i)) }) : /* @__PURE__ */ jsx5(
-        TextInput,
-        {
-          value: input,
-          onChange: setInput,
-          onSubmit: handleSubmit,
-          placeholder: ""
-        }
-      ) })
+    selectingModel ? /* @__PURE__ */ jsxs5(Box5, { marginTop: 1, flexDirection: "column", children: [
+      /* @__PURE__ */ jsx5(Text6, { color: "#00ffff", dimColor: true, children: "\u2500".repeat(dividerWidth) }),
+      /* @__PURE__ */ jsxs5(Box5, { flexDirection: "column", marginTop: 1, children: [
+        /* @__PURE__ */ jsx5(Text6, { bold: true, color: "#00ffff", children: "\u9009\u62E9\u6A21\u578B\uFF1A" }),
+        modelOptions.map((id, i) => {
+          const meta = SUPPORTED_MODELS[id];
+          const isCurrent = id === activeModel;
+          const isSelected = i === modelSelectIndex;
+          const marker = isSelected ? " > " : "   ";
+          const suffix = isCurrent ? " (\u5F53\u524D)" : "";
+          return /* @__PURE__ */ jsxs5(Box5, { children: [
+            /* @__PURE__ */ jsxs5(
+              Text6,
+              {
+                color: isSelected ? "#00ff41" : void 0,
+                bold: isSelected,
+                children: [
+                  marker,
+                  meta.displayName,
+                  suffix
+                ]
+              }
+            ),
+            isSelected && /* @__PURE__ */ jsxs5(Text6, { color: "#808080", children: [
+              " \u2014 ",
+              id
+            ] })
+          ] }, id);
+        }),
+        /* @__PURE__ */ jsx5(Box5, { marginTop: 1, children: /* @__PURE__ */ jsx5(Text6, { color: "#808080", dimColor: true, children: "\u2191\u2193 \u9009\u62E9 \xB7 Enter \u786E\u8BA4 \xB7 Esc \u53D6\u6D88" }) })
+      ] }),
+      /* @__PURE__ */ jsx5(Text6, { color: "#00ffff", dimColor: true, children: "\u2500".repeat(dividerWidth) })
+    ] }) : /* @__PURE__ */ jsxs5(Fragment, { children: [
+      /* @__PURE__ */ jsx5(Box5, { marginTop: 1, children: /* @__PURE__ */ jsx5(Text6, { color: "#00ffff", dimColor: true, children: "\u2500".repeat(dividerWidth) }) }),
+      /* @__PURE__ */ jsxs5(Box5, { children: [
+        /* @__PURE__ */ jsx5(Box5, { width: 4, flexShrink: 0, children: /* @__PURE__ */ jsx5(Text6, { bold: true, color: "#00ff41", children: "\u26A1" }) }),
+        /* @__PURE__ */ jsx5(Box5, { flexGrow: 1, children: !input && !isStreaming && idlePlaceholder && gradientColors.length > 0 ? /* @__PURE__ */ jsx5(Text6, { children: idlePlaceholder.split("").map((ch, i) => /* @__PURE__ */ jsx5(Text6, { color: gradientColors[i] ?? void 0, children: ch }, i)) }) : !input && isStreaming && streamingPlaceholder && streamingGradientColors.length > 0 ? /* @__PURE__ */ jsx5(Text6, { children: streamingPlaceholder.split("").map((ch, i) => /* @__PURE__ */ jsx5(Text6, { color: streamingGradientColors[i] ?? void 0, children: ch }, i)) }) : /* @__PURE__ */ jsx5(
+          TextInput,
+          {
+            value: input,
+            onChange: setInput,
+            onSubmit: handleSubmit,
+            placeholder: ""
+          }
+        ) })
+      ] }),
+      /* @__PURE__ */ jsx5(Box5, { children: /* @__PURE__ */ jsx5(Text6, { color: "#00ffff", dimColor: true, children: "\u2500".repeat(dividerWidth) }) })
     ] }),
-    /* @__PURE__ */ jsx5(Box5, { children: /* @__PURE__ */ jsx5(Text6, { color: "#00ffff", dimColor: true, children: "\u2500".repeat(dividerWidth) }) }),
     doubleCtrlC && !isStreaming && /* @__PURE__ */ jsx5(Box5, { marginTop: 1, children: /* @__PURE__ */ jsx5(Text6, { color: "#ff1493", bold: true, children: "  \u26A0 \u518D\u6309\u4E00\u6B21 Ctrl+C \u9000\u51FA dskcode" }) }),
     isStreaming && /* @__PURE__ */ jsx5(Box5, { marginTop: 1, children: /* @__PURE__ */ jsx5(Text6, { color: "yellow", dimColor: true, children: "  \u63D0\u793A\uFF1A\u6309 Ctrl+C \u53D6\u6D88\u5F53\u524D\u8BF7\u6C42" }) })
   ] });