nolo-cli 0.1.10 → 0.1.12

package/ai/agent/streamAgentChatTurn.ts

@@ -1,1278 +0,0 @@
-// 文件路径: packages/ai/agent/streamAgentChatTurn.ts
-import { extractCustomId } from "core/prefix";
-import { createDialogMessageKeyAndId } from "database/keys";
-import { DataType } from "create/types";
-
-import type { RootState } from "app/store";
-import { patch, read } from "database/dbSlice";
-import { generateRequestBody } from "ai/llm/generateRequestBody";
-import {
-    selectCurrentDialogConfig,
-    selectDialogConfigByKey,
-    addActiveController,
-    removeActiveController,
-    selectPendingUserInputQueue,
-    dequeueUserInput,
-    clearPendingUserInputQueue,
-} from "chat/dialog/dialogSlice";
-import { removeTransientMessage, selectAllMsgs } from "chat/messages/messageSlice";
-import {
-    selectContextRetention,
-    selectMaxExecutionTime,
-    selectCurrentServer,
-} from "app/settings/settingSlice";
-import { filterAndCleanMessages } from "integrations/openai/filterAndCleanMessages";
-import {
-    getFullChatContextKeys,
-    deduplicateContextKeys,
-} from "ai/agent/getFullChatContextKeys";
-import type { Agent } from "app/types";
-import { isResponseAPIModel } from "ai/llm/isResponseAPIModel";
-import { getModelContextWindow } from "ai/llm/getModelContextWindow";
-
-import {
-    sendOpenAICompletionsRequest,
-    type CompletionMeta,
-} from "../chat/sendOpenAICompletionsRequest";
-import { sendOpenAIResponseRequest } from "../chat/sendOpenAIResponseRequest";
-
-import type { AgentRuntimeOptions } from "./types";
-import { buildAgentViewMessages } from "./cleanAgentMessages";
-import { extractCategorizedMentions, type CategorizedMentions } from "create/editor/utils/slateUtils";
-import { resolveReferenceAssets, resolveToolsFromKeys } from "./referenceUtils";
-import { estimateTokenCount } from "ai/context/tokenUtils";
-import {
-    applyImageConfigRuntimeOverride,
-    buildStaticContexts,
-    compressOldToolResults,
-    buildDynamicContexts,
-    mergeContexts,
-    hasImageInMessages,
-    mergeAgentToolsWithRuntime,
-    trimMessagesWithSummary,
-    validateAccessAndBalance,
-} from "./streamAgentChatTurnUtils";
-import { buildCliPrompt } from "./cliPrompt";
-import { createCliChatTurnStream } from "./cliChatClient";
-import { getCliChatSession, startCliChatSession } from "./cliChatClient";
-import { messageStreaming, prepareAndPersistUserMessage } from "chat/messages/messageSlice";
-import { selectCurrentToken, selectUserId } from "auth/authSlice";
-import { persistMessageWithFixedId } from "./persistMessageWithFixedId";
-import { updateTotalUsage } from "../chat/updateTotalUsage";
-import { createSSEParser } from "../chat/parseMultilineSSE";
-import {
-    extractAgentRuntimeServerBase,
-    normalizeServerOrigin,
-} from "./runtimeServerBase";
-
-const buildParallelMessageMetadata = (
-    agentConfig: Pick<Agent, "dbKey" | "name">,
-    runtimeOptions?: AgentRuntimeOptions,
-) => {
-    const rawName =
-        typeof agentConfig?.name === "string" ? agentConfig.name.trim() : "";
-    return {
-        agentKey: agentConfig.dbKey,
-        cybotKey: agentConfig.dbKey,
-        ...(rawName ? { agentName: rawName } : {}),
-        ...(runtimeOptions?.parallelSessionId
-            ? { parallelSessionId: runtimeOptions.parallelSessionId }
-            : {}),
-        ...(runtimeOptions?.parallelBranchId
-            ? { parallelBranchId: runtimeOptions.parallelBranchId }
-            : {}),
-        ...(runtimeOptions?.parallelLabel
-            ? { parallelLabel: runtimeOptions.parallelLabel }
-            : {}),
-        ...(runtimeOptions?.parallelIndex !== undefined
-            ? { parallelIndex: runtimeOptions.parallelIndex }
-            : {}),
-    };
-};
-
-const filterMessagesForParallelBranch = (
-    messages: any[],
-    runtimeOptions?: AgentRuntimeOptions,
-) => {
-    if (!runtimeOptions?.parallelSessionId) return messages;
-
-    return messages.filter((message: any) => {
-        const sessionId = message?.parallelSessionId;
-        if (sessionId !== runtimeOptions.parallelSessionId) {
-            return true;
-        }
-        return message?.parallelBranchId === runtimeOptions.parallelBranchId;
-    });
-};
-
-const formatMachineAgentRunError = async (response: Response): Promise<string> => {
-    const errorText = await response.text();
-    let payload: any = null;
-    try {
-        payload = errorText ? JSON.parse(errorText) : null;
-    } catch {
-        payload = null;
-    }
-
-    const reason = typeof payload?.reason === "string" ? payload.reason : "";
-    if (response.status === 409) {
-        if (reason === "bound_machine_unavailable") {
-            return "绑定的电脑不在线。请确认这台电脑已开机并重新运行连接命令。";
-        }
-        if (reason === "connector_offline") {
-            return "电脑在线，但连接器未连接。请在这台电脑上重新运行连接命令后再试。";
-        }
-        if (reason === "missing_capability") {
-            return "这台电脑没有对应的 CLI 能力。请安装对应 CLI，或把 Agent 绑定到另一台电脑。";
-        }
-    }
-
-    const message =
-        typeof payload?.message === "string" && payload.message.trim()
-            ? payload.message.trim()
-            : typeof payload?.error === "string" && payload.error.trim()
-                ? payload.error.trim()
-                : errorText.trim();
-    return message || `Machine agent run failed (${response.status})`;
-};
-
-/** streamAgentChatTurn 参数（聊天轮次专用） */
-export interface StreamAgentChatTurnArgs {
-    agentKey: string;
-    userInput: string | any[];
-    serverBase?: string;
-    dialogKey?: string; // 可选。显式指定目标对话，不传则使用当前活跃对话。
-    isStreaming?: boolean;
-    parentMessageId?: string;
-    runtimeOptions?: AgentRuntimeOptions;
-}
-
-/**
- * 真正用于“聊天轮次”的流式 Agent 调用（带 Agent Loop）：
- * - 每轮检查权限 & 余额
- * - 每轮重建上下文 & 消息
- * - 调用 Completions/Response API
- * - 对 Completions 模型，基于 tool_calls / handoff / pending 决定是否继续
- */
-export const streamAgentChatTurnHandler = async (
-    args: StreamAgentChatTurnArgs,
-    thunkApi: any,
-) => {
-    const { agentKey, userInput, dialogKey: explicitDialogKey, parentMessageId, runtimeOptions } = args;
-    const { getState, dispatch, rejectWithValue } = thunkApi;
-    const state = getState() as RootState;
-
-    // 🚀 额外引入一个 Loop 控制器，用于中止整个 Agent 循环
-    const loopController = new AbortController();
-    const isAbortError = (error?: { name?: string } | null) =>
-        error?.name === "AbortError" || loopController.signal.aborted || thunkApi.signal.aborted;
-    const onAbort = () => loopController.abort();
-    thunkApi.signal.addEventListener("abort", onAbort);
-    let loopKey: string | null = null;
-    let runtimeDialogKey: string | null = explicitDialogKey ?? null;
-    let remoteTransientMessageId: string | null = null;
-    let remoteTransientMessageFinalized = false;
-
-    try {
-        let totalTurnUsage: any = null;
-        // 1. 读取 Agent 配置
-        const agentConfig = (await dispatch(read({ dbKey: agentKey })).unwrap()) as Agent;
-        if (!agentConfig) {
-            return rejectWithValue(`Agent config not found for ID: ${agentKey}`);
-        }
-
-        // ── CLI Agent 专用路由 ────────────────────────────────────────────────
-        // CLI 共享 prompt / model 这些入口能力，但不复用本地 tool-call 循环。
-        if (agentConfig.apiSource === "cli") {
-            const currentState = getState() as RootState;
-            const w =
-                typeof globalThis !== "undefined" && (globalThis as any).window
-                    ? (globalThis as any).window
-                    : null;
-            if (w) w.__LOOP_STOP_REASON__ = null;
-
-            const userText = typeof userInput === "string"
-                ? userInput
-                : (userInput as any[]).find((p) => p.type === "text")?.text ?? "";
-
-            const prompt = buildCliPrompt(agentConfig.prompt, userText);
-
-            // 生成消息 key
-            const dialogConfig =
-                selectDialogConfigByKey(currentState, explicitDialogKey) ??
-                selectCurrentDialogConfig(currentState);
-            if (!dialogConfig) {
-                return rejectWithValue("Dialog config not found");
-            }
-
-            const dialogKey = explicitDialogKey || dialogConfig.dbKey;
-            if (!dialogKey) {
-                return rejectWithValue("当前对话不存在，无法发送消息。");
-            }
-            runtimeDialogKey = dialogKey;
-            const dialogId = extractCustomId(dialogKey);
-            loopKey = `loop:${dialogId}`;
-            dispatch(addActiveController({ messageId: loopKey, controller: loopController, dialogKey }));
-
-            const { key: msgKey, messageId } = createDialogMessageKeyAndId(dialogId);
-            const cliMessageMetadata = buildParallelMessageMetadata(
-                agentConfig,
-                runtimeOptions,
-            );
-            const boundMachineId =
-                typeof (agentConfig as any).runtimeBinding?.machineId === "string"
-                    ? (agentConfig as any).runtimeBinding.machineId.trim()
-                    : "";
-
-            if (boundMachineId) {
-                const token = selectCurrentToken(currentState);
-                const authHeader = token ? `Bearer ${token}` : "";
-                const rawMessages = filterMessagesForParallelBranch(
-                    selectAllMsgs(currentState, dialogId),
-                    runtimeOptions,
-                );
-                const visibleMessages = buildAgentViewMessages(
-                    rawMessages as any,
-                    agentConfig.dbKey,
-                );
-                const cleanedMessages = filterAndCleanMessages(visibleMessages);
-                const currentServer = selectCurrentServer(currentState);
-                remoteTransientMessageId = messageId;
-                let accumulated = "";
-                let totalTurnUsage: any = undefined;
-                const buildMachineAssistantMessage = () => ({
-                    id: messageId,
-                    dbKey: msgKey,
-                    role: "assistant" as const,
-                    content: accumulated,
-                    ...cliMessageMetadata,
-                    userId: selectUserId(getState() as RootState),
-                });
-
-                dispatch(messageStreaming({
-                    id: messageId,
-                    dialogId,
-                    dbKey: msgKey,
-                    content: "",
-                    role: "assistant",
-                    ...cliMessageMetadata,
-                }));
-
-                const rejectMachineStream = async (message: string) => {
-                    if (accumulated.length > 0) {
-                        await persistMessageWithFixedId(dispatch, buildMachineAssistantMessage());
-                    } else {
-                        dispatch(removeTransientMessage(messageId));
-                    }
-                    remoteTransientMessageFinalized = true;
-                    return rejectWithValue(message);
-                };
-
-                const machineResponse = await fetch(`${currentServer.replace(/\/+$/, "")}/api/agent/run`, {
-                    method: "POST",
-                    headers: {
-                        "Content-Type": "application/json",
-                        Accept: "text/event-stream",
-                        ...(authHeader ? { Authorization: authHeader } : {}),
-                    },
-                    body: JSON.stringify({
-                        agentKey,
-                        userInput: userText,
-                        messages: cleanedMessages,
-                        stream: true,
-                        runtimeContext: {
-                            surface: "web",
-                            host: "browser",
-                            runtime: "react",
-                            entrypoint: "chat-dialog",
-                            capabilities: ["streaming", "dialog-ui", "machine-bound-cli"],
-                        },
-                        ...((dialogConfig as any)?.spaceId ? { spaceId: (dialogConfig as any).spaceId } : {}),
-                    }),
-                    signal: loopController.signal,
-                });
-
-                if (!machineResponse.ok) {
-                    return await rejectMachineStream(
-                        await formatMachineAgentRunError(machineResponse),
-                    );
-                }
-
-                const reader = machineResponse.body?.getReader();
-                if (!reader) {
-                    return await rejectMachineStream("无法读取电脑端 Agent 流式响应");
-                }
-
-                const decoder = new TextDecoder();
-                const parseSSE = createSSEParser();
-                const abortMachineStream = async () => {
-                    if (w) w.__LOOP_STOP_REASON__ = "aborted";
-                    if (accumulated.length <= 0) return;
-                    await persistMessageWithFixedId(dispatch, buildMachineAssistantMessage());
-                };
-
-                try {
-                    while (true) {
-                        const { done, value } = await reader.read();
-                        if (done) break;
-                        if (loopController.signal.aborted || thunkApi.signal.aborted) {
-                            await abortMachineStream();
-                            return;
-                        }
-
-                        const payloads = parseSSE(
-                            decoder.decode(value, { stream: true }),
-                        );
-                        for (const payload of payloads) {
-                            if (loopController.signal.aborted || thunkApi.signal.aborted) {
-                                await abortMachineStream();
-                                return;
-                            }
-                            if (payload.type === "error") {
-                                return await rejectMachineStream(
-                                    payload.message || "电脑端 Agent 执行失败",
-                                );
-                            }
-                            if (payload.type === "text" && typeof payload.content === "string") {
-                                accumulated += payload.content;
-                                dispatch(messageStreaming({
-                                    id: messageId,
-                                    dialogId,
-                                    dbKey: msgKey,
-                                    content: accumulated,
-                                    role: "assistant",
-                                    ...cliMessageMetadata,
-                                }));
-                            }
-                            if (payload.type === "done") {
-                                totalTurnUsage = payload.usage;
-                            }
-                        }
-                    }
-
-                    if (loopController.signal.aborted || thunkApi.signal.aborted) {
-                        await abortMachineStream();
-                        return;
-                    }
-
-                    await persistMessageWithFixedId(dispatch, buildMachineAssistantMessage());
-                    remoteTransientMessageFinalized = true;
-                } finally {
-                    try {
-                        await reader.cancel();
-                    } catch {
-                        // ignore
-                    }
-                }
-
-                return {
-                    usage: totalTurnUsage ?? undefined,
-                };
-            }
-
-            let cliSessionId = dialogConfig.cliSessionId ?? null;
-
-            // 先创建一条空的流式消息（让用户立刻看到 loading 状态）
-            dispatch(messageStreaming({
-                id: messageId,
-                dialogId,
-                dbKey: msgKey,
-                content: "",
-                role: "assistant",
-                ...cliMessageMetadata,
-            }));
-
-            const ensureCliSession = async () => {
-                if (cliSessionId) {
-                    const existing = await getCliChatSession(
-                        { getState },
-                        { sessionId: cliSessionId },
-                    ).catch(() => null);
-                    if (existing?.ok && existing?.session?.sessionId) {
-                        return cliSessionId;
-                    }
-                }
-
-                const started = await startCliChatSession(
-                    { getState },
-                    {
-                        cliProvider: agentConfig.cliProvider || "copilot",
-                        model: agentConfig.model || undefined,
-                        systemPrompt: agentConfig.prompt || undefined,
-                    },
-                );
-
-                const newSessionId =
-                    typeof started?.sessionId === "string" ? started.sessionId : null;
-                if (!newSessionId) {
-                    throw new Error("无法创建 CLI session。");
-                }
-
-                cliSessionId = newSessionId;
-                const patchResult = dispatch(
-                    patch({
-                        dbKey: dialogKey,
-                        changes: {
-                            cliSessionId: newSessionId,
-                        },
-                    })
-                ) as any;
-                try {
-                    if (typeof patchResult?.unwrap === "function") {
-                        await patchResult.unwrap();
-                    } else {
-                        await patchResult;
-                    }
-                } catch {
-                    // Best effort only. Session still exists server-side even if dialog persistence fails.
-                }
-                return newSessionId;
-            };
-
-            const initialSessionId = await ensureCliSession();
-            let resp = await createCliChatTurnStream(
-                {
-                    getState,
-                },
-                {
-                    sessionId: initialSessionId,
-                    prompt,
-                    model: agentConfig.model || undefined,
-                },
-                loopController.signal,
-            );
-
-            if (!resp.ok && resp.status === 404) {
-                cliSessionId = null;
-                const renewedSessionId = await ensureCliSession();
-                resp = await createCliChatTurnStream(
-                    {
-                        getState,
-                    },
-                    {
-                        sessionId: renewedSessionId,
-                        prompt,
-                        model: agentConfig.model || undefined,
-                    },
-                    loopController.signal,
-                );
-            }
-
-            if (!resp.ok) {
-                const err = await resp.json().catch(() => ({ error: resp.statusText }));
-                return rejectWithValue(err.error || "CLI 执行失败");
-            }
-
-            // 读取 SSE 流并逐步更新消息内容
-            const reader = resp.body?.getReader();
-            if (!reader) {
-                return rejectWithValue("无法读取流式响应");
-            }
-
-            let accumulated = "";
-            const decoder = new TextDecoder();
-            const buildCliAssistantMessage = () => ({
-                id: messageId,
-                dbKey: msgKey,
-                role: "assistant" as const,
-                content: accumulated,
-                ...cliMessageMetadata,
-                userId: selectUserId(getState() as RootState),
-            });
-            const abortCliStream = async () => {
-                if (w) w.__LOOP_STOP_REASON__ = "aborted";
-                if (accumulated.length <= 0) {
-                    return;
-                }
-                await persistMessageWithFixedId(dispatch, buildCliAssistantMessage());
-            };
-
-            try {
-                while (true) {
-                    const { done, value } = await reader.read();
-                    if (done) break;
-                    if (loopController.signal.aborted || thunkApi.signal.aborted) {
-                        await abortCliStream();
-                        return;
-                    }
-
-                    const raw = decoder.decode(value, { stream: true });
-                    // 解析 SSE 格式 "data: {...}\n\n"
-                    const lines = raw.split("\n");
-                    for (const line of lines) {
-                        if (!line.startsWith("data: ")) continue;
-                        if (loopController.signal.aborted || thunkApi.signal.aborted) {
-                            await abortCliStream();
-                            return;
-                        }
-                        try {
-                            const payload = JSON.parse(line.slice(6));
-                            if (payload.error) {
-                                return rejectWithValue(payload.error);
-                            }
-                            if (payload.chunk) {
-                                if (loopController.signal.aborted || thunkApi.signal.aborted) {
-                                    await abortCliStream();
-                                    return;
-                                }
-                                accumulated += payload.chunk;
-                                if (loopController.signal.aborted || thunkApi.signal.aborted) {
-                                    await abortCliStream();
-                                    return;
-                                }
-                                dispatch(messageStreaming({
-                                    id: messageId,
-                                    dialogId,
-                                    dbKey: msgKey,
-                                    content: accumulated,
-                                    role: "assistant",
-                                    ...cliMessageMetadata,
-                                }));
-                            }
-                            // payload.done 时 stream 自然关闭，while loop 会退出
-                        } catch {
-                            // 忽略解析失败的行
-                        }
-                    }
-                }
-
-                if (loopController.signal.aborted || thunkApi.signal.aborted) {
-                    await abortCliStream();
-                    return;
-                }
-
-                // 持久化最终消息：用已有 ID，避免 prepareAndPersistMessage 重新生成 ID 导致重复
-                await persistMessageWithFixedId(dispatch, buildCliAssistantMessage());
-            } finally {
-                try {
-                    await reader.cancel();
-                } catch {
-                    // ignore
-                }
-            }
-
-            return;
-        }
-        // ─────────────────────────────────────────────────────────────────────
-
-        const currentDialog =
-            selectDialogConfigByKey(state, explicitDialogKey) ??
-            selectCurrentDialogConfig(state);
-        const activeDialogKey = currentDialog?.dbKey;
-        const dialogKey = explicitDialogKey || activeDialogKey;
-
-        if (!dialogKey) {
-            return rejectWithValue("当前对话不存在，无法发送消息。");
-        }
-        runtimeDialogKey = dialogKey;
-        const dialogId = extractCustomId(dialogKey);
-
-        let userInputText = "";
-        if (typeof userInput === "string") {
-            userInputText = userInput;
-        } else if (Array.isArray(userInput)) {
-            const textPart = userInput.find((p) => p.type === "text");
-            if (textPart && textPart.text) {
-                userInputText = textPart.text;
-            }
-        }
-
-        const explicitServerBase =
-            typeof args.serverBase === "string" && args.serverBase.trim()
-                ? args.serverBase.trim()
-                : null;
-        const declaredRuntimeServerBase = extractAgentRuntimeServerBase(agentConfig);
-        const requestedServerBase = explicitServerBase ?? declaredRuntimeServerBase;
-        const normalizedRequestedServerBase =
-            requestedServerBase && normalizeServerOrigin(requestedServerBase);
-        const normalizedCurrentServer = normalizeServerOrigin(
-            selectCurrentServer(state),
-        );
-        const canAutoRouteRemotely =
-            !Array.isArray(userInput) &&
-            !runtimeOptions?.extraTools?.length &&
-            !runtimeOptions?.editingTarget &&
-            !runtimeOptions?.imageConfigOverride;
-        if (requestedServerBase && canAutoRouteRemotely) {
-            if (
-                normalizedRequestedServerBase &&
-                normalizedCurrentServer &&
-                normalizedRequestedServerBase === normalizedCurrentServer
-            ) {
-                // same server as current workspace; keep local flow
-            } else {
-                const token = selectCurrentToken(state);
-                const authHeader = token ? `Bearer ${token}` : "";
-                const rawMessages = filterMessagesForParallelBranch(
-                    selectAllMsgs(state, dialogId),
-                    runtimeOptions,
-                );
-                const visibleMessages = buildAgentViewMessages(
-                    rawMessages as any,
-                    agentConfig.dbKey,
-                );
-                const cleanedMessages = filterAndCleanMessages(visibleMessages);
-                const { key: msgKey, messageId } = createDialogMessageKeyAndId(dialogId);
-                remoteTransientMessageId = messageId;
-                const remoteMessageMetadata = buildParallelMessageMetadata(
-                    agentConfig,
-                    runtimeOptions,
-                );
-                let accumulated = "";
-                let totalTurnUsage: any = undefined;
-                const buildRemoteAssistantMessage = () => ({
-                    id: messageId,
-                    dbKey: msgKey,
-                    role: "assistant" as const,
-                    content: accumulated,
-                    ...remoteMessageMetadata,
-                    userId: selectUserId(getState() as RootState),
-                });
-
-                loopKey = `loop:${dialogId}`;
-                dispatch(addActiveController({ messageId: loopKey, controller: loopController, dialogKey }));
-                dispatch(messageStreaming({
-                    id: messageId,
-                    dialogId,
-                    dbKey: msgKey,
-                    content: "",
-                    role: "assistant",
-                    ...remoteMessageMetadata,
-                }));
-
-                const rejectRemoteStream = async (message: string) => {
-                    if (accumulated.length > 0) {
-                        await persistMessageWithFixedId(dispatch, buildRemoteAssistantMessage());
-                    } else {
-                        dispatch(removeTransientMessage(messageId));
-                    }
-                    remoteTransientMessageFinalized = true;
-                    return rejectWithValue(message);
-                };
-
-                const remoteResponse = await fetch(`${requestedServerBase.replace(/\/+$/, "")}/api/agent/run`, {
-                    method: "POST",
-                    headers: {
-                        "Content-Type": "application/json",
-                        Accept: "text/event-stream",
-                        ...(authHeader ? { Authorization: authHeader } : {}),
-                    },
-                    body: JSON.stringify({
-                        agentKey,
-                        userInput: userInputText,
-                        messages: cleanedMessages,
-                        stream: true,
-                        runtimeContext: {
-                            surface: "web",
-                            host: "browser",
-                            runtime: "react",
-                            entrypoint: "chat-dialog",
-                            capabilities: ["streaming", "dialog-ui", "tool-cards"],
-                        },
-                        ...(currentDialog?.spaceId ? { spaceId: currentDialog.spaceId } : {}),
-                    }),
-                    signal: loopController.signal,
-                });
-
-                if (!remoteResponse.ok) {
-                    const errorText = await remoteResponse.text();
-                    return await rejectRemoteStream(
-                        errorText || `Remote agent run failed (${remoteResponse.status})`,
-                    );
-                }
-
-                const reader = remoteResponse.body?.getReader();
-                if (!reader) {
-                    return await rejectRemoteStream("无法读取远端流式响应");
-                }
-
-                const decoder = new TextDecoder();
-                const parseSSE = createSSEParser();
-                const abortRemoteStream = async () => {
-                    if (accumulated.length <= 0) return;
-                    await persistMessageWithFixedId(dispatch, buildRemoteAssistantMessage());
-                };
-
-                try {
-                    while (true) {
-                        const { done, value } = await reader.read();
-                        if (done) break;
-                        if (loopController.signal.aborted || thunkApi.signal.aborted) {
-                            await abortRemoteStream();
-                            return;
-                        }
-
-                        const payloads = parseSSE(
-                            decoder.decode(value, { stream: true }),
-                        );
-                        for (const payload of payloads) {
-                            if (payload.type === "error") {
-                                return await rejectRemoteStream(
-                                    payload.message || "远端 Agent 执行失败",
-                                );
-                            }
-                            if (payload.type === "text" && typeof payload.content === "string") {
-                                accumulated += payload.content;
-                                dispatch(messageStreaming({
-                                    id: messageId,
-                                    dialogId,
-                                    dbKey: msgKey,
-                                    content: accumulated,
-                                    role: "assistant",
-                                    ...remoteMessageMetadata,
-                                }));
-                            }
-                            if (payload.type === "done") {
-                                totalTurnUsage = payload.usage;
-                            }
-                        }
-                    }
-                } finally {
-                    try {
-                        await reader.cancel();
-                    } catch {
-                        // ignore
-                    }
-                }
-
-                await persistMessageWithFixedId(dispatch, buildRemoteAssistantMessage());
-                remoteTransientMessageFinalized = true;
-                return {
-                    usage: totalTurnUsage ?? undefined,
-                };
-            }
-        }
-
-        // Extract Mentions from userInput if it's potentially Slate content
-        let extractedMentions: CategorizedMentions | undefined;
-        if (Array.isArray(userInput)) {
-            // Basic check if it looks like Slate nodes (has children) or just assume safe to traverse
-            // extractCategorizedMentions handles traversal safely.
-            extractedMentions = extractCategorizedMentions(userInput as any);
-        }
-
-        const mentionedTools = extractedMentions?.tools ?? [];
-
-        // 2. 解析引用：包含 tools 的页面自动升级为 instruction
-        const {
-            references: normalizedReferences,
-            contentByKey: referenceContentCache,
-            referencedTools: referenceTools,
-            recommendedSkillTools: referenceRecommendedSkillTools,
-            recommendedSkillHints: referenceRecommendedSkillHints,
-            skillPromptPatches: referenceSkillPromptPatches,
-        } = await resolveReferenceAssets(agentConfig.references, dispatch);
-
-        const agentConfigWithReferences = {
-            ...agentConfig,
-            references: normalizedReferences,
-            referencedTools: referenceTools,
-            recommendedSkillTools: referenceRecommendedSkillTools,
-            recommendedSkillHints: referenceRecommendedSkillHints,
-            skillPromptPatches: referenceSkillPromptPatches,
-        };
-
-        // --- [新增] 提取本次 Handler 启动前的稳定历史消息 ID 集合 ---
-        const initialRawMsgs = selectAllMsgs(state, dialogId);
-        const initialHistoryIds = new Set(initialRawMsgs.map((m: any) => m.id));
-
-        const keySets = await getFullChatContextKeys(
-            state,
-            dispatch,
-            agentConfigWithReferences,
-            userInput,
-            currentDialog ?? undefined,
-        );
-        const finalKeys = deduplicateContextKeys(keySets);
-        const allContextKeys = new Set<string>([
-            ...finalKeys.botInstructionsContext,
-            ...finalKeys.currentInputContext,
-            ...finalKeys.historyContext,
-            ...finalKeys.botKnowledgeContext,
-        ]);
-
-        // 4. 上下文页面里提取 tools 并缓存内容
-        const {
-            tools: contextTools,
-            contentByKey: contextContentCache,
-            recommendedSkillTools: contextRecommendedSkillTools = [],
-            recommendedSkillHints: contextRecommendedSkillHints = [],
-            skillPromptPatches: contextSkillPromptPatches = [],
-        } = await resolveToolsFromKeys(
-            Array.from(allContextKeys),
-            dispatch,
-            referenceContentCache,
-        );
-
-        const mergedContentCache = new Map<string, any>([
-            ...referenceContentCache,
-            ...contextContentCache,
-        ]);
-
-        // 4. 合并工具 (Base + Default + Context + Mentioned + Runtime) + 图片配置
-        const agentConfigWithTools = mergeAgentToolsWithRuntime(
-            {
-                ...agentConfigWithReferences,
-                recommendedSkillTools: [
-                    ...(((agentConfigWithReferences as any).recommendedSkillTools ?? []) as string[]),
-                    ...contextRecommendedSkillTools,
-                ],
-                recommendedSkillHints: [
-                    ...(((agentConfigWithReferences as any).recommendedSkillHints ?? []) as string[]),
-                    ...contextRecommendedSkillHints,
-                ],
-                skillPromptPatches: [
-                    ...(((agentConfigWithReferences as any).skillPromptPatches ?? []) as string[]),
-                    ...contextSkillPromptPatches,
-                ],
-            },
-            contextTools,
-            mentionedTools,
-            runtimeOptions,
-            state,
-        );
-        const agentConfigForCall = applyImageConfigRuntimeOverride(
-            agentConfigWithTools,
-            runtimeOptions,
-        );
-
-        // 如果对话设置了 maxTokens，覆盖 agent 的 max_tokens
-        const dialogMaxTokens = currentDialog?.maxTokens;
-        const effectiveAgentConfig = dialogMaxTokens
-            ? { ...agentConfigForCall, max_tokens: dialogMaxTokens }
-            : agentConfigForCall;
-
-        const isRespModel = isResponseAPIModel(agentConfigForCall);
-
-        // 🔹 Response-style 模型：与 completions 一样走完整 Agent Loop
-        if (isRespModel) {
-            const maxExecutionTime = selectMaxExecutionTime(state);
-            const MAX_TIME_MS = maxExecutionTime > 0 ? maxExecutionTime : 240_000;
-            const startTime = Date.now();
-
-            const staticContexts = await buildStaticContexts(
-                state,
-                dispatch,
-                agentConfigForCall,
-                currentDialog ?? undefined,
-                mergedContentCache,
-            );
-
-            let appendTempUserInput = true;
-            let currentParentMessageId = parentMessageId ?? undefined;
-
-            const w = typeof globalThis !== "undefined" && (globalThis as any).window ? (globalThis as any).window : null;
-            if (w) w.__LOOP_STOP_REASON__ = null;
-
-            loopKey = `loop:${dialogId}`;
-            dispatch(addActiveController({ messageId: loopKey, controller: loopController, dialogKey }));
-
-            for (;;) {
-                const requestParentMessageId = currentParentMessageId;
-                if (loopController.signal.aborted || thunkApi.signal.aborted) {
-                    if (w) w.__LOOP_STOP_REASON__ = "aborted";
-                    break;
-                }
-
-                const loopState = getState() as RootState;
-                const now = Date.now();
-                if (now - startTime > MAX_TIME_MS) {
-                    if (w) w.__LOOP_STOP_REASON__ = "timeout";
-                    break;
-                }
-
-                const accessError = validateAccessAndBalance(
-                    agentConfigForCall,
-                    loopState,
-                );
-                if (accessError) {
-                    return rejectWithValue(accessError);
-                }
-
-                const dynamicContexts = await buildDynamicContexts(
-                    loopState,
-                    dispatch,
-                    agentConfigForCall,
-                    userInput,
-                    runtimeOptions,
-                    mergedContentCache,
-                    dialogKey,
-                );
-                const contexts = mergeContexts(staticContexts, dynamicContexts);
-
-                const rawMessages = filterMessagesForParallelBranch(
-                    selectAllMsgs(loopState, dialogId),
-                    runtimeOptions,
-                );
-                let visibleMessages = buildAgentViewMessages(
-                    rawMessages as any,
-                    agentConfigForCall.dbKey,
-                );
-
-                if (
-                    appendTempUserInput &&
-                    userInputText &&
-                    userInputText.trim().length > 0
-                ) {
-                    visibleMessages = [
-                        ...visibleMessages,
-                        {
-                            id: `__tmp_user_${Date.now()}`,
-                            dbKey: "",
-                            role: "user",
-                            content: userInputText,
-                            thinkContent: "",
-                            cybotKey: agentConfigForCall.dbKey,
-                            isStreaming: false,
-                        } as any,
-                    ];
-                }
-
-                const cleanedMessages = filterAndCleanMessages(visibleMessages);
-                const ctxWindow =
-                    getModelContextWindow(agentConfigForCall.model) || 128000;
-                const retention = selectContextRetention(loopState);
-                const summaryTokenCount = contexts.dialogSummary
-                    ? estimateTokenCount(contexts.dialogSummary)
-                    : 0;
-                const processedMessages = trimMessagesWithSummary(
-                    compressOldToolResults(cleanedMessages),
-                    ctxWindow,
-                    summaryTokenCount,
-                    retention,
-                );
-
-                let firstDynamicIdx = processedMessages.findIndex(
-                    (m) => m.id && !initialHistoryIds.has(m.id),
-                );
-                if (firstDynamicIdx === -1) firstDynamicIdx = processedMessages.length;
-
-                const stableMessages = processedMessages.slice(0, firstDynamicIdx);
-                const dynamicMessages = processedMessages.slice(firstDynamicIdx);
-
-                if (appendTempUserInput) {
-                    const agentHasVision =
-                        (agentConfigForCall as any).hasVision === undefined
-                            ? true
-                            : Boolean((agentConfigForCall as any).hasVision);
-
-                    if (!agentHasVision && hasImageInMessages(processedMessages)) {
-                        return rejectWithValue(
-                            "当前 Agent 不支持图片输入，请改用文本或文档。",
-                        );
-                    }
-                }
-
-                const bodyData = generateRequestBody({
-                    agentConfig: effectiveAgentConfig,
-                    messages: dynamicMessages,
-                    stableMessages,
-                    userInput: userInputText,
-                    contexts,
-                });
-
-                const meta: CompletionMeta = await sendOpenAIResponseRequest({
-                    bodyData,
-                    agentConfig: agentConfigForCall,
-                    thunkApi,
-                    dialogKey,
-                    parentMessageId: currentParentMessageId,
-                    messageMetadata: buildParallelMessageMetadata(
-                        agentConfigForCall,
-                        runtimeOptions,
-                    ),
-                });
-
-                appendTempUserInput = false;
-                currentParentMessageId = undefined;
-                totalTurnUsage = updateTotalUsage(totalTurnUsage, meta.usage);
-
-                if (meta.hasHandedOff) {
-                    if (!requestParentMessageId && meta.messageId) {
-                        dispatch(removeTransientMessage(meta.messageId));
-                    }
-                    if (w) w.__LOOP_STOP_REASON__ = "handoff";
-                    break;
-                }
-
-                if (meta.hasPendingInteraction) {
-                    if (w) w.__LOOP_STOP_REASON__ = "pending";
-                    break;
-                }
-
-                const afterTurnState = getState() as RootState;
-                const queuedMessages = selectPendingUserInputQueue(afterTurnState, dialogKey);
-                if (queuedMessages.length > 0) {
-                    const queuedText = queuedMessages[0];
-
-                    const currentDialogConfig =
-                        selectDialogConfigByKey(afterTurnState, dialogKey) ??
-                        selectCurrentDialogConfig(afterTurnState);
-                    if (!currentDialogConfig) {
-                        dispatch(clearPendingUserInputQueue({ dialogKey }));
-                        break;
-                    }
-                    await dispatch(
-                        prepareAndPersistUserMessage({
-                            userInput: queuedText,
-                            dialogConfig: currentDialogConfig,
-                        })
-                    ).unwrap();
-                    dispatch(dequeueUserInput({ dialogKey }));
-                    continue;
-                }
-
-                if (!meta.hasToolCalls) {
-                    if (w) w.__LOOP_STOP_REASON__ = "done";
-                    break;
-                }
-            }
-
-            return {
-                usage: totalTurnUsage ?? undefined,
-            };
-        }
-
-        // 🔹 Completions-style 模型：Agent Loop
-        const maxExecutionTime = selectMaxExecutionTime(state);
-
-        const MAX_TIME_MS = maxExecutionTime > 0 ? maxExecutionTime : 240_000;
-        const startTime = Date.now();
-
-        // 🚀 优化：在 Loop 外构建静态上下文（只执行一次）
-        // 静态上下文包含：botInstructions、botKnowledge、spaceContext、userGlobalPrompt
-        // 这些内容在 Loop 期间是稳定的，不需要每轮重新构建
-        const staticContexts = await buildStaticContexts(
-            state,
-            dispatch,
-            agentConfigForCall,
-            currentDialog ?? undefined,
-            mergedContentCache,
-        );
-
-        let appendTempUserInput = true;
-        let currentParentMessageId = parentMessageId ?? undefined;
-
-        const w = typeof globalThis !== "undefined" && (globalThis as any).window ? (globalThis as any).window : null;
-        if (w) w.__LOOP_STOP_REASON__ = null;
-
-        if (!isRespModel) {
-            loopKey = `loop:${dialogId}`;
-            dispatch(addActiveController({ messageId: loopKey, controller: loopController, dialogKey }));
-        }
-
-        for (;;) {
-            const requestParentMessageId = currentParentMessageId;
-            // 每轮开始前检查是否已中止
-            if (loopController.signal.aborted || thunkApi.signal.aborted) {
-                if (w) w.__LOOP_STOP_REASON__ = "aborted";
-                break;
-            }
-
-            const loopState = getState() as RootState;
-            const now = Date.now();
-            if (now - startTime > MAX_TIME_MS) {
-                if (w) w.__LOOP_STOP_REASON__ = "timeout";
-                break;
-            }
-
-            // 每轮检查权限 & 余额
-            const accessError = validateAccessAndBalance(
-                agentConfigForCall,
-                loopState,
-            );
-            if (accessError) {
-                return rejectWithValue(accessError);
-            }
-
-            // 🚀 优化：每轮只构建动态上下文（currentInput、history、editingContext、dialogSummary）
-            const dynamicContexts = await buildDynamicContexts(
-                loopState,
-                dispatch,
-                agentConfigForCall,
-                userInput,
-                runtimeOptions,
-                mergedContentCache,
-                dialogKey,
-            );
-
-            // 合并静态和动态上下文
-            const contexts = mergeContexts(staticContexts, dynamicContexts);
-
-            const rawMessages = filterMessagesForParallelBranch(
-                selectAllMsgs(loopState, dialogId),
-                runtimeOptions,
-            );
-            let visibleMessages = buildAgentViewMessages(
-                rawMessages as any,
-                agentConfigForCall.dbKey,
-            );
-
-            if (
-                appendTempUserInput &&
-                userInputText &&
-                userInputText.trim().length > 0
-            ) {
-                visibleMessages = [
-                    ...visibleMessages,
-                    {
-                        id: `__tmp_user_${Date.now()}`,
-                        dbKey: "",
-                        role: "user",
-                        content: userInputText,
-                        thinkContent: "",
-                        cybotKey: agentConfigForCall.dbKey,
-                        isStreaming: false,
-                    } as any,
-                ];
-            }
-
-            const cleanedMessages = filterAndCleanMessages(visibleMessages);
-            const ctxWindow =
-                getModelContextWindow(agentConfigForCall.model) || 128000;
-            const retention = selectContextRetention(loopState);
-            const summaryTokenCount = contexts.dialogSummary
-                ? estimateTokenCount(contexts.dialogSummary)
-                : 0;
-            const processedMessages = trimMessagesWithSummary(
-                compressOldToolResults(cleanedMessages),
-                ctxWindow,
-                summaryTokenCount,
-                retention,
-            );
-
-            // --- [优化 P1] 使用 findIndex + slice 确保顺序和无 ID 稳定消息的保留 ---
-            let firstDynamicIdx = processedMessages.findIndex(
-                (m) => m.id && !initialHistoryIds.has(m.id),
-            );
-            if (firstDynamicIdx === -1) firstDynamicIdx = processedMessages.length;
-
-            const stableMessages = processedMessages.slice(0, firstDynamicIdx);
-            const dynamicMessages = processedMessages.slice(firstDynamicIdx);
-
-            if (appendTempUserInput) {
-                const agentHasVision =
-                    (agentConfigForCall as any).hasVision === undefined
-                        ? true
-                        : Boolean((agentConfigForCall as any).hasVision);
-
-                if (!agentHasVision && hasImageInMessages(processedMessages)) {
-                    return rejectWithValue(
-                        "当前 Agent 不支持图片输入，请改用文本或文档。",
-                    );
-                }
-            }
-
-            const bodyData = generateRequestBody({
-                agentConfig: effectiveAgentConfig,
-                messages: dynamicMessages,
-                stableMessages,
-                userInput: userInputText,
-                contexts,
-            });
-
-            const meta: CompletionMeta = await sendOpenAICompletionsRequest({
-                bodyData,
-                agentConfig: agentConfigForCall,
-                thunkApi,
-                dialogKey,
-                parentMessageId: currentParentMessageId,
-                messageMetadata: buildParallelMessageMetadata(
-                    agentConfigForCall,
-                    runtimeOptions,
-                ),
-            });
-
-            appendTempUserInput = false;
-            currentParentMessageId = undefined;
-            totalTurnUsage = updateTotalUsage(totalTurnUsage, meta.usage);
-
-            // handoff（例如 runStreamingAgent）：当前 Agent 停止，后续由子 Agent 自动续跑
-            if (meta.hasHandedOff) {
-                if (!requestParentMessageId && meta.messageId) {
-                    dispatch(removeTransientMessage(meta.messageId));
-                }
-                if (w) w.__LOOP_STOP_REASON__ = "handoff";
-                break;
-            }
-
-            if (meta.hasPendingInteraction) {
-                if (w) w.__LOOP_STOP_REASON__ = "pending";
-                break;
-            }
-
-            // 检查是否有用户在 loop 期间发送的排队消息
-            const afterTurnState = getState() as RootState;
-            const queuedMessages = selectPendingUserInputQueue(afterTurnState, dialogKey);
-            if (queuedMessages.length > 0) {
-                const queuedText = queuedMessages[0];
-
-                const currentDialogConfig =
-                    selectDialogConfigByKey(afterTurnState, dialogKey) ??
-                    selectCurrentDialogConfig(afterTurnState);
-                if (!currentDialogConfig) {
-                    // 对话已切换/销毁，无法持久化；清空队列并终止 loop，避免死循环重试
-                    dispatch(clearPendingUserInputQueue({ dialogKey }));
-                    break;
-                }
-                await dispatch(
-                    prepareAndPersistUserMessage({
-                        userInput: queuedText,
-                        dialogConfig: currentDialogConfig,
-                    })
-                ).unwrap();
-                // 持久化成功后再出队，避免 persist 失败时丢消息
-                dispatch(dequeueUserInput({ dialogKey }));
-                // 用户消息已持久化到 DB，下一轮 selectAllMsgs 会自动包含它
-                // 不设置 appendTempUserInput，直接继续下一轮
-                continue;
-            }
-
-            if (!meta.hasToolCalls) {
-                if (w) w.__LOOP_STOP_REASON__ = "done";
-                break;
-            }
-
-            // 否则：存在 tool_calls 且没有 handoff / pending，基于新的 history 继续下一轮
-        }
-
-        return {
-            usage: totalTurnUsage ?? undefined,
-        };
-    } catch (error: any) {
-        if (isAbortError(error)) {
-            if (remoteTransientMessageId && !remoteTransientMessageFinalized) {
-                dispatch(removeTransientMessage(remoteTransientMessageId));
-                remoteTransientMessageFinalized = true;
-            }
-            const w =
-                typeof globalThis !== "undefined" && (globalThis as any).window
-                    ? (globalThis as any).window
-                    : null;
-            if (w) w.__LOOP_STOP_REASON__ = "aborted";
-            return;
-        }
-        console.error(
-            `Error in streamAgentChatTurn for [${agentKey}]:`,
-            error,
-        );
-        if (remoteTransientMessageId && !remoteTransientMessageFinalized) {
-            dispatch(removeTransientMessage(remoteTransientMessageId));
-            remoteTransientMessageFinalized = true;
-        }
-
-        return rejectWithValue(
-            error?.message ||
-            "An unexpected error occurred in streamAgentChatTurn.",
-        );
-    } finally {
-        if (loopKey && runtimeDialogKey) {
-            dispatch(removeActiveController({ messageId: loopKey, dialogKey: runtimeDialogKey }));
-        } else if (loopKey) {
-            dispatch(removeActiveController(loopKey));
-        }
-        // 清空排队的用户消息（loop 结束或异常时，未消费的消息不应继续保留）
-        dispatch(clearPendingUserInputQueue(runtimeDialogKey ? { dialogKey: runtimeDialogKey } : undefined));
-        thunkApi.signal.removeEventListener("abort", onAbort);
-    }
-};