Merge branch 'master-nj' into 'master'

Master nj

See merge request knowledge-base-bj/deerflow-code!23

frontend-web/src/components/workspace/input-box.tsx

@@ -22,10 +22,10 @@ import {
   useEffect,
   useMemo,
   useRef,
-  useState,
-  type ComponentProps,
-  type ReactNode,
-} from "react";
+  useState,
+  type ComponentProps,
+  type ReactNode,
+} from "react";
 
 import {
   PromptInput,
@@ -62,10 +62,10 @@ import {
 } from "@/components/ui/dropdown-menu";
 import { apiFetch } from "@/core/api";
 import { getBackendBaseURL } from "@/core/config";
-import { useI18n } from "@/core/i18n/hooks";
-import { isHiddenFromUIMessage } from "@/core/messages/utils";
-import { useModels } from "@/core/models/hooks";
-import type { Model } from "@/core/models/types";
+import { useI18n } from "@/core/i18n/hooks";
+import { isHiddenFromUIMessage } from "@/core/messages/utils";
+import { useModels } from "@/core/models/hooks";
+import type { Model } from "@/core/models/types";
 import type { AgentThreadContext } from "@/core/threads";
 import { textOfMessage } from "@/core/threads/utils";
 import { desensitize } from "@/lib/desensitize";
@@ -134,11 +134,11 @@ export function InputBox({
   onFollowupsVisibilityChange,
   onSubmit,
   onStop,
-  promptPrefix,
-  agentSelector,
-  knowledgeSelector,
-  extraModels = [],
-  assistantId,
+  promptPrefix,
+  agentSelector,
+  knowledgeSelector,
+  extraModels = [],
+  assistantId,
   hideAdminRecommendedQuestions,
   compact,
   ...props
@@ -194,28 +194,28 @@ export function InputBox({
    * homepage (``ChatPage``) passes this; ``AgentChatPage`` omits it because the
    * thread is already bound to one agent.
    */
-  agentSelector?: {
-    selectedAgentId: string | null;
-    onAgentChange: (agentId: string | null) => void;
-  };
-  /** WeKnora-style knowledge-space picker rendered inside the composer toolbar. */
-  knowledgeSelector?: ReactNode;
-  /** Page-scoped model choices implemented outside the DeerFlow model API. */
-  extraModels?: Model[];
-}) {
+  agentSelector?: {
+    selectedAgentId: string | null;
+    onAgentChange: (agentId: string | null) => void;
+  };
+  /** WeKnora-style knowledge-space picker rendered inside the composer toolbar. */
+  knowledgeSelector?: ReactNode;
+  /** Page-scoped model choices implemented outside the DeerFlow model API. */
+  extraModels?: Model[];
+}) {
   const { t } = useI18n();
   const [localSettings, setLocalSettings] = useLocalSettings();
   const searchParams = useSearchParams();
-  const [modelDialogOpen, setModelDialogOpen] = useState(false);
-  const { models } = useModels();
-  const singleRoundRagActive =
-    context.execution_profile === "single_round_rag";
-  const multiRoundRagActive =
-    context.execution_profile === "multi_round_rag";
-  const frontendRagActive = singleRoundRagActive || multiRoundRagActive;
-  const modelOptions = useMemo(() => {
-    return singleRoundRagActive ? extraModels : models;
-  }, [extraModels, models, singleRoundRagActive]);
+  const [modelDialogOpen, setModelDialogOpen] = useState(false);
+  const { models } = useModels();
+  const singleRoundRagActive =
+    context.execution_profile === "single_round_rag";
+  const multiRoundRagActive =
+    context.execution_profile === "multi_round_rag";
+  const frontendRagActive = singleRoundRagActive || multiRoundRagActive;
+  const modelOptions = useMemo(() => {
+    return singleRoundRagActive ? extraModels : models;
+  }, [extraModels, models, singleRoundRagActive]);
   const { thread, isMock } = useThread();
   const { textInput } = usePromptInputController();
   const promptRootRef = useRef<HTMLDivElement | null>(null);
@@ -238,46 +238,46 @@ export function InputBox({
     !userPreferences.hide_admin_prompt_prefix &&
     recommendedQuestions.length > 0;
 
-  useEffect(() => {
-    // Multi-round RAG is a frontend execution profile, not a model option.
-    // Do not let the normal model fallback clear it on every render.
-    if (multiRoundRagActive) {
-      return;
-    }
-    if (modelOptions.length === 0) {
-      return;
-    }
-    const currentModel = modelOptions.find((m) => m.name === context.model_name);
-    const fallbackModel = currentModel ?? modelOptions[0]!;
+  useEffect(() => {
+    // Multi-round RAG is a frontend execution profile, not a model option.
+    // Do not let the normal model fallback clear it on every render.
+    if (multiRoundRagActive) {
+      return;
+    }
+    if (modelOptions.length === 0) {
+      return;
+    }
+    const currentModel = modelOptions.find((m) => m.name === context.model_name);
+    const fallbackModel = currentModel ?? modelOptions[0]!;
     const supportsThinking = fallbackModel.supports_thinking ?? false;
     const nextModelName = fallbackModel.name;
     const nextMode = getResolvedMode(context.mode, supportsThinking);
 
-    if (
-      context.model_name === nextModelName &&
-      context.mode === nextMode &&
-      context.execution_profile === fallbackModel.execution_profile
-    ) {
-      return;
-    }
+    if (
+      context.model_name === nextModelName &&
+      context.mode === nextMode &&
+      context.execution_profile === fallbackModel.execution_profile
+    ) {
+      return;
+    }
 
     onContextChange?.({
-      ...context,
-      model_name: nextModelName,
-      mode: nextMode,
-      execution_profile: fallbackModel.execution_profile,
-    });
-  }, [context, modelOptions, multiRoundRagActive, onContextChange]);
-
-  const selectedModel = useMemo(() => {
-    if (modelOptions.length === 0) {
-      return undefined;
-    }
-    return modelOptions.find((m) => m.name === context.model_name) ?? modelOptions[0];
-  }, [context.model_name, modelOptions]);
+      ...context,
+      model_name: nextModelName,
+      mode: nextMode,
+      execution_profile: fallbackModel.execution_profile,
+    });
+  }, [context, modelOptions, multiRoundRagActive, onContextChange]);
 
-  const resolvedModelName = selectedModel?.name;
-  const selectedSingleRoundRag = singleRoundRagActive;
+  const selectedModel = useMemo(() => {
+    if (modelOptions.length === 0) {
+      return undefined;
+    }
+    return modelOptions.find((m) => m.name === context.model_name) ?? modelOptions[0];
+  }, [context.model_name, modelOptions]);
+
+  const resolvedModelName = selectedModel?.name;
+  const selectedSingleRoundRag = singleRoundRagActive;
 
   const supportThinking = useMemo(
     () => selectedModel?.supports_thinking ?? false,
@@ -291,72 +291,72 @@ export function InputBox({
 
   const handleModelSelect = useCallback(
     (model_name: string) => {
-      const model = modelOptions.find((m) => m.name === model_name);
+      const model = modelOptions.find((m) => m.name === model_name);
       if (!model) {
         return;
       }
       onContextChange?.({
         ...context,
         model_name,
-        mode: getResolvedMode(context.mode, model.supports_thinking ?? false),
-        reasoning_effort: context.reasoning_effort,
-        execution_profile: model.execution_profile,
-      });
-      setModelDialogOpen(false);
-    },
-    [onContextChange, context, modelOptions],
+        mode: getResolvedMode(context.mode, model.supports_thinking ?? false),
+        reasoning_effort: context.reasoning_effort,
+        execution_profile: model.execution_profile,
+      });
+      setModelDialogOpen(false);
+    },
+    [onContextChange, context, modelOptions],
   );
 
-  const handleModeSelect = useCallback(
-    (mode: InputMode) => {
-      const ordinaryModel =
-        models.find((model) => model.name === context.model_name) ?? models[0];
-      onContextChange?.({
-        ...context,
-        model_name: ordinaryModel?.name ?? context.model_name,
-        mode: getResolvedMode(
-          mode,
-          ordinaryModel?.supports_thinking ?? supportThinking,
-        ),
-        reasoning_effort:
-          mode === "ultra"
-            ? "high"
+  const handleModeSelect = useCallback(
+    (mode: InputMode) => {
+      const ordinaryModel =
+        models.find((model) => model.name === context.model_name) ?? models[0];
+      onContextChange?.({
+        ...context,
+        model_name: ordinaryModel?.name ?? context.model_name,
+        mode: getResolvedMode(
+          mode,
+          ordinaryModel?.supports_thinking ?? supportThinking,
+        ),
+        reasoning_effort:
+          mode === "ultra"
+            ? "high"
             : mode === "pro"
               ? "medium"
               : mode === "thinking"
-                ? "low"
-                : "minimal",
-        execution_profile: undefined,
-      });
-    },
-    [onContextChange, context, models, supportThinking],
-  );
-
-  const handleSingleRoundRagSelect = useCallback(() => {
-    const ragModel =
-      extraModels.find((model) => model.name === context.model_name) ??
-      extraModels[0];
-    if (!ragModel) {
-      return;
-    }
-
-    onContextChange?.({
-      ...context,
-      model_name: ragModel.name,
-      mode: getResolvedMode(
-        context.mode,
-        ragModel.supports_thinking ?? false,
-      ),
-      execution_profile: "single_round_rag",
-    });
-  }, [context, extraModels, onContextChange]);
-
-  const handleMultiRoundRagSelect = useCallback(() => {
-    onContextChange?.({
-      ...context,
-      execution_profile: "multi_round_rag",
-    });
-  }, [context, onContextChange]);
+                ? "low"
+                : "minimal",
+        execution_profile: undefined,
+      });
+    },
+    [onContextChange, context, models, supportThinking],
+  );
+
+  const handleSingleRoundRagSelect = useCallback(() => {
+    const ragModel =
+      extraModels.find((model) => model.name === context.model_name) ??
+      extraModels[0];
+    if (!ragModel) {
+      return;
+    }
+
+    onContextChange?.({
+      ...context,
+      model_name: ragModel.name,
+      mode: getResolvedMode(
+        context.mode,
+        ragModel.supports_thinking ?? false,
+      ),
+      execution_profile: "single_round_rag",
+    });
+  }, [context, extraModels, onContextChange]);
+
+  const handleMultiRoundRagSelect = useCallback(() => {
+    onContextChange?.({
+      ...context,
+      execution_profile: "multi_round_rag",
+    });
+  }, [context, onContextChange]);
 
   const handleReasoningEffortSelect = useCallback(
     (effort: "minimal" | "low" | "medium" | "high") => {
@@ -422,11 +422,11 @@ export function InputBox({
         onContextChange?.({
           ...context,
           model_name: resolvedModelName,
-          mode: getResolvedMode(
-            context.mode,
-            selectedModel?.supports_thinking ?? false,
-          ),
-          execution_profile: selectedModel?.execution_profile,
+          mode: getResolvedMode(
+            context.mode,
+            selectedModel?.supports_thinking ?? false,
+          ),
+          execution_profile: selectedModel?.execution_profile,
         });
         setTimeout(() => onSubmit?.(message), 0);
         return;
@@ -441,8 +441,8 @@ export function InputBox({
       onStop,
       resolvedModelName,
       ragModeEnabled,
-      selectedModel?.supports_thinking,
-      selectedModel?.execution_profile,
+      selectedModel?.supports_thinking,
+      selectedModel?.execution_profile,
       status,
       threadId,
     ],
@@ -500,9 +500,9 @@ export function InputBox({
   }, [pendingSuggestion, requestFormSubmit, textInput]);
 
   const showFollowups =
-    !disabled &&
-    !isNewThread &&
-    !selectedSingleRoundRag &&
+    !disabled &&
+    !isNewThread &&
+    !selectedSingleRoundRag &&
     !followupsHidden &&
     (followupsLoading || followups.length > 0);
 
@@ -528,19 +528,19 @@ export function InputBox({
       return;
     }
 
-    if (disabled || isMock || selectedSingleRoundRag) {
+    if (disabled || isMock || selectedSingleRoundRag) {
       return;
     }
 
-    const visibleMessages = thread.messages.filter((m) => !isHiddenFromUIMessage(m));
-    const lastAi = [...visibleMessages].reverse().find((m) => m.type === "ai");
+    const visibleMessages = thread.messages.filter((m) => !isHiddenFromUIMessage(m));
+    const lastAi = [...visibleMessages].reverse().find((m) => m.type === "ai");
     const lastAiId = lastAi?.id ?? null;
     if (!lastAiId || lastAiId === lastGeneratedForAiIdRef.current) {
       return;
     }
     lastGeneratedForAiIdRef.current = lastAiId;
 
-    const recent = visibleMessages
+    const recent = visibleMessages
       .filter((m) => m.type === "human" || m.type === "ai")
       .map((m) => {
         const role = m.type === "human" ? "user" : "assistant";
@@ -590,7 +590,7 @@ export function InputBox({
       });
 
     return () => controller.abort();
-  }, [context.model_name, disabled, isMock, selectedSingleRoundRag, status, thread.messages, threadId]);
+  }, [context.model_name, disabled, isMock, selectedSingleRoundRag, status, thread.messages, threadId]);
 
   return (
     <div ref={promptRootRef} className="relative flex flex-col gap-6">
@@ -677,7 +677,7 @@ export function InputBox({
             : "*:data-[slot='input-group']:rounded-3xl",
           // 紧凑落地态（CM 参考图）：白卡 + 细边 + 轻阴影 + 主色聚焦环（无旋转彩边）
           compact && isNewThread
-            ? "rounded-2xl border border-gray-200/70 bg-white overflow-hidden shadow-[0_4.5px_22px_rgba(0,0,0,0.025)] transition-all duration-300 focus-within:border-[rgb(var(--legacy-primary-rgb)/35%)] focus-within:ring-4 focus-within:ring-[rgb(var(--legacy-primary-rgb)/5%)] focus-within:shadow-[0_10px_35px_rgb(var(--legacy-primary-rgb)/4%)] dark:border-slate-700 dark:bg-slate-800"
+            ? "rounded-2xl border border-gray-200/70 bg-white overflow-hidden shadow-[0_4.5px_22px_rgba(0,0,0,0.025)] transition-all duration-300 focus-within:border-[rgb(var(--legacy-primary-rgb)/35%)] focus-within:ring-4 focus-within:ring-[rgb(var(--legacy-primary-rgb)/5%)] focus-within:shadow-[0_10px_35px_rgb(var(--legacy-primary-rgb)/4%)] dark:border-slate-700 dark:bg-slate-800"
             : // landing 态：透明，让外层 gemini-gradient-border 提供白卡 + 旋转描边
               // 对话态：常规白卡 + 中性边框 + 轻阴影
               isNewThread
@@ -727,16 +727,16 @@ export function InputBox({
               />
             </PromptInputActionMenuContent>
           </PromptInputActionMenu> */}
-            {!frontendRagActive && (
-              <AddAttachmentsButton
-                className={cn(
-                  "rounded-full border-0 bg-transparent px-2.5! shadow-none text-muted-foreground hover:bg-muted/70 hover:text-foreground dark:text-slate-400 dark:hover:bg-slate-700/70 dark:hover:text-slate-100",
-                  compact &&
-                    "gap-1.5! rounded-lg px-2.5! py-1! text-[11px] font-medium",
-                )}
-              />
-            )}
-            {!frontendRagActive && <Tooltip
+            {!frontendRagActive && (
+              <AddAttachmentsButton
+                className={cn(
+                  "rounded-full border-0 bg-transparent px-2.5! shadow-none text-muted-foreground hover:bg-muted/70 hover:text-foreground dark:text-slate-400 dark:hover:bg-slate-700/70 dark:hover:text-slate-100",
+                  compact &&
+                    "gap-1.5! rounded-lg px-2.5! py-1! text-[11px] font-medium",
+                )}
+              />
+            )}
+            {!frontendRagActive && <Tooltip
               content={
                 context.writing_mode
                   ? "写作模式已开启：最终输出会强制保存为 Markdown 文件"
@@ -746,23 +746,23 @@ export function InputBox({
               <PromptInputButton
                 className={cn(
                   "gap-1.5! rounded-full px-3! shadow-none",
-                  compact && "rounded-lg px-2.5! py-1! text-[11px] font-medium",
-                  context.writing_mode
-                    ? "border border-amber-300 bg-amber-50 text-amber-700 ring-1 ring-amber-300/35 hover:bg-amber-100 dark:border-amber-300/70 dark:bg-amber-400/20 dark:text-amber-100 dark:ring-amber-300/35 dark:shadow-[0_0_0_1px_rgba(251,191,36,0.18),0_0_18px_rgba(251,191,36,0.18)] dark:hover:bg-amber-400/25"
-                    : compact
-                      ? "border border-transparent bg-muted/40 text-muted-foreground hover:bg-muted hover:text-foreground"
-                      : "border-0 bg-transparent text-muted-foreground hover:bg-muted/70 hover:text-foreground dark:text-slate-400 dark:hover:bg-slate-700/70 dark:hover:text-slate-100",
-                )}
-                onClick={handleWritingModeToggle}
-              >
-                {context.writing_mode && (
-                  <CheckIcon className="size-3 text-amber-700 dark:text-amber-100" />
-                )}
-                <FileTextIcon className="size-3" />
-                <span className="text-xs font-normal">写作</span>
-              </PromptInputButton>
-            </Tooltip>}
-            {!frontendRagActive && <Tooltip
+                  compact && "rounded-lg px-2.5! py-1! text-[11px] font-medium",
+                  context.writing_mode
+                    ? "border border-amber-300 bg-amber-50 text-amber-700 ring-1 ring-amber-300/35 hover:bg-amber-100 dark:border-amber-300/70 dark:bg-amber-400/20 dark:text-amber-100 dark:ring-amber-300/35 dark:shadow-[0_0_0_1px_rgba(251,191,36,0.18),0_0_18px_rgba(251,191,36,0.18)] dark:hover:bg-amber-400/25"
+                    : compact
+                      ? "border border-transparent bg-muted/40 text-muted-foreground hover:bg-muted hover:text-foreground"
+                      : "border-0 bg-transparent text-muted-foreground hover:bg-muted/70 hover:text-foreground dark:text-slate-400 dark:hover:bg-slate-700/70 dark:hover:text-slate-100",
+                )}
+                onClick={handleWritingModeToggle}
+              >
+                {context.writing_mode && (
+                  <CheckIcon className="size-3 text-amber-700 dark:text-amber-100" />
+                )}
+                <FileTextIcon className="size-3" />
+                <span className="text-xs font-normal">写作</span>
+              </PromptInputButton>
+            </Tooltip>}
+            {!frontendRagActive && <Tooltip
               content={
                 memoryInjectionEnabled
                   ? "记忆注入已开启：对话前会注入历史记忆"
@@ -771,24 +771,24 @@ export function InputBox({
             >
               <PromptInputButton
                 className={cn(
-                  "gap-1.5! rounded-full px-3! shadow-none",
-                  compact && "rounded-lg px-2.5! py-1! text-[11px] font-medium",
-                  memoryInjectionEnabled
-                    ? "border border-blue-300 bg-blue-50 text-blue-700 ring-1 ring-blue-300/35 hover:bg-blue-100 dark:border-cyan-300/70 dark:bg-cyan-400/20 dark:text-cyan-50 dark:ring-cyan-300/35 dark:shadow-[0_0_0_1px_rgba(34,211,238,0.18),0_0_18px_rgba(34,211,238,0.18)] dark:hover:bg-cyan-400/25"
-                    : compact
-                      ? "border border-transparent bg-muted/40 text-muted-foreground hover:bg-muted hover:text-foreground"
-                      : "border-0 bg-transparent text-muted-foreground hover:bg-muted/70 hover:text-foreground dark:text-slate-400 dark:hover:bg-slate-700/70 dark:hover:text-slate-100",
-                )}
-                onClick={handleMemoryInjectionToggle}
-              >
-                {memoryInjectionEnabled && (
-                  <CheckIcon className="size-3 text-blue-700 dark:text-cyan-50" />
-                )}
-                <BrainCircuitIcon className="size-3" />
-                <span className="text-xs font-normal">记忆</span>
-              </PromptInputButton>
-            </Tooltip>}
-            {!frontendRagActive && <Tooltip
+                  "gap-1.5! rounded-full px-3! shadow-none",
+                  compact && "rounded-lg px-2.5! py-1! text-[11px] font-medium",
+                  memoryInjectionEnabled
+                    ? "border border-blue-300 bg-blue-50 text-blue-700 ring-1 ring-blue-300/35 hover:bg-blue-100 dark:border-cyan-300/70 dark:bg-cyan-400/20 dark:text-cyan-50 dark:ring-cyan-300/35 dark:shadow-[0_0_0_1px_rgba(34,211,238,0.18),0_0_18px_rgba(34,211,238,0.18)] dark:hover:bg-cyan-400/25"
+                    : compact
+                      ? "border border-transparent bg-muted/40 text-muted-foreground hover:bg-muted hover:text-foreground"
+                      : "border-0 bg-transparent text-muted-foreground hover:bg-muted/70 hover:text-foreground dark:text-slate-400 dark:hover:bg-slate-700/70 dark:hover:text-slate-100",
+                )}
+                onClick={handleMemoryInjectionToggle}
+              >
+                {memoryInjectionEnabled && (
+                  <CheckIcon className="size-3 text-blue-700 dark:text-cyan-50" />
+                )}
+                <BrainCircuitIcon className="size-3" />
+                <span className="text-xs font-normal">记忆</span>
+              </PromptInputButton>
+            </Tooltip>}
+            {!frontendRagActive && <Tooltip
               content={
                 ragModeEnabled
                   ? "参考文献模式已开启：检索类结果会按编号引用展示"
@@ -817,8 +817,8 @@ export function InputBox({
                 <BookMarkedIcon className="size-3" />
                 <span className="text-xs font-normal">参考文献</span>
               </PromptInputButton>
-            </Tooltip>}
-            {!frontendRagActive && <Tooltip
+            </Tooltip>}
+            {!frontendRagActive && <Tooltip
               content={
                 contextCompressionEnabled
                   ? "上下文压缩已开启：超长对话会自动压缩历史以节省 token"
@@ -846,70 +846,70 @@ export function InputBox({
                 <ShrinkIcon className="size-3" />
                 <span className="text-xs font-normal">上下文压缩</span>
               </PromptInputButton>
-            </Tooltip>}
+            </Tooltip>}
             <PromptInputActionMenu>
-              <ModeHoverGuide
-                mode={
+              <ModeHoverGuide
+                mode={
                   context.mode === "flash" ||
                   context.mode === "thinking" ||
                   context.mode === "pro" ||
                   context.mode === "ultra"
-                    ? context.mode
-                    : "flash"
-                }
-                tooltipContent={
-                  singleRoundRagActive
-                    ? "单轮 RAG 问答：基于知识库检索生成答案，并展示参考资料"
-                    : multiRoundRagActive
-                      ? "多轮 RAG 问答：多次检索并展示检索过程与参考资料"
-                      : undefined
-                }
-              >
+                    ? context.mode
+                    : "flash"
+                }
+                tooltipContent={
+                  singleRoundRagActive
+                    ? "单轮 RAG 问答：基于知识库检索生成答案，并展示参考资料"
+                    : multiRoundRagActive
+                      ? "多轮 RAG 问答：多次检索并展示检索过程与参考资料"
+                      : undefined
+                }
+              >
                 <PromptInputActionMenuTrigger
                   className={cn(
                     "gap-1.5! rounded-full border-0 bg-transparent px-3! text-muted-foreground shadow-none hover:bg-muted/70 hover:text-foreground dark:text-slate-400 dark:hover:bg-slate-700/70 dark:hover:text-slate-100",
                     compact && "rounded-lg border px-2.5! py-1! text-[11px] font-medium",
-                    compact &&
-                      (frontendRagActive
-                        ? "border-emerald-200 bg-emerald-50/60 text-emerald-700 hover:bg-emerald-100 dark:border-emerald-700 dark:bg-emerald-950/40 dark:text-emerald-200"
-                        : context.mode === "thinking"
-                        ? "border-indigo-200 bg-indigo-50/60 text-indigo-700 hover:bg-indigo-100 dark:border-indigo-700 dark:bg-indigo-950/40 dark:text-indigo-200"
+                    compact &&
+                      (frontendRagActive
+                        ? "border-emerald-200 bg-emerald-50/60 text-emerald-700 hover:bg-emerald-100 dark:border-emerald-700 dark:bg-emerald-950/40 dark:text-emerald-200"
+                        : context.mode === "thinking"
+                        ? "border-indigo-200 bg-indigo-50/60 text-indigo-700 hover:bg-indigo-100 dark:border-indigo-700 dark:bg-indigo-950/40 dark:text-indigo-200"
                         : context.mode === "pro"
                           ? "border-violet-200 bg-violet-50/60 text-violet-700 hover:bg-violet-100 dark:border-violet-700 dark:bg-violet-950/40 dark:text-violet-200"
                           : context.mode === "ultra"
                             ? "border-amber-200 bg-amber-50/60 text-amber-700 hover:bg-amber-100 dark:border-amber-700 dark:bg-amber-950/40 dark:text-amber-200"
                             : "border-transparent bg-muted/40 hover:bg-muted hover:text-foreground"),
                   )}
-                  >
-                    <div>
-                    {frontendRagActive ? (
-                      <BookMarkedIcon className="size-3" />
-                    ) : context.mode === "flash" ? (
-                      <ZapIcon className="size-3" />
-                    ) : context.mode === "thinking" ? (
-                      <LightbulbIcon className="size-3" />
-                    ) : context.mode === "pro" ? (
-                      <GraduationCapIcon className="size-3" />
-                    ) : context.mode === "ultra" ? (
-                      <RocketIcon className="size-3 text-[#dabb5e]" />
-                    ) : null}
-                    </div>
+                  >
+                    <div>
+                    {frontendRagActive ? (
+                      <BookMarkedIcon className="size-3" />
+                    ) : context.mode === "flash" ? (
+                      <ZapIcon className="size-3" />
+                    ) : context.mode === "thinking" ? (
+                      <LightbulbIcon className="size-3" />
+                    ) : context.mode === "pro" ? (
+                      <GraduationCapIcon className="size-3" />
+                    ) : context.mode === "ultra" ? (
+                      <RocketIcon className="size-3 text-[#dabb5e]" />
+                    ) : null}
+                    </div>
                   <div
                     className={cn(
                       "text-xs font-normal",
                       context.mode === "ultra" ? "golden-text" : "",
                     )}
                   >
-                    {frontendRagActive
-                      ? multiRoundRagActive
-                        ? "多轮 RAG"
-                        : "单轮 RAG"
-                      : (context.mode === "flash" && t.inputBox.flashMode) ||
-                      (context.mode === "thinking" &&
-                        t.inputBox.reasoningMode) ||
-                      (context.mode === "pro" && t.inputBox.proMode) ||
-                      (context.mode === "ultra" && t.inputBox.ultraMode)}
-                  </div>
+                    {frontendRagActive
+                      ? multiRoundRagActive
+                        ? "多轮 RAG"
+                        : "单轮 RAG"
+                      : (context.mode === "flash" && t.inputBox.flashMode) ||
+                      (context.mode === "thinking" &&
+                        t.inputBox.reasoningMode) ||
+                      (context.mode === "pro" && t.inputBox.proMode) ||
+                      (context.mode === "ultra" && t.inputBox.ultraMode)}
+                  </div>
                 </PromptInputActionMenuTrigger>
               </ModeHoverGuide>
               <PromptInputActionMenuContent className="w-80">
@@ -918,11 +918,11 @@ export function InputBox({
                     {t.inputBox.mode}
                   </DropdownMenuLabel>
                   <PromptInputActionMenu>
-                    <PromptInputActionMenuItem
-                      className={cn(
-                        !frontendRagActive && context.mode === "flash"
-                          ? "text-accent-foreground"
-                          : "text-muted-foreground/65",
+                    <PromptInputActionMenuItem
+                      className={cn(
+                        !frontendRagActive && context.mode === "flash"
+                          ? "text-accent-foreground"
+                          : "text-muted-foreground/65",
                       )}
                       onSelect={() => handleModeSelect("flash")}
                     >
@@ -930,9 +930,9 @@ export function InputBox({
                         <div className="flex items-center gap-1 font-bold">
                           <ZapIcon
                             className={cn(
-                              "mr-2 size-4",
-                              !frontendRagActive && context.mode === "flash" &&
-                                "text-accent-foreground",
+                              "mr-2 size-4",
+                              !frontendRagActive && context.mode === "flash" &&
+                                "text-accent-foreground",
                             )}
                           />
                           {t.inputBox.flashMode}
@@ -941,7 +941,7 @@ export function InputBox({
                           {t.inputBox.flashModeDescription}
                         </div>
                       </div>
-                      {!frontendRagActive && context.mode === "flash" ? (
+                      {!frontendRagActive && context.mode === "flash" ? (
                         <CheckIcon className="ml-auto size-4" />
                       ) : (
                         <div className="ml-auto size-4" />
@@ -949,9 +949,9 @@ export function InputBox({
                     </PromptInputActionMenuItem>
                     {supportThinking && (
                       <PromptInputActionMenuItem
-                        className={cn(
-                          !frontendRagActive && context.mode === "thinking"
-                            ? "text-accent-foreground"
+                        className={cn(
+                          !frontendRagActive && context.mode === "thinking"
+                            ? "text-accent-foreground"
                             : "text-muted-foreground/65",
                         )}
                         onSelect={() => handleModeSelect("thinking")}
@@ -960,9 +960,9 @@ export function InputBox({
                           <div className="flex items-center gap-1 font-bold">
                             <LightbulbIcon
                               className={cn(
-                                "mr-2 size-4",
-                                !frontendRagActive && context.mode === "thinking" &&
-                                  "text-accent-foreground",
+                                "mr-2 size-4",
+                                !frontendRagActive && context.mode === "thinking" &&
+                                  "text-accent-foreground",
                               )}
                             />
                             {t.inputBox.reasoningMode}
@@ -971,7 +971,7 @@ export function InputBox({
                             {t.inputBox.reasoningModeDescription}
                           </div>
                         </div>
-                        {!frontendRagActive && context.mode === "thinking" ? (
+                        {!frontendRagActive && context.mode === "thinking" ? (
                           <CheckIcon className="ml-auto size-4" />
                         ) : (
                           <div className="ml-auto size-4" />
@@ -979,8 +979,8 @@ export function InputBox({
                       </PromptInputActionMenuItem>
                     )}
                     <PromptInputActionMenuItem
-                      className={cn(
-                        !frontendRagActive && context.mode === "pro"
+                      className={cn(
+                        !frontendRagActive && context.mode === "pro"
                           ? "text-accent-foreground"
                           : "text-muted-foreground/65",
                       )}
@@ -990,8 +990,8 @@ export function InputBox({
                         <div className="flex items-center gap-1 font-bold">
                           <GraduationCapIcon
                             className={cn(
-                              "mr-2 size-4",
-                              !frontendRagActive && context.mode === "pro" &&
+                              "mr-2 size-4",
+                              !frontendRagActive && context.mode === "pro" &&
                                 "text-accent-foreground",
                             )}
                           />
@@ -1001,15 +1001,15 @@ export function InputBox({
                           {t.inputBox.proModeDescription}
                         </div>
                       </div>
-                      {!frontendRagActive && context.mode === "pro" ? (
+                      {!frontendRagActive && context.mode === "pro" ? (
                         <CheckIcon className="ml-auto size-4" />
                       ) : (
                         <div className="ml-auto size-4" />
                       )}
                     </PromptInputActionMenuItem>
                     <PromptInputActionMenuItem
-                      className={cn(
-                        !frontendRagActive && context.mode === "ultra"
+                      className={cn(
+                        !frontendRagActive && context.mode === "ultra"
                           ? "text-accent-foreground"
                           : "text-muted-foreground/65",
                       )}
@@ -1019,17 +1019,17 @@ export function InputBox({
                         <div className="flex items-center gap-1 font-bold">
                           <RocketIcon
                             className={cn(
-                              "mr-2 size-4",
-                              !frontendRagActive &&
-                                context.mode === "ultra" &&
-                                "text-[#dabb5e]",
+                              "mr-2 size-4",
+                              !frontendRagActive &&
+                                context.mode === "ultra" &&
+                                "text-[#dabb5e]",
                             )}
                           />
                           <div
                             className={cn(
-                              !frontendRagActive &&
-                                context.mode === "ultra" &&
-                                "golden-text",
+                              !frontendRagActive &&
+                                context.mode === "ultra" &&
+                                "golden-text",
                             )}
                           >
                             {t.inputBox.ultraMode}
@@ -1039,78 +1039,78 @@ export function InputBox({
                           {t.inputBox.ultraModeDescription}
                         </div>
                       </div>
-                      {!frontendRagActive && context.mode === "ultra" ? (
+                      {!frontendRagActive && context.mode === "ultra" ? (
                         <CheckIcon className="ml-auto size-4" />
                       ) : (
                         <div className="ml-auto size-4" />
                       )}
-                    </PromptInputActionMenuItem>
-                    {extraModels.length > 0 && (
-                      <>
-                        <DropdownMenuSeparator className="my-2" />
-                        <PromptInputActionMenuItem
-                          className={cn(
-                            selectedSingleRoundRag
-                              ? "bg-emerald-50 text-emerald-800 hover:bg-emerald-100 focus:bg-emerald-100 dark:bg-emerald-950/40 dark:text-emerald-200 dark:hover:bg-emerald-950/60 dark:focus:bg-emerald-950/60"
-                              : "text-muted-foreground/65",
-                          )}
-                          onSelect={handleSingleRoundRagSelect}
-                        >
-                          <div className="flex flex-col gap-2">
-                            <div className="flex items-center gap-1 font-bold">
-                              <BookMarkedIcon
-                                className={cn(
-                                  "mr-2 size-4",
-                                  selectedSingleRoundRag && "text-emerald-700 dark:text-emerald-200",
-                                )}
-                              />
-                              单轮 RAG 问答
-                            </div>
-                            <div className="pl-7 text-xs">
-                              基于知识库检索生成答案，并展示思考内容与参考资料
-                            </div>
-                          </div>
-                          {selectedSingleRoundRag ? (
-                            <CheckIcon className="ml-auto size-4" />
-                          ) : (
-                            <div className="ml-auto size-4" />
-                          )}
-                        </PromptInputActionMenuItem>
-                      </>
-                    )}
-                    <PromptInputActionMenuItem
-                      className={cn(
-                        multiRoundRagActive
-                          ? "bg-emerald-50 text-emerald-800 hover:bg-emerald-100 focus:bg-emerald-100 dark:bg-emerald-950/40 dark:text-emerald-200 dark:hover:bg-emerald-950/60 dark:focus:bg-emerald-950/60"
-                          : "text-muted-foreground/65",
-                      )}
-                      onSelect={handleMultiRoundRagSelect}
-                    >
-                      <div className="flex flex-col gap-2">
-                        <div className="flex items-center gap-1 font-bold">
-                          <BookMarkedIcon
-                            className={cn(
-                              "mr-2 size-4",
-                              multiRoundRagActive && "text-emerald-700 dark:text-emerald-200",
-                            )}
-                          />
-                          多轮 RAG 问答
-                        </div>
-                        <div className="pl-7 text-xs">
-                          多次检索、展示过程与参考资料，适合复杂问题
-                        </div>
-                      </div>
-                      {multiRoundRagActive ? (
-                        <CheckIcon className="ml-auto size-4" />
-                      ) : (
-                        <div className="ml-auto size-4" />
-                      )}
-                    </PromptInputActionMenuItem>
-                  </PromptInputActionMenu>
+                    </PromptInputActionMenuItem>
+                    {extraModels.length > 0 && (
+                      <>
+                        <DropdownMenuSeparator className="my-2" />
+                        <PromptInputActionMenuItem
+                          className={cn(
+                            selectedSingleRoundRag
+                              ? "bg-emerald-50 text-emerald-800 hover:bg-emerald-100 focus:bg-emerald-100 dark:bg-emerald-950/40 dark:text-emerald-200 dark:hover:bg-emerald-950/60 dark:focus:bg-emerald-950/60"
+                              : "text-muted-foreground/65",
+                          )}
+                          onSelect={handleSingleRoundRagSelect}
+                        >
+                          <div className="flex flex-col gap-2">
+                            <div className="flex items-center gap-1 font-bold">
+                              <BookMarkedIcon
+                                className={cn(
+                                  "mr-2 size-4",
+                                  selectedSingleRoundRag && "text-emerald-700 dark:text-emerald-200",
+                                )}
+                              />
+                              单轮 RAG 问答
+                            </div>
+                            <div className="pl-7 text-xs">
+                              基于知识库检索生成答案，并展示思考内容与参考资料
+                            </div>
+                          </div>
+                          {selectedSingleRoundRag ? (
+                            <CheckIcon className="ml-auto size-4" />
+                          ) : (
+                            <div className="ml-auto size-4" />
+                          )}
+                        </PromptInputActionMenuItem>
+                      </>
+                    )}
+                    <PromptInputActionMenuItem
+                      className={cn(
+                        multiRoundRagActive
+                          ? "bg-emerald-50 text-emerald-800 hover:bg-emerald-100 focus:bg-emerald-100 dark:bg-emerald-950/40 dark:text-emerald-200 dark:hover:bg-emerald-950/60 dark:focus:bg-emerald-950/60"
+                          : "text-muted-foreground/65",
+                      )}
+                      onSelect={handleMultiRoundRagSelect}
+                    >
+                      <div className="flex flex-col gap-2">
+                        <div className="flex items-center gap-1 font-bold">
+                          <BookMarkedIcon
+                            className={cn(
+                              "mr-2 size-4",
+                              multiRoundRagActive && "text-emerald-700 dark:text-emerald-200",
+                            )}
+                          />
+                          多轮 RAG 问答
+                        </div>
+                        <div className="pl-7 text-xs">
+                          多次检索、展示过程与参考资料，适合复杂问题
+                        </div>
+                      </div>
+                      {multiRoundRagActive ? (
+                        <CheckIcon className="ml-auto size-4" />
+                      ) : (
+                        <div className="ml-auto size-4" />
+                      )}
+                    </PromptInputActionMenuItem>
+                  </PromptInputActionMenu>
                 </DropdownMenuGroup>
               </PromptInputActionMenuContent>
             </PromptInputActionMenu>
-            {!frontendRagActive && supportReasoningEffort && context.mode !== "flash" && (
+            {!frontendRagActive && supportReasoningEffort && context.mode !== "flash" && (
               <PromptInputActionMenu>
                 <PromptInputActionMenuTrigger className={cn("gap-1.5! rounded-full border-0 bg-transparent px-3! text-muted-foreground shadow-none hover:bg-muted/70 hover:text-foreground dark:text-slate-400 dark:hover:bg-slate-700/70 dark:hover:text-slate-100", compact && "rounded-lg border border-border px-2.5! py-1! text-[13px]")}>
                   <div className="text-xs font-normal">
@@ -1226,10 +1226,9 @@ export function InputBox({
                 </PromptInputActionMenuContent>
               </PromptInputActionMenu>
             )}
-          </PromptInputTools>
-          <PromptInputTools className={cn("gap-1.5", compact && "gap-2")}>
-            {!frontendRagActive && agentSelector && (
-            {!frontendRagActive && knowledgeSelector}
+          </PromptInputTools>
+          <PromptInputTools className={cn("gap-1.5", compact && "gap-2")}>
+            {!frontendRagActive && agentSelector && (
               <AgentSelector
                 selectedAgentId={agentSelector.selectedAgentId}
                 onAgentChange={agentSelector.onAgentChange}
@@ -1237,7 +1236,8 @@ export function InputBox({
                 compact={compact}
               />
             )}
-            {!frontendRagActive && promptPrefix && (
+            {!frontendRagActive && knowledgeSelector}
+            {!frontendRagActive && promptPrefix && (
               <Tooltip
                 content={
                   promptPrefix.on
@@ -1265,7 +1265,7 @@ export function InputBox({
                 </PromptInputButton>
               </Tooltip>
             )}
-            {!multiRoundRagActive && <ModelSelector
+            {!multiRoundRagActive && <ModelSelector
               open={modelDialogOpen}
               onOpenChange={setModelDialogOpen}
             >
@@ -1292,7 +1292,7 @@ export function InputBox({
               <ModelSelectorContent>
                 <ModelSelectorInput placeholder={t.inputBox.searchModels} />
                 <ModelSelectorList>
-                  {modelOptions.map((m) => (
+                  {modelOptions.map((m) => (
                     <ModelSelectorItem
                       key={m.name}
                       value={m.name}
@@ -1300,11 +1300,11 @@ export function InputBox({
                     >
                       <div className="flex min-w-0 flex-1 flex-col">
                         <ModelSelectorName>{m.display_name}</ModelSelectorName>
-                        <span className="text-muted-foreground truncate text-[10px]">
-                          {selectedSingleRoundRag
-                            ? `单轮知识库问答 · ${m.name}`
-                            : m.model}
-                        </span>
+                        <span className="text-muted-foreground truncate text-[10px]">
+                          {selectedSingleRoundRag
+                            ? `单轮知识库问答 · ${m.name}`
+                            : m.model}
+                        </span>
                       </div>
                       {m.name === context.model_name ? (
                         <CheckIcon className="ml-auto size-4" />
@@ -1312,15 +1312,15 @@ export function InputBox({
                         <div className="ml-auto size-4" />
                       )}
                     </ModelSelectorItem>
-                  ))}
-                </ModelSelectorList>
+                  ))}
+                </ModelSelectorList>
               </ModelSelectorContent>
-            </ModelSelector>}
+            </ModelSelector>}
             <PromptInputSubmit
               className={cn(
                 "rounded-full border-border bg-background shadow-sm hover:bg-muted dark:bg-blue-500 dark:border-blue-500 dark:text-white dark:hover:bg-blue-400 dark:shadow-none",
                 compact &&
-                  "size-7 rounded-lg border-0 bg-[var(--primary)] text-[var(--button-foreground,var(--primary-foreground))] hover:opacity-90 disabled:bg-muted disabled:text-muted-foreground/40 dark:bg-[var(--primary)] dark:border-0 dark:text-[var(--button-foreground,var(--primary-foreground))] dark:hover:opacity-90 dark:disabled:bg-muted",
+                  "size-7 rounded-lg border-0 bg-[var(--primary)] text-[var(--button-foreground,var(--primary-foreground))] hover:opacity-90 disabled:bg-muted disabled:text-muted-foreground/40 dark:bg-[var(--primary)] dark:border-0 dark:text-[var(--button-foreground,var(--primary-foreground))] dark:hover:opacity-90 dark:disabled:bg-muted",
               )}
               disabled={disabled}
               variant="outline"

frontend-web/src/components/workspace/messages/message-list.tsx

@@ -30,12 +30,13 @@ import { useSkills } from "@/core/skills/hooks";
 import {
   buildSkillDisplayMap,
   collectRuntimeDisplaySources,
+  type RuntimeDisplaySource,
 } from "@/core/skill-display/runtime";
 import type { Subtask } from "@/core/tasks";
 import { useUpdateSubtask } from "@/core/tasks/context";
-import type { AgentThreadState } from "@/core/threads";
-import { isSingleRoundRagMessage } from "@/core/threads/use-single-round-rag";
-import { isMultiRoundRagMessage } from "@/core/threads/use-multi-round-rag";
+import type { AgentThreadState } from "@/core/threads";
+import { isSingleRoundRagMessage } from "@/core/threads/use-single-round-rag";
+import { isMultiRoundRagMessage } from "@/core/threads/use-multi-round-rag";
 import { cn } from "@/lib/utils";
 import { resolveAllowMultiple } from "@/roundtable-planning/lib/clarification";
 
@@ -99,18 +100,18 @@ export function MessageList({
   tokenUsageEnabled = false,
   extraRehypePlugins,
   scrollButtonClassName,
-  onClarificationSubmit,
-  onEditResubmit,
-  onEditSave,
-  messageContentOverrides,
-  externallyHiddenMessageIds,
-  onSetupAgentApprove,
-  onSetupAgentReject,
-  onSetupWritingApprove,
-  onSetupWritingReject,
-  onSetupWritingStart,
-  onOpenArtifact,
-}: {
+  onClarificationSubmit,
+  onEditResubmit,
+  onEditSave,
+  messageContentOverrides,
+  externallyHiddenMessageIds,
+  onSetupAgentApprove,
+  onSetupAgentReject,
+  onSetupWritingApprove,
+  onSetupWritingReject,
+  onSetupWritingStart,
+  onOpenArtifact,
+}: {
   className?: string;
   contentClassName?: string;
   fitMessageRowsToContent?: boolean;
@@ -121,19 +122,19 @@ export function MessageList({
   /** 额外定位「回到最新」悬浮按钮——不同页面输入框高度不同，需上移让位。 */
   scrollButtonClassName?: string;
   extraRehypePlugins?: ReturnType<typeof useRehypeSplitWordsIntoSpans>;
-  onClarificationSubmit?: (answer: string) => void | Promise<void>;
-  onEditResubmit?: (text: string) => void;
-  onEditSave?: (messageId: string, text: string) => void | Promise<void>;
-  messageContentOverrides?: Record<string, string>;
-  externallyHiddenMessageIds?: Set<string>;
-  onSetupAgentApprove?: (payload: SetupAgentApprovalPayload) => void | Promise<void>;
+  onClarificationSubmit?: (answer: string) => void | Promise<void>;
+  onEditResubmit?: (text: string) => void;
+  onEditSave?: (messageId: string, text: string) => void | Promise<void>;
+  messageContentOverrides?: Record<string, string>;
+  externallyHiddenMessageIds?: Set<string>;
+  onSetupAgentApprove?: (payload: SetupAgentApprovalPayload) => void | Promise<void>;
   onSetupAgentReject?: (payload: SetupAgentApprovalPayload) => void | Promise<void>;
-  onSetupWritingApprove?: (payload: SetupWritingApprovalPayload) => void | Promise<void>;
-  onSetupWritingReject?: (payload: SetupWritingApprovalPayload) => void | Promise<void>;
+  onSetupWritingApprove?: (payload: SetupWritingApprovalPayload) => void | Promise<void>;
+  onSetupWritingReject?: (payload: SetupWritingApprovalPayload) => void | Promise<void>;
   /** 审批卡「直接开始写作」回调：应用配置（含已确认大纲）并立即启动写作流程（仅写作页提供，模板编辑等场景缺省隐藏该按钮）。 */
-  onSetupWritingStart?: (payload: SetupWritingApprovalPayload) => void | Promise<void>;
-  onOpenArtifact?: (filepath: string) => void;
-}) {
+  onSetupWritingStart?: (payload: SetupWritingApprovalPayload) => void | Promise<void>;
+  onOpenArtifact?: (filepath: string) => void;
+}) {
   const baseRehypePlugins = useRehypeSplitWordsIntoSpans(thread.isLoading);
   const rehypePlugins = useMemo(
     () => (extraRehypePlugins?.length ? [...baseRehypePlugins, ...extraRehypePlugins] : baseRehypePlugins),
@@ -165,33 +166,33 @@ export function MessageList({
     });
   }, []);
 
-  const allMessages = useMemo(() => {
-    const hasOverrides = messageContentOverrides && Object.keys(messageContentOverrides).length > 0;
-    return thread.messages
-      .filter((m) => !isHiddenFromUIMessage(m))
-      .map((message) => {
-        const override =
-          hasOverrides && message.id
-            ? messageContentOverrides?.[message.id]
-            : undefined;
-        return override === undefined
-          ? message
-          : ({ ...message, content: override } as Message);
-      });
-  }, [messageContentOverrides, thread.messages]);
-  const messages = useMemo(
-    () => {
-      const externalHidden = externallyHiddenMessageIds ?? new Set<string>();
-      return hiddenMessageIds.size === 0 && externalHidden.size === 0
-        ? allMessages
-        : allMessages.filter(
-            (m) =>
-              !m.id ||
-              (!hiddenMessageIds.has(m.id) && !externalHidden.has(m.id)),
-          );
-    },
-    [allMessages, externallyHiddenMessageIds, hiddenMessageIds],
-  );
+  const allMessages = useMemo(() => {
+    const hasOverrides = messageContentOverrides && Object.keys(messageContentOverrides).length > 0;
+    return thread.messages
+      .filter((m) => !isHiddenFromUIMessage(m))
+      .map((message) => {
+        const override =
+          hasOverrides && message.id
+            ? messageContentOverrides?.[message.id]
+            : undefined;
+        return override === undefined
+          ? message
+          : ({ ...message, content: override } as Message);
+      });
+  }, [messageContentOverrides, thread.messages]);
+  const messages = useMemo(
+    () => {
+      const externalHidden = externallyHiddenMessageIds ?? new Set<string>();
+      return hiddenMessageIds.size === 0 && externalHidden.size === 0
+        ? allMessages
+        : allMessages.filter(
+            (m) =>
+              !m.id ||
+              (!hiddenMessageIds.has(m.id) && !externalHidden.has(m.id)),
+          );
+    },
+    [allMessages, externallyHiddenMessageIds, hiddenMessageIds],
+  );
 
   const askTimes = useChatAskTimes();
 
@@ -237,21 +238,21 @@ export function MessageList({
   }, [messages, askTimes]);
 
   const [settings] = useLocalSettings();
-  const ragModeEnabled =
-    settings.display.ragModeEnabled || isThreadRagModeLocked(threadId);
-  const hasSingleRoundRag = useMemo(
-    () =>
-      messages.some(
-        (message) =>
-          isSingleRoundRagMessage(message) || isMultiRoundRagMessage(message),
-      ),
-    [messages],
-  );
-  const referencesEnabled = ragModeEnabled || hasSingleRoundRag;
-  const { skills } = useSkills();
-  const skillDisplayMap = useMemo(
-    () => (referencesEnabled ? buildSkillDisplayMap(skills) : new Map()),
-    [referencesEnabled, skills],
+  const ragModeEnabled =
+    settings.display.ragModeEnabled || isThreadRagModeLocked(threadId);
+  const hasSingleRoundRag = useMemo(
+    () =>
+      messages.some(
+        (message) =>
+          isSingleRoundRagMessage(message) || isMultiRoundRagMessage(message),
+      ),
+    [messages],
+  );
+  const referencesEnabled = ragModeEnabled || hasSingleRoundRag;
+  const { skills } = useSkills();
+  const skillDisplayMap = useMemo(
+    () => (referencesEnabled ? buildSkillDisplayMap(skills) : new Map()),
+    [referencesEnabled, skills],
   );
   // Same per-turn reference batches the right-side panel renders (backend
   // batches preferred). Inline [n] tokens must resolve against these so badge
@@ -259,6 +260,53 @@ export function MessageList({
   // frontend tool-message scan below comes up empty (e.g. after a reload).
   const referenceBatches = useContext(ReferenceBatchesContext);
 
+  // 单轮/多轮 RAG 的参考资料存放在一条 hide_from_ui 的 tool 消息里（紧挨在
+  // assistant 消息之前）。allMessages 会把 hide_from_ui 消息过滤掉，导致通用
+  // 的 collectRuntimeDisplaySources 扫描不到它们。这里直接在原始消息流里按
+  // assistant 消息提取其对应的参考资料，保证 `[1]` 等编号能解析成可点击引用。
+  const frontendRagSourcesByAiId = useMemo(() => {
+    const map = new Map<string, RuntimeDisplaySource[]>();
+    const raw = thread.messages;
+    for (let i = 0; i < raw.length; i++) {
+      const msg = raw[i];
+      if (!msg || msg.type !== "ai" || !msg.id) continue;
+      const kwargs = msg.additional_kwargs;
+      if (!kwargs || (kwargs.single_round_rag !== true && kwargs.multi_round_rag !== true)) {
+        continue;
+      }
+      let sources: RuntimeDisplaySource[] = [];
+      for (let j = i - 1; j >= 0; j--) {
+        const prev = raw[j];
+        if (!prev) break;
+        // 前一条 human/ai 消息即为本轮边界，跨边界即停止。
+        if (prev.type === "human" || prev.type === "ai") break;
+        if (prev.type !== "tool") continue;
+        const prevKwargs = prev.additional_kwargs;
+        if (
+          !prevKwargs ||
+          (prevKwargs.single_round_rag !== true && prevKwargs.multi_round_rag !== true)
+        ) {
+          continue;
+        }
+        const results = extractFrontendRagResults(prev.content);
+        if (results && results.length > 0) {
+          const toolName = String(
+            (prev as { name?: unknown }).name ??
+              (prevKwargs.multi_round_rag === true
+                ? "multi_round_rag_references"
+                : "knowledge_base_search"),
+          );
+          sources = results.map((item, index) =>
+            toRuntimeDisplaySource(item, index + 1, toolName),
+          );
+        }
+        break;
+      }
+      map.set(msg.id, sources);
+    }
+    return map;
+  }, [thread.messages]);
+
   // Track which clarification messages are still awaiting an answer.
   // Pending = clarification tool messages that appear after the last human message
   // (no user reply has been sent for them yet).
@@ -384,7 +432,7 @@ export function MessageList({
           if (group.type === "human" || group.type === "assistant") {
             return group.messages.map((msg) => {
               const isAssistantMessage =
-                referencesEnabled &&
+                referencesEnabled &&
                 group.type === "assistant" &&
                 msg.type === "ai";
               const batchSources = isAssistantMessage
@@ -392,14 +440,21 @@ export function MessageList({
                     (batch) => batch.assistant_message_id === msg.id,
                   )?.sources ?? []
                 : [];
+              // 单轮/多轮 RAG 的参考资料来自隐藏的 tool 消息，优先于通用扫描。
+              const frontendRagSources =
+                isAssistantMessage && msg.id
+                  ? frontendRagSourcesByAiId.get(msg.id) ?? []
+                  : [];
               const displaySources = isAssistantMessage
                 ? batchSources.length > 0
                   ? batchSources
-                  : collectRuntimeDisplaySources(messages, skillDisplayMap, {
-                      afterMessageId: findPreviousHumanMessageId(messages, msg.id),
-                      beforeMessageId: msg.id,
-                      includeBeforeMessage: true,
-                    })
+                  : frontendRagSources.length > 0
+                    ? frontendRagSources
+                    : collectRuntimeDisplaySources(messages, skillDisplayMap, {
+                        afterMessageId: findPreviousHumanMessageId(messages, msg.id),
+                        beforeMessageId: msg.id,
+                        includeBeforeMessage: true,
+                      })
                 : [];
               return (
                 <MessageListItem
@@ -415,11 +470,11 @@ export function MessageList({
                   threadId={threadId}
                   tokenUsageEnabled={tokenUsageEnabled}
                   displaySources={displaySources}
-                  extraRehypePlugins={extraRehypePlugins}
-                  onEditResubmit={onEditResubmit}
-                  onEditSave={onEditSave}
-                  onMessageDeleted={
-                    allowDelete ? handleMessageDeleted : undefined
+                  extraRehypePlugins={extraRehypePlugins}
+                  onEditResubmit={onEditResubmit}
+                  onEditSave={onEditSave}
+                  onMessageDeleted={
+                    allowDelete ? handleMessageDeleted : undefined
                   }
                   turnTime={msg.id ? turnTimes.get(msg.id) : undefined}
                 />
@@ -524,11 +579,11 @@ export function MessageList({
                     className="mb-4"
                   />
                 )}
-                <ArtifactFileList
-                  files={files}
-                  threadId={threadId}
-                  onOpenArtifact={onOpenArtifact}
-                />
+                <ArtifactFileList
+                  files={files}
+                  threadId={threadId}
+                  onOpenArtifact={onOpenArtifact}
+                />
                 <MessageTokenUsageList
                   enabled={tokenUsageEnabled}
                   isLoading={thread.isLoading}
@@ -728,6 +783,46 @@ function findPreviousHumanMessageId(
   return undefined;
 }
 
+/** 从单轮/多轮 RAG 的 tool 消息体里取出 `content.results`。 */
+function extractFrontendRagResults(content: unknown): unknown[] | null {
+  if (content && typeof content === "object" && !Array.isArray(content)) {
+    const results = (content as { results?: unknown }).results;
+    if (Array.isArray(results)) return results;
+  }
+  return null;
+}
+
+/** 把单轮/多轮 RAG 的一条参考资料归一化为展示用的 RuntimeDisplaySource。 */
+function toRuntimeDisplaySource(
+  item: unknown,
+  index: number,
+  skillName: string,
+): RuntimeDisplaySource {
+  const record =
+    item && typeof item === "object" && !Array.isArray(item)
+      ? (item as Record<string, unknown>)
+      : {};
+  const raw =
+    (record.raw as Record<string, unknown> | undefined) ?? record;
+  const content = String(record.content ?? record.snippet ?? "");
+  return {
+    id: typeof record.id === "string" && record.id ? record.id : undefined,
+    title: String(record.title ?? `参考资料 ${index}`),
+    type: String(record.type ?? "result"),
+    snippet: String(record.snippet ?? content),
+    content,
+    source:
+      typeof record.source === "string" && record.source ? record.source : undefined,
+    author: undefined,
+    time: typeof record.time === "string" && record.time ? record.time : undefined,
+    url: typeof record.url === "string" && record.url ? record.url : undefined,
+    raw,
+    index,
+    skillName,
+    mode: "citation",
+  };
+}
+
 function getSetupAgentApprovalPayload(
   message: unknown,
 ): SetupAgentApprovalPayload | null {

frontend-web/src/pages/ChatPage.tsx

@@ -815,11 +815,7 @@ export default function ChatPage() {
                             : "ready"
                       }
                       context={settings.context}
-                      extraModels={
-                        singleRoundRagSelected
-                          ? SINGLE_ROUND_RAG_MODELS
-                          : undefined
-                      }
+                      extraModels={SINGLE_ROUND_RAG_MODELS}
                       disabled={
                         env.NEXT_PUBLIC_STATIC_WEBSITE_ONLY === "true" ||
                         isUploading ||

offline-backend-20260512/backend/config.yaml

@@ -364,8 +364,7 @@ auth_login:
 # 点击后在新标签页打开下面的 url。enabled 关掉则不显示该入口。
 
 database:
-  backend: postgres
-  postgres_url: $POSTGRES_DATABASE_URL
+  backend: sqlite
   sqlite_dir: .deer-flow/data
   pool_size: 20 # 基础连接池(postgres/mysql 生效;sqlite 忽略)
   max_overflow: 20 # 高峰期可临时超出 pool_size 的额外连接数,真实上限 = pool_size + max_overflow
@@ -374,13 +373,8 @@ database:
   echo_sql: false
 
 checkpointer:
-  type: postgres
-  connection_string: $POSTGRES_CHECKPOINT_URL
-  use_postgres: true
-  # PostgreSQL 优先；SQLite 仅作历史记录读取兜底，确认稳定后可移除。
-  dual_read_enabled: true
-  dual_read_sqlite_paths:
-    - .deer-flow/data/deerflow.db
+  type: sqlite
+  connection_string: .deer-flow/data/deerflow.db
 
 # 记忆子系统:跨会话持久化用户画像与 agent 私有记忆。
 # 三层覆盖优先级(高->低):