feat(frontend): add workspace single and multi round rag

frontend-web/src/core/models/single-round-rag.ts

@@ -19,3 +19,18 @@ export function isSingleRoundRagProfile(
       "single_round_rag"
   );
 }
+
+export function isMultiRoundRagProfile(
+  context: unknown,
+): boolean {
+  return (
+    typeof context === "object" &&
+    context !== null &&
+    (context as { execution_profile?: unknown }).execution_profile ===
+      "multi_round_rag"
+  );
+}
+
+export function isFrontendRagProfile(context: unknown): boolean {
+  return isSingleRoundRagProfile(context) || isMultiRoundRagProfile(context);
+}

frontend-web/src/core/models/types.ts

@@ -7,7 +7,7 @@ export interface Model {
   supports_thinking?: boolean;
   supports_reasoning_effort?: boolean;
   /** Execution profiles handled by the frontend instead of DeerFlow. */
-  execution_profile?: "single_round_rag";
+  execution_profile?: "single_round_rag" | "multi_round_rag";
 }
 
 export interface TokenUsageSettings {

frontend-web/src/core/skill-display/inline-citations.test.ts

@@ -71,6 +71,12 @@ test("keeps half-width syntax working", () => {
   assert.equal(out.includes(citation(3)), true);
 });
 
+test("converts double-bracket RAG references", () => {
+  const out = applyInlineCitations("RAG answer [[1]] and [[2]].", makeSources(2));
+  assert.equal(out.includes(citation(1)), true);
+  assert.equal(out.includes(citation(2)), true);
+});
+
 test("accepts spaced and full-width square brackets", () => {
   const out = applyInlineCitations("参考[ 1 ]和［2］。", makeSources(2));
   assert.equal(out.includes(citation(1)), true);

frontend-web/src/core/threads/types.ts

@@ -27,8 +27,8 @@ export interface AgentThreadContext extends Record<string, unknown> {
   knowledge_rag_enabled?: boolean;
   /** DeerFlow LLMWiki mapping ids selected for this conversation. */
   llmwiki_knowledge_base_ids?: string[];
-  /** Direct knowledge-base RAG mode; this request never enters Lead Agent. */
-  execution_profile?: "single_round_rag";
+  /** Direct RAG modes handled in the frontend; these requests never enter Lead Agent. */
+  execution_profile?: "single_round_rag" | "multi_round_rag";
   agent_id?: string;
   /** @deprecated kept for reading legacy thread metadata; prefer agent_id */
   agent_name?: string;

frontend-web/src/core/threads/use-multi-round-rag.ts

@@ -0,0 +1,483 @@
+import type { Message } from "@langchain/langgraph-sdk";
+import { useCallback, useEffect, useRef, useState } from "react";
+
+import { readRuntimeEnv } from "@/env";
+
+const STORAGE_PREFIX = "workspace.multi-round-rag.";
+
+export type MultiRoundRagRetrieval = {
+  id?: string;
+  title: string;
+  content: string;
+  source?: string;
+  time?: string;
+  recUuid?: string;
+  url?: string;
+};
+
+export type MultiRoundRagRound = {
+  id: string;
+  title: string;
+  thought: string;
+  /** Model text emitted for this round, before its retrieval results. */
+  body: string;
+  query: string;
+  isSearching: boolean;
+  retrievals: MultiRoundRagRetrieval[];
+};
+
+export type MultiRoundRagTrace = {
+  rounds: MultiRoundRagRound[];
+  answer: string;
+  isComplete: boolean;
+};
+
+type MultiRoundRagSource = {
+  id?: string;
+  title: string;
+  type: string;
+  snippet: string;
+  content: string;
+  source?: string;
+  time?: string;
+  url?: string;
+  raw: Record<string, unknown>;
+};
+
+type MultiRoundEvent = {
+  event?: unknown;
+  data?: unknown;
+};
+
+function storageKey(threadId: string) {
+  return `${STORAGE_PREFIX}${threadId}`;
+}
+
+function readStoredMessages(threadId: string): Message[] {
+  if (typeof window === "undefined" || !threadId) return [];
+  try {
+    const raw = window.sessionStorage.getItem(storageKey(threadId));
+    const parsed: unknown = raw ? JSON.parse(raw) : [];
+    return Array.isArray(parsed) ? (parsed as Message[]) : [];
+  } catch {
+    return [];
+  }
+}
+
+function writeStoredMessages(threadId: string, messages: Message[]) {
+  if (typeof window === "undefined" || !threadId) return;
+  try {
+    window.sessionStorage.setItem(storageKey(threadId), JSON.stringify(messages));
+  } catch {
+    // Storage quota failures must not interrupt a running answer.
+  }
+}
+
+function createId(prefix: string) {
+  return `${prefix}-${crypto.randomUUID()}`;
+}
+
+function isAbortError(error: unknown) {
+  return error instanceof DOMException && error.name === "AbortError";
+}
+
+function asRecord(value: unknown): Record<string, unknown> | null {
+  return value && typeof value === "object" && !Array.isArray(value)
+    ? (value as Record<string, unknown>)
+    : null;
+}
+
+function asString(value: unknown): string {
+  return typeof value === "string" ? value : "";
+}
+
+function parseJson(value: unknown): unknown {
+  if (typeof value !== "string") return value;
+  try {
+    return JSON.parse(value) as unknown;
+  } catch {
+    return value;
+  }
+}
+
+function parseEventLine(line: string): MultiRoundEvent | null {
+  const trimmed = line.trim();
+  if (!trimmed.startsWith("data:")) return null;
+  try {
+    return JSON.parse(trimmed.slice("data:".length).trim()) as MultiRoundEvent;
+  } catch {
+    return null;
+  }
+}
+
+function initialTrace(): MultiRoundRagTrace {
+  return {
+    rounds: [
+      {
+        id: createId("multi-rag-round"),
+        title: "正在准备检索",
+        thought: "",
+        body: "",
+        query: "",
+        isSearching: true,
+        retrievals: [],
+      },
+    ],
+    answer: "",
+    isComplete: false,
+  };
+}
+
+function latestRound(trace: MultiRoundRagTrace): MultiRoundRagRound {
+  const current = trace.rounds.at(-1);
+  if (current) return current;
+  const next = initialTrace().rounds[0]!;
+  trace.rounds.push(next);
+  return next;
+}
+
+function addThinkingRound(trace: MultiRoundRagTrace, thought: string) {
+  const current = latestRound(trace);
+  if (current.title === "正在准备检索" && !current.thought && !current.query) {
+    current.title = "思考过程";
+    current.thought = thought;
+    return;
+  }
+  current.isSearching = false;
+  trace.rounds.push({
+    id: createId("multi-rag-round"),
+    title: "思考过程",
+    thought,
+    body: "",
+    query: "",
+    isSearching: true,
+    retrievals: [],
+  });
+}
+
+function appendRoundBody(trace: MultiRoundRagTrace, body: string) {
+  if (!body) return;
+  const current = latestRound(trace);
+  // Stored traces created before `body` was introduced remain readable.
+  current.body = `${current.body ?? ""}${body}`;
+}
+
+function mapRetrievals(value: unknown): MultiRoundRagRetrieval[] {
+  const root = asRecord(parseJson(value));
+  const list =
+    (Array.isArray(root?.results) && root.results) ||
+    (Array.isArray(root?.organic) && root.organic) ||
+    [];
+  return list.flatMap((item, index) => {
+    const record = asRecord(item);
+    if (!record) return [];
+    const recUuid = asString(record.recUuid ?? record.rec_uuid);
+    const url = asString(record.url ?? record.link ?? record.href);
+    return [
+      {
+        id: asString(record.id) || recUuid || undefined,
+        title: asString(record.title) || `检索资料 ${index + 1}`,
+        content: asString(record.content ?? record.snippet ?? record.text),
+        source: asString(record.source ?? record.source1 ?? url) || undefined,
+        time: asString(record.publish_time ?? record.m_publish ?? record.date) || undefined,
+        recUuid: recUuid || undefined,
+        url: /^https?:\/\//i.test(url) ? url : undefined,
+      },
+    ];
+  });
+}
+
+function mapReferences(value: unknown): MultiRoundRagSource[] {
+  const record = asRecord(value);
+  const list = Array.isArray(record?.references) ? record.references : [];
+  return list.flatMap((item, index) => {
+    const source = asRecord(item);
+    if (!source) return [];
+    const content = asString(source.content ?? source.snippet);
+    const rawUrl = asString(source.url ?? source.link ?? source.href);
+    const recUuid = asString(source.recUuid ?? source.rec_uuid ?? source.uuid);
+    return [
+      {
+        id: asString(source.id) || recUuid || undefined,
+        title: asString(source.title) || `参考资料 ${index + 1}`,
+        type: asString(source.type) || "参考资料",
+        snippet: asString(source.snippet) || content,
+        content,
+        source: asString(source.source ?? source.source1) || undefined,
+        time: asString(source.time ?? source.m_publish ?? source.publish_time) || undefined,
+        url: /^https?:\/\//i.test(rawUrl) ? rawUrl : undefined,
+        raw: source,
+      },
+    ];
+  });
+}
+
+function cloneTrace(trace: MultiRoundRagTrace): MultiRoundRagTrace {
+  return {
+    ...trace,
+    rounds: trace.rounds.map((round) => ({
+      ...round,
+      retrievals: [...round.retrievals],
+    })),
+  };
+}
+
+function isSingleRoundRagMessage(message: Message) {
+  return message.additional_kwargs?.single_round_rag === true;
+}
+
+export function isMultiRoundRagMessage(message: Message): boolean {
+  return message.additional_kwargs?.multi_round_rag === true;
+}
+
+export function getMultiRoundRagTrace(
+  message: Message,
+): MultiRoundRagTrace | null {
+  const trace = message.additional_kwargs?.multi_round_rag_trace;
+  return asRecord(trace)
+    ? (trace as unknown as MultiRoundRagTrace)
+    : null;
+}
+
+/**
+ * Frontend adapter for GeneralQA's multi-round endpoint. It keeps the external
+ * SSE protocol outside DeerFlow and stores only its UI messages in sessionStorage.
+ */
+export function useMultiRoundRagMessages(threadId: string) {
+  const [messages, setMessages] = useState<Message[]>(() => readStoredMessages(threadId));
+  const messagesRef = useRef(messages);
+  const [isLoading, setIsLoading] = useState(false);
+  const [error, setError] = useState<string | null>(null);
+  const controllerRef = useRef<AbortController | null>(null);
+
+  useEffect(() => {
+    const stored = readStoredMessages(threadId);
+    messagesRef.current = stored;
+    setMessages(stored);
+    setError(null);
+  }, [threadId]);
+
+  useEffect(() => {
+    messagesRef.current = messages;
+  }, [messages]);
+
+  useEffect(() => () => controllerRef.current?.abort(), []);
+
+  const updateMessages = useCallback(
+    (targetThreadId: string, updater: (current: Message[]) => Message[]) => {
+      setMessages((current) => {
+        const next = updater(current);
+        messagesRef.current = next;
+        writeStoredMessages(targetThreadId, next);
+        return next;
+      });
+    },
+    [],
+  );
+
+  const stop = useCallback(() => controllerRef.current?.abort(), []);
+
+  const submit = useCallback(
+    async ({ targetThreadId, query }: { targetThreadId: string; query: string }) => {
+      if (isLoading) return;
+      const question = query.trim();
+      if (!question) return;
+
+      const humanId = createId("multi-rag-human");
+      const assistantId = createId("multi-rag-ai");
+      const sourceId = createId("multi-rag-sources");
+      const trace = initialTrace();
+      let sources: MultiRoundRagSource[] = [];
+      const clientTs = new Date().toISOString();
+      const userMessage: Message = {
+        id: humanId,
+        type: "human",
+        content: [{ type: "text", text: question }],
+        additional_kwargs: { client_ts: clientTs, multi_round_rag: true },
+      };
+      const assistantMessage: Message = {
+        id: assistantId,
+        type: "ai",
+        content: "",
+        additional_kwargs: {
+          multi_round_rag: true,
+          multi_round_rag_trace: trace,
+        },
+      };
+      updateMessages(targetThreadId, (current) => [
+        ...current,
+        userMessage,
+        assistantMessage,
+      ]);
+      setError(null);
+      setIsLoading(true);
+
+      const controller = new AbortController();
+      controllerRef.current = controller;
+      const refreshAssistant = () => {
+        const traceSnapshot = cloneTrace(trace);
+        updateMessages(targetThreadId, (current) => {
+          const next = current.filter((message) => message.id !== sourceId);
+          const assistantIndex = next.findIndex((message) => message.id === assistantId);
+          if (assistantIndex < 0) return current;
+          next[assistantIndex] = {
+            ...next[assistantIndex],
+            content: traceSnapshot.answer,
+            additional_kwargs: {
+              ...next[assistantIndex]!.additional_kwargs,
+              multi_round_rag: true,
+              multi_round_rag_trace: traceSnapshot,
+              reasoning_content: traceSnapshot.rounds
+                .map((round) => round.thought)
+                .filter(Boolean)
+                .join("\n\n") || null,
+            },
+          } as Message;
+          if (sources.length > 0) {
+            next.splice(assistantIndex, 0, {
+              id: sourceId,
+              type: "tool",
+              name: "multi_round_rag_references",
+              content: { results: sources },
+              additional_kwargs: {
+                hide_from_ui: true,
+                multi_round_rag: true,
+              },
+            } as unknown as Message);
+          }
+          return next;
+        });
+      };
+
+      try {
+        const baseUrl = readRuntimeEnv("VITE_AIPI_GENERATE_CHAT_URL") || "";
+        if (!baseUrl) {
+          throw new Error("未配置 VITE_AIPI_GENERATE_CHAT_URL，无法发起多轮 RAG 问答。");
+        }
+        const response = await fetch(`${baseUrl.replace(/\/$/, "")}/generate`, {
+          method: "POST",
+          headers: { "Content-Type": "application/json" },
+          signal: controller.signal,
+          body: JSON.stringify({ prompt: question }),
+        });
+        if (!response.ok) {
+          throw new Error(`多轮 RAG 服务请求失败（${response.status}）。`);
+        }
+        if (!response.body) {
+          throw new Error("多轮 RAG 服务未返回流式响应。");
+        }
+
+        const reader = response.body.getReader();
+        const decoder = new TextDecoder();
+        let buffer = "";
+        while (true) {
+          const { done, value } = await reader.read();
+          if (done) break;
+          buffer += decoder.decode(value, { stream: true });
+          const lines = buffer.split(/\r?\n/);
+          buffer = lines.pop() ?? "";
+          for (const line of lines) {
+            const event = parseEventLine(line);
+            if (!event) continue;
+            const eventType = asString(event.event);
+            const data = event.data;
+            if (eventType === "tool_call") {
+              const tool = asRecord(data);
+              const toolName = asString(tool?.tool_name);
+              const input = asRecord(tool?.tool_input) ?? {};
+              const current = latestRound(trace);
+              if (toolName === "show_text") {
+                const text = asString(input.text);
+                if (text.includes("<think>")) {
+                  const thinkEnd = text.indexOf("</think>");
+                  addThinkingRound(
+                    trace,
+                    (thinkEnd >= 0
+                      ? text.slice(0, thinkEnd + "</think>".length)
+                      : text
+                    )
+                      .replace(/<\/?think>/g, "")
+                      .trim(),
+                  );
+                  if (thinkEnd >= 0) {
+                    const roundBody = text
+                      .slice(thinkEnd + "</think>".length)
+                      .trim();
+                    if (roundBody) {
+                      appendRoundBody(trace, roundBody);
+                    }
+                  }
+                } else {
+                  // A non-thinking `show_text` immediately after a thought is
+                  // this round's narrative; the terminal `message` event is
+                  // still rendered only as the final answer below all rounds.
+                  current.isSearching = false;
+                  trace.answer += text;
+                  trace.isComplete = true;
+                }
+              } else if (
+                toolName === "local_search" ||
+                toolName === "tavily_search" ||
+                toolName === "google_search"
+              ) {
+                const queryData = asRecord(parseJson(input.result));
+                current.title =
+                  toolName === "local_search" ? "本地知识检索" : "联网检索";
+                current.query =
+                  asString(input.q ?? input.query) ||
+                  asString(queryData?.query) ||
+                  current.query;
+                const retrievals = mapRetrievals(input.result);
+                if (retrievals.length > 0) current.retrievals = retrievals;
+                current.isSearching = false;
+              } else if (toolName === "get_textid_content") {
+                const article = asRecord(parseJson(input.result));
+                const body =
+                  asString(article?.content ?? article?.text) ||
+                  asString(input.content);
+                if (body) appendRoundBody(trace, body);
+                current.isSearching = false;
+              }
+              refreshAssistant();
+            } else if (eventType === "message" && typeof data === "string") {
+              latestRound(trace).isSearching = false;
+              trace.answer += data;
+              trace.isComplete = true;
+              refreshAssistant();
+            } else if (eventType === "references") {
+              sources = mapReferences(data);
+              refreshAssistant();
+            } else if (eventType === "done") {
+              trace.isComplete = true;
+              for (const round of trace.rounds) round.isSearching = false;
+              refreshAssistant();
+            }
+          }
+        }
+        trace.isComplete = true;
+        for (const round of trace.rounds) round.isSearching = false;
+        refreshAssistant();
+      } catch (caught) {
+        trace.isComplete = true;
+        for (const round of trace.rounds) round.isSearching = false;
+        if (!isAbortError(caught)) {
+          const message =
+            caught instanceof Error ? caught.message : "多轮 RAG 问答失败。";
+          setError(message);
+          if (!trace.answer) trace.answer = message;
+        }
+        refreshAssistant();
+      } finally {
+        if (controllerRef.current === controller) controllerRef.current = null;
+        setIsLoading(false);
+      }
+    },
+    [isLoading, updateMessages],
+  );
+
+  return { messages, isLoading, error, submit, stop };
+}
+
+export function isFrontendRagMessage(message: Message): boolean {
+  return isSingleRoundRagMessage(message) || isMultiRoundRagMessage(message);
+}

frontend-web/src/core/threads/use-single-round-rag.ts

@@ -22,6 +22,13 @@ type RagSource = {
   raw: Record<string, unknown>;
 };
 
+type RagHistoryEntry = {
+  role: "user" | "assistant";
+  content: string;
+};
+
+const MAX_HISTORY_MESSAGES = 12;
+
 /** Same JSON-object framing used by the established GeneralQA single-RAG flow. */
 function extractJsonObjects(raw: string): { objects: string[]; rest: string } {
   const objects: string[] = [];
@@ -199,6 +206,44 @@ function isAbortError(error: unknown) {
   return error instanceof DOMException && error.name === "AbortError";
 }
 
+function textFromMessage(message: Message): string {
+  if (typeof message.content === "string") {
+    return message.content.trim();
+  }
+  if (!Array.isArray(message.content)) {
+    return "";
+  }
+  return message.content
+    .map((part) => {
+      if (typeof part === "string") return part;
+      if (!part || typeof part !== "object") return "";
+      const text = (part as { text?: unknown }).text;
+      return typeof text === "string" ? text : "";
+    })
+    .join("\n")
+    .trim();
+}
+
+/**
+ * The general-QA endpoint accepts the usual `{ role, content }` history
+ * format. Keep it scoped to completed single-round-RAG turns so a normal
+ * DeerFlow conversation never leaks into this external endpoint.
+ */
+function buildRagHistory(messages: Message[]): RagHistoryEntry[] {
+  return messages
+    .filter(
+      (message) =>
+        message.additional_kwargs?.single_round_rag === true &&
+        (message.type === "human" || message.type === "ai"),
+    )
+    .map((message): RagHistoryEntry => ({
+      role: message.type === "human" ? "user" : "assistant",
+      content: textFromMessage(message),
+    }))
+    .filter((entry) => entry.content.length > 0)
+    .slice(-MAX_HISTORY_MESSAGES);
+}
+
 export function isSingleRoundRagMessage(message: Message): boolean {
   return message.additional_kwargs?.single_round_rag === true;
 }
@@ -209,21 +254,29 @@ export function isSingleRoundRagMessage(message: Message): boolean {
  */
 export function useSingleRoundRagMessages(threadId: string) {
   const [messages, setMessages] = useState<Message[]>(() => readStoredMessages(threadId));
+  const messagesRef = useRef(messages);
   const [isLoading, setIsLoading] = useState(false);
   const [error, setError] = useState<string | null>(null);
   const controllerRef = useRef<AbortController | null>(null);
 
   useEffect(() => {
-    setMessages(readStoredMessages(threadId));
+    const stored = readStoredMessages(threadId);
+    messagesRef.current = stored;
+    setMessages(stored);
     setError(null);
   }, [threadId]);
 
+  useEffect(() => {
+    messagesRef.current = messages;
+  }, [messages]);
+
   useEffect(() => () => controllerRef.current?.abort(), []);
 
   const updateMessages = useCallback(
     (targetThreadId: string, updater: (current: Message[]) => Message[]) => {
       setMessages((current) => {
         const next = updater(current);
+        messagesRef.current = next;
         writeStoredMessages(targetThreadId, next);
         return next;
       });
@@ -251,6 +304,7 @@ export function useSingleRoundRagMessages(threadId: string) {
       const assistantId = createId("single-rag-ai");
       const sourceId = createId("single-rag-sources");
       const clientTs = new Date().toISOString();
+      const history = buildRagHistory(messagesRef.current);
       const userMessage: Message = {
         id: humanId,
         type: "human",
@@ -319,7 +373,7 @@ export function useSingleRoundRagMessages(threadId: string) {
               knowledge_base_name: [],
               top_k: 5,
               score_threshold: 0.5,
-              history: [],
+              history,
               stream: true,
               model_name: modelName,
               temperature: 0.3,