From 5b9d4d19b3877fc1d7508b5a54b388ea3086c2ef Mon Sep 17 00:00:00 2001 From: thzxx <1440196015@qq.com> Date: Sat, 5 Sep 2026 20:06:26 +0800 Subject: [PATCH] =?UTF-8?q?feat:=20v0.8.0=20=E6=B5=81=E8=AF=AD=E4=B9=89?= =?UTF-8?q?=E8=A1=A5=E5=85=A8=20=C2=B7=20=E4=BC=9A=E8=AF=9D=E5=8F=AF?= =?UTF-8?q?=E9=9D=A0=20=C2=B7=20=E6=81=A2=E5=A4=8D=E5=8A=9B=20=E2=80=94=20?= =?UTF-8?q?finish=5Freason=20=E5=85=A8=E9=93=BE=E8=B7=AF=E8=B4=AF=E9=80=9A?= =?UTF-8?q?=E6=A0=B9=E6=B2=BB"=E6=80=9D=E8=80=83=E4=B8=AD=E5=81=9C?= =?UTF-8?q?=E6=AD=A2"=20=C2=B7=202445=20=E7=94=A8=E4=BE=8B=E5=85=A8?= =?UTF-8?q?=E9=87=8F=E5=9B=9E=E5=BD=92?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit P0 会话可靠性收口(根治"模型思考着会话就停止"): - P0-1 finish_reason 全链路贯通:DONE 事件与 IterationStep 新增 finishReason,OpenAI 共享 SSE / Anthropic message_delta.stop_reason / Ollama done_reason 三路采集,TRACE 层弃用硬编码 'stop' 记录真值 - P0-2 空响应守卫 + 降级重试:零产出流→可重试错误走退避;思考耗尽输出预算(reasoning-only + length)→自动关闭思考降级重试一次;仍失败→OUTPUT_LENGTH_EXCEEDED 结构化错误 + 故障转移;附带根治 abort 恰逢零工具调用轮被 COMPLETED 抢占的真实缺陷 - P0-3 思考×能力×预算三对齐:DeepSeek/MiMo/Agnes/Ollama 四家 supportsThinking=false 强制不发思考参数;小输出预算告警;设置页联动提示 - P0-4 渲染层可见性:截断/空完成/友好错误三类提示,i18n 全部出层 - P0-5 回归四件套:reasoning-only 终止判定、集成级空闲超时、504 引擎重试归类、思考中 abort→USER_INTERRUPT、P4-2 强制收尾路径 FEAT-1:LLM 设置新增「最大输出上限」——Provider 支持矩阵显隐 + 模型上限钳制提示 + 超限保存警告 + llm.maxTokens 热生效 P1 修复面收口: - 渲染层三缺陷根治:后台会话回放缓冲(2000 条/4MB 有界 + agent:getReplayState + 事件总线)+ abort 双层自愈 + sendMessage 收尾兜底 + 中断卡片清扫 - 工具 abort 信号全覆盖:web_search/web_fetch/http_request/code_search/git 系列/delegate_task 全部接入引擎中断;web_search 时间预算收敛(720s→≤240s);移除伪造 ToolExecutionContext 与死代码 - 安全:本地 Pinned CONNECT 代理根治浏览器通道 DNS rebinding(校验期 IP pinning,可注入 resolver 表测);配置 URL 域名解析深校验(DeepCheckSoftFailure 软失败);SSE 空 error 帧防御修复;Ollama generate/embed AbortSignal.any 合并 - 缺陷清单:UTF-16 BOM 读取、tmp 同毫秒碰撞(nanoid 后缀)、code_search JS 回退参数对称(case_sensitive/前后文独立)、list_directory include_node_modules、崩溃自愈退避(60s 窗 ≥3 次停 reload)、MemoryViewer/Sidebar i18n 收口 P2 能力演进: - 会话回收站:SCHEMA_VERSION 3 + 迁移 10(deleted_at,存在性守卫),软删除/恢复/彻底删除/30 天自动清理(启动+24h),searchMessages 聚合剔除,Sidebar 回收站面板 - 会话回放播放器:sessions:listRecordings/readRecording(白名单+目录边界+20MB 上限),SessionReplayPlayer 时间轴/步进/变速,Trace 面板入口 - electron-updater 自动更新:双轨(手动 feed 比对保留),生产环境启动静默检查 + update:status 广播 + app:updateInstall + LogsSettings UpdatePanel + builder publish 配置 - @ 文件提及:workspace.listFiles/readFileClip(边界/512KB/NUL 拒绝/MEMORY.md 保护),ChatInput Fuse 联想+键盘导航+附件管线注入 - MCP Resources/Prompts 发现:可选能力 try/catch 降级,mcp:listServerContents,MCPSettings 展开视图 - 文档对齐:内部 API 标准 HTML(Adapter 清单补 MiMo/已实现注记/STREAM_RESET/DONE.finishReason/ repetition_truncation 映射);README v0.8.0 亮点表 P3 测试基建: - 新增 4 个测试文件:engine-stream-contract(6)、engine-stream-reliability(4:集成空闲超时/504 重试/思考中 abort/P4-2 强制收尾)、thinking-capability-gate(7)、pinned-proxy(9,含深校验 5)、session-trash(5,DB 域)、use-agent-stream hook 级(5)、agent.test 回放缓冲(2) - 契约更新:orchestrator 被中断 SubAgent success=false(abort 优先级修复语义)、SSE 空 error 帧、UTF-16 正常读取、DeepSeek 未配置思考显式 disabled、迁移矩阵 v2→3 - 弱断言根治:registry WEBP 单向断言、hooks-contracts 自比恒真、memory 空 token 补强 全量验证:typecheck 0 错误 / lint 0 问题 / 系统 Node 2144 通过(301 DB 用例按 ABI 跳过)/ Electron ABI 2445/2445 全量通过 0 跳过 --- README.md | 19 +- ...etonaAI-Desktop 内部API请求与响应标准.html | 36 +- electron-builder.yml | 4 + .../__tests__/deepseek-vision.test.ts | 6 +- .../__tests__/provider-request-shapes.test.ts | 7 +- .../adapters/__tests__/sse-stream.test.ts | 32 +- .../thinking-capability-gate.test.ts | 159 ++++++++ electron/harness/adapters/agnes-ai.adapter.ts | 16 +- .../harness/adapters/anthropic.adapter.ts | 47 ++- electron/harness/adapters/deepseek.adapter.ts | 23 +- electron/harness/adapters/mimo.adapter.ts | 20 +- electron/harness/adapters/ollama.adapter.ts | 79 +++- electron/harness/adapters/openai.adapter.ts | 17 +- .../harness/adapters/shared/sse-stream.ts | 33 +- .../__tests__/engine-stream-contract.test.ts | 230 ++++++++++++ .../engine-stream-reliability.test.ts | 243 ++++++++++++ electron/harness/agent-loop/engine.ts | 173 ++++++++- electron/harness/agent-loop/index.ts | 2 + electron/harness/agent-loop/types.ts | 7 + .../hooks/__tests__/hooks-contracts.test.ts | 4 +- .../memory/__tests__/memory-manager.test.ts | 2 + .../__tests__/orchestrator.test.ts | 20 +- .../harness/orchestration/orchestrator.ts | 22 ++ .../harness/tools/__tests__/registry.test.ts | 17 +- .../__tests__/filesystem-tools.test.ts | 19 +- .../tools/built-in/browser-window-manager.ts | 76 +++- .../harness/tools/built-in/code-search.ts | 97 ++++- .../harness/tools/built-in/delegate-task.ts | 11 +- .../harness/tools/built-in/file-editor.ts | 4 +- electron/harness/tools/built-in/filesystem.ts | 81 +++- electron/harness/tools/built-in/git.ts | 53 ++- .../harness/tools/built-in/http-request.ts | 6 +- .../harness/tools/built-in/network-utils.ts | 10 + electron/harness/tools/built-in/ssrf-guard.ts | 90 ++++- electron/harness/tools/built-in/web-fetch.ts | 71 +++- electron/harness/tools/built-in/web-search.ts | 101 ++++- electron/harness/types/metona-response.ts | 12 + electron/ipc/__tests__/agent.test.ts | 70 ++++ electron/ipc/agent.ts | 141 ++++++- electron/ipc/app.ts | 12 + electron/ipc/config.ts | 6 +- electron/ipc/mcp.ts | 8 + electron/ipc/sessions.ts | 90 +++++ electron/ipc/shared.ts | 23 +- electron/ipc/workspace.ts | 213 ++++++----- electron/main.ts | 38 ++ electron/preload.ts | 24 ++ .../__tests__/database-migration.test.ts | 17 +- .../services/__tests__/session-search.test.ts | 27 +- .../__tests__/session-summary.test.ts | 6 + .../services/__tests__/session-trash.test.ts | 157 ++++++++ .../__tests__/session-truncate-trace.test.ts | 1 + electron/services/database.service.ts | 20 +- electron/services/mcp-manager.service.ts | 70 ++++ electron/services/session.service.ts | 97 ++++- electron/services/update.service.ts | 122 +++++- electron/services/window-manager.service.ts | 52 +++ electron/services/workspace.service.ts | 112 +++++- electron/utils/__tests__/pinned-proxy.test.ts | 194 ++++++++++ electron/utils/network-proxy.ts | 21 +- electron/utils/pinned-proxy.ts | 267 +++++++++++++ package-lock.json | 117 +++++- package.json | 3 +- src/__tests__/use-agent-stream.test.tsx | 129 +++++++ src/components/chat/ChatInput.tsx | 212 ++++++++++- src/components/layout/DetailPanel.tsx | 7 + src/components/layout/Sidebar.tsx | 171 +++++++++ src/components/memory/MemoryViewer.tsx | 2 +- src/components/settings/LLMSettings.tsx | 76 +++- src/components/settings/LogsSettings.tsx | 105 ++++++ src/components/settings/MCPSettings.tsx | 199 ++++++---- src/components/trace/SessionReplayPlayer.tsx | 353 ++++++++++++++++++ src/hooks/useAgentStream.ts | 119 +++++- src/lib/i18n-strings.ts | 126 +++++++ src/lib/model-capabilities.ts | 31 ++ src/lib/stream-event-bus.ts | 46 +++ src/stores/agent-store.ts | 121 +++++- src/types/global.d.ts | 85 ++++- 78 files changed, 5128 insertions(+), 411 deletions(-) create mode 100644 electron/harness/adapters/__tests__/thinking-capability-gate.test.ts create mode 100644 electron/harness/agent-loop/__tests__/engine-stream-contract.test.ts create mode 100644 electron/harness/agent-loop/__tests__/engine-stream-reliability.test.ts create mode 100644 electron/services/__tests__/session-trash.test.ts create mode 100644 electron/utils/__tests__/pinned-proxy.test.ts create mode 100644 electron/utils/pinned-proxy.ts create mode 100644 src/__tests__/use-agent-stream.test.tsx create mode 100644 src/components/trace/SessionReplayPlayer.tsx create mode 100644 src/lib/stream-event-bus.ts diff --git a/README.md b/README.md index 0369615..2ecabbc 100644 --- a/README.md +++ b/README.md @@ -10,7 +10,7 @@

- Version + Version License Electron React @@ -54,6 +54,23 @@ --- +## 🆕 v0.8.0 更新亮点 + +| 特性 | 说明 | +|:---|:---| +| 🔒 **流结束契约(根治"思考中会话停止")** | finish_reason 全链路贯通:`length` 截断不再被静默判定为完成;零产出流自动重试;思考耗尽输出预算时自动降级(关闭思考)重试一次,仍失败则给出 OUTPUT_LENGTH_EXCEEDED 结构化错误与修复建议 | +| 🧠 **思考参数 × 模型能力门控** | `supportsThinking:false` 的模型(如 deepseek-v4-flash-vision-exp)不再发送思考参数,防止思考烧光输出预算;设置页同步提示 | +| 🪟 **后台会话回放缓冲** | 切走会话后其运行内容不再丢失,切回时自动恢复;abort 失败自愈、收尾兜底、中断卡片清扫三重防线 | +| 🗑️ **会话回收站** | 删除改为软删除,30 天自动清理;支持恢复与彻底删除 | +| 🎬 **会话回放播放器** | TRACE JSONL 录制可视化回放(时间轴 / 步进 / 变速) | +| 🛡️ **浏览器通道 SSRF 根治** | Agent 浏览器分区流量经本地 Pinned CONNECT 代理(校验期 IP pinning),关闭 DNS rebinding 残余窗口;配置类 URL 增加域名解析深校验 | +| 📦 **electron-updater 自动更新** | 生产环境启动静默检查、可用即通知,一键下载安装 | +| ✏️ **@ 文件提及** | 输入 `@` 联想工作空间文件,发送时以附件管线注入文本片段(512KB 上限 + 二进制拒绝) | +| 🔍 **MCP Resources/Prompts 发现** | 连接后自动发现并列出 Server 的资源与提示词(可选能力,失败静默降级) | +| ⚡ **工具中断全覆盖** | web_search / web_fetch / http_request / code_search / git 系列 / delegate_task 全部接入引擎 abort 信号;web_search 自动抓取增加时间预算收敛 | + +--- + ## ✨ 核心亮点 diff --git a/docs/MetonaAI-Desktop 内部API请求与响应标准.html b/docs/MetonaAI-Desktop 内部API请求与响应标准.html index 33b1d3a..ef99c06 100644 --- a/docs/MetonaAI-Desktop 内部API请求与响应标准.html +++ b/docs/MetonaAI-Desktop 内部API请求与响应标准.html @@ -639,8 +639,8 @@ REASONING_DELTA = 'reasoning_delta', // 推理内容增量TOOL_CALL_DELTA = 'tool_call_delta', // 工具调用增量TOOL_CALL_COMPLETE = 'tool_call_complete', - THINKING_START = 'thinking_start', // 思考开始 - THINKING_END = 'thinking_end', // 思考结束 + // v0.7.2 注:THINKING_START/THINKING_END 已移除(全链路零发送方零消费者) + STREAM_RESET = 'stream_reset', // v0.7.4: 重试/故障转移后清空前端增量缓冲ERROR = 'error', // 流中错误DONE = 'done', // 流结束USAGE = 'usage', // Token 统计(通常在 DONE 前) @@ -663,6 +663,9 @@ }; // TOOL_CALL_DELTAtoolCall?: MetonaToolCall; // TOOL_CALL_COMPLETE(拼接完成后的完整调用)usage?: MetonaTokenUsage; // USAGE + // v0.8.0: DONE 事件新增 finishReason(归一化 OpenAI 语义:stop/length/ + // tool_calls/content_filter;MiMo repetition_truncation→stop)。引擎据此 + // 区分自然完成与输出上限截断(空响应守卫 + 降级重试)。error?: MetonaError; // ERROR } @@ -1000,16 +1003,22 @@ - + - + - + - + + + + + + +
Ollama 原生 JSON / NDJSON
AnthropicAdapter(未来计划,未实现)AnthropicAdapter(v0.8.0 注:已实现) anthropic https://api.anthropic.com/v1/messagesAnthropic 原生 JSONAnthropic 原生 JSON(message_start/content_block_delta/message_delta 事件机)
OpenAIAdapter(未来计划,未实现)OpenAIAdapter(v0.8.0 注:已实现) openai https://api.openai.com/v1/chat/completionsOpenAI 原生 JSONOpenAI 原生 JSON(o 系列走 max_completion_tokens + reasoning_effort)
MimoAdapter(v0.8.0 注:文档此前遗漏,已实现)mimohttps://api.xiaomimimo.com/v1/chat/completionsOpenAI 兼容 JSON(max_completion_tokens 含推理 token;finish_reason 含 repetition_truncation)
@@ -1051,14 +1060,17 @@

MVP 优先级

-

当前优先实现以下 3 个 Adapter,Anthropic 和 OpenAI 为未来计划:

+

v0.8.0 注:六家 Adapter(DeepSeek / Agnes / MiMo / Ollama / OpenAI / Anthropic)均已实现。 + 流式 DONE 事件自 v0.8.0 起携带归一化 finishReason(stop / length / tool_calls / content_filter), + MiMo 的 repetition_truncation 映射为 stop;引擎以 finishReason=length 驱动空响应守卫与降级重试。

- - - - - + + + + + +
优先级Adapter状态
P0DeepSeekAdapterMVP 必须实现
P0AgnesAdapterMVP 必须实现
P1OllamaAdapterMVP 必须实现(本地推理)
P2AnthropicAdapter未来计划
P2OpenAIAdapter未来计划
P0DeepSeekAdapter已实现
P0AgnesAdapter已实现
P1OllamaAdapter已实现(本地推理)
P2AnthropicAdapter已实现
P2OpenAIAdapter已实现
P2MimoAdapter已实现

Provider 故障转移策略

diff --git a/electron-builder.yml b/electron-builder.yml index 41acd8a..ee3ad65 100644 --- a/electron-builder.yml +++ b/electron-builder.yml @@ -43,5 +43,9 @@ "allowToChangeInstallationDirectory": true, "createDesktopShortcut": true, "createStartMenuShortcut": true + }, + "publish": { + "provider": "generic", + "url": "https://git.metona.cn/MetonaTeam/metona-updates/latest/" } } diff --git a/electron/harness/adapters/__tests__/deepseek-vision.test.ts b/electron/harness/adapters/__tests__/deepseek-vision.test.ts index 264e520..20764aa 100644 --- a/electron/harness/adapters/__tests__/deepseek-vision.test.ts +++ b/electron/harness/adapters/__tests__/deepseek-vision.test.ts @@ -238,8 +238,10 @@ describe('DeepSeek vision 模型多模态请求格式(v0.5.4)', () => { } as MetonaRequest); const body = requestBody(); - // vision 模型 supportsThinking=false,但参数仍透传 thinking enabled(引擎决定) - expect(body.thinking).toEqual({ type: 'enabled' }); + // v0.8.0 P0-3 能力门控: vision 模型 supportsThinking=false → 显式 disabled, + // 且不发送 reasoning_effort(思考会耗尽该模型 8192 输出预算 —— 生产事故根因) + expect(body.thinking).toEqual({ type: 'disabled' }); + expect(body.reasoning_effort).toBeUndefined(); }); it('vision 模型 messages 数组首位始终为 system 消息', async () => { diff --git a/electron/harness/adapters/__tests__/provider-request-shapes.test.ts b/electron/harness/adapters/__tests__/provider-request-shapes.test.ts index d788e9f..fa5a969 100644 --- a/electron/harness/adapters/__tests__/provider-request-shapes.test.ts +++ b/electron/harness/adapters/__tests__/provider-request-shapes.test.ts @@ -757,11 +757,14 @@ describe('DeepSeekAdapter — thinking 映射矩阵', () => { expect(bodies[0].reasoning_effort).toBeUndefined(); }); - it('thinkingEnabled 未配置 → 不发送 thinking 字段', async () => { + it('thinkingEnabled 未配置 → 显式 disabled(v0.8.0 P0-3 确定性契约)', async () => { + // v0.8.0 P0-3 契约变更: DeepSeek 服务端默认 thinking.enabled,未配置即发请求 + // 会得到隐式思考 —— 现在未配置一律显式 disabled,行为不依赖服务端隐式默认 + //(与 Agnes/MiMo 的显式口径对齐;引擎路径恒传布尔值,此处为兜底确定性)。 const adapter = makeAdapter(); const { bodies } = captureFetch(); await adapter.send(makeRequest({ params: { maxTokens: 4096, temperature: 0, stream: false } })); - expect(bodies[0].thinking).toBeUndefined(); + expect(bodies[0].thinking).toEqual({ type: 'disabled' }); }); it('effort 未配置缺省 high → reasoning_effort=high', async () => { diff --git a/electron/harness/adapters/__tests__/sse-stream.test.ts b/electron/harness/adapters/__tests__/sse-stream.test.ts index 45f5e70..ac411f9 100644 --- a/electron/harness/adapters/__tests__/sse-stream.test.ts +++ b/electron/harness/adapters/__tests__/sse-stream.test.ts @@ -11,7 +11,7 @@ import { readStreamChunkWithIdleTimeout, SseUpstreamError, } from '../shared/sse-stream'; -import { MetonaStreamEventType } from '../../types'; +import { MetonaStreamEventType, type MetonaStreamEvent } from '../../types'; /** 构造 SSE 测试流 */ function makeStream(chunks: string[]): ReadableStream { @@ -586,13 +586,33 @@ describe('parseSSEStream — 上游错误帧 status 推断矩阵', () => { expect((err as SseUpstreamError).status).toBe(503); }); - it('error 为空对象 → 回退 message 为 {}(JSON.stringify 兜底使其仍被识别为错误帧)', async () => { - // 源码注明的防御意图是无害空对象不报错,但 message 兜底取 JSON.stringify({})='{}', - // 恒真值使该防御分支失效 —— 此处锁定实际行为并留档(见报告)。 - const err = await collectError({ error: {} }); + it('error 为空对象 → 不构成错误帧(v0.8.0 P1-3.1 防御分支根治后契约)', async () => { + // v0.8.0 P1-3.1 根治: 旧实现 message 兜底取 JSON.stringify({})='{}' 恒真值, + // `{"error":{}}` 被误判为错误帧抛 SseUpstreamError 并触发不必要的重试。 + // 新契约:无 message 且无 status/code 标识的空对象为无害帧,静默跳过。 + const events: MetonaStreamEvent[] = []; + let error: Error | null = null; + try { + for await (const ev of parseSSEStream( + makeStream(['data: {"error":{}}\n\n', 'data: [DONE]\n\n']), + 'r', + 's', + 1, + )) { + events.push(ev); + } + } catch (e) { + error = e as Error; + } + expect(error).toBeNull(); + expect(events.some((ev) => ev.type === MetonaStreamEventType.DONE)).toBe(true); + }); + + it('error 有 code 无 message → 仍构成错误帧(标识存在,占位文案兜底)', async () => { + const err = await collectError({ error: { code: 'rate_limit_exceeded' } }); expect(err).toBeInstanceOf(SseUpstreamError); expect((err as Error).message).toContain('upstream_error'); - expect((err as SseUpstreamError).status).toBeUndefined(); + expect((err as SseUpstreamError).status).toBe(429); }); it('error 为空白字符串 → 不算错误帧', async () => { diff --git a/electron/harness/adapters/__tests__/thinking-capability-gate.test.ts b/electron/harness/adapters/__tests__/thinking-capability-gate.test.ts new file mode 100644 index 0000000..ae97ddc --- /dev/null +++ b/electron/harness/adapters/__tests__/thinking-capability-gate.test.ts @@ -0,0 +1,159 @@ +/** + * v0.8.0 P0-3: 思考参数 × 模型能力 门控矩阵。 + * + * 根因回顾:MODEL_INFO 标注 supportsThinking:false 的模型(如 + * deepseek-v4-flash-vision-exp)此前仍被发送 thinking 参数 —— 思考耗尽输出 + * 预算(8192 上限)导致 finish_reason=length 空回复、会话静默停止。 + * 本文件钉住四家 Provider 的能力门控行为与输出预算告警前置条件。 + */ + +import { describe, it, expect } from 'vitest'; +import { DeepSeekAdapter } from '../deepseek.adapter'; +import { AgnesAdapter } from '../agnes-ai.adapter'; +import { MimoAdapter } from '../mimo.adapter'; +import { OllamaAdapter } from '../ollama.adapter'; +import type { MetonaRequest } from '../../types'; + +const SYSTEM_PROMPT = { + roleDefinition: 'test', + outputConstraints: '', + safetyGuidelines: '', +}; + +function makeRequest(overrides?: Partial): MetonaRequest { + return { + meta: { + sessionId: 's', + iteration: 1, + requestId: 'r', + timestamp: Date.now(), + agentVersion: '1.0.0', + }, + systemPrompt: SYSTEM_PROMPT, + messages: [{ role: 'user', content: 'hi', timestamp: Date.now() }], + params: { + maxTokens: 63488, + temperature: 0, + stream: true, + thinkingEnabled: true, + thinkingEffort: 'max', + ...overrides, + }, + }; +} + +function asNative( + adapter: unknown, +): (req: MetonaRequest, stream: boolean) => Record { + return ( + adapter as { toNativeRequest: (r: MetonaRequest, s: boolean) => Record } + ).toNativeRequest.bind(adapter); +} + +describe('P0-3 thinking capability gate', () => { + it('DeepSeek: vision-exp (supportsThinking:false) → thinking disabled, no reasoning_effort, max_tokens clamped to 8192', async () => { + const adapter = new DeepSeekAdapter({ + provider: 'deepseek', + baseURL: 'https://api.deepseek.com', + apiKey: 'k', + defaultModel: 'deepseek-v4-flash-vision-exp', + }); + const body = asNative(adapter)(makeRequest(), false); + expect(body.thinking).toEqual({ type: 'disabled' }); + expect(body.reasoning_effort).toBeUndefined(); + expect(body.max_tokens).toBe(8192); + }); + + it('DeepSeek: pro (supportsThinking:true) → thinking enabled + reasoning_effort mapped (max→max)', async () => { + const adapter = new DeepSeekAdapter({ + provider: 'deepseek', + baseURL: 'https://api.deepseek.com', + apiKey: 'k', + defaultModel: 'deepseek-v4-pro', + }); + const body = asNative(adapter)(makeRequest(), false); + expect(body.thinking).toEqual({ type: 'enabled' }); + expect(body.reasoning_effort).toBe('max'); + }); + + it('Agnes: supportsThinking:false model → enable_thinking:false', async () => { + const adapter = new AgnesAdapter({ + provider: 'agnes', + baseURL: 'https://apihub.agnes-ai.com/v1', + apiKey: 'k', + defaultModel: 'agnes-2.0-flash', + }); + // 临时改写元信息模拟"不支持思考"的模型,结束后恢复 + const table = AgnesAdapter['MODEL_INFO'] as Record; + const original = table['agnes-2.0-flash'].supportsThinking; + table['agnes-2.0-flash'].supportsThinking = false; + try { + const body = asNative(adapter)(makeRequest(), false); + expect(body.chat_template_kwargs).toEqual({ enable_thinking: false }); + } finally { + table['agnes-2.0-flash'].supportsThinking = original; + } + }); + + it('Agnes: supportsThinking:true model → enable_thinking:true', async () => { + const adapter = new AgnesAdapter({ + provider: 'agnes', + baseURL: 'https://apihub.agnes-ai.com/v1', + apiKey: 'k', + defaultModel: 'agnes-2.0-flash', + }); + const body = asNative(adapter)(makeRequest(), false); + expect(body.chat_template_kwargs).toEqual({ enable_thinking: true }); + }); + + it('MiMo: supportsThinking:false model → thinking disabled + temperature passthrough', async () => { + const adapter = new MimoAdapter({ + provider: 'mimo', + baseURL: 'https://api.xiaomimimo.com/v1', + apiKey: 'k', + defaultModel: 'mimo-v2.5-pro', + }); + const table = MimoAdapter['MODEL_INFO'] as Record; + const original = table['mimo-v2.5-pro'].supportsThinking; + table['mimo-v2.5-pro'].supportsThinking = false; + try { + const body = asNative(adapter)(makeRequest(), false); + expect(body.thinking).toEqual({ type: 'disabled' }); + expect(body.temperature).toBe(0); + } finally { + table['mimo-v2.5-pro'].supportsThinking = original; + } + }); + + it('Ollama: probed no-thinking (cachedThinkingSupport=false) → no think parameter', async () => { + const adapter = new OllamaAdapter({ + provider: 'ollama', + baseURL: 'http://localhost:11434', + apiKey: '', + defaultModel: 'qwen3:latest', + }); + (adapter as unknown as { cachedThinkingSupport: boolean | null }).cachedThinkingSupport = false; + const body = await ( + adapter as unknown as { + toNativeRequest: (r: MetonaRequest) => Promise>; + } + ).toNativeRequest(makeRequest()); + expect(body.think).toBeUndefined(); + }); + + it('Ollama: probe unknown (null) fails open → think parameter present', async () => { + const adapter = new OllamaAdapter({ + provider: 'ollama', + baseURL: 'http://localhost:11434', + apiKey: '', + defaultModel: 'qwen3:latest', + }); + (adapter as unknown as { cachedThinkingSupport: boolean | null }).cachedThinkingSupport = null; + const body = await ( + adapter as unknown as { + toNativeRequest: (r: MetonaRequest) => Promise>; + } + ).toNativeRequest(makeRequest()); + expect(body.think).toBe(true); // effort=max → true + }); +}); diff --git a/electron/harness/adapters/agnes-ai.adapter.ts b/electron/harness/adapters/agnes-ai.adapter.ts index 74889e5..e9d73ce 100644 --- a/electron/harness/adapters/agnes-ai.adapter.ts +++ b/electron/harness/adapters/agnes-ai.adapter.ts @@ -56,7 +56,10 @@ export class AgnesAdapter extends OpenAICompatibleAdapter { // ========== 协议参数映射(Agnes 差异点) ========== - protected override toNativeRequest(request: MetonaRequest, stream: boolean): Record { + protected override toNativeRequest( + request: MetonaRequest, + stream: boolean, + ): Record { // v0.6.2: images 处理收敛至共享层(原索引对齐循环在孤立 tool 过滤后会错位) const messages = buildOpenAICompatibleMessages(request, true); const tools = buildOpenAICompatibleTools(request.tools); @@ -96,12 +99,21 @@ export class AgnesAdapter extends OpenAICompatibleAdapter { // 显式发送 enable_thinking:false —— 原实现只在 thinkingEnabled===true 时写该字段, // 若服务端默认开启思考,客户端没有任何路径把它关掉(DeepSeek/MiMo 均显式发送 // disabled 保持对称,唯独此处漏了)。 + // v0.8.0 P0-3: 模型能力门控 —— supportsThinking===false 的模型强制关闭思考 { const effort = request.params.thinkingEffort ?? 'high'; + const modelThinkingSupported = + AgnesAdapter.MODEL_INFO[this.config.defaultModel]?.supportsThinking !== false; // 未配置 thinkingEnabled 一律显式关闭 —— 与 DeepSeek/MiMo 的"服务端默认开启, // 必须显式发送 disabled"口径对齐,让行为确定性不依赖服务端隐式默认。 - const wantThinking = request.params.thinkingEnabled === true && effort !== 'low'; + const wantThinking = + request.params.thinkingEnabled === true && effort !== 'low' && modelThinkingSupported; body.chat_template_kwargs = { enable_thinking: wantThinking }; + if (request.params.thinkingEnabled === true && !modelThinkingSupported) { + log.warn( + `[Agnes] model "${this.config.defaultModel}" does not support thinking — sending enable_thinking:false (P0-3 capability gate)`, + ); + } } // 停止序列 diff --git a/electron/harness/adapters/anthropic.adapter.ts b/electron/harness/adapters/anthropic.adapter.ts index 77641fa..686ed59 100644 --- a/electron/harness/adapters/anthropic.adapter.ts +++ b/electron/harness/adapters/anthropic.adapter.ts @@ -24,6 +24,28 @@ import type { MetonaRequest, MetonaResponse, MetonaStreamEvent } from '../types' import { MetonaFinishReason, MetonaStreamEventType } from '../types'; import type { MetonaModelInfo } from '../types/metona-adapter'; +/** + * v0.8.0 P0-1: Anthropic stop_reason → 归一化 OpenAI 语义(与 MetonaFinishReason + * 的非流式映射语义一致)。pause_turn(长回复暂停续传标记)视为自然停止。 + */ +function mapAnthropicStopReason(reason: string): string { + switch (reason) { + case 'max_tokens': + return 'length'; + case 'tool_use': + return 'tool_calls'; + case 'refusal': + case 'content_filter': + return 'content_filter'; + case 'end_turn': + case 'pause_turn': + case 'stop_sequence': + return 'stop'; + default: + return 'stop'; + } +} + export class AnthropicAdapter extends BaseAdapter { override readonly providerId: string = 'anthropic'; readonly supportedModels = ['claude-sonnet-4-5', 'claude-opus-4-1', 'claude-haiku-4-5']; @@ -116,6 +138,10 @@ export class AnthropicAdapter extends BaseAdapter { let buffer = ''; let eventName = ''; let streamEndedNormally = false; + // v0.8.0 P0-1: 采集 message_delta.delta.stop_reason —— Anthropic 的停止原因 + // 在 message_delta(而非 message_stop)事件携带;旧实现只读 usage, + // max_tokens 截断在流式路径完全不可见(与 OpenAI 共享层 finish_reason 缺口同源) + let streamStopReason: string | undefined; // 工具调用缓冲:content block index → { id, name, argsBuffer } const toolBlocks = new Map(); @@ -256,11 +282,22 @@ export class AnthropicAdapter extends BaseAdapter { }, }); } + // v0.8.0 P0-1: 采集停止原因(message_delta.delta.stop_reason,可能在 + // 多个 message_delta 中重复出现,取任一即可;归一化为 OpenAI 语义) + const delta = data.delta as Record | undefined; + if (delta && typeof delta.stop_reason === 'string') { + streamStopReason = mapAnthropicStopReason(delta.stop_reason); + } break; } case 'message_stop': { streamEndedNormally = true; - events.push({ type: MetonaStreamEventType.DONE, ...base() }); + events.push({ + type: MetonaStreamEventType.DONE, + ...base(), + // v0.8.0 P0-1: 携带归一化停止原因 + ...(streamStopReason ? { finishReason: streamStopReason } : {}), + }); break; } case 'error': { @@ -362,7 +399,13 @@ export class AnthropicAdapter extends BaseAdapter { log.warn('[Anthropic] Stream ended without message_stop (connection likely dropped)'); } toolBlocks.clear(); - yield { type: MetonaStreamEventType.DONE, ...base() }; + yield { + type: MetonaStreamEventType.DONE, + ...base(), + // v0.8.0 P0-1: 断流合成路径同样携带已观察到的停止原因 + //(断流时多为 undefined —— 引擎据此走空响应守卫/重试而非误判自然结束) + ...(streamStopReason ? { finishReason: streamStopReason } : {}), + }; } } diff --git a/electron/harness/adapters/deepseek.adapter.ts b/electron/harness/adapters/deepseek.adapter.ts index 7298e5d..e140bc8 100644 --- a/electron/harness/adapters/deepseek.adapter.ts +++ b/electron/harness/adapters/deepseek.adapter.ts @@ -164,7 +164,10 @@ export class DeepSeekAdapter extends OpenAICompatibleAdapter { // ========== 协议参数映射(DeepSeek 差异点) ========== - protected override toNativeRequest(request: MetonaRequest, stream: boolean): Record { + protected override toNativeRequest( + request: MetonaRequest, + stream: boolean, + ): Record { // v0.6.2: images 处理收敛至共享层(includeImages = vision 模型才转换, // 非 vision 静默丢弃——正确行为,见 openai-format.ts #27 记录) const messages = buildOpenAICompatibleMessages(request, this.isVisionModel()); @@ -200,10 +203,17 @@ export class DeepSeekAdapter extends OpenAICompatibleAdapter { } // Thinking 模式 + // v0.8.0 P0-3 根治: 模型能力门控 —— MODEL_INFO.supportsThinking === false 的 + // 模型(如 deepseek-v4-flash-vision-exp)一律不发思考参数并显式 disabled, + // 防止思考消耗输出预算(生产事故:vision-exp 8192 输出预算被 max 档思考 + // 全部烧光 → finish_reason=length 空回复 → 会话静默停止)。 + const modelThinkingSupported = + DeepSeekAdapter.MODEL_INFO[this.config.defaultModel]?.supportsThinking !== false; + const wantThinking = request.params.thinkingEnabled === true && modelThinkingSupported; // API 默认 thinking.type = "enabled",必须显式发送 disabled 才能关闭 - if (request.params.thinkingEnabled === false) { + if (!wantThinking) { body.thinking = { type: 'disabled' }; - } else if (request.params.thinkingEnabled) { + } else { body.thinking = { type: 'enabled' }; const effortMap: Record = { low: 'high', @@ -213,6 +223,13 @@ export class DeepSeekAdapter extends OpenAICompatibleAdapter { }; // DeepSeek API 仅支持 high / max 两档,low/medium 映射为 high body.reasoning_effort = effortMap[request.params.thinkingEffort ?? 'high'] ?? 'high'; + // v0.8.0 P0-3: 思考会占用输出预算 —— 钳制后预算过小时显式告警 + //(思考 token 计入 max_tokens,预算过小会出现"思考耗尽正文为零"截断) + if (maxTokens < 8192) { + log.warn( + `[DeepSeek] thinking enabled with small output budget (${maxTokens} tokens after model clamp) — reasoning may consume the entire budget and truncate the answer`, + ); + } } // 停止序列 diff --git a/electron/harness/adapters/mimo.adapter.ts b/electron/harness/adapters/mimo.adapter.ts index 1a9acb1..09cd972 100644 --- a/electron/harness/adapters/mimo.adapter.ts +++ b/electron/harness/adapters/mimo.adapter.ts @@ -11,6 +11,7 @@ * @see apis/mimo-api-docs-20260715.html */ +import log from 'electron-log'; import type { MetonaRequest } from '../types'; import type { MetonaModelInfo } from '../types/metona-adapter'; import { buildOpenAICompatibleMessages, buildOpenAICompatibleTools } from './shared/openai-format'; @@ -72,7 +73,10 @@ export class MimoAdapter extends OpenAICompatibleAdapter { // ========== 协议参数映射(MiMo 差异点) ========== - protected override toNativeRequest(request: MetonaRequest, stream: boolean): Record { + protected override toNativeRequest( + request: MetonaRequest, + stream: boolean, + ): Record { // v0.6.2: images 处理收敛至共享层(原索引对齐循环在孤立 tool 过滤后会错位) const messages = buildOpenAICompatibleMessages(request, true); const tools = buildOpenAICompatibleTools(request.tools); @@ -122,7 +126,12 @@ export class MimoAdapter extends OpenAICompatibleAdapter { // Thinking 模式(与 DeepSeek 参数结构一致) // MiMo API 默认 thinking.type = "enabled",必须显式发送 disabled 才能关闭 - if (request.params.thinkingEnabled === false) { + // v0.8.0 P0-3: 模型能力门控 —— supportsThinking===false 的模型强制 disabled + //(思考 token 计入 max_completion_tokens,能力不符的模型上会耗尽输出预算) + const modelThinkingSupported = + MimoAdapter.MODEL_INFO[this.config.defaultModel]?.supportsThinking !== false; + const wantThinking = request.params.thinkingEnabled !== false && modelThinkingSupported; + if (!wantThinking) { // 显式禁用思考:传 disabled + temperature/top_p(非思考模式下这两个参数有效) body.thinking = { type: 'disabled' }; body.temperature = request.params.temperature; @@ -131,6 +140,13 @@ export class MimoAdapter extends OpenAICompatibleAdapter { // 启用思考(包括 undefined,因为 MiMo 默认 enabled) // 思考模式下 temperature/top_p 被 API 强制覆盖为 1.0/0.95,不传 body.thinking = { type: 'enabled' }; + // v0.8.0 P0-3: 思考占用输出预算 —— 钳制后预算过小时显式告警 + const effectiveMax = body.max_completion_tokens as number; + if (typeof effectiveMax === 'number' && effectiveMax < 8192) { + log.warn( + `[MiMo] thinking enabled with small output budget (${effectiveMax} tokens after model clamp) — reasoning may consume the entire budget and truncate the answer`, + ); + } } // 停止序列 diff --git a/electron/harness/adapters/ollama.adapter.ts b/electron/harness/adapters/ollama.adapter.ts index ff0d0ce..67f8e84 100644 --- a/electron/harness/adapters/ollama.adapter.ts +++ b/electron/harness/adapters/ollama.adapter.ts @@ -192,6 +192,14 @@ export class OllamaAdapter extends BaseAdapter { // 流结束 if (chunk.done) { streamEndedNormally = true; + // v0.8.0 P0-1: 采集 done_reason —— Ollama 的停止原因在最终 done chunk + // 携带;旧实现流式路径完全忽略(length 截断不可见),现归一化后随 + // DONE 事件上交引擎 + const doneReason = mapOllamaDoneReason( + chunk.done_reason as string | undefined, + Array.isArray(chunk.message?.tool_calls) && chunk.message.tool_calls.length > 0, + ); + const finishReason = doneReason === MetonaFinishReason.LENGTH ? 'length' : undefined; // 发送 usage 信息 yield { type: MetonaStreamEventType.USAGE, @@ -214,6 +222,9 @@ export class OllamaAdapter extends BaseAdapter { iteration: request.meta.iteration, seq: seq++, timestamp: Date.now(), + // v0.8.0 P0-1: 仅 length 需要显式上交(stop/tool_calls 语义由 + // 事件流本身表达;load/unload 属本地引擎状态非响应语义) + ...(finishReason ? { finishReason } : {}), }; return; } @@ -252,6 +263,8 @@ export class OllamaAdapter extends BaseAdapter { format?: string | object; images?: string[]; options?: Record; + /** v0.8.0 P1-3.2: 外部取消信号(与固定 300s 超时合并,任一触发即中止) */ + signal?: AbortSignal; }): Promise<{ response: string; thinking?: string; @@ -262,8 +275,13 @@ export class OllamaAdapter extends BaseAdapter { const response = await fetch(`${this.baseURL}/api/generate`, { method: 'POST', headers: { 'Content-Type': 'application/json' }, - body: JSON.stringify({ ...params, stream: false }), - signal: AbortSignal.timeout(300_000), + body: JSON.stringify({ ...params, stream: false, signal: undefined }), + // v0.8.0 P1-3.2 根治: 旧实现固定 AbortSignal.timeout(300_000) —— 外部中断 + // 无法取消该请求(最长 5 分钟资源悬挂)。现用 AbortSignal.any 合并外部 + // signal 与超时信号,任一触发即中止(Node ≥20.3 / Electron 35 满足)。 + signal: params.signal + ? AbortSignal.any([params.signal, AbortSignal.timeout(300_000)]) + : AbortSignal.timeout(300_000), }); if (!response.ok) throw new Error(`Ollama generate error: ${response.status}`); @@ -290,12 +308,17 @@ export class OllamaAdapter extends BaseAdapter { model: string; input: string | string[]; dimensions?: number; + /** v0.8.0 P1-3.2: 外部取消信号(与固定 60s 超时合并) */ + signal?: AbortSignal; }): Promise<{ embeddings: number[][]; totalDuration: number }> { const response = await fetch(`${this.baseURL}/api/embed`, { method: 'POST', headers: { 'Content-Type': 'application/json' }, body: JSON.stringify(params), - signal: AbortSignal.timeout(60_000), + // v0.8.0 P1-3.2: 与 generate 同口径 —— 外部 signal 与超时合并 + signal: params.signal + ? AbortSignal.any([params.signal, AbortSignal.timeout(60_000)]) + : AbortSignal.timeout(60_000), }); if (!response.ok) throw new Error(`Ollama embed error: ${response.status}`); @@ -381,9 +404,18 @@ export class OllamaAdapter extends BaseAdapter { */ private cachedContextWindow: number | null = null; private refreshingContextWindow = false; + /** v0.8.0 P0-3: /api/show 探测只发一次(构造函数发起),失败不重试(fail-open) */ + private probeAttempted = false; + /** + * v0.8.0 P0-3: /api/show capabilities 探测缓存 —— 模型是否支持思考。 + * null = 未探测/探测失败(fail-open 放行,与 listModels 能力回退策略一致); + * false = 服务端明确不支持 → toNativeRequest 不发 think 参数。 + */ + private cachedThinkingSupport: boolean | null = null; private refreshContextWindow(): void { - if (this.refreshingContextWindow) return; + if (this.refreshingContextWindow || this.probeAttempted) return; + this.probeAttempted = true; this.refreshingContextWindow = true; void this.showModel(this.config.defaultModel) .then((info) => { @@ -396,6 +428,10 @@ export class OllamaAdapter extends BaseAdapter { log.info(`[Ollama] Context window (num_ctx) detected: ${value}`); } } + // v0.8.0 P0-3: 同一次探测顺带缓存思考能力(供 toNativeRequest 同步门控) + if (Array.isArray(info.capabilities)) { + this.cachedThinkingSupport = info.capabilities.map(String).includes('thinking'); + } }) .catch(() => { /* 模型探测失败不阻塞对话 */ @@ -573,6 +609,9 @@ export class OllamaAdapter extends BaseAdapter { } private async toNativeRequest(request: MetonaRequest): Promise> { + // 说明:能力/num_ctx 探测由构造函数 fire-and-forget 发起(probeAttempts 上限 + // 1 次)—— 此处不再重复探测,避免每次请求都打 /api/show(也保持测试环境的 + // fetch 捕获不受污染)。探测失败时 cachedThinkingSupport=null → 门控 fail-open。 const messages: Record[] = [ { role: 'system', @@ -662,14 +701,32 @@ export class OllamaAdapter extends BaseAdapter { } // Thinking 模式 + // v0.8.0 P0-3: 模型能力门控 —— /api/show capabilities 探测为不支持思考 + //(cachedThinkingSupport === false)时不发 think 参数(Ollama 服务端默认关闭); + // 未探测/探测失败(null)fail-open 放行,与 listModels 能力回退策略一致。 + // 探测在适配器实例创建时 fire-and-forget 发起(refreshContextWindow)。 if (request.params.thinkingEnabled) { - const effortMap: Record = { - low: 'low', - medium: 'medium', - high: 'high', - max: true, - }; - body.think = effortMap[request.params.thinkingEffort ?? 'high'] ?? true; + const modelThinkingSupported = this.cachedThinkingSupport !== false; + if (modelThinkingSupported) { + const effortMap: Record = { + low: 'low', + medium: 'medium', + high: 'high', + max: true, + }; + body.think = effortMap[request.params.thinkingEffort ?? 'high'] ?? true; + // v0.8.0 P0-3: 思考占用 num_predict 输出预算 —— 预算过小时显式告警 + const numPredict = request.params.maxTokens; + if (typeof numPredict === 'number' && numPredict < 8192) { + log.warn( + `[Ollama] thinking enabled with small num_predict budget (${numPredict}) — reasoning may consume the entire budget and truncate the answer`, + ); + } + } else { + log.warn( + `[Ollama] model "${this.config.defaultModel}" does not support thinking (per /api/show capabilities) — omitting think parameter (P0-3 capability gate)`, + ); + } } return body; diff --git a/electron/harness/adapters/openai.adapter.ts b/electron/harness/adapters/openai.adapter.ts index db627d8..738bec5 100644 --- a/electron/harness/adapters/openai.adapter.ts +++ b/electron/harness/adapters/openai.adapter.ts @@ -12,13 +12,11 @@ * @see https://platform.openai.com/docs/api-reference/chat */ +import log from 'electron-log'; import type { MetonaRequest } from '../types'; import type { MetonaModelInfo } from '../types/metona-adapter'; import { buildOpenAICompatibleMessages, buildOpenAICompatibleTools } from './shared/openai-format'; -import { - ModelCapabilityError, - OpenAICompatibleAdapter, -} from './shared/openai-compatible-base'; +import { ModelCapabilityError, OpenAICompatibleAdapter } from './shared/openai-compatible-base'; export class OpenAIAdapter extends OpenAICompatibleAdapter { override readonly providerId: string = 'openai'; @@ -110,7 +108,10 @@ export class OpenAIAdapter extends OpenAICompatibleAdapter { // ========== 协议参数映射(OpenAI 差异点) ========== - protected override toNativeRequest(request: MetonaRequest, stream: boolean): Record { + protected override toNativeRequest( + request: MetonaRequest, + stream: boolean, + ): Record { // 推理模型检测(o 系列使用新参数名) const model = this.config.defaultModel; const isReasoningModel = /^(o\d|gpt-5)/.test(model); @@ -168,6 +169,12 @@ export class OpenAIAdapter extends OpenAICompatibleAdapter { max: 'high', }; body.reasoning_effort = effortMap[request.params.thinkingEffort ?? 'high'] ?? 'high'; + // v0.8.0 P0-3: 推理 token 计入 max_completion_tokens —— 钳制后预算过小时告警 + if (oaMaxTokens < 8192) { + log.warn( + `[OpenAI] reasoning enabled with small output budget (${oaMaxTokens} tokens after model clamp) — reasoning may consume the entire budget and truncate the answer`, + ); + } } else if (!isReasoningModel) { // 非推理模型使用温度控制 body.temperature = request.params.temperature; diff --git a/electron/harness/adapters/shared/sse-stream.ts b/electron/harness/adapters/shared/sse-stream.ts index 7ff5b2c..f0b421c 100644 --- a/electron/harness/adapters/shared/sse-stream.ts +++ b/electron/harness/adapters/shared/sse-stream.ts @@ -139,17 +139,17 @@ function extractUpstreamErrorFrame( ? e.status_code : undefined; const rawCode = typeof e.code === 'string' ? e.code : typeof e.type === 'string' ? e.type : ''; - const message = - typeof e.message === 'string' - ? e.message - : typeof e.msg === 'string' - ? e.msg - : JSON.stringify(errObj); - // 无消息且无状态码的无害空对象不算错误(防御性) - if (!message && rawStatus === undefined && !rawCode) return null; + // v0.8.0 P1-3.1 根治: 旧实现 message 兜底取 JSON.stringify(errObj) —— '{}' 恒为 + // 真值,使本防御分支永不可达,`{"error":{}}` 被误判为错误帧抛 SseUpstreamError + // (原行为被 sse-stream.test.ts 锁定留档,本次随修复同步改约)。 + // 现契约:仅当存在显式 message/msg 或 status/code 标识之一才构成错误帧; + // 描述文本只在有标识但无显式文本时由序列化/占位生成。 + const explicitMessage = + typeof e.message === 'string' ? e.message : typeof e.msg === 'string' ? e.msg : ''; + if (!explicitMessage && rawStatus === undefined && !rawCode) return null; return { - message: message || `upstream error (${rawCode || rawStatus})`, + message: explicitMessage || `upstream error (${rawCode || rawStatus})`, status: rawStatus, code: rawCode || undefined, }; @@ -326,6 +326,10 @@ export async function* parseSSEStream( let buffer = ''; // v0.6.3: 是否收到过 [DONE](流断开兜底用) let sawDone = false; + // v0.8.0 P0-1: 跟踪 Provider 原生 finish_reason,随 DONE 事件携带给引擎 + // ('stop' | 'length' | 'tool_calls' | 'content_filter' 等;引擎据此区分 + // 自然完成与输出上限截断,此前 length 仅落日志、引擎不可见) + let lastFinishReason: string | undefined; // v0.7.4 P1-2: 流空闲超时 — 服务器保活但不再推送数据(连接挂死)时, // reader.read() 会无限挂起,totalTimeoutMs 只在迭代之间检查,无法兜底。 @@ -369,6 +373,8 @@ export async function* parseSSEStream( iteration, seq: seqRef.seq++, timestamp: Date.now(), + // v0.8.0 P0-1: 携带 Provider 原生 finish_reason(未见任何帧时 undefined) + ...(lastFinishReason ? { finishReason: lastFinishReason } : {}), }; return; } @@ -480,6 +486,11 @@ export async function* parseSSEStream( // 非 [DONE] 但 finish_reason 为 tool_calls 时提前 flush 缓冲区 const finishReason = chunk.choices?.[0]?.finish_reason as string | undefined; + // v0.8.0 P0-1: 记录最近一次 finish_reason(最终帧通常携带于最后一个 + // 含 choices 的 chunk),随 DONE 事件上交通知引擎与前端 + if (finishReason) { + lastFinishReason = finishReason; + } if (finishReason === 'tool_calls') { // L-4 修复: 使用 flushToolCallBuffer 替代重复的遍历代码 yield* flushToolCallBuffer(toolCallsBuffer, requestId, sessionId, iteration, seqRef); @@ -528,6 +539,10 @@ export async function* parseSSEStream( iteration, seq: seqRef.seq++, timestamp: Date.now(), + // v0.8.0 P0-1: 断流合成路径同样携带已观察到的 finish_reason + //(注意:断流时 finishReason 多为 undefined —— 引擎据此走空响应守卫/重试, + // 而不是误判为模型自然说完了) + ...(lastFinishReason ? { finishReason: lastFinishReason } : {}), }; } } diff --git a/electron/harness/agent-loop/__tests__/engine-stream-contract.test.ts b/electron/harness/agent-loop/__tests__/engine-stream-contract.test.ts new file mode 100644 index 0000000..862403a --- /dev/null +++ b/electron/harness/agent-loop/__tests__/engine-stream-contract.test.ts @@ -0,0 +1,230 @@ +/** + * v0.8.0 P0-5: 引擎流结束契约测试 —— "模型思考着会话停止"根治回归防线。 + * + * 钉住 chatStreamWithRetry 的流结束契约(P0-2)与 finish_reason 贯通(P0-1): + * 1. 零产出流 → EmptyResponseError → 退避重试 → 重试耗尽 ERROR; + * 2. reasoning-only + finish_reason=length → 一次性降级重试(thinkingEnabled=false) + * → 降级后成功 COMPLETED; + * 3. 降级重试仍 length → OutputTruncatedError → ERROR 事件携带 + * MetonaErrorCode.OUTPUT_LENGTH_EXCEEDED; + * 4. 有正文 + length → COMPLETED 且最终 DONE 事件携带 finishReason='length'; + * 5. 引擎最终 DONE 贯通最后一轮 step.finishReason。 + */ + +import { describe, it, expect } from 'vitest'; +import { AgentLoopEngine } from '../engine'; +import type { AgentLoopOutput } from '../types'; +import type { + MetonaRequest, + MetonaStreamEvent, + IMetonaProviderAdapter, + MetonaResponse, +} from '../../types'; +import { MetonaStreamEventType, MetonaErrorCode } from '../../types'; + +const SYSTEM_PROMPT = { + roleDefinition: 'test', + outputConstraints: '', + safetyGuidelines: '', +}; + +function makeAdapter( + script: (request: MetonaRequest) => MetonaStreamEvent[], +): IMetonaProviderAdapter & { calls: MetonaRequest[] } { + const calls: MetonaRequest[] = []; + const adapter = { + providerId: 'mock' as const, + supportedModels: ['m'], + supportsToolCalling: true, + supportsThinking: true, + calls, + getContextWindow(): number { + return 128_000; + }, + async send(_request: MetonaRequest): Promise { + throw new Error('send() not expected in this test'); + }, + async *sendStream(request: MetonaRequest): AsyncIterable { + calls.push(request); + for (const ev of script(request)) { + yield ev; + } + }, + }; + return adapter as unknown as IMetonaProviderAdapter & { calls: MetonaRequest[] }; +} + +function makeEngine(adapter: IMetonaProviderAdapter, retryCount = 1): AgentLoopEngine { + return new AgentLoopEngine( + { + maxIterations: 3, + totalTimeoutMs: 10_000, + retryCount, + thinkingEnabled: true, + thinkingEffort: 'high', + contextWindow: 128_000, + }, + adapter, + ); +} + +function textDelta(t: string): MetonaStreamEvent { + return { + type: MetonaStreamEventType.TEXT_DELTA, + requestId: 'r', + sessionId: 's', + iteration: 1, + seq: 0, + timestamp: Date.now(), + delta: t, + }; +} + +function reasoningDelta(t: string): MetonaStreamEvent { + return { + type: MetonaStreamEventType.REASONING_DELTA, + requestId: 'r', + sessionId: 's', + iteration: 1, + seq: 0, + timestamp: Date.now(), + delta: t, + }; +} + +function doneEvent(finishReason?: string): MetonaStreamEvent { + return { + type: MetonaStreamEventType.DONE, + requestId: 'r', + sessionId: 's', + iteration: 1, + seq: 0, + timestamp: Date.now(), + ...(finishReason ? { finishReason } : {}), + }; +} + +describe('engine stream-end contract (v0.8.0 P0-2)', () => { + it('scenario 2: reasoning-only + length → one degraded retry with thinking disabled → COMPLETED', async () => { + let call = 0; + const adapter = makeAdapter((req) => { + call++; + if (call === 1) { + // 第一次:思考烧满输出预算(reasoning-only + length 截断) + expect(req.params.thinkingEnabled).toBe(true); + return [reasoningDelta('thinking...'), doneEvent('length')]; + } + // 降级重试:thinkingEnabled 必须已被引擎置 false + expect(req.params.thinkingEnabled).toBe(false); + return [textDelta('answer'), doneEvent('stop')]; + }); + const engine = makeEngine(adapter); + const events: MetonaStreamEvent[] = []; + engine.on('streamEvent', (e: MetonaStreamEvent) => events.push(e)); + + const output = await engine.runStream( + { role: 'user', content: 'hi', timestamp: Date.now() }, + 's1', + [], + SYSTEM_PROMPT, + ); + + expect(output.terminationReason).toBe('completed'); + expect(output.finalAnswer).toBe('answer'); + expect(adapter.calls.length).toBe(2); + // 降级重试经过 RETRY 事件(引擎消费后向渲染层转 STREAM_RESET) + expect(events.some((e) => e.type === MetonaStreamEventType.STREAM_RESET)).toBe(true); + }); + + it('scenario 2b: degraded retry still length → OUTPUT_LENGTH_EXCEEDED error (no infinite retry)', async () => { + const adapter = makeAdapter(() => [reasoningDelta('thinking...'), doneEvent('length')]); + const engine = makeEngine(adapter); + const events: MetonaStreamEvent[] = []; + engine.on('streamEvent', (e: MetonaStreamEvent) => events.push(e)); + + const output: AgentLoopOutput = await engine.runStream( + { role: 'user', content: 'hi', timestamp: Date.now() }, + 's1', + [], + SYSTEM_PROMPT, + ); + + // 恰好两次调用(原始 + 一次性降级重试),随后终止 + expect(adapter.calls.length).toBe(2); + expect(output.terminationReason).toBe('error'); + const errorEvent = events.find((e) => e.type === MetonaStreamEventType.ERROR); + expect(errorEvent).toBeDefined(); + expect(errorEvent?.error?.code).toBe(MetonaErrorCode.OUTPUT_LENGTH_EXCEEDED); + }); + + it('scenario 1: zero-content stream → retryable → retry exhausted → ERROR', async () => { + const adapter = makeAdapter(() => []); + const engine = makeEngine(adapter, 1); + const events: MetonaStreamEvent[] = []; + engine.on('streamEvent', (e: MetonaStreamEvent) => events.push(e)); + + const output = await engine.runStream( + { role: 'user', content: 'hi', timestamp: Date.now() }, + 's1', + [], + SYSTEM_PROMPT, + ); + + expect(output.terminationReason).toBe('error'); + expect(adapter.calls.length).toBe(2); // 原始 1 次 + 重试 1 次(retryCount=1) + const errorEvent = events.find((e) => e.type === MetonaStreamEventType.ERROR); + expect(errorEvent?.error?.message).toContain('empty stream'); + }); + + it('scenario 4: text + length → COMPLETED with DONE.finishReason=length (no retry)', async () => { + const adapter = makeAdapter(() => [textDelta('partial answer'), doneEvent('length')]); + const engine = makeEngine(adapter); + const events: MetonaStreamEvent[] = []; + engine.on('streamEvent', (e: MetonaStreamEvent) => events.push(e)); + + const output = await engine.runStream( + { role: 'user', content: 'hi', timestamp: Date.now() }, + 's1', + [], + SYSTEM_PROMPT, + ); + + expect(output.terminationReason).toBe('completed'); + expect(output.finalAnswer).toBe('partial answer'); + expect(adapter.calls.length).toBe(1); // 有产出不重试 + const done = events.find((e) => e.type === MetonaStreamEventType.DONE); + expect(done?.finishReason).toBe('length'); + }); + + it('P0-1: normal stop → DONE carries finishReason=stop from adapter', async () => { + const adapter = makeAdapter(() => [textDelta('ok'), doneEvent('stop')]); + const engine = makeEngine(adapter); + const events: MetonaStreamEvent[] = []; + engine.on('streamEvent', (e: MetonaStreamEvent) => events.push(e)); + + const output = await engine.runStream( + { role: 'user', content: 'hi', timestamp: Date.now() }, + 's1', + [], + SYSTEM_PROMPT, + ); + + expect(output.terminationReason).toBe('completed'); + const done = events.find((e) => e.type === MetonaStreamEventType.DONE); + expect(done?.finishReason).toBe('stop'); + }); + + it('reasoning-only + natural stop (no length) → retryable empty response path', async () => { + const adapter = makeAdapter(() => [reasoningDelta('hmm'), doneEvent('stop')]); + const engine = makeEngine(adapter, 1); + const output = await engine.runStream( + { role: 'user', content: 'hi', timestamp: Date.now() }, + 's1', + [], + SYSTEM_PROMPT, + ); + // 无产出且非 length 截断 → EmptyResponseError(可重试)→ 重试耗尽 ERROR + expect(output.terminationReason).toBe('error'); + expect(adapter.calls.length).toBe(2); + }); +}); diff --git a/electron/harness/agent-loop/__tests__/engine-stream-reliability.test.ts b/electron/harness/agent-loop/__tests__/engine-stream-reliability.test.ts new file mode 100644 index 0000000..8106bbe --- /dev/null +++ b/electron/harness/agent-loop/__tests__/engine-stream-reliability.test.ts @@ -0,0 +1,243 @@ +/** + * v0.8.0 P0-5 回归测试(第二批):流可靠性四件套的剩余项。 + * + * 1. parseSSEStream 集成级空闲超时:挂死 ReadableStream → 60s 无数据 → + * SseUpstreamError(504)(P1-2 防线的集成真实性,此前仅纯函数层覆盖); + * 2. SseUpstreamError(504) 经引擎 isRetryableError 归类 → 退避重试后成功; + * 3. REASONING_DELTA 持续输出中 abort → USER_INTERRUPT("思考中点停止"场景); + * 4. 引擎 P4-2:上一 run 卡死(不响应 abort)→ 新 run 等待 30s → 强制 abort + * → 再等 5s → 抛错(防"会话永久排队")。 + */ + +import { describe, it, expect, vi, afterEach } from 'vitest'; +import { parseSSEStream, SseUpstreamError } from '../../adapters/shared/sse-stream'; +import { AgentLoopEngine } from '../engine'; +import type { MetonaRequest, MetonaStreamEvent, IMetonaProviderAdapter } from '../../types'; +import { MetonaStreamEventType } from '../../types'; + +const SYSTEM_PROMPT = { + roleDefinition: 'test', + outputConstraints: '', + safetyGuidelines: '', +}; + +afterEach(() => { + vi.useRealTimers(); +}); + +// ===== 1. 集成级空闲超时 ===== + +describe('parseSSEStream integrated idle timeout (P0-5 #2)', () => { + function makeHungBody(): ReadableStream { + return { + getReader: () => ({ + // 永不 resolve 的 read —— 模拟"服务器保活但不再推数据" + read: () => new Promise(() => {}), + releaseLock: () => {}, + }), + } as unknown as ReadableStream; + } + + it('挂死流 60s 无数据 → SseUpstreamError(504)', async () => { + vi.useFakeTimers(); + const iterator = parseSSEStream(makeHungBody(), 'r', 's', 1); + let thrown: unknown = null; + const firstRead = iterator.next().catch((e: unknown) => { + thrown = e; + return { done: true, value: undefined } as const; + }); + // 推进 61s:IDLE_TIMEOUT_MS=60_000 触发 idle controller → read() 竞速失败 + await vi.advanceTimersByTimeAsync(61_000); + await firstRead; + expect(thrown).toBeInstanceOf(SseUpstreamError); + expect((thrown as SseUpstreamError).status).toBe(504); + expect((thrown as Error).message).toContain('idle timeout'); + }); +}); + +// ===== 2. SseUpstreamError(504) → 引擎重试 ===== + +describe('engine: SseUpstreamError(504) retry classification (P0-5 #3)', () => { + it('504 首败 → 退避重试 → 第二次流成功 → COMPLETED', async () => { + let call = 0; + const adapter = { + providerId: 'mock' as const, + supportedModels: ['m'], + supportsToolCalling: true, + supportsThinking: true, + getContextWindow: () => 128_000, + async send(): Promise { + throw new Error('not used'); + }, + async *sendStream(_r: MetonaRequest): AsyncIterable { + call++; + if (call === 1) { + throw new SseUpstreamError('Stream idle timeout after 60000ms (no data received)', { + status: 504, + }); + } + yield { + type: MetonaStreamEventType.TEXT_DELTA, + requestId: 'r', + sessionId: 's', + iteration: 1, + seq: 0, + timestamp: Date.now(), + delta: 'recovered', + }; + yield { + type: MetonaStreamEventType.DONE, + requestId: 'r', + sessionId: 's', + iteration: 1, + seq: 1, + timestamp: Date.now(), + finishReason: 'stop', + }; + }, + } as unknown as IMetonaProviderAdapter; + + const engine = new AgentLoopEngine( + { maxIterations: 3, totalTimeoutMs: 15_000, retryCount: 2, contextWindow: 128_000 }, + adapter, + ); + const events: MetonaStreamEvent[] = []; + engine.on('streamEvent', (e: MetonaStreamEvent) => events.push(e)); + const output = await engine.runStream( + { role: 'user', content: 'hi', timestamp: Date.now() }, + 's1', + [], + SYSTEM_PROMPT, + ); + expect(output.terminationReason).toBe('completed'); + expect(output.finalAnswer).toBe('recovered'); + expect(call).toBe(2); + // 重试经过 RETRY 事件(引擎消费后向渲染层转 STREAM_RESET) + expect(events.some((e) => e.type === MetonaStreamEventType.STREAM_RESET)).toBe(true); + }); +}); + +// ===== 3. REASONING_DELTA 中途 abort ===== + +describe('engine: abort during reasoning stream (P0-5 #4)', () => { + it('思考增量持续输出中 abort → USER_INTERRUPT,迟到 reasoning 不再产出事件', async () => { + const adapter = { + providerId: 'mock' as const, + supportedModels: ['m'], + supportsToolCalling: true, + supportsThinking: true, + getContextWindow: () => 128_000, + async send(): Promise { + throw new Error('not used'); + }, + // 持续每 10ms 推送一个 reasoning delta,直到 abort 或 500 个(挂起) + async *sendStream(_r: MetonaRequest): AsyncIterable { + for (let i = 0; i < 500; i++) { + if (_r.params.thinkingEnabled === false) return; + yield { + type: MetonaStreamEventType.REASONING_DELTA, + requestId: 'r', + sessionId: 's', + iteration: 1, + seq: i, + timestamp: Date.now(), + delta: `t${i};`, + }; + await new Promise((r) => setTimeout(r, 10)); + } + yield { + type: MetonaStreamEventType.DONE, + requestId: 'r', + sessionId: 's', + iteration: 1, + seq: 999, + timestamp: Date.now(), + }; + }, + } as unknown as IMetonaProviderAdapter; + + const engine = new AgentLoopEngine( + { maxIterations: 3, totalTimeoutMs: 30_000, retryCount: 0, contextWindow: 128_000 }, + adapter, + ); + const events: MetonaStreamEvent[] = []; + engine.on('streamEvent', (e: MetonaStreamEvent) => events.push(e)); + + const run = engine.runStream( + { role: 'user', content: 'hi', timestamp: Date.now() }, + 's1', + [], + SYSTEM_PROMPT, + ); + // 等待至少 3 个 reasoning delta 到达(确认流确实在思考态输出中) + await new Promise((r) => setTimeout(r, 80)); + expect( + events.filter((e) => e.type === MetonaStreamEventType.REASONING_DELTA).length, + ).toBeGreaterThanOrEqual(3); + engine.abort(); + const output = await run; + expect(output.terminationReason).toBe('user_interrupt'); + const lastReasoningIdx = events + .map((e) => e.type) + .lastIndexOf(MetonaStreamEventType.REASONING_DELTA); + const doneIdx = events.findIndex((e) => e.type === MetonaStreamEventType.DONE); + // abort 生效后不再有迟到 reasoning(abort 尾巴过滤在引擎侧) + expect(lastReasoningIdx).toBeLessThan(doneIdx); + }, 15_000); +}); + +// ===== 4. P4-2: 上一 run 卡死 → 30s 强制 abort + 5s 再等待 → 抛错 ===== + +describe('engine: stuck previous run force-abort path (P4-2)', () => { + it('上一 run 不响应 abort → 新 run 30s 超时强制 abort → 5s 后仍卡 → 抛错', async () => { + vi.useFakeTimers(); + let released = false; + const stuckAdapter = { + providerId: 'mock' as const, + supportedModels: ['m'], + supportsToolCalling: true, + supportsThinking: true, + getContextWindow: () => 128_000, + async send(): Promise { + throw new Error('not used'); + }, + // 第一次 run:永远挂起且不响应 abort(模拟卡死的底层 fetch); + // 挂起生成器无 yield 属场景定义(require-yield 由下行指令豁免) + // eslint-disable-next-line require-yield + async *sendStream(): AsyncIterable { + while (!released) { + await new Promise(() => {}); + } + return; + }, + } as unknown as IMetonaProviderAdapter; + + const engine = new AgentLoopEngine( + { maxIterations: 3, totalTimeoutMs: 600_000, retryCount: 0, contextWindow: 128_000 }, + stuckAdapter, + ); + const firstRun = engine.runStream( + { role: 'user', content: 'first', timestamp: Date.now() }, + 's1', + [], + SYSTEM_PROMPT, + ); + firstRun.catch(() => {}); + + // 新消息进入 → 等待上一 run 结束(30s 超时)→ 强制 abort → 5s 再等待 → 抛错 + const secondRun = engine.runStream( + { role: 'user', content: 'second', timestamp: Date.now() }, + 's1', + [], + SYSTEM_PROMPT, + ); + const expectation = expect(secondRun).rejects.toThrow(/Previous run did not finish/i); + + // 驱动 30s 等待超时 + 5s 强制 abort 等待 + await vi.advanceTimersByTimeAsync(30_500); + await vi.advanceTimersByTimeAsync(5_500); + // 清理:放行挂死流,让 firstRun 收敛 + released = true; + await expectation; + }, 20_000); +}); diff --git a/electron/harness/agent-loop/engine.ts b/electron/harness/agent-loop/engine.ts index 11f483f..6d4b45d 100644 --- a/electron/harness/agent-loop/engine.ts +++ b/electron/harness/agent-loop/engine.ts @@ -48,6 +48,37 @@ class DeadLoopError extends Error { } } +/** + * v0.8.0 P0-2: 空响应错误 — 流结束但零产出(无 reasoning、无正文、无工具调用)。 + * 属可重试的瞬时故障(上游断流被 [DONE] 兜底合成 / 网关干净空响应),携带 + * status=503 使 engine.isRetryableError 的 5xx 分支命中,走指数退避重试; + * 重试耗尽后以 ERROR 收尾并给用户明确文案(此前该形态被静默判定 COMPLETED, + * 即"模型思考着会话就停止"的根因之一)。 + */ +export class EmptyResponseError extends Error { + readonly status = 503; + constructor(message: string) { + super(message); + this.name = 'EmptyResponseError'; + } +} + +/** + * v0.8.0 P0-2: 输出截断错误 — 思考耗尽型截断(reasoning 烧满输出预算、正文零字节, + * finish_reason=length),一次性降级重试(关闭思考)后仍复现时抛出。 + * 不可重试(status=400),携带 code=OUTPUT_LENGTH_EXCEEDED(IR 预留错误码首次 + * 投产),engine.finish() 据此向前端发送结构化错误;非可重试语义同时触发 + * Provider 故障转移(备用 Provider 输出预算更大时即构成自愈)。 + */ +export class OutputTruncatedError extends Error { + readonly status = 400; + readonly code = MetonaErrorCode.OUTPUT_LENGTH_EXCEEDED; + constructor(message: string) { + super(message); + this.name = 'OutputTruncatedError'; + } +} + const DEFAULT_CONFIG: AgentLoopConfig = { maxIterations: 20, totalTimeoutMs: 600_000, @@ -88,6 +119,12 @@ export class AgentLoopEngine extends EventEmitter { private lastRealInputTokens = 0; /** 当前 run 的唯一标识(用于前端过滤旧流事件) */ private runId: string = ''; + /** + * v0.8.0 P0-1: 最近一轮 LLM 流的 Provider 原生停止原因(归一化 OpenAI 语义)。 + * 由 adapter DONE 事件携带、引擎捕获;finish() 据此在最终 DONE 事件上转发, + * 前端/TRACE 据此区分自然完成与输出上限截断。 + */ + private lastIterationFinishReason: string | undefined; /** 正在进行的 run Promise(用于 run lock,防止并发 run 污染状态) */ private currentRunPromise: Promise | null = null; @@ -269,6 +306,8 @@ export class AgentLoopEngine extends EventEmitter { this.toolCallHistory = []; // v0.7.4 P4-1: 重置工具结果历史(进度信号,与调用历史按轮对应) this.toolResultHistory = []; + // v0.8.0 P0-1: 重置最近一轮停止原因 + this.lastIterationFinishReason = undefined; try { await this.transitionTo(AgentLoopState.INIT); @@ -316,6 +355,16 @@ export class AgentLoopEngine extends EventEmitter { const step = await this.executeOneIterationStream(request, sessionId); this.iterations.push(step); + // v0.8.0 P0-1: 记录最近一轮停止原因(供 finish() 的 DONE 事件转发) + this.lastIterationFinishReason = step.finishReason; + + // v0.8.0 P0-2 回归根治: abort 优先级 —— 旧实现该检查位于 while 循环 + // 末尾,若 abort 恰逢"零工具调用轮"(如思考中点停止),空轮会先命中 + // 下方 COMPLETED 分支,把 USER_INTERRUPT 静默吞成 completed(与新 + // 回归测试 engine-stream-reliability 同时发现)。此处前置检查修复。 + if (this.aborted) { + return this.finish(TerminationReason.USER_INTERRUPT); + } // 将 assistant 回复加入消息历史 // 崩溃修复(会话停止根因): 原条件 `if (step.thought)` 在模型发起纯工具调用 @@ -499,13 +548,20 @@ export class AgentLoopEngine extends EventEmitter { let reasoningContent = ''; const toolCallsBuffer = new Map(); let tokenUsage: TokenUsage | undefined; + // v0.8.0 P0-1: 本轮流的 Provider 原生停止原因(adapter DONE 携带) + let iterationFinishReason: string | undefined; // 流式接收响应 for await (const event of this.chatStreamWithRetry(request)) { if (this.aborted) break; // 过滤掉每轮的 DONE 事件 — 只在全部迭代完成后发送一个最终 DONE - if (event.type === MetonaStreamEventType.DONE) continue; + // v0.8.0 P0-1: 先捕获 adapter 携带的 Provider 原生停止原因再过滤 —— + // 引擎与 TRACE 据此区分自然完成与输出上限截断(length 此前完全不可见) + if (event.type === MetonaStreamEventType.DONE) { + if (event.finishReason) iterationFinishReason = event.finishReason; + continue; + } // 过滤掉 RETRY 类型的 ERROR 事件 — 不转发到前端,避免触发虚假错误 UI // RETRY 事件仅用于 Engine 内部清空缓冲区(见下方 switch 分支) @@ -598,6 +654,11 @@ export class AgentLoopEngine extends EventEmitter { // === v0.2.0: PARSING 状态 — 解析流式缓冲区中的工具调用 === await this.transitionTo(AgentLoopState.PARSING); + // v0.8.0 P0-1: 记录本轮停止原因到迭代步骤(TRACE/审计/前端提示的输入) + if (iterationFinishReason) { + step.finishReason = iterationFinishReason; + } + // L-19 修复: 提取 finalizeToolCallsFromBuffer 子方法(PARSING 阶段) this.finalizeToolCallsFromBuffer(step, toolCallsBuffer); @@ -1031,6 +1092,11 @@ export class AgentLoopEngine extends EventEmitter { let attempt = 0; let currentAdapter = this.adapter; let failoverUsed = false; + // v0.8.0 P0-2: 降级重试状态。思考耗尽型截断(reasoning 烧满输出预算、正文 + // 零字节)时以 thinkingEnabled=false 重发一次(每 run 最多一次);activeRequest + // 被替换后,后续的重试与故障转移均使用降级版请求(降级跨故障转移保持)。 + let degradedThinkingRetryUsed = false; + let activeRequest = request; while (true) { try { @@ -1038,9 +1104,9 @@ export class AgentLoopEngine extends EventEmitter { // 重试时:先发送一个 retry 事件,让 UI 清空已接收的 delta yield { type: MetonaStreamEventType.ERROR, - requestId: request.meta.requestId, - sessionId: request.meta.sessionId, - iteration: request.meta.iteration, + requestId: activeRequest.meta.requestId, + sessionId: activeRequest.meta.sessionId, + iteration: activeRequest.meta.iteration, seq: 0, timestamp: Date.now(), error: { @@ -1050,7 +1116,86 @@ export class AgentLoopEngine extends EventEmitter { }, }; } - yield* currentAdapter.sendStream(request); + // v0.8.0 P0-2: 逐事件跟踪内容产出与停止原因(原 yield* 直通改显式转发)。 + // yield* 正常返回后执行流结束契约检查;异常路径不进入检查(走 catch 重试)。 + // - sawThought: 思考有产出(REASONING_DELTA) + // - sawOutput: 正文/工具调用有产出(TEXT_DELTA / TOOL_CALL_*) + // - lastFinishReason: adapter DONE 携带的 Provider 原生停止原因 + let sawThought = false; + let sawOutput = false; + let lastFinishReason: string | undefined; + for await (const event of currentAdapter.sendStream(activeRequest)) { + if (event.type === MetonaStreamEventType.REASONING_DELTA) { + sawThought = true; + } + if ( + event.type === MetonaStreamEventType.TEXT_DELTA || + event.type === MetonaStreamEventType.TOOL_CALL_DELTA || + event.type === MetonaStreamEventType.TOOL_CALL_COMPLETE + ) { + sawOutput = true; + } + if (event.type === MetonaStreamEventType.DONE && event.finishReason) { + lastFinishReason = event.finishReason; + } + yield event; + } + + // ===== 流结束契约检查(根治"空回复 → 静默 COMPLETED → 会话无声停止")===== + if (!sawOutput) { + if (!sawThought) { + // 零产出流:连接中断被 [DONE] 兜底合成 / 上游干净空响应 —— 可重试 + // 瞬时故障(status=503 命中 5xx 重试分支),重试耗尽后 ERROR 收尾 + throw new EmptyResponseError( + 'Model returned an empty stream (no reasoning, no text, no tool calls). ' + + 'This is usually caused by an upstream connection drop or an output-limit cutoff.', + ); + } + if (lastFinishReason === 'length') { + // 思考耗尽型截断(v0.7.4 生产事故形态:finish_reason=length、正文零 + // 字节)—— 先做一次性降级重试(临时关闭思考重发),仍复现则终止。 + if (!degradedThinkingRetryUsed && activeRequest.params.thinkingEnabled === true) { + degradedThinkingRetryUsed = true; + attempt = 0; + activeRequest = { + ...activeRequest, + params: { ...activeRequest.params, thinkingEnabled: false }, + }; + log.warn( + '[AgentLoop] Output budget exhausted by reasoning (finish_reason=length, empty content) — retrying once with thinking disabled', + ); + yield { + type: MetonaStreamEventType.ERROR, + requestId: activeRequest.meta.requestId, + sessionId: activeRequest.meta.sessionId, + iteration: activeRequest.meta.iteration, + seq: 0, + timestamp: Date.now(), + error: { + code: MetonaErrorCode.RETRY, + message: 'Thinking consumed the output budget — retrying with thinking disabled', + retryable: true, + }, + }; + continue; + } + // 降级重试不可用(已用 / 思考本就关闭)仍截断 —— 终止:携带 IR 预留 + // 的 OUTPUT_LENGTH_EXCEEDED 码,finish() 映射为结构化错误事件, + // 非 retryable 同时触发 Provider 故障转移(备用预算更大即自愈)。 + throw new OutputTruncatedError( + 'The model consumed the entire output token budget on reasoning and produced no answer ' + + '(finish_reason=length). Raise the max output limit, disable thinking, or switch to a model with a larger output budget.', + ); + } + // 思考有产出但自然结束且零正文(罕见的服务端异常收尾)—— 与零产出流 + // 同按可重试空响应处理(retryCount 兜底防循环) + throw new EmptyResponseError( + 'Model stream ended after reasoning without any answer content.', + ); + } + // sawOutput=true:正文/工具调用已有产出。length 截断场景不再重试 —— + // 工具参数截断由 _truncatedArguments 自愈(v0.6.3 契约)、正文截断由 + // DONE.finishReason='length' 驱动前端截断提醒(P0-4)。 return; } catch (error) { if (this.aborted) throw error; @@ -1136,6 +1281,11 @@ export class AgentLoopEngine extends EventEmitter { /** 判断错误是否可重试 */ private isRetryableError(error: unknown): boolean { + // v0.8.0 P0-2: 流结束契约错误的显式归类(EmptyResponseError 自带 status=503、 + // OutputTruncatedError 自带 status=400,本可由下方 5xx/4xx 分支覆盖;此处 + // 显式 instanceof 使两类新错误的语义不依赖 status 字段约定) + if (error instanceof EmptyResponseError) return true; + if (error instanceof OutputTruncatedError) return false; const err = error as { status?: number; code?: string; message?: string }; // 429 Too Many Requests — 可重试 if (err.status === 429) return true; @@ -1482,9 +1632,17 @@ export class AgentLoopEngine extends EventEmitter { if ((reason === TerminationReason.ERROR || reason === TerminationReason.DEAD_LOOP) && error) { // v0.3.17: 识别 ContentFilterError,映射为 CONTENT_FILTERED 错误码 + 友好消息 // v0.6.4: 流式路径的拦截以 err.code='content_filtered' 抵达(无 instanceof 上下文),一并识别 + // v0.8.0 P0-2: 透传流结束契约错误的结构化错误码(OUTPUT_LENGTH_EXCEEDED 等), + // 前端据此展示针对性文案而非泛化 UNKNOWN const isContentFilter = error instanceof ContentFilterError || (error as Error & { code?: string }).code === MetonaErrorCode.CONTENT_FILTERED; + const rawErrorCode = (error as Error & { code?: string }).code; + const errorCode = isContentFilter + ? MetonaErrorCode.CONTENT_FILTERED + : rawErrorCode === MetonaErrorCode.OUTPUT_LENGTH_EXCEEDED + ? MetonaErrorCode.OUTPUT_LENGTH_EXCEEDED + : MetonaErrorCode.UNKNOWN; this.emit('streamEvent', { type: MetonaStreamEventType.ERROR, requestId: this.currentRequestId, @@ -1494,7 +1652,7 @@ export class AgentLoopEngine extends EventEmitter { timestamp: Date.now(), runId: this.runId, error: { - code: isContentFilter ? MetonaErrorCode.CONTENT_FILTERED : MetonaErrorCode.UNKNOWN, + code: errorCode, message: isContentFilter ? '内容被 Provider 安全审核拦截,请修改图片或文本后重试' : error.message, @@ -1513,6 +1671,9 @@ export class AgentLoopEngine extends EventEmitter { timestamp: Date.now(), runId: this.runId, terminationReason: reason, + // v0.8.0 P0-1: 转发最后一轮的 Provider 原生停止原因(前端据此展示 + // "回答可能因输出上限被截断"等提示;无流式轮次时为 undefined) + ...(this.lastIterationFinishReason ? { finishReason: this.lastIterationFinishReason } : {}), }); // H-3: 通过 stateChange 发射 TERMINATED 状态(前端可据此清理 UI) diff --git a/electron/harness/agent-loop/index.ts b/electron/harness/agent-loop/index.ts index 13f99c0..aa0207d 100644 --- a/electron/harness/agent-loop/index.ts +++ b/electron/harness/agent-loop/index.ts @@ -1,4 +1,6 @@ export { AgentLoopEngine } from './engine'; +// v0.8.0 P0-2: 流结束契约错误(空响应重试 / 思考耗尽截断终止)—— 导出供测试断言 +export { EmptyResponseError, OutputTruncatedError } from './engine'; export { AgentLoopState, TerminationReason, diff --git a/electron/harness/agent-loop/types.ts b/electron/harness/agent-loop/types.ts index c44da6b..3a8758a 100644 --- a/electron/harness/agent-loop/types.ts +++ b/electron/harness/agent-loop/types.ts @@ -48,6 +48,13 @@ export interface IterationStep { toolCalls?: MetonaToolCall[]; toolResults?: MetonaToolResult[]; tokenUsage?: TokenUsage; + /** + * v0.8.0 P0-1: 本轮 LLM 流的 Provider 原生停止原因(归一化 OpenAI 语义: + * 'stop' | 'length' | 'tool_calls' | 'content_filter')。 + * 由 adapter DONE 事件携带、引擎捕获;TRACE 层据此记录真值, + * 引擎据此区分"自然完成"与"输出上限截断"(P0-2 空响应守卫的输入)。 + */ + finishReason?: string; } export interface TokenUsage { diff --git a/electron/harness/hooks/__tests__/hooks-contracts.test.ts b/electron/harness/hooks/__tests__/hooks-contracts.test.ts index 0ba15c2..7375112 100644 --- a/electron/harness/hooks/__tests__/hooks-contracts.test.ts +++ b/electron/harness/hooks/__tests__/hooks-contracts.test.ts @@ -248,7 +248,9 @@ describe('SecurityScanHook — 分级防护矩阵', () => { const hook = new SecurityScanHook(asDefender(sd)); await hook.afterExecute(toolCall('web_fetch'), failRes, 's'); await hook.afterExecute(toolCall('web_fetch'), zeroScoreRes, 's'); - expect(failRes.result).toBe(failRes.result); + // v0.8.0 P3 根治: 旧断言 failRes.result 自比恒真 —— 现断言"失败结果不触发 + // 扫描、内容保持原样"(与下方 zero-score 计数互补) + expect(failRes.result).toBe(longText('__low_score_')); expect( sd.detectSemantic.mock.calls.filter((c: unknown[]) => String(c[0]).includes('__zero')).length, ).toBe(1); diff --git a/electron/harness/memory/__tests__/memory-manager.test.ts b/electron/harness/memory/__tests__/memory-manager.test.ts index 2df0752..d5e297c 100644 --- a/electron/harness/memory/__tests__/memory-manager.test.ts +++ b/electron/harness/memory/__tests__/memory-manager.test.ts @@ -157,6 +157,8 @@ describe.skipIf(!dbAvailable)('MemoryManager — tokenize 分词', () => { it('数字/纯符号文本产生空 token 列表(search 返回空)', () => { const tokens = tokensOf('12345 !@#'); expect(Array.isArray(tokens)).toBe(true); + // v0.8.0 P3: 补强断言 —— 旧断言只验数组类型未验为空 + expect(tokens).toHaveLength(0); }); }); diff --git a/electron/harness/orchestration/__tests__/orchestrator.test.ts b/electron/harness/orchestration/__tests__/orchestrator.test.ts index f27d4c1..d73108f 100644 --- a/electron/harness/orchestration/__tests__/orchestrator.test.ts +++ b/electron/harness/orchestration/__tests__/orchestrator.test.ts @@ -39,12 +39,24 @@ const seenRequests: MetonaRequest[] = []; function doneOnlyStream(request: MetonaRequest): AsyncGenerator { return (async function* () { + // v0.8.0 P0-2 契约对齐: 引擎的流结束契约要求至少有一个产出事件 + //(TEXT/REASONING/TOOL_CALL),纯 DONE 空流会被 EmptyResponseError 守卫 + // 判为可重试空响应并触发退避重试。此处补一个 TEXT_DELTA 使流构成合法完成。 + yield { + type: MetonaStreamEventType.TEXT_DELTA, + requestId: request.meta.requestId, + sessionId: request.meta.sessionId, + iteration: request.meta.iteration, + seq: 0, + timestamp: Date.now(), + delta: 'sub-agent done', + } as MetonaStreamEvent; yield { type: MetonaStreamEventType.DONE, requestId: request.meta.requestId, sessionId: request.meta.sessionId, iteration: request.meta.iteration, - seq: 0, + seq: 1, timestamp: Date.now(), } as MetonaStreamEvent; })(); @@ -314,8 +326,10 @@ describe('TaskOrchestrator — 中断契约', () => { releases.forEach((r) => r()); const results = await Promise.all([a, b]); - // A 被中断、B 正常完成,两者都最终收尾 - expect(results.map((r) => r.success)).toEqual([true, true]); + // A 被中断、B 正常完成,两者都最终收尾。 + // v0.8.0 P0-2 abort 优先级修复后:被中断的 A 以 USER_INTERRUPT 终止 → + // success=false(旧实现 COMPLETED 抢占使 A 误报 success=true,属缺陷语义)。 + expect(results.map((r) => r.success)).toEqual([false, true]); expect(orchestrator.getActiveAgentsStatus()).toHaveLength(0); }); diff --git a/electron/harness/orchestration/orchestrator.ts b/electron/harness/orchestration/orchestrator.ts index 7fde493..ed050dc 100644 --- a/electron/harness/orchestration/orchestrator.ts +++ b/electron/harness/orchestration/orchestrator.ts @@ -129,6 +129,12 @@ export class TaskOrchestrator extends EventEmitter { parentSessionId: string; maxIterations?: number; tools?: string[]; + /** + * v0.8.0 P1-2: 引擎级取消信号 —— delegate_task 工具透传 context.signal, + * 用户中断父会话时 SubEngine 联动终止(此前工具被 registry 兜底超时杀掉, + * SubEngine 仍在后台跑满剩余轮次)。 + */ + signal?: AbortSignal; }): Promise { const taskId = params.taskId ?? `sub_${nanoid(8)}`; const startMs = Date.now(); @@ -183,6 +189,18 @@ export class TaskOrchestrator extends EventEmitter { subEngine.setFallbackAdapter(this.engines.getFallbackAdapter()); subEngine.setWorkspacePath(this.engines.getWorkspacePath()); + // v0.8.0 P1-2: 外部取消信号联动 —— signal 已中止或运行中触发 abort 时 + // 终止 SubEngine;finally 中解除监听防泄漏 + let onExternalAbort: (() => void) | null = null; + if (params.signal) { + if (params.signal.aborted) { + subEngine.abort(); + } else { + onExternalAbort = () => subEngine.abort(); + params.signal.addEventListener('abort', onExternalAbort, { once: true }); + } + } + // v0.5.0: 转发 SubEngine 的流式/状态事件(供 TRACE 录制与前端可观测) // 事件以 taskId 为 sessionId,录制到独立 JSONL 文件,不污染父会话的流 this.forwardSubEngineEvents(subEngine, taskId); @@ -283,6 +301,10 @@ export class TaskOrchestrator extends EventEmitter { return result; } finally { + // v0.8.0 P1-2: 解除外部取消信号监听(防事件循环泄漏) + if (params.signal && onExternalAbort) { + params.signal.removeEventListener('abort', onExternalAbort); + } // #5 修复: 统一在 finally 块恢复 sessionDepth,覆盖正常完成/异常/abort 所有路径 // 审查修复: 如果 abortAll 已 clear sessionDepth,不再恢复(避免覆盖紧急清理)。 // 场景:用户紧急中断时 abortAll 先 clear,若 SubEngine 随后才返回执行 finally, diff --git a/electron/harness/tools/__tests__/registry.test.ts b/electron/harness/tools/__tests__/registry.test.ts index f9deec7..dd2ded2 100644 --- a/electron/harness/tools/__tests__/registry.test.ts +++ b/electron/harness/tools/__tests__/registry.test.ts @@ -104,16 +104,13 @@ describe('ToolRegistry.truncateResult', () => { expect(truncate(shot)).toBe(shot); }); - it('WEBP 魔数(RIFF....WEBP)的裸 base64 放行', () => { - const shot = { image: `UklGRg==${'A'.repeat(200_000)}WEBP` }; - // RIFF 头 base64 'UklGRg==' 解码为 'RIFF\x00\x00\x00\x00',WEBP 魔数在 8-11 字节 - const result = truncate(shot) as { image?: string; _truncated?: boolean }; - if (result._truncated === true) { - // WEBP 魔数未命中(base64 头部 64 字节内未含完整 RIFF+WEBP)→ 走常规截断(保守) - expect(result.image).toBeUndefined(); - } else { - expect(result).toBe(shot); - } + it('WEBP 魔数(RIFF....WEBP)的裸 base64 放行(v0.8.0 P3: 单向断言替代恒真分支)', () => { + // v0.8.0 P3 根治: 旧断言按 _truncated 分支双向放行(无论实现如何都通过,恒真)。 + // 现构造 8-11 字节确为 'WEBP' 的确定性载荷:base64('RIFF\0\0\0\0WEBP') 前缀, + // isInlineImagePayload 必命中白名单 → 整段放行。 + const head = Buffer.from('RIFF\0\0\0\0WEBP', 'latin1').toString('base64'); + const shot = { image: head + 'A'.repeat(200_000) }; + expect(truncate(shot)).toBe(shot); }); it('image 字段但非图片内容(普通长文本)仍按常规 50KB 截断(堵住旧白名单漏洞)', () => { diff --git a/electron/harness/tools/built-in/__tests__/filesystem-tools.test.ts b/electron/harness/tools/built-in/__tests__/filesystem-tools.test.ts index 9e9e27c..8bfbb74 100644 --- a/electron/harness/tools/built-in/__tests__/filesystem-tools.test.ts +++ b/electron/harness/tools/built-in/__tests__/filesystem-tools.test.ts @@ -248,22 +248,25 @@ describe('filesystem 工具 — read_file', () => { expect(String(r.content).charCodeAt(0)).not.toBe(0xfeff); }); - it('UTF-16LE 文件被二进制检测拒绝(0x00 字节触发,编码探测死代码)', async () => { - // 已知源码缺陷:isBinaryFile 的 NUL 字节检测拒绝一切 UTF-16 文件, - // decodeBufferWithDetection 的 UTF-16 分支因此不可达。此处锁定实际契约。 + it('UTF-16LE 文件正常读取(v0.8.0 P1-3.3 BOM 检测根治后契约)', async () => { + // v0.8.0 P1-3.3 根治: isBinaryFile 此前的 NUL 字节启发式拒绝一切 UTF-16 + // 文件,decodeBufferWithDetection 的 UTF-16 分支不可达(原缺陷被本测试 + // 锁定留档)。现 UTF-16 BOM(FF FE / FE FF)视为文本,编码探测生效。 const r = (await tool.execute({ file_path: 'utf16le.txt' }, ctxFor(ws))) as { success: boolean; - error?: string; + content?: string; + encoding?: string; }; - expect(r.success).toBe(false); - expect(String((r as { error?: string }).error)).toContain('Binary'); + expect(r.success).toBe(true); + expect(r.encoding).toBe('utf-16le'); }); - it('UTF-16BE 文件同样被二进制检测拒绝(实况契约)', async () => { + it('UTF-16BE 文件正常读取(v0.8.0 P1-3.3)', async () => { const r = (await tool.execute({ file_path: 'utf16be.txt' }, ctxFor(ws))) as { success: boolean; + content?: string; }; - expect(r.success).toBe(false); + expect(r.success).toBe(true); }); it('GBK 字节样本 → 降级 gbk 编码并正确解码', async () => { diff --git a/electron/harness/tools/built-in/browser-window-manager.ts b/electron/harness/tools/built-in/browser-window-manager.ts index c171ce1..fcc82f2 100644 --- a/electron/harness/tools/built-in/browser-window-manager.ts +++ b/electron/harness/tools/built-in/browser-window-manager.ts @@ -11,6 +11,9 @@ import { BrowserWindow, session } from 'electron'; import log from 'electron-log'; // v0.7.3 P2-2: CORS Origin 回显(纯函数在 network-utils,可表测) import { corsAllowOrigin, extractOriginHeader } from './network-utils'; +// v0.8.0 P1-4: 浏览器通道 SSRF 根治 —— 分区流量走本地 Pinned CONNECT 代理 +import { ensurePinnedProxy } from '../../../utils/pinned-proxy'; +import { isProxyActive } from '../../../utils/network-proxy'; /** Agent 浏览器专用 session partition — 与主应用 default session 完全隔离 */ const AGENT_PARTITION = 'persist:metona-agent-browser'; @@ -146,6 +149,28 @@ export class BrowserWindowManager { // 跨域 Origin / 无 Origin 一律不加 ACAO 头 —— 保持默认同源策略阻止读取, // 彻底堵死"第三方页面借该分区凭据读取跨域资源"的通道。 const agentSession = session.fromPartition(AGENT_PARTITION); + + // v0.8.0 P1-4 根治: 浏览器通道 SSRF —— 分区全部流量经本地 Pinned CONNECT + // 代理(入口同源校验 + 校验期 IP pinning,Chromium 端 TLS/SNI 端到端不变), + // 关闭"入口校验通过 → Chromium 独立 DNS 解析"之间的 rebinding 残余窗口。 + // 用户已显式配置 network.proxyUrl 时不启用(代理模式下 DNS 在代理端解析、 + // pinning 不可实现,保持 M7 已知限制口径);代理启动失败降级可用性优先。 + if (!isProxyActive()) { + const proxyPort = await ensurePinnedProxy(); + if (proxyPort) { + try { + await agentSession.setProxy({ + proxyRules: `http=127.0.0.1:${proxyPort};https=127.0.0.1:${proxyPort}`, + }); + log.info( + `[BrowserWindowManager] agent partition routed via pinned proxy :${proxyPort}`, + ); + } catch (err) { + log.warn('[BrowserWindowManager] pinned proxy setProxy failed:', err); + } + } + } + agentSession.webRequest.onHeadersReceived((details, callback) => { // Electron 类型在此版本的 OnHeadersReceivedListenerDetails 上不暴露 // requestHeaders —— 显式声明读取面(Origin 大小写不敏感提取) @@ -308,21 +333,48 @@ export class BrowserWindowManager { * 各自直接 open/evaluate 共享单例,后到者销毁前者的窗口(崩溃根因, * 见 openChain 注释)。排队后同一时刻只有一个抓取在使用窗口。 * + * v0.8.0 P1-2: 新增可选 abort signal —— 中断时停止页面加载(webContents.stop) + * 并在序列各步骤间检查,终止整个 open→等待→evaluate 链。 + * * @returns 提取的页面正文;失败/内容过短返回 null(调用方走失败路径) */ - async fetchPageText(url: string): Promise { + async fetchPageText(url: string, signal?: AbortSignal): Promise { const run = async (): Promise => { - await this.openInternal({ url }); - await this.sleep(2_500); - const text = (await this.evaluate(` - (function() { - var clone = document.body.cloneNode(true); - var noise = clone.querySelectorAll('script, style, noscript, nav, header, footer, aside, iframe, svg'); - noise.forEach(function(el) { el.remove(); }); - return clone.innerText || ''; - })(); - `)) as string; - return text && text.trim() ? text : null; + const throwIfAborted = (): void => { + if (signal?.aborted) throw new Error('Browser fetch aborted'); + }; + // abort 时停止页面加载 —— loadURL/evaluate 会以 ERR_ABORTED 拒绝, + // 序列随之终止(窗口由单例管理,不在此销毁) + let onAbort: (() => void) | null = null; + if (signal && !signal.aborted) { + onAbort = () => { + try { + this.win?.webContents.stop(); + } catch { + /* 窗口可能已销毁 */ + } + }; + signal.addEventListener('abort', onAbort, { once: true }); + } + try { + throwIfAborted(); + await this.openInternal({ url }); + throwIfAborted(); + await this.sleep(2_500); + throwIfAborted(); + const text = (await this.evaluate(` + (function() { + var clone = document.body.cloneNode(true); + var noise = clone.querySelectorAll('script, style, noscript, nav, header, footer, aside, iframe, svg'); + noise.forEach(function(el) { el.remove(); }); + return clone.innerText || ''; + })(); + `)) as string; + throwIfAborted(); + return text && text.trim() ? text : null; + } finally { + if (signal && onAbort) signal.removeEventListener('abort', onAbort); + } }; this.openChain = this.openChain.then(run, run); return this.openChain as Promise; diff --git a/electron/harness/tools/built-in/code-search.ts b/electron/harness/tools/built-in/code-search.ts index d383250..5c9243c 100644 --- a/electron/harness/tools/built-in/code-search.ts +++ b/electron/harness/tools/built-in/code-search.ts @@ -40,7 +40,8 @@ export class CodeSearchTool implements IMetonaTool { readonly definition: MetonaToolDef = { name: 'code_search', - description: 'Search code using ripgrep. Supports regex patterns, file type filtering, and context lines. Much faster than search_files for large codebases. Falls back to JS implementation if ripgrep is not installed.', + description: + 'Search code using ripgrep. Supports regex patterns, file type filtering, and context lines. Much faster than search_files for large codebases. Falls back to JS implementation if ripgrep is not installed.', parameters: { type: 'object', properties: { @@ -48,8 +49,14 @@ export class CodeSearchTool implements IMetonaTool { path: { type: 'string', description: 'Search directory (default: workspace root)' }, file_glob: { type: 'string', description: 'File name glob filter (e.g., "*.ts", "*.py")' }, case_sensitive: { type: 'boolean', description: 'Case sensitive search (default false)' }, - context_before: { type: 'number', description: 'Lines of context before match (default 0, max 5)' }, - context_after: { type: 'number', description: 'Lines of context after match (default 0, max 5)' }, + context_before: { + type: 'number', + description: 'Lines of context before match (default 0, max 5)', + }, + context_after: { + type: 'number', + description: 'Lines of context after match (default 0, max 5)', + }, max_results: { type: 'number', description: 'Maximum results (default 50, max 200)' }, }, required: ['pattern'], @@ -98,7 +105,13 @@ export class CodeSearchTool implements IMetonaTool { private async searchWithRipgrep( pattern: string, searchPath: string, - opts: { fileGlob?: string; caseSensitive: boolean; contextBefore: number; contextAfter: number; maxResults: number }, + opts: { + fileGlob?: string; + caseSensitive: boolean; + contextBefore: number; + contextAfter: number; + maxResults: number; + }, context: ToolExecutionContext, ): Promise { const rgArgs: string[] = ['--json']; @@ -118,12 +131,31 @@ export class CodeSearchTool implements IMetonaTool { const { stdout } = await execFileAsync('rg', rgArgs, { maxBuffer: 10 * 1024 * 1024, timeout: 25_000, + // v0.8.0 P1-2: 引擎级 abort 信号 —— 用户中断会话时立即 kill rg 子进程 + //(Node execFile 原生支持 signal 选项,触发时以 AbortError 拒绝) + signal: context.signal, }); const results = this.parseRipgrepJsonOutput(stdout); - return { results: results.slice(0, opts.maxResults), count: results.length, engine: 'ripgrep' }; + return { + results: results.slice(0, opts.maxResults), + count: results.length, + engine: 'ripgrep', + }; } catch (error) { - const err = error as { code?: number; signal?: string; stdout?: string; stderr?: string; killed?: boolean; message?: string }; + const err = error as { + code?: number; + signal?: string; + stdout?: string; + stderr?: string; + killed?: boolean; + message?: string; + name?: string; + }; + // v0.8.0 P1-2: 引擎 abort → 直接以中断语义返回(不再回退 JS 搜索) + if (err.name === 'AbortError' || context.signal?.aborted) { + return { results: [], count: 0, error: 'Search aborted', engine: 'ripgrep', aborted: true }; + } // rg 退出码 1 = 无匹配,不是错误 if (err.code === 1) { return { results: [], count: 0, engine: 'ripgrep' }; @@ -148,10 +180,24 @@ export class CodeSearchTool implements IMetonaTool { before?: string[]; after?: string[]; }> { - const results: Array<{ path: string; line: number; column: number; match: string; before?: string[]; after?: string[] }> = []; + const results: Array<{ + path: string; + line: number; + column: number; + match: string; + before?: string[]; + after?: string[]; + }> = []; const lines = output.split('\n').filter((l) => l.trim()); - let currentMatch: { path: string; line: number; column: number; match: string; before?: string[]; after?: string[] } | null = null; + let currentMatch: { + path: string; + line: number; + column: number; + match: string; + before?: string[]; + after?: string[]; + } | null = null; let beforeBuffer: string[] = []; let afterBuffer: string[] = []; @@ -185,7 +231,9 @@ export class CodeSearchTool implements IMetonaTool { } const text = (data.lines as { text?: string } | undefined)?.text ?? ''; - const submatches = (data.submatches as Array<{ match: { text?: string }; start?: number }> | undefined) ?? []; + const submatches = + (data.submatches as Array<{ match: { text?: string }; start?: number }> | undefined) ?? + []; const matchText = submatches[0]?.match?.text ?? text; const column = (submatches[0]?.start ?? 0) + 1; @@ -214,18 +262,33 @@ export class CodeSearchTool implements IMetonaTool { private async searchWithJs( pattern: string, searchPath: string, - opts: { fileGlob?: string; caseSensitive: boolean; contextBefore: number; contextAfter: number; maxResults: number }, + opts: { + fileGlob?: string; + caseSensitive: boolean; + contextBefore: number; + contextAfter: number; + maxResults: number; + }, context: ToolExecutionContext, ): Promise { // 动态导入以避免循环依赖 const { SearchFilesTool } = await import('./filesystem'); const searchTool = new SearchFilesTool(); - return searchTool.execute({ - pattern, - target: 'content', - path: searchPath, - file_glob: opts.fileGlob, - limit: opts.maxResults, - }, context); + // v0.8.0 P1-3.5 根治: 回退路径此前静默丢失 case_sensitive / context_before / + // context_after —— 同一调用在 ripgrep 存在与否会产生不同结果(行为不对称)。 + // SearchFilesTool 现已支持这三个参数(filesystem.ts P1-3.5),原样透传。 + return searchTool.execute( + { + pattern, + target: 'content', + path: searchPath, + file_glob: opts.fileGlob, + limit: opts.maxResults, + case_sensitive: opts.caseSensitive, + context_before: opts.contextBefore, + context_after: opts.contextAfter, + }, + context, + ); } } diff --git a/electron/harness/tools/built-in/delegate-task.ts b/electron/harness/tools/built-in/delegate-task.ts index 6303bb5..39dc921 100644 --- a/electron/harness/tools/built-in/delegate-task.ts +++ b/electron/harness/tools/built-in/delegate-task.ts @@ -15,18 +15,21 @@ import type { TaskOrchestrator } from '../../orchestration/orchestrator'; export class DelegateTaskTool implements IMetonaTool { readonly definition: MetonaToolDef = { name: 'delegate_task', - description: 'Delegate a sub-task to an independent SubAgent for parallel or isolated execution. The SubAgent runs its own ReAct loop and returns the final result. Use this for complex sub-tasks that benefit from focused reasoning.', + description: + 'Delegate a sub-task to an independent SubAgent for parallel or isolated execution. The SubAgent runs its own ReAct loop and returns the final result. Use this for complex sub-tasks that benefit from focused reasoning.', parameters: { type: 'object', properties: { description: { type: 'string', - description: 'Clear and detailed description of the sub-task to delegate. This will be the SubAgent\'s user message.', + description: + "Clear and detailed description of the sub-task to delegate. This will be the SubAgent's user message.", }, tools: { type: 'array', items: { type: 'string', description: 'Tool name' }, - description: 'Optional whitelist of tool names the SubAgent can use (e.g. ["read_file", "web_search"]). If omitted, SubAgent inherits all tools.', + description: + 'Optional whitelist of tool names the SubAgent can use (e.g. ["read_file", "web_search"]). If omitted, SubAgent inherits all tools.', }, maxIterations: { type: 'number', @@ -58,6 +61,8 @@ export class DelegateTaskTool implements IMetonaTool { parentSessionId: context.sessionId, maxIterations, tools, + // v0.8.0 P1-2: 引擎级 abort 信号透传 —— 用户中断会话时联动终止 SubAgent + signal: context.signal, }); return { diff --git a/electron/harness/tools/built-in/file-editor.ts b/electron/harness/tools/built-in/file-editor.ts index b87c400..1fa8977 100644 --- a/electron/harness/tools/built-in/file-editor.ts +++ b/electron/harness/tools/built-in/file-editor.ts @@ -19,6 +19,7 @@ import { readFile, writeFile, rename, mkdir, stat, unlink } from 'fs/promises'; import { dirname } from 'path'; import { existsSync } from 'fs'; +import { nanoid } from 'nanoid'; import type { IMetonaTool, ToolExecutionContext } from '../../types/metona-tool'; import type { MetonaToolDef } from '../../../harness/types'; import { MetonaToolCategory, MetonaRiskLevel } from '../../../harness/types'; @@ -422,7 +423,8 @@ export class FileEditorTool implements IMetonaTool { const newContent = newLines.join('\n'); // F1.7: 原子写入 - 临时文件 + rename - const tmpPath = `${filePath}.tmp_${Date.now()}`; + // v0.8.0 P1-3.4: tmp 文件名追加 nanoid 随机段(同毫秒并发写防碰撞) + const tmpPath = `${filePath}.tmp_${Date.now()}_${nanoid(6)}`; try { await writeFile(tmpPath, newContent, 'utf-8'); await rename(tmpPath, filePath); diff --git a/electron/harness/tools/built-in/filesystem.ts b/electron/harness/tools/built-in/filesystem.ts index cc213c7..3123708 100644 --- a/electron/harness/tools/built-in/filesystem.ts +++ b/electron/harness/tools/built-in/filesystem.ts @@ -38,6 +38,7 @@ import { } from 'fs/promises'; import { join, relative, resolve, dirname } from 'path'; import { existsSync, realpathSync } from 'fs'; +import { nanoid } from 'nanoid'; import type { IMetonaTool, ToolExecutionContext } from '../../types/metona-tool'; import type { MetonaToolDef } from '../../../harness/types'; import { MetonaToolCategory, MetonaRiskLevel } from '../../../harness/types'; @@ -61,6 +62,13 @@ import { * v0.3.2 修复 FAIL: 使用 bytesRead 限制循环上界,而非 buffer.length。 * Buffer.alloc(8192) 初始化为全 0,fd.read 只覆盖 [0, bytesRead) 区间, * 若用 buffer.length 遍历会误判所有 < 8192 字节的纯文本文件为二进制。 + * + * v0.8.0 P1-3.3 根治: UTF-16 文件(LE: FF FE / BE: FE FF BOM)几乎每个字符 + * 都含 0x00 字节 —— NULL 字节启发式把一切 UTF-16 文本判为二进制并整体拒绝, + * 使 decodeBufferWithDetection 的 UTF-16 解码分支不可达(原缺陷被 + * filesystem-tools.test.ts 锁定留档,本次随修复同步改约)。现先检测 UTF-16 + * BOM:命中即视为文本(UTF-16 解码由 decodeBufferWithDetection 承担), + * 未命中 BOM 才走 NULL 字节启发式。 */ async function isBinaryFile(filePath: string): Promise { // M-20 修复: 使用 finally 块确保文件句柄关闭,防止 fd 泄漏 @@ -69,6 +77,9 @@ async function isBinaryFile(filePath: string): Promise { const buffer = Buffer.alloc(8192); fd = await open(filePath, 'r'); const { bytesRead } = await fd.read(buffer, 0, 8192, 0); + // v0.8.0 P1-3.3: UTF-16 LE/BE BOM → 文本(NUL 启发式对其无效) + if (bytesRead >= 2 && buffer[0] === 0xff && buffer[1] === 0xfe) return false; // UTF-16 LE BOM + if (bytesRead >= 2 && buffer[0] === 0xfe && buffer[1] === 0xff) return false; // UTF-16 BE BOM // 仅检查实际读取的字节范围 [0, bytesRead) // 含 NULL 字节 → 二进制;空文件(bytesRead=0)→ 非二进制 for (let i = 0; i < bytesRead; i++) { @@ -346,7 +357,9 @@ export class WriteFileTool implements IMetonaTool { } // v0.3.2: overwrite 模式使用原子写入(临时文件 + rename,与 file_editor 一致) - const tmpPath = `${filePath}.tmp_${Date.now()}`; + // v0.8.0 P1-3.4: tmp 文件名追加 nanoid 随机段 —— 同毫秒并发写同一文件时 + // Date.now() 精度的 tmp 名会互相覆盖(并行工具调用场景) + const tmpPath = `${filePath}.tmp_${Date.now()}_${nanoid(6)}`; try { await writeFile(tmpPath, content, 'utf-8'); await rename(tmpPath, filePath); @@ -395,6 +408,11 @@ export class ListDirectoryTool implements IMetonaTool { type: 'boolean', description: 'Include hidden files/dirs starting with "." (default false)', }, + include_node_modules: { + type: 'boolean', + description: + 'Include node_modules directories (default false — skipped for performance; set true to inspect dependency packages)', + }, }, }, category: MetonaToolCategory.FILESYSTEM, @@ -411,6 +429,8 @@ export class ListDirectoryTool implements IMetonaTool { const depth = Math.min(5, Math.max(1, (args.depth as number) ?? 1)); const glob = args.glob as string | undefined; const includeHidden = (args.include_hidden as boolean) ?? false; + // v0.8.0 P1-3.6: node_modules 可选纳入(默认跳过保持兼容) + const includeNodeModules = (args.include_node_modules as boolean) ?? false; // v0.3.2 修复 WARN-3: listDir 内部提前终止,避免大目录全量遍历 const MAX_ENTRIES = 1000; @@ -421,7 +441,17 @@ export class ListDirectoryTool implements IMetonaTool { size?: number; modified?: string; }> = []; - await this.listDir(dirPath, dirPath, depth, glob, includeHidden, 0, entries, MAX_ENTRIES); + await this.listDir( + dirPath, + dirPath, + depth, + glob, + includeHidden, + includeNodeModules, + 0, + entries, + MAX_ENTRIES, + ); return { entries, count: entries.length, @@ -439,6 +469,7 @@ export class ListDirectoryTool implements IMetonaTool { maxDepth: number, glob: string | undefined, includeHidden: boolean, + includeNodeModules: boolean, currentDepth: number, results: Array<{ name: string; path: string; type: string; size?: number; modified?: string }>, maxEntries: number, @@ -453,8 +484,9 @@ export class ListDirectoryTool implements IMetonaTool { if (results.length >= maxEntries) return; // v0.3.2: include_hidden 参数控制隐藏文件 if (!includeHidden && entry.name.startsWith('.')) continue; - // node_modules 始终跳过(太大) - if (entry.name === 'node_modules') continue; + // v0.8.0 P1-3.6: node_modules 默认跳过(太大);include_node_modules=true + // 时纳入(分析依赖包场景;上限保护仍生效) + if (entry.name === 'node_modules' && !includeNodeModules) continue; const fullPath = join(dirPath, entry.name); // 相对路径始终以根请求目录为基准 @@ -470,6 +502,7 @@ export class ListDirectoryTool implements IMetonaTool { maxDepth, glob, includeHidden, + includeNodeModules, currentDepth + 1, results, maxEntries, @@ -526,6 +559,19 @@ export class SearchFilesTool implements IMetonaTool { description: 'Lines of context to show around content matches (default 0, max 5). Only for target="content".', }, + case_sensitive: { + type: 'boolean', + description: + 'Case sensitive search (default false — v0.8.0 P1-3.5: previously hardcoded case-insensitive, breaking code_search fallback symmetry)', + }, + context_before: { + type: 'number', + description: 'Lines of context before match (default = context_lines, max 5)', + }, + context_after: { + type: 'number', + description: 'Lines of context after match (default = context_lines, max 5)', + }, include_hidden: { type: 'boolean', description: 'Include hidden files/dirs starting with "." (default false)', @@ -549,6 +595,15 @@ export class SearchFilesTool implements IMetonaTool { const fileGlob = args.file_glob as string | undefined; const limit = Math.min(200, Math.max(1, (args.limit as number) ?? 50)); const contextLines = Math.min(5, Math.max(0, (args.context_lines as number) ?? 0)); + // v0.8.0 P1-3.5: 与 code_search 参数对称 —— case_sensitive 此前被硬编码 + // 'gi' 忽略;context_before/after 此前在 code_search JS 回退路径静默丢失。 + // 缺省时回退 context_lines(向后兼容)。 + const caseSensitive = (args.case_sensitive as boolean) ?? false; + const contextBefore = Math.min( + 5, + Math.max(0, (args.context_before as number) ?? contextLines), + ); + const contextAfter = Math.min(5, Math.max(0, (args.context_after as number) ?? contextLines)); const includeHidden = (args.include_hidden as boolean) ?? false; if (target === 'files') { @@ -566,7 +621,9 @@ export class SearchFilesTool implements IMetonaTool { pattern, fileGlob, limit, - contextLines, + contextBefore, + contextAfter, + caseSensitive, includeHidden, context.workspacePath, ); @@ -606,7 +663,9 @@ export class SearchFilesTool implements IMetonaTool { pattern: string, fileGlob: string | undefined, limit: number, - contextLines: number, + contextBefore: number, + contextAfter: number, + caseSensitive: boolean, includeHidden: boolean, workspacePath: string, ): Promise { @@ -635,7 +694,8 @@ export class SearchFilesTool implements IMetonaTool { } let regex: RegExp; try { - regex = new RegExp(pattern, 'gi'); + // v0.8.0 P1-3.5: case_sensitive 参数生效(此前硬编码 'gi' 恒大小写不敏感) + regex = new RegExp(pattern, caseSensitive ? 'g' : 'gi'); } catch { return { results: [], count: 0, error: `Invalid regex pattern: ${pattern}`, success: false }; } @@ -668,10 +728,11 @@ export class SearchFilesTool implements IMetonaTool { if (regex.test(lines[i])) { const match = lines[i].trim().slice(0, 200); // v0.3.2: context lines 支持 + // v0.8.0 P1-3.5: 前后文行数独立(code_search 对称性) let context: string[] | undefined; - if (contextLines > 0) { - const start = Math.max(0, i - contextLines); - const end = Math.min(lines.length - 1, i + contextLines); + if (contextBefore > 0 || contextAfter > 0) { + const start = Math.max(0, i - contextBefore); + const end = Math.min(lines.length - 1, i + contextAfter); context = lines.slice(start, end + 1); } results.push({ diff --git a/electron/harness/tools/built-in/git.ts b/electron/harness/tools/built-in/git.ts index 2246383..2a0754f 100644 --- a/electron/harness/tools/built-in/git.ts +++ b/electron/harness/tools/built-in/git.ts @@ -28,18 +28,23 @@ const MAX_DIFF_SIZE = 50 * 1024; * * v0.3.1 修复 WARN-4: 允许调用方覆盖 maxBuffer,默认 1MB; * git_diff 传 5MB 以支持超大变更集(避免 maxBuffer exceeded 报错而非截断返回)。 + * + * v0.8.0 P1-2: 新增可选 signal —— 引擎中断(用户停止/超时)时立即 kill git + * 子进程(Node execFile 原生支持 AbortSignal 选项),各工具经 context 透传。 */ async function runGit( args: string[], cwd: string, timeout = 10_000, maxBuffer = 1024 * 1024, // 默认 1MB + signal?: AbortSignal, ): Promise { const { stdout } = await execFileAsync('git', args, { cwd, maxBuffer, timeout, encoding: 'utf-8', + ...(signal ? { signal } : {}), }); return stdout; } @@ -84,11 +89,15 @@ function extractRenamedFile(file: string): string { export class GitStatusTool implements IMetonaTool { readonly definition: MetonaToolDef = { name: 'git_status', - description: 'Show the working tree status. Returns current branch, ahead/behind counts, and lists of staged, unstaged, and untracked files.', + description: + 'Show the working tree status. Returns current branch, ahead/behind counts, and lists of staged, unstaged, and untracked files.', parameters: { type: 'object', properties: { - pathspec: { type: 'string', description: 'Limit to a path scope, e.g. "src/" or "package.json"' }, + pathspec: { + type: 'string', + description: 'Limit to a path scope, e.g. "src/" or "package.json"', + }, }, }, category: MetonaToolCategory.FILESYSTEM, @@ -191,11 +200,15 @@ export class GitStatusTool implements IMetonaTool { export class GitDiffTool implements IMetonaTool { readonly definition: MetonaToolDef = { name: 'git_diff', - description: 'Show changes between the working tree and the index (or staged changes). Returns unified diff text. Output is truncated at 50KB.', + description: + 'Show changes between the working tree and the index (or staged changes). Returns unified diff text. Output is truncated at 50KB.', parameters: { type: 'object', properties: { - cached: { type: 'boolean', description: 'Show only staged (cached) changes (default false)' }, + cached: { + type: 'boolean', + description: 'Show only staged (cached) changes (default false)', + }, pathspec: { type: 'string', description: 'Limit to a path scope, e.g. "src/"' }, contextLines: { type: 'number', description: 'Number of context lines (default 3)' }, }, @@ -269,7 +282,8 @@ export class GitDiffTool implements IMetonaTool { export class GitLogTool implements IMetonaTool { readonly definition: MetonaToolDef = { name: 'git_log', - description: 'Show commit history. Returns commits with hash, message, and optionally author/date. Supports filtering by author and path.', + description: + 'Show commit history. Returns commits with hash, message, and optionally author/date. Supports filtering by author and path.', parameters: { type: 'object', properties: { @@ -357,17 +371,25 @@ export class GitLogTool implements IMetonaTool { export class GitCommitTool implements IMetonaTool { readonly definition: MetonaToolDef = { name: 'git_commit', - description: 'Stage files and create a git commit. Supports amending the previous commit. In amend mode, message is optional (omitting it keeps the original commit message). Requires user confirmation.', + description: + 'Stage files and create a git commit. Supports amending the previous commit. In amend mode, message is optional (omitting it keeps the original commit message). Requires user confirmation.', parameters: { type: 'object', properties: { - message: { type: 'string', description: 'Commit message. Required for new commits. Optional in amend mode (omitting keeps original message).' }, + message: { + type: 'string', + description: + 'Commit message. Required for new commits. Optional in amend mode (omitting keeps original message).', + }, files: { type: 'array', items: { type: 'string', description: 'File path to stage' }, description: 'Files to stage (default: all changes via git add -A)', }, - amend: { type: 'boolean', description: 'Amend the previous commit instead of creating a new one (default false)' }, + amend: { + type: 'boolean', + description: 'Amend the previous commit instead of creating a new one (default false)', + }, }, // F1-3: message 不再硬性 required — amend 模式下可省略(保留原 message) // 校验在 execute 内根据 amend 标志动态进行 @@ -385,11 +407,17 @@ export class GitCommitTool implements IMetonaTool { // F1-3: 非 amend 模式必须有 message;amend 模式可省略(保留原 message) if (!amend && (!message || message.trim().length === 0)) { - return { success: false, error: 'Commit message is required (use amend: true to reuse the previous message)' }; + return { + success: false, + error: 'Commit message is required (use amend: true to reuse the previous message)', + }; } // amend 模式下若提供 message,需校验非空字符串 if (amend && message !== undefined && message.trim().length === 0) { - return { success: false, error: 'Commit message must not be empty (or omit message field to keep original)' }; + return { + success: false, + error: 'Commit message must not be empty (or omit message field to keep original)', + }; } try { @@ -402,7 +430,10 @@ export class GitCommitTool implements IMetonaTool { return { success: false, error: `File path outside workspace: ${f}`, - commit: '', branch: '', message: '', filesChanged: 0, + commit: '', + branch: '', + message: '', + filesChanged: 0, }; } } diff --git a/electron/harness/tools/built-in/http-request.ts b/electron/harness/tools/built-in/http-request.ts index 3a2709b..c5411b7 100644 --- a/electron/harness/tools/built-in/http-request.ts +++ b/electron/harness/tools/built-in/http-request.ts @@ -82,7 +82,7 @@ export class HttpRequestTool implements IMetonaTool { timeoutMs: 30_000, }; - async execute(args: Record, _context: ToolExecutionContext): Promise { + async execute(args: Record, context: ToolExecutionContext): Promise { try { const url = args.url as string; const method = ((args.method as string) ?? 'GET').toUpperCase(); @@ -127,7 +127,9 @@ export class HttpRequestTool implements IMetonaTool { // v0.7.3 P2-1: pinned fetch —— 校验通过的 IP pin 到连接层, // 关闭校验-连接之间的 DNS rebinding 窗口 - const response = await ssrfPinnedFetch(url, fetchOptions, timeout); + // v0.8.0 P1-2: 第 4 参透传引擎级 abort signal —— 用户中断会话时底层 + // fetch 立即取消(此前最长 60s 资源悬挂) + const response = await ssrfPinnedFetch(url, fetchOptions, timeout, context.signal); const text = await response.text(); // 截断到 50KB diff --git a/electron/harness/tools/built-in/network-utils.ts b/electron/harness/tools/built-in/network-utils.ts index 20b04c2..1814019 100644 --- a/electron/harness/tools/built-in/network-utils.ts +++ b/electron/harness/tools/built-in/network-utils.ts @@ -83,13 +83,23 @@ export async function fetchWithTimeout( url: string, options: RequestInit = {}, timeoutMs = 20_000, + externalSignal?: AbortSignal, ): Promise { + // v0.8.0 P1-2: 合并外部 abort 信号(引擎中断时立即取消,不再跑到自身超时) const controller = new AbortController(); + const onExternalAbort = (): void => controller.abort(); + if (externalSignal) { + if (externalSignal.aborted) controller.abort(); + else externalSignal.addEventListener('abort', onExternalAbort, { once: true }); + } const timer = setTimeout(() => controller.abort(), timeoutMs); try { return await fetch(url, { ...options, signal: controller.signal }); } finally { clearTimeout(timer); + if (externalSignal) { + externalSignal.removeEventListener('abort', onExternalAbort); + } } } diff --git a/electron/harness/tools/built-in/ssrf-guard.ts b/electron/harness/tools/built-in/ssrf-guard.ts index 2003d42..254ce4c 100644 --- a/electron/harness/tools/built-in/ssrf-guard.ts +++ b/electron/harness/tools/built-in/ssrf-guard.ts @@ -17,6 +17,12 @@ import { lookup } from 'node:dns/promises'; import { isIP } from 'node:net'; +/** + * v0.8.0 P1-5: DNS lookup 可替换点 —— 生产恒为 node:dns/promises 的 lookup; + * 测试通过替换 current 注入受控解析结果(ESM namespace 只读,无法直接打桩)。 + */ +export const __dnsLookup: { current: typeof lookup } = { current: lookup }; + /** * 检查 IP 是否为私有/内网/回环/元数据地址 * @@ -98,7 +104,7 @@ export async function resolvePublicAddresses(url: string): Promise { // 域名 — DNS 解析后检测所有 IP let addresses: Array<{ address: string }>; try { - addresses = await lookup(hostname, { all: true }); + addresses = await __dnsLookup.current(hostname, { all: true }); } catch (err) { throw new Error( `Blocked SSRF: DNS resolution failed for ${hostname}: ${(err as Error).message}`, @@ -231,3 +237,85 @@ export function assertSafeConfigTarget(url: string): void { } // 域名(非 IP):允许(DNS 可能解析到内网,但用户显式配置的本地服务是合法场景) } + +/** + * v0.8.0 P1-5: 配置类 URL 的**深校验** —— 在 assertSafeConfigTarget 的静态规则 + * 之上对域名做真实 DNS 解析,任一解析结果命中高危段(链路本地/云元数据、0/8、 + * 组播保留、::ffff: 映射等价段)即拒绝。 + * + * 与 assertSafeConfigTarget 的差异:后者对"域名放行"(不解析 DNS,本地实例 + * 域名合法);本函数补上"域名解析到云元数据 IP"的绕过窗口 + * (如 attacker.example 解析到 169.254.169.254)。 + * + * 放行语义不变:127.0.0.1 / RFC1918 私网(本地 MCP/SearXNG 合法);DNS 解析 + * 失败也放行(配置期校验为纵深手段 —— 离线配置合法,且运行时工具路径仍有 + * 独立校验;失败仅 WARN 由调用方留痕)。 + * + * @throws 如果 URL 解析结果指向高危目标或协议不被允许 + */ +export async function assertSafeConfigTargetDeep(url: string): Promise { + // 先执行静态规则(协议白名单/元数据主机名/IP 字面量段位) + assertSafeConfigTarget(url); + + let parsed: URL; + try { + parsed = new URL(url); + } catch { + return; // assertSafeConfigTarget 已抛,此处不可达(防御) + } + const rawHostname = parsed.hostname.toLowerCase(); + // IPv6 字面量去方括号(与 assertSafeConfigTarget 同口径) + const hostname = + rawHostname.startsWith('[') && rawHostname.endsWith(']') + ? rawHostname.slice(1, -1) + : rawHostname; + if (isIP(hostname) !== 0) return; // IP 字面量已由静态规则判定,无需解析 + + let addresses: Array<{ address: string }>; + try { + addresses = await __dnsLookup.current(hostname, { all: true }); + } catch (err) { + // DNS 解析失败:放行(配置期纵深校验不做可用性裁决),由调用方日志留痕 + throw new DeepCheckSoftFailure( + `config target DNS resolution failed for ${hostname}: ${(err as Error).message}`, + ); + } + for (const { address } of addresses) { + // 复用 isPrivateIP 的完整段位表做"高危段"判定?——不行:isPrivateIP 把 + // 回环/RFC1918 也判为私有,而配置路径放行它们。此处仅拦"静态规则拦不到、 + // 但解析后才暴露"的高危段:链路本地(含云元数据)、0/8、组播保留。 + if (isHighRiskResolvedIP(address)) { + throw new Error( + `Blocked: ${hostname} resolves to high-risk address ${address} (link-local/metadata/multicast)`, + ); + } + } +} + +/** 高危段判定(仅配置深校验使用):链路本地/0/8/组播保留,放行回环与 RFC1918 */ +function isHighRiskResolvedIP(ip: string): boolean { + if (isIP(ip) === 4) { + const parts = ip.split('.').map(Number); + return (parts[0] === 169 && parts[1] === 254) || parts[0] === 0 || parts[0] >= 224; + } + if (isIP(ip) === 6) { + const lower = ip.toLowerCase(); + if (lower.startsWith('fe80:') || lower.startsWith('ff')) return true; + if (lower === '::') return true; + const v4MappedMatch = lower.match(/::ffff:(\d+\.\d+\.\d+\.\d+)$/); + if (v4MappedMatch) return isHighRiskResolvedIP(v4MappedMatch[1]); + return false; + } + return false; +} + +/** + * v0.8.0 P1-5: 深校验的"软失败"信号 —— DNS 解析失败不算配置错误(放行), + * 但调用方需要区分"校验通过"与"跳过校验"以便日志留痕。 + */ +export class DeepCheckSoftFailure extends Error { + constructor(message: string) { + super(message); + this.name = 'DeepCheckSoftFailure'; + } +} diff --git a/electron/harness/tools/built-in/web-fetch.ts b/electron/harness/tools/built-in/web-fetch.ts index d53772b..6e1f878 100644 --- a/electron/harness/tools/built-in/web-fetch.ts +++ b/electron/harness/tools/built-in/web-fetch.ts @@ -84,7 +84,7 @@ export class WebFetchTool implements IMetonaTool { timeoutMs: 240_000, }; - async execute(args: Record, _context: ToolExecutionContext): Promise { + async execute(args: Record, context: ToolExecutionContext): Promise { const url = args.url as string; const mobileUA = (args.mobile_ua as boolean) ?? false; const enableRetry = (args.retry as boolean) ?? true; @@ -96,6 +96,11 @@ export class WebFetchTool implements IMetonaTool { return { url, content: '', success: false, error: 'URL must start with http:// or https://' }; } + // v0.8.0 P1-2: abort 立即生效 —— 引擎中断(超时/用户停止)时不再继续后续阶段 + if (context.signal?.aborted) { + return { url, content: '', success: false, error: 'Fetch aborted' }; + } + // v0.6.4 P2-2: SSRF 校验 —— 覆盖 Phase1 HTTP 与 Phase3 浏览器两条通道的入口。 // 协议白名单 / 私有段 IP / 云元数据地址一律拒绝。 try { @@ -117,7 +122,12 @@ export class WebFetchTool implements IMetonaTool { logTool('web_fetch', `Fetching: ${url}`); // ===== Phase 1: HTTP 抓取 ===== - const phase1Result = await this.httpFetch(url, mobileUA, enableRetry); + const phase1Result = await this.httpFetch(url, mobileUA, enableRetry, context.signal); + + // v0.8.0 P1-2: HTTP 阶段被 abort —— 直接终止,不进浏览器回退 + if (context.signal?.aborted) { + return { url, content: '', success: false, error: 'Fetch aborted' }; + } if (phase1Result.success && !phase1Result.intercepted) { // 根据 extract_mode 选择返回内容:'html' 模式返回清理后的 HTML,'text' 模式返回纯文本 @@ -136,7 +146,7 @@ export class WebFetchTool implements IMetonaTool { 'web_fetch', `Phase 2: Content too short (${phase1Content.length} chars), upgrading to browser`, ); - const browserResult = await this.browserFetch(url); + const browserResult = await this.browserFetch(url, context.signal); if (browserResult) { return this.buildSuccess(url, browserResult, 'browser', maxChars); } @@ -159,7 +169,7 @@ export class WebFetchTool implements IMetonaTool { }; } logTool('web_fetch', `Phase 3: Falling back to browser (${phase1Result.reason})`); - const browserResult = await this.browserFetch(url); + const browserResult = await this.browserFetch(url, context.signal); if (browserResult) { return this.buildSuccess(url, browserResult, 'browser', maxChars); } @@ -179,6 +189,7 @@ export class WebFetchTool implements IMetonaTool { url: string, mobileUA: boolean, enableRetry: boolean, + signal?: AbortSignal, ): Promise<{ success: boolean; html: string; @@ -206,7 +217,14 @@ export class WebFetchTool implements IMetonaTool { let redirectBlocked: string | null = null; for (let hop = 0; hop <= MAX_REDIRECT_HOPS; hop++) { - response = await ssrfPinnedFetch(currentUrl, { headers, redirect: 'manual' }, 20_000); + // v0.8.0 P1-2: 每跳透传引擎级 abort signal —— 用户中断时底层 fetch + // 立即取消(含重定向链中间跳) + response = await ssrfPinnedFetch( + currentUrl, + { headers, redirect: 'manual' }, + 20_000, + signal, + ); const next = resolveRedirectTarget(response, currentUrl); if (next === null) break; // 非重定向(或无/非法 Location)—— 当前响应即终态 if (hop === MAX_REDIRECT_HOPS) { @@ -261,6 +279,7 @@ export class WebFetchTool implements IMetonaTool { if (response.status >= 500 && attempt < maxRetries - 1) { await this.sleep( backoffBase * Math.pow(2, attempt) + Math.random() * backoffBase * 0.6, + signal, ); continue; } @@ -293,9 +312,22 @@ export class WebFetchTool implements IMetonaTool { return { success: true, html, text, intercepted: false, reason: '' }; } catch (err) { const errorMsg = (err as Error).message; + // v0.8.0 P1-2: abort 语义 —— 不重试、不睡眠,直接终止本阶段 + if (signal?.aborted || (err as Error & { name?: string }).name === 'AbortError') { + return { + success: false, + html: '', + text: '', + intercepted: false, + reason: 'Fetch aborted', + }; + } if (attempt < maxRetries - 1) { logTool('web_fetch', `Attempt ${attempt + 1} failed: ${errorMsg}, retrying...`); - await this.sleep(backoffBase * Math.pow(2, attempt) + Math.random() * backoffBase * 0.6); + await this.sleep( + backoffBase * Math.pow(2, attempt) + Math.random() * backoffBase * 0.6, + signal, + ); continue; } return { success: false, html: '', text: '', intercepted: false, reason: errorMsg }; @@ -313,7 +345,7 @@ export class WebFetchTool implements IMetonaTool { // ===== Phase 2/3: 浏览器回退(使用共享 BrowserWindowManager 单例) ===== - private async browserFetch(url: string): Promise { + private async browserFetch(url: string, signal?: AbortSignal): Promise { // 查缓存(浏览器阶段产出的是纯文本,与 text 模式同键) const cached = fetchCache.get(`text:${url}`); if (cached) { @@ -322,10 +354,13 @@ export class WebFetchTool implements IMetonaTool { } try { + // v0.8.0 P1-2: 引擎级 abort 透传 —— 中断时 BrowserWindowManager 停止页面 + // 加载并终止抓取序列(此前浏览器回退完全不受 abort 控制) + if (signal?.aborted) return null; // 崩溃修复: 走 manager.fetchPageText(内部串行化完整的 open→等待→evaluate 序列)。 // 原实现直接 open/evaluate 共享单例 —— web_search 并行抓取触发多个回退同时进入时, // 后到者销毁前者的窗口(ERR_ABORTED ×3 = 应用崩溃 ×3,见 manager 注释)。 - const text = await getBrowserManager().fetchPageText(url); + const text = await getBrowserManager().fetchPageText(url, signal); if (text && text.trim().length >= 80) { // 拦截检测(浏览器渲染后仍可能是验证码挑战页) @@ -383,7 +418,23 @@ export class WebFetchTool implements IMetonaTool { }; } - private sleep(ms: number): Promise { - return new Promise((resolve) => setTimeout(resolve, ms)); + /** v0.8.0 P1-2: abort 感知的 sleep —— 中断期间不再等待退避间隔 */ + private sleep(ms: number, signal?: AbortSignal): Promise { + return new Promise((resolve) => { + if (signal?.aborted) return resolve(); + const cleanup = (): void => { + clearTimeout(timer); + signal?.removeEventListener('abort', onAbort); + }; + const onAbort = (): void => { + cleanup(); + resolve(); + }; + const timer = setTimeout(() => { + cleanup(); + resolve(); + }, ms); + signal?.addEventListener('abort', onAbort, { once: true }); + }); } } diff --git a/electron/harness/tools/built-in/web-search.ts b/electron/harness/tools/built-in/web-search.ts index 70260cb..f184733 100644 --- a/electron/harness/tools/built-in/web-search.ts +++ b/electron/harness/tools/built-in/web-search.ts @@ -338,9 +338,15 @@ function parse360Regex(html: string): SearchResult[] { // 探测经 ssrfPinnedFetch(DNS pinning),重定向不自动跟随(3xx 即视为可达 —— // 链接活性已证明,且跟跳目标不再绕过校验)。 -async function checkReachability(urls: string[], concurrency = 5): Promise> { +async function checkReachability( + urls: string[], + signal?: AbortSignal, + concurrency = 5, +): Promise> { const result = new Map(); for (let i = 0; i < urls.length; i += concurrency) { + // v0.8.0 P1-2: abort 后立即终止预检批次 + if (signal?.aborted) break; const batch = urls.slice(i, i + concurrency); const checks = batch.map(async (url) => { try { @@ -349,7 +355,13 @@ async function checkReachability(urls: string[], concurrency = 5): Promise= 300 && resp.status < 400)); } catch { result.set(url, false); @@ -424,9 +436,14 @@ export class WebSearchTool implements IMetonaTool { private webFetchTool: WebFetchTool, ) {} - async execute(args: Record, _context: ToolExecutionContext): Promise { + async execute(args: Record, context: ToolExecutionContext): Promise { const query = args.query as string; + // v0.8.0 P1-2: abort 立即生效(后续各阶段均携带本 signal) + if (context.signal?.aborted) { + return { success: false, query, results: [], error: 'Search aborted' }; + } + // 读取 SearXNG 配置 const searxngConfig = this.readSearXNGConfig(); const useSearXNG = searxngConfig.enabled && !!searxngConfig.url; @@ -463,12 +480,23 @@ export class WebSearchTool implements IMetonaTool { let engineStats: Record; if (useSearXNG) { - const searxResult = await this.searchSearXNG(query, searxngConfig, maxResults, timeRange); + const searxResult = await this.searchSearXNG( + query, + searxngConfig, + maxResults, + timeRange, + context.signal, + ); results = searxResult.results; mode = 'searxng'; engineStats = searxResult.engineStats; } else { - const builtinResult = await this.searchBuiltinEngines(query, maxResults, timeRange); + const builtinResult = await this.searchBuiltinEngines( + query, + maxResults, + timeRange, + context.signal, + ); results = builtinResult.results; mode = 'builtin'; engineStats = builtinResult.engineStats; @@ -479,7 +507,7 @@ export class WebSearchTool implements IMetonaTool { // 可达性预检 const topUrls = deduped.slice(0, Math.min(20, deduped.length)).map((r) => r.url); - const reachabilityMap = await checkReachability(topUrls); + const reachabilityMap = await checkReachability(topUrls, context.signal); for (const r of deduped) { r.reachable = reachabilityMap.get(r.url) ?? false; } @@ -489,11 +517,17 @@ export class WebSearchTool implements IMetonaTool { // 摘要增强 if (enhanceSnippets) { - await this.enhanceSnippets(sorted, 3); + await this.enhanceSnippets(sorted, 3, context); } // 自动抓取完整内容 - const fetchedContent = await this.autoFetch(query, sorted, searxngConfig.fetch_mode, fetchTop); + const fetchedContent = await this.autoFetch( + query, + sorted, + searxngConfig.fetch_mode, + fetchTop, + context, + ); // 格式化输出 const formatted = this.formatResults(query, sorted); @@ -536,6 +570,7 @@ export class WebSearchTool implements IMetonaTool { config: SearXNGConfig, maxResults: number, timeRange: string, + signal?: AbortSignal, ): Promise<{ results: SearchResult[]; engineStats: Record }> { const results: SearchResult[] = []; const engineStats: Record = {}; @@ -549,6 +584,8 @@ export class WebSearchTool implements IMetonaTool { let page = 1; while (results.length < maxResults && page <= maxPages) { + // v0.8.0 P1-2: abort 后停止翻页 + if (signal?.aborted) break; const params = new URLSearchParams(); params.set('q', query); params.set('format', config.format || 'json'); @@ -564,7 +601,8 @@ export class WebSearchTool implements IMetonaTool { const pageResults: SearchResult[] = []; try { - const response = await fetchWithTimeout(searchUrl, { headers }, 15_000); + // v0.8.0 P1-2: SearXNG 请求携带引擎级 abort signal + const response = await fetchWithTimeout(searchUrl, { headers }, 15_000, signal); if (!response.ok) { throw new Error(`SearXNG API error: ${response.status} ${response.statusText}`); } @@ -625,6 +663,7 @@ export class WebSearchTool implements IMetonaTool { query: string, maxResults: number, timeRange: string, + signal?: AbortSignal, ): Promise<{ results: SearchResult[]; engineStats: Record }> { const searchPromises = ENGINES.map(async (engine) => { try { @@ -639,6 +678,7 @@ export class WebSearchTool implements IMetonaTool { }, }, 8_000, + signal, ); if (!response.ok) { @@ -689,16 +729,23 @@ export class WebSearchTool implements IMetonaTool { // ===== 摘要增强(委托给 WebFetchTool) ===== - private async enhanceSnippets(results: SearchResult[], maxEnhance: number): Promise { + private async enhanceSnippets( + results: SearchResult[], + maxEnhance: number, + context: ToolExecutionContext, + ): Promise { let enhanced = 0; for (const r of results) { + // v0.8.0 P1-2: abort 后停止增强 + if (context.signal?.aborted) break; if (enhanced >= maxEnhance) break; if (r.snippet.length < 30 && r.reachable) { try { - const fetchResult = (await this.webFetchTool.execute( - { url: r.url }, - { sessionId: '', workspacePath: '', iteration: 0, requestId: '' }, - )) as { success: boolean; content?: string }; + // v0.8.0 P1-2: 透传真实执行上下文(signal)—— 移除"伪造空上下文" + const fetchResult = (await this.webFetchTool.execute({ url: r.url }, context)) as { + success: boolean; + content?: string; + }; if (fetchResult.success && fetchResult.content) { const text = fetchResult.content.slice(0, 200); @@ -729,13 +776,21 @@ export class WebSearchTool implements IMetonaTool { results: SearchResult[], fetchMode: string, fetchTop: number, + context: ToolExecutionContext, ): Promise> { + // v0.8.0 P1-2 根治: 时间预算收敛 —— 旧实现最坏耗时(8 条 × 3 并发 × 单次 + // web_fetch 240s ≈ 720s)远超工具自身 300s 超时,registry 层超时后整次调用 + // 作废(已抓取内容全部丢弃)。现以 deadline 为界:剩余预算不足一次最小 + // 抓取(30s)即停,已抓取内容照常返回。 + const BUDGET_RATIO = 0.8; + const MIN_FETCH_BUDGET_MS = 30_000; + const deadlineMs = Date.now() + 300_000 * BUDGET_RATIO; + // 相关性评分(不过滤,relevance=0 的结果也参与抓取候选) - const withRelevance = results.map((r) => ({ + const filtered = results.map((r) => ({ result: r, relevance: computeRelevance(query, r), })); - const filtered = withRelevance.length > 0 ? withRelevance : []; // 确定抓取数量:fetchTop 已在 execute() 中综合了配置面板和工具参数 const topN = Math.min(fetchTop, 8, filtered.length); @@ -761,9 +816,10 @@ export class WebSearchTool implements IMetonaTool { }): Promise<{ url: string; title: string; content: string } | null> => { try { // 委托给 WebFetchTool — 享受三阶段回退策略(HTTP + 反爬 + 浏览器渲染) + // v0.8.0 P1-2: 透传真实执行上下文(signal)—— 移除"伪造空上下文" const fetchResult = (await this.webFetchTool.execute( { url: item.result.url }, - { sessionId: '', workspacePath: '', iteration: 0, requestId: '' }, + context, )) as { success: boolean; content?: string }; if (fetchResult.success && fetchResult.content) { @@ -779,9 +835,18 @@ export class WebSearchTool implements IMetonaTool { } }; - // 并行批次抓取(并发 3) + // 并行批次抓取(并发 3,时间预算收敛) const CONCURRENCY = 3; for (let i = 0; i < toFetch.length; i += CONCURRENCY) { + // v0.8.0 P1-2: 剩余预算不足一次最小抓取 → 停止,保留已抓取内容 + const remaining = deadlineMs - Date.now(); + if (context.signal?.aborted || remaining < MIN_FETCH_BUDGET_MS) { + logTool( + 'web_search', + `Auto-fetch budget stop: fetched=${fetched.length}, remaining=${Math.max(0, remaining)}ms, aborted=${!!context.signal?.aborted}`, + ); + break; + } const batch = toFetch.slice(i, i + CONCURRENCY); const settled = await Promise.allSettled(batch.map((item) => fetchOne(item))); for (const r of settled) { diff --git a/electron/harness/types/metona-response.ts b/electron/harness/types/metona-response.ts index 8f6c96a..3535f82 100644 --- a/electron/harness/types/metona-response.ts +++ b/electron/harness/types/metona-response.ts @@ -144,6 +144,18 @@ export interface MetonaStreamEvent { /** DONE — 终止原因(前端可据此区分正常完成/错误/中断) */ terminationReason?: string; + + /** + * DONE — Provider 原生停止原因(v0.8.0 P0-1 finish_reason 全链路贯通)。 + * + * 取值域为归一化后的 OpenAI 语义:'stop' | 'length' | 'tool_calls' | + * 'content_filter'(Anthropic max_tokens→length、refusal→content_filter; + * Ollama done_reason 同映射;MiMo repetition_truncation→stop)。 + * adapter 在流结束(真实 [DONE]/message_stop/done:true 或断流合成)时携带; + * 引擎转发到最终 DONE 事件,前端据此展示截断/过滤提示;TRACE 层记录真值 + * (v0.7.4 及之前硬编码 'stop',length 截断在录制文件中不可归因)。 + */ + finishReason?: string; } /** v0.4.1: 输出验证事件载荷 — OutputValidator 检出的疑似问题(幻觉/事实矛盾/敏感信息/格式) */ diff --git a/electron/ipc/__tests__/agent.test.ts b/electron/ipc/__tests__/agent.test.ts index 6870438..b6a73a3 100644 --- a/electron/ipc/__tests__/agent.test.ts +++ b/electron/ipc/__tests__/agent.test.ts @@ -1054,3 +1054,73 @@ describe('llm:ollamaPull / ollamaPullCancel — 模型下载', () => { expect(result.success).toBe(false); }); }); + +// ===== v0.8.0 P1-1a: 后台会话回放缓冲 ===== + +describe('agent:getReplayState — v0.8.0 P1-1a 后台会话回放缓冲', () => { + it('streamEvent + stateChange 按序入缓冲,runId 记录,INIT 清空上一 run', async () => { + const { ctx } = makeCtx(); + registerAgentHandlers(ctx); + const em = ctx.agentEngineManager as unknown as EventEmitter; + + em.emit('streamEvent', { + type: 'text_delta', + sessionId: 'sess_1', + seq: 0, + timestamp: Date.now(), + delta: 'a', + }); + em.emit('stateChange', { + sessionId: 'sess_1', + state: 'THINKING', + iteration: 1, + runId: 'run_x', + }); + em.emit('streamEvent', { + type: 'text_delta', + sessionId: 'sess_1', + seq: 1, + timestamp: Date.now(), + delta: 'b', + runId: 'run_x', + }); + + const r = (await getHandler('agent:getReplayState')(null, 'sess_1')) as { + success: boolean; + data: { + isRunning: boolean; + runId: string | null; + truncated: boolean; + events: Array<{ channel: string }>; + }; + }; + expect(r.success).toBe(true); + expect(r.data.events.length).toBe(3); + expect(r.data.runId).toBe('run_x'); + expect(r.data.events.map((e) => e.channel)).toEqual([ + 'streamEvent', + 'stateChange', + 'streamEvent', + ]); + + // 新 run INIT → 上一 run 缓冲作废 + em.emit('stateChange', { + sessionId: 'sess_1', + state: 'INIT', + iteration: 0, + runId: 'run_y', + }); + const r2 = (await getHandler('agent:getReplayState')(null, 'sess_1')) as { + success: boolean; + data: { events: unknown[] }; + }; + expect(r2.data.events.length).toBe(0); + }); + + it('无效 sessionId → 明确失败', async () => { + const { ctx } = makeCtx(); + registerAgentHandlers(ctx); + const r = (await getHandler('agent:getReplayState')(null, '')) as { success: boolean }; + expect(r.success).toBe(false); + }); +}); diff --git a/electron/ipc/agent.ts b/electron/ipc/agent.ts index f4368cb..dffb7ef 100644 --- a/electron/ipc/agent.ts +++ b/electron/ipc/agent.ts @@ -59,8 +59,43 @@ interface IterationTrace { text: string; usage?: { input: number; output: number; total: number }; responded: boolean; // 本轮 llm_response 是否已记录(PARSING 与下一轮 THINKING 去重) + /** + * v0.8.0 P0-1: 本轮 Provider 原生停止原因(DONE 事件携带的 finishReason)。 + * v0.7.4 及之前 recordLLMResponse 硬编码 'stop',length 截断在 TRACE 文件中 + * 不可归因;现记录真值,缺省回退 'stop'。 + */ + finishReason?: string; } +/** + * v0.8.0 P1-1a: 单会话的流式回放缓冲 —— 渲染层切走会话期间,事件管道仍照常 + * 投递(按会话隔离被渲染层准入拒绝),此缓冲按序保存 streamEvent + stateChange + * 双通道事件;用户切回会话时经 agent:getReplayState 拉取并灌入渲染层事件总线, + * 实现"后台会话运行内容不丢失"。 + * + * 有界设计:单会话上限 MAX_REPLAY_EVENTS 条 / MAX_REPLAY_BYTES 字节,溢出丢弃 + * 最旧并置 truncated 标记;TERMINATED 后保留(供"完成后切回"看到最终内容), + * 下一次 run 启动(INIT stateChange)时清空重建。 + */ +interface ReplayEntry { + channel: 'streamEvent' | 'stateChange'; + payload: unknown; + ts: number; +} + +interface ReplayBuffer { + events: ReplayEntry[]; + bytes: number; + truncated: boolean; + /** 最近一次观察到的 runId(供渲染层回放后对齐 run 守卫) */ + runId?: string; + terminated: boolean; +} + +const MAX_REPLAY_EVENTS = 2000; +const MAX_REPLAY_BYTES = 4 * 1024 * 1024; +const MAX_REPLAY_SESSIONS = 50; + export function registerAgentHandlers(ctx: IPCContext): void { const { agentEngineManager, @@ -83,6 +118,54 @@ export function registerAgentHandlers(ctx: IPCContext): void { // ===== 常驻事件管道:text_delta 按会话节流(F8) ===== const throttleStates = new Map(); const iterationTraces = new Map(); + // v0.8.0 P1-1a: 每会话流式回放缓冲(后台会话内容恢复) + const replayBuffers = new Map(); + + /** 追加一条事件到会话回放缓冲(有界:条数/字节双上限,溢出丢最旧) */ + const appendReplay = (sessionId: string, entry: ReplayEntry): void => { + let buf = replayBuffers.get(sessionId); + if (!buf) { + // 会话数上限保护:超出后淘汰最早的缓冲(Map 迭代序 = 插入序) + if (replayBuffers.size >= MAX_REPLAY_SESSIONS) { + const oldest = replayBuffers.keys().next().value as string | undefined; + if (oldest !== undefined) replayBuffers.delete(oldest); + } + buf = { events: [], bytes: 0, truncated: false, terminated: false }; + replayBuffers.set(sessionId, buf); + } + // run 启动(INIT)后旧 run 缓冲作废 —— 由 stateChange 监听器在 INIT 时清空, + // 此处仅在 terminated 缓冲上遇到新内容时兜底重置(正常路径 INIT 先行) + if (buf.terminated) { + replayBuffers.delete(sessionId); + buf = { events: [], bytes: 0, truncated: false, terminated: false }; + replayBuffers.set(sessionId, buf); + } + let size = 0; + try { + size = JSON.stringify(entry.payload).length; + } catch { + size = 256; // 序列化失败按保守值计入 + } + buf.events.push(entry); + buf.bytes += size; + while ( + buf.events.length > MAX_REPLAY_EVENTS || + (buf.bytes > MAX_REPLAY_BYTES && buf.events.length > 1) + ) { + const dropped = buf.events.shift(); + buf.truncated = true; + if (dropped) { + try { + buf.bytes -= JSON.stringify(dropped.payload).length; + } catch { + buf.bytes -= 256; + } + } + } + // 记录 runId(首个携带者) + const payload = entry.payload as { runId?: string } | null; + if (!buf.runId && payload?.runId) buf.runId = payload.runId; + }; // v0.7.3 P1-5: 会话级记忆固化时间戳(consolidation-policy 频率门控的状态持有方) // v0.7.4 P3-5: LRU 化 —— 旧实现每会话一条永不清除,长期运行后无界增长。 @@ -136,6 +219,9 @@ export function registerAgentHandlers(ctx: IPCContext): void { agentEngineManager.on('streamEvent', (event: MetonaStreamEvent) => { if (!event.sessionId) return; const sessionId = event.sessionId; + // v0.8.0 P1-1a: 全量事件进入回放缓冲(含 text_delta —— 必须在节流分支 + // return 之前入队,否则切回会话时已缓冲的正文增量缺失) + appendReplay(sessionId, { channel: 'streamEvent', payload: event, ts: Date.now() }); const trace = iterationTraces.get(sessionId); // F8: text_delta 聚合,其他事件立即转发(先 flush 保证顺序) @@ -224,6 +310,8 @@ export function registerAgentHandlers(ctx: IPCContext): void { // 旧实现在 DONE 时删除 iterationTraces → 最终迭代的 TRACE 完整性缺失。 // 现仅在 TERMINATED 分支统一收尾(flush throttle buffer + 删 iterationTraces)。 // 注意:此处仍需转发 DONE 事件到前端(下方 broadcast 统一执行)。 + // v0.8.0 P0-1: 捕获 Provider 原生停止原因(recordLLMResponse 记录真值) + if (trace && event.finishReason) trace.finishReason = event.finishReason; break; default: break; @@ -247,6 +335,20 @@ export function registerAgentHandlers(ctx: IPCContext): void { broadcast('agent:stateChange', data); const sessionId = data.sessionId; + // v0.8.0 P1-1a: stateChange 事件进入回放缓冲。INIT(run 启动)清空上一 run + // 的缓冲;TERMINATED 置终态标记(保留缓冲供"完成后切回"回放最终内容)。 + if (sessionId) { + const stateRaw = data.state ?? data.current ?? ''; + if (stateRaw === 'INIT') { + replayBuffers.delete(sessionId); + } else { + appendReplay(sessionId, { channel: 'stateChange', payload: data, ts: Date.now() }); + if (stateRaw === 'TERMINATED') { + const buf = replayBuffers.get(sessionId); + if (buf) buf.terminated = true; + } + } + } if (!sessionId || data.iteration == null) return; const stateValue = data.state ?? data.current ?? ''; const trace = iterationTraces.get(sessionId); @@ -258,7 +360,8 @@ export function registerAgentHandlers(ctx: IPCContext): void { sessionId, iteration: trace.iteration, content: trace.text, - finishReason: 'stop', + // v0.8.0 P0-1: 记录真实停止原因(旧实现硬编码 'stop',length 截断不可归因) + finishReason: trace.finishReason ?? 'stop', tokenUsage: trace.usage ?? { input: 0, output: 0, total: 0 }, }); } @@ -298,7 +401,8 @@ export function registerAgentHandlers(ctx: IPCContext): void { sessionId, iteration: trace.iteration, content: trace.text, - finishReason: 'stop', + // v0.8.0 P0-1: 记录真实停止原因(旧实现硬编码 'stop',length 截断不可归因) + finishReason: trace.finishReason ?? 'stop', tokenUsage: trace.usage ?? { input: 0, output: 0, total: 0 }, }); } @@ -991,6 +1095,27 @@ export function registerAgentHandlers(ctx: IPCContext): void { return { success: true }; }); + // ===== v0.8.0 P1-1a: 后台会话流回放(切回会话时恢复运行内容) ===== + // 渲染层 setCurrentSession 检测到目标会话在 sessionRunStates 中运行(或刚结束) + // 时调用本通道:返回回放缓冲中按序保存的 streamEvent + stateChange 事件, + // 渲染层经事件总线灌入 useAgentStream 管线重建消息/Trace 状态,实时事件随后 + // 无缝衔接(runId 对齐由缓冲记录的 runId 提供)。 + ipcMain.handle('agent:getReplayState', (_event, sessionId: unknown) => { + if (typeof sessionId !== 'string' || !sessionId) { + return { success: false, error: 'Invalid sessionId' }; + } + const buf = replayBuffers.get(sessionId); + return { + success: true, + data: { + isRunning: agentEngineManager.isRunning(sessionId), + runId: buf?.runId ?? null, + truncated: buf?.truncated ?? false, + events: buf?.events ?? [], + }, + }; + }); + // ===== v0.5.0: SubAgent 可观测性 ===== // 1) 生命周期事件广播给前端(AgentMonitor 的 SubAgent 状态区) // 2) SubEngine 流事件录制到独立 TRACE 文件(sessionId = taskId),不污染父会话的流 @@ -1005,6 +1130,8 @@ export function registerAgentHandlers(ctx: IPCContext): void { text: string; usage: { input: number; output: number; total: number }; responded: boolean; + /** v0.8.0 P0-1: 本轮 Provider 原生停止原因(DONE 事件携带) */ + finishReason?: string; } const subTraces = new Map(); @@ -1117,7 +1244,8 @@ export function registerAgentHandlers(ctx: IPCContext): void { sessionId: taskId, iteration: st.iteration, content: st.text, - finishReason: 'stop', + // v0.8.0 P0-1: 记录真实停止原因(与主管道同口径) + finishReason: st.finishReason ?? 'stop', tokenUsage: st.usage, }); sessionRecorder.recordIterationEnd(taskId, { @@ -1152,7 +1280,8 @@ export function registerAgentHandlers(ctx: IPCContext): void { sessionId: taskId, iteration: st.iteration, content: st.text, - finishReason: 'stop', + // v0.8.0 P0-1: 记录真实停止原因(与主管道同口径) + finishReason: st.finishReason ?? 'stop', tokenUsage: st.usage, }); } @@ -1184,6 +1313,10 @@ export function registerAgentHandlers(ctx: IPCContext): void { }; } break; + case MetonaStreamEventType.DONE: + // v0.8.0 P0-1: 捕获 Provider 原生停止原因(SubAgent TRACE 真值) + if (event.finishReason) st.finishReason = event.finishReason; + break; case MetonaStreamEventType.TOOL_CALL_COMPLETE: if (event.toolCall) { sessionRecorder.recordToolCall({ diff --git a/electron/ipc/app.ts b/electron/ipc/app.ts index 146d9bd..109c9e6 100644 --- a/electron/ipc/app.ts +++ b/electron/ipc/app.ts @@ -35,6 +35,18 @@ export function registerAppHandlers(ctx: IPCContext): void { return result; }); + // ===== v0.8.0 P2-3: 下载并安装更新(electron-updater;未启用时明确失败) ===== + ipcMain.handle('app:updateInstall', async () => { + const { getAutoUpdaterHandle } = await import('../services/update.service'); + const handle = getAutoUpdaterHandle(); + if (!handle) { + return { success: false, error: '自动更新未启用(未配置更新源或开发模式)' }; + } + // fire-and-forget:进度经 update:status 广播,完成后 quitAndInstall + void handle.downloadAndInstall(); + return { success: true }; + }); + // ===== v0.7.3 P3-4: 健康快照(SLO 指标 + 最近健康检查报告)===== ipcMain.handle('app:healthSnapshot', async () => { try { diff --git a/electron/ipc/config.ts b/electron/ipc/config.ts index a7bed21..5378576 100644 --- a/electron/ipc/config.ts +++ b/electron/ipc/config.ts @@ -42,8 +42,9 @@ export function registerConfigHandlers(ctx: IPCContext): void { } // v0.7.4 P2-9-C: URL 类配置键高危目标校验(llm.baseURL/searxng.url/updateFeedUrl) + // v0.8.0 P1-5: 升级为 async 深校验(域名 DNS 解析拦云元数据绕过) try { - assertSafeConfigUrls([{ key, value }]); + await assertSafeConfigUrls([{ key, value }]); } catch (err) { log.warn(`[CONFIG] config:set rejected URL for ${key}: ${(err as Error).message}`); return { success: false, error: (err as Error).message }; @@ -104,8 +105,9 @@ export function registerConfigHandlers(ctx: IPCContext): void { } // v0.7.4 P2-9-C: 批量写入同样校验 URL 类配置键 + // v0.8.0 P1-5: 升级为 async 深校验 try { - assertSafeConfigUrls(entries as Array<{ key: string; value: unknown }>); + await assertSafeConfigUrls(entries as Array<{ key: string; value: unknown }>); } catch (err) { log.warn(`[CONFIG] config:setBatch rejected URL entry: ${(err as Error).message}`); return { success: false, error: (err as Error).message }; diff --git a/electron/ipc/mcp.ts b/electron/ipc/mcp.ts index 8b180dd..efda5d4 100644 --- a/electron/ipc/mcp.ts +++ b/electron/ipc/mcp.ts @@ -144,4 +144,12 @@ export function registerMCPHandlers(ctx: IPCContext): void { return { success: false, error: error instanceof Error ? error.message : String(error) }; } }); + + // ===== v0.8.0 P2-5: MCP Resources / Prompts 发现 ===== + ipcMain.handle('mcp:listServerContents', async (_event, name: unknown) => { + if (typeof name !== 'string' || !name.trim()) { + return { success: false, error: 'Invalid server name' }; + } + return { success: true, data: mcpManager.getServerContents(name) }; + }); } diff --git a/electron/ipc/sessions.ts b/electron/ipc/sessions.ts index d709c7f..5508cee 100644 --- a/electron/ipc/sessions.ts +++ b/electron/ipc/sessions.ts @@ -6,12 +6,19 @@ */ import { ipcMain } from 'electron'; +import { join, resolve as resolvePath, sep } from 'path'; +import { existsSync, readdirSync, readFileSync, statSync } from 'fs'; import type { IPCContext } from './context'; /** M-34 修复: 统一 sessionId 校验辅助函数 */ const isValidSessionId = (id: unknown): id is string => typeof id === 'string' && id.length > 0 && id.length <= 200; +/** v0.8.0 P2-2: 会话录制文件名白名单(防路径逃逸/任意文件读取) */ +const RECORDING_NAME_RE = /^session_[A-Za-z0-9_-]+\.jsonl$/; +/** v0.8.0 P2-2: 单个录制文件读取上限(防超大文件撑爆 IPC) */ +const MAX_RECORDING_BYTES = 20 * 1024 * 1024; + export function registerSessionHandlers(ctx: IPCContext): void { const { sessionService, confirmationHook, agentEngineManager } = ctx; @@ -40,6 +47,8 @@ export function registerSessionHandlers(ctx: IPCContext): void { agentEngineManager.abort(sessionId); await agentEngineManager.waitForAbort(sessionId); } + // v0.8.0 P2-1: 软删除 —— 会话进入回收站(deleted_at),30 天后自动清理, + // 可经 sessions:restore 恢复 / sessions:purge 彻底删除 const result = { success: sessionService.delete(sessionId) }; // v0.7.3 P2-3: 会话删除 = 会话终态 —— 决策记忆/pending 确认一并清理 // (防 rememberedDecisions 随会话数累积泄漏) @@ -52,6 +61,27 @@ export function registerSessionHandlers(ctx: IPCContext): void { return result; }); + // ===== v0.8.0 P2-1: 回收站 ===== + ipcMain.handle('sessions:listTrash', async () => { + return sessionService.listDeleted(); + }); + + ipcMain.handle('sessions:restore', async (_event, sessionId: unknown) => { + if (!isValidSessionId(sessionId)) return { success: false, error: 'Invalid sessionId' }; + return { success: sessionService.restore(sessionId) }; + }); + + ipcMain.handle('sessions:purge', async (_event, sessionId: unknown) => { + if (!isValidSessionId(sessionId)) return { success: false, error: 'Invalid sessionId' }; + // 彻底删除 = 终态:决策记忆/引擎与软删除同口径清理 + const result = { success: sessionService.purge(sessionId) }; + if (result.success) { + confirmationHook.forgetSession(sessionId); + agentEngineManager.disposeEngine(sessionId); + } + return result; + }); + ipcMain.handle('sessions:getMessages', async (_event, sessionId: unknown) => { // M-34 修复: 校验 sessionId if (!isValidSessionId(sessionId)) return []; @@ -161,4 +191,64 @@ export function registerSessionHandlers(ctx: IPCContext): void { if (!isValidSessionId(sessionId)) return null; return sessionService.getTraceData(sessionId); }); + + // ===== v0.8.0 P2-2: 会话回放(JSONL 录制读取) ===== + ipcMain.handle('sessions:listRecordings', async (_event, sessionId: unknown) => { + if (!isValidSessionId(sessionId)) return { success: false, error: 'Invalid sessionId' }; + try { + const logsDir = join(ctx.workspaceService.getPath(), 'logs'); + if (!existsSync(logsDir)) return { success: true, data: [] }; + const prefix = `session_${sessionId}_`; + const files = readdirSync(logsDir) + .filter((n) => n.startsWith(prefix) && RECORDING_NAME_RE.test(n)) + .map((name) => { + try { + const st = statSync(join(logsDir, name)); + return { name, size: st.size, mtime: st.mtimeMs }; + } catch { + return null; + } + }) + .filter((f): f is { name: string; size: number; mtime: number } => f !== null) + .sort((a, b) => b.mtime - a.mtime); + return { success: true, data: files }; + } catch (error) { + return { success: false, error: (error as Error).message }; + } + }); + + ipcMain.handle('sessions:readRecording', async (_event, name: unknown) => { + if (typeof name !== 'string' || !RECORDING_NAME_RE.test(name)) { + return { success: false, error: 'Invalid recording file name' }; + } + try { + const logsDir = resolvePath(ctx.workspaceService.getPath(), 'logs'); + const filePath = resolvePath(logsDir, name); + // 目录边界校验(双保险:文件名白名单已杜绝子目录,此处防 symlink/拼接逃逸) + if (!filePath.startsWith(logsDir + sep)) { + return { success: false, error: 'Recording path escapes logs directory' }; + } + if (!existsSync(filePath)) return { success: false, error: 'Recording file not found' }; + const size = statSync(filePath).size; + if (size > MAX_RECORDING_BYTES) { + return { success: false, error: `Recording file too large (${size} bytes)` }; + } + const raw = readFileSync(filePath, 'utf-8'); + // 逐行解析,坏行跳过并计数(录制文件允许尾部截断) + const events: Array> = []; + let skipped = 0; + for (const line of raw.split('\n')) { + const trimmed = line.trim(); + if (!trimmed) continue; + try { + events.push(JSON.parse(trimmed) as Record); + } catch { + skipped++; + } + } + return { success: true, data: { name, events, skipped } }; + } catch (error) { + return { success: false, error: (error as Error).message }; + } + }); } diff --git a/electron/ipc/shared.ts b/electron/ipc/shared.ts index 39ce910..a2c5936 100644 --- a/electron/ipc/shared.ts +++ b/electron/ipc/shared.ts @@ -10,7 +10,11 @@ import log from 'electron-log'; import type { IPCContext } from './context'; import { broadcast } from './context'; import { isSensitiveConfigKey } from '../utils/secure-config'; -import { assertSafeConfigTarget } from '../harness/tools/built-in/ssrf-guard'; +// v0.8.0 P1-5: 配置 URL 深校验(域名真实 DNS 解析,拦"解析到云元数据 IP"绕过) +import { + assertSafeConfigTargetDeep, + DeepCheckSoftFailure, +} from '../harness/tools/built-in/ssrf-guard'; /** * v0.7.4 P2-9-C: URL 类配置键 —— 写入时须过 assertSafeConfigTarget 高危目标校验。 @@ -33,12 +37,25 @@ const URL_CONFIG_KEYS = new Set([ /** * 校验一批配置写入中的 URL 类键值。任一非法即抛错(调用方整体拒绝)。 * 非 URL 类键 / 空值 / 非字符串值跳过(空 baseURL 表示"未配置",允许)。 + * + * v0.8.0 P1-5: 升级为深校验 —— 静态规则(assertSafeConfigTarget)之外对域名 + * 做真实 DNS 解析,任一解析结果命中链路本地/云元数据/组播保留段即拒绝; + * DNS 解析失败视为软失败放行(离线配置合法),仅 WARN 留痕。 */ -export function assertSafeConfigUrls(entries: Array<{ key: string; value: unknown }>): void { +export async function assertSafeConfigUrls( + entries: Array<{ key: string; value: unknown }>, +): Promise { for (const { key, value } of entries) { if (!URL_CONFIG_KEYS.has(key)) continue; if (typeof value !== 'string' || value.trim() === '') continue; - assertSafeConfigTarget(value); + await assertSafeConfigTargetDeep(value).catch((err) => { + // 软失败(DNS 解析失败)放行,仅日志留痕;硬失败(高危目标)向上抛 + if (err instanceof DeepCheckSoftFailure) { + log.warn(`[CONFIG] URL deep check skipped for ${key}: ${(err as Error).message}`); + return; + } + throw err; + }); } } diff --git a/electron/ipc/workspace.ts b/electron/ipc/workspace.ts index d6490fe..38ffad3 100644 --- a/electron/ipc/workspace.ts +++ b/electron/ipc/workspace.ts @@ -12,6 +12,34 @@ import log from 'electron-log'; export function registerWorkspaceHandlers(ctx: IPCContext): void { const { workspaceService } = ctx; + // ===== v0.8.0 P2-4: @ 文件提及 ===== + + // 工作空间文件名索引(联想数据源;renderer 侧 Fuse 模糊过滤) + ipcMain.handle('workspace:listFiles', async (_event, limit: unknown) => { + const maxFiles = + typeof limit === 'number' && Number.isFinite(limit) && limit > 0 + ? Math.min(5000, Math.floor(limit)) + : 2000; + try { + return { success: true, data: workspaceService.listFiles(maxFiles) }; + } catch (error) { + return { success: false, error: (error as Error).message }; + } + }); + + // 读取提及文件的文本片段(路径边界 + 512KB 上限 + 二进制拒绝 + MEMORY.md 保护) + ipcMain.handle('workspace:readFileClip', async (_event, relPath: unknown, maxBytes: unknown) => { + if (typeof relPath !== 'string' || !relPath.trim()) { + return { success: false, error: 'Invalid path' }; + } + const cap = + typeof maxBytes === 'number' && Number.isFinite(maxBytes) && maxBytes > 0 + ? Math.min(1024 * 1024, Math.floor(maxBytes)) + : 512 * 1024; + log.info(`[WORKSPACE] readFileClip: ${relPath.slice(0, 120)} (cap=${cap})`); + return workspaceService.readFileClip(relPath, cap); + }); + // 校验工作空间路径:检测路径有效性 + 必需文件状态 + 数据库是否存在 ipcMain.handle('workspace:check', async (_event, targetPath: string) => { if (!targetPath || typeof targetPath !== 'string') { @@ -67,96 +95,111 @@ export function registerWorkspaceHandlers(ctx: IPCContext): void { }); // 从旧工作空间继承文件到新工作空间 - ipcMain.handle('workspace:inheritFiles', async (_event, params: { - targetPath: string; - sourcePath: string; - files: string[]; - }) => { - const { targetPath, sourcePath, files } = params; - if (!targetPath || !sourcePath || !Array.isArray(files)) { - return { success: false, error: '参数无效' }; - } - - const { existsSync, mkdirSync, copyFileSync } = await import('fs'); - const { resolve } = await import('path'); - - const resolvedTarget = resolve(targetPath); - const resolvedSource = resolve(sourcePath); - - // 确保目标目录存在 - if (!existsSync(resolvedTarget)) { - mkdirSync(resolvedTarget, { recursive: true }); - } - - // 继承白名单:键为前端传入的标识,值为实际相对路径分段 - // 白名单映射防止路径遍历攻击(不直接使用用户传入的路径拼接到 fs 调用) - const INHERIT_WHITELIST: Record = { - 'SOUL.md': ['SOUL.md'], - '.metona/agent.db': ['.metona', 'agent.db'], - }; - - const inherited: string[] = []; - const failed: Array<{ file: string; error: string }> = []; - - for (const fileName of files) { - const pathSegments = INHERIT_WHITELIST[fileName]; - // 不在白名单中:拒绝(防止路径遍历) - if (!pathSegments) { - failed.push({ file: fileName, error: '不在继承白名单中' }); - continue; + ipcMain.handle( + 'workspace:inheritFiles', + async ( + _event, + params: { + targetPath: string; + sourcePath: string; + files: string[]; + }, + ) => { + const { targetPath, sourcePath, files } = params; + if (!targetPath || !sourcePath || !Array.isArray(files)) { + return { success: false, error: '参数无效' }; } - const srcFile = join(resolvedSource, ...pathSegments); - const dstFile = join(resolvedTarget, ...pathSegments); - try { - // 确保目标文件的父目录存在(如 .metona/) - const dstDir = join(resolvedTarget, ...pathSegments.slice(0, -1)); - if (!existsSync(dstDir)) { - mkdirSync(dstDir, { recursive: true }); - } - if (!existsSync(srcFile)) { - failed.push({ file: fileName, error: '源文件不存在' }); + + const { existsSync, mkdirSync, copyFileSync } = await import('fs'); + const { resolve } = await import('path'); + + const resolvedTarget = resolve(targetPath); + const resolvedSource = resolve(sourcePath); + + // 确保目标目录存在 + if (!existsSync(resolvedTarget)) { + mkdirSync(resolvedTarget, { recursive: true }); + } + + // 继承白名单:键为前端传入的标识,值为实际相对路径分段 + // 白名单映射防止路径遍历攻击(不直接使用用户传入的路径拼接到 fs 调用) + const INHERIT_WHITELIST: Record = { + 'SOUL.md': ['SOUL.md'], + '.metona/agent.db': ['.metona', 'agent.db'], + }; + + const inherited: string[] = []; + const failed: Array<{ file: string; error: string }> = []; + + for (const fileName of files) { + const pathSegments = INHERIT_WHITELIST[fileName]; + // 不在白名单中:拒绝(防止路径遍历) + if (!pathSegments) { + failed.push({ file: fileName, error: '不在继承白名单中' }); continue; } - - // === 数据库文件特殊处理 === - // agent.db 启用了 WAL 模式,直接 copyFileSync 会丢失 WAL 中未 checkpoint 的事务; - // 用 better-sqlite3 的 backup API(自带 checkpoint + 一致性快照) - if (fileName === '.metona/agent.db') { - try { - const Database = (await import('better-sqlite3')).default; - const srcDb = new Database(srcFile, { readonly: true, fileMustExist: true }); - try { - srcDb.backup(dstFile); - inherited.push(fileName); - log.info(`[WORKSPACE] Inherited ${fileName} (via SQLite backup): ${resolvedSource} → ${resolvedTarget}`); - } finally { - srcDb.close(); - } - } catch (err) { - // backup 失败时回退到 copyFileSync(至少保证基本可用) - log.warn(`[WORKSPACE] SQLite backup failed, fallback to copyFileSync: ${(err as Error).message}`); - try { - copyFileSync(srcFile, dstFile); - inherited.push(fileName); - log.info(`[WORKSPACE] Inherited ${fileName} (fallback copyFileSync): ${resolvedSource} → ${resolvedTarget}`); - } catch (err2) { - failed.push({ file: fileName, error: `backup 和 fallback 均失败: ${(err2 as Error).message}` }); - } + const srcFile = join(resolvedSource, ...pathSegments); + const dstFile = join(resolvedTarget, ...pathSegments); + try { + // 确保目标文件的父目录存在(如 .metona/) + const dstDir = join(resolvedTarget, ...pathSegments.slice(0, -1)); + if (!existsSync(dstDir)) { + mkdirSync(dstDir, { recursive: true }); } - continue; + if (!existsSync(srcFile)) { + failed.push({ file: fileName, error: '源文件不存在' }); + continue; + } + + // === 数据库文件特殊处理 === + // agent.db 启用了 WAL 模式,直接 copyFileSync 会丢失 WAL 中未 checkpoint 的事务; + // 用 better-sqlite3 的 backup API(自带 checkpoint + 一致性快照) + if (fileName === '.metona/agent.db') { + try { + const Database = (await import('better-sqlite3')).default; + const srcDb = new Database(srcFile, { readonly: true, fileMustExist: true }); + try { + srcDb.backup(dstFile); + inherited.push(fileName); + log.info( + `[WORKSPACE] Inherited ${fileName} (via SQLite backup): ${resolvedSource} → ${resolvedTarget}`, + ); + } finally { + srcDb.close(); + } + } catch (err) { + // backup 失败时回退到 copyFileSync(至少保证基本可用) + log.warn( + `[WORKSPACE] SQLite backup failed, fallback to copyFileSync: ${(err as Error).message}`, + ); + try { + copyFileSync(srcFile, dstFile); + inherited.push(fileName); + log.info( + `[WORKSPACE] Inherited ${fileName} (fallback copyFileSync): ${resolvedSource} → ${resolvedTarget}`, + ); + } catch (err2) { + failed.push({ + file: fileName, + error: `backup 和 fallback 均失败: ${(err2 as Error).message}`, + }); + } + } + continue; + } + + // === 普通文件直接复制 === + copyFileSync(srcFile, dstFile); + inherited.push(fileName); + log.info(`[WORKSPACE] Inherited ${fileName}: ${resolvedSource} → ${resolvedTarget}`); + } catch (err) { + failed.push({ file: fileName, error: (err as Error).message }); } - - // === 普通文件直接复制 === - copyFileSync(srcFile, dstFile); - inherited.push(fileName); - log.info(`[WORKSPACE] Inherited ${fileName}: ${resolvedSource} → ${resolvedTarget}`); - } catch (err) { - failed.push({ file: fileName, error: (err as Error).message }); } - } - return { success: true, inherited, failed }; - }); + return { success: true, inherited, failed }; + }, + ); // #51 修复: 校验源数据库完整性,防止继承损坏的数据库导致新工作空间数据丢失 ipcMain.handle('workspace:checkDatabaseIntegrity', async (_event, sourcePath: string) => { @@ -175,7 +218,9 @@ export function registerWorkspaceHandlers(ctx: IPCContext): void { const result = db.prepare('PRAGMA integrity_check').get() as { integrity_check: string }; const ok = result.integrity_check === 'ok'; if (!ok) { - log.warn(`[WORKSPACE] Source database integrity check failed: ${result.integrity_check} (source=${sourcePath})`); + log.warn( + `[WORKSPACE] Source database integrity check failed: ${result.integrity_check} (source=${sourcePath})`, + ); } return { success: true, ok, detail: result.integrity_check }; } finally { diff --git a/electron/main.ts b/electron/main.ts index 74c560e..c4ce5b7 100644 --- a/electron/main.ts +++ b/electron/main.ts @@ -321,6 +321,24 @@ async function initialize(): Promise { // v0.7.3 P3-3: 启动期清理旧 JSONL 录制文件(按 mtime 保留最近 200 个) void sessionRecorder.pruneOldRecordings(); + // v0.8.0 P2-1: 回收站 30 天自动清理(启动 + 每 24h;软删除会话超期硬删除) + const TRASH_RETENTION_DAYS = 30; + const purgeTrash = (): void => { + try { + const purged = sessionService.purgeExpiredTrash(TRASH_RETENTION_DAYS); + if (purged > 0) { + log.info( + `[Trash] Purged ${purged} session(s) older than ${TRASH_RETENTION_DAYS} days from trash`, + ); + } + } catch (err) { + log.warn('[Trash] Purge failed:', err); + } + }; + purgeTrash(); + const trashTimer = setInterval(purgeTrash, 24 * 60 * 60 * 1000); + trashTimer.unref?.(); + // ===== v0.2.0: 安全模块(必须在工具注册之前) ===== const policyEngine = new PolicyEngine(); const sandboxManager = new SandboxManager({ @@ -759,6 +777,26 @@ async function initialize(): Promise { trayManager = new TrayManager(resourcesPath); trayManager.initialize(mainWindow); + // ===== v0.8.0 P2-3: electron-updater 自动更新(仅生产环境 + 已配置 feed) ===== + // dev 模式 / 未配置 app.updateFeedUrl 时静默禁用;状态事件广播 update:status + if (!process.env['ELECTRON_RENDERER_URL']) { + const feedUrl = configService.get('app.updateFeedUrl') ?? ''; + if (feedUrl) { + const { startAutoUpdater } = await import('./services/update.service'); + await startAutoUpdater(feedUrl, (event) => { + for (const win of BrowserWindow.getAllWindows()) { + if (!win.isDestroyed()) win.webContents.send('update:status', event); + } + if (event.status === 'available') { + trayManager?.sendNotification( + 'MetonaAI 更新可用', + `新版本 ${event.latestVersion} 已发布,可在 设置 → 日志与数据 中下载安装`, + ); + } + }); + } + } + // ===== 全局快捷键 ===== windowManager.registerGlobalShortcuts(); diff --git a/electron/preload.ts b/electron/preload.ts index b4d3cab..f4e5f6a 100644 --- a/electron/preload.ts +++ b/electron/preload.ts @@ -36,6 +36,8 @@ const metonaAPI = { }, /** 中断会话 */ abortSession: (sessionId: string) => ipcRenderer.invoke('agent:abortSession', sessionId), + /** v0.8.0 P1-1a: 拉取后台会话流式回放缓冲(切回运行中会话时恢复内容) */ + getReplayState: (sessionId: string) => ipcRenderer.invoke('agent:getReplayState', sessionId), /** 监听 Provider 切换通知 */ onProviderSwitched: (callback: (data: unknown) => void) => { const listener = (_event: Electron.IpcRendererEvent, data: unknown) => callback(data); @@ -72,6 +74,13 @@ const metonaAPI = { /** P2-11: 截断消息(编辑重发/重新生成——删除锚点消息之后的所有消息) */ truncateAfter: (sessionId: string, messageId: string, inclusive?: boolean) => ipcRenderer.invoke('sessions:truncateAfter', sessionId, messageId, inclusive), + // v0.8.0 P2-1: 回收站(软删除 30 天保留) + listTrash: () => ipcRenderer.invoke('sessions:listTrash'), + restore: (sessionId: string) => ipcRenderer.invoke('sessions:restore', sessionId), + purge: (sessionId: string) => ipcRenderer.invoke('sessions:purge', sessionId), + // v0.8.0 P2-2: 会话回放(JSONL 录制读取) + listRecordings: (sessionId: string) => ipcRenderer.invoke('sessions:listRecordings', sessionId), + readRecording: (name: string) => ipcRenderer.invoke('sessions:readRecording', name), /** v0.5.0: 全文搜索会话内容(FTS5,跨会话,按会话聚合) */ searchContent: (query: string) => ipcRenderer.invoke('sessions:searchContent', query) as Promise<{ @@ -99,6 +108,8 @@ const metonaAPI = { removeServer: (name: string) => ipcRenderer.invoke('mcp:removeServer', name), toggleServer: (name: string, enabled: boolean) => ipcRenderer.invoke('mcp:toggleServer', name, enabled), + // v0.8.0 P2-5: Resources / Prompts 发现 + listServerContents: (name: string) => ipcRenderer.invoke('mcp:listServerContents', name), }, // ===== 记忆系统 ===== @@ -239,6 +250,15 @@ const metonaAPI = { | { status: 'up-to-date'; latestVersion: string } | { status: 'available'; latestVersion: string; downloadUrl?: string; notes?: string } >, + // v0.8.0 P2-3: 下载并安装更新(electron-updater;仅生产环境可用) + updateInstall: () => + ipcRenderer.invoke('app:updateInstall') as Promise<{ success: boolean; error?: string }>, + // v0.8.0 P2-3: 订阅更新状态事件(checking/available/downloading/downloaded/error) + onUpdateStatus: (callback: (event: unknown) => void) => { + const listener = (_event: Electron.IpcRendererEvent, data: unknown) => callback(data); + ipcRenderer.on('update:status', listener); + return () => ipcRenderer.removeListener('update:status', listener); + }, // #49 修复: 前端错误上报通道(主进程可注册 'error:report' handler 记录到 electron-log/审计日志) // 使用 send(单向)而非 invoke,即使主进程未注册 handler 也不会 reject reportError: (payload: unknown) => ipcRenderer.send('error:report', payload), @@ -296,6 +316,10 @@ const metonaAPI = { // #51 修复: 校验源数据库完整性(PRAGMA integrity_check),用于 inheritDatabase 前置校验 checkDatabaseIntegrity: (sourcePath: string) => ipcRenderer.invoke('workspace:checkDatabaseIntegrity', sourcePath), + // v0.8.0 P2-4: @ 文件提及(文件名索引 + 文本片段读取) + listFiles: (limit?: number) => ipcRenderer.invoke('workspace:listFiles', limit), + readFileClip: (relPath: string, maxBytes?: number) => + ipcRenderer.invoke('workspace:readFileClip', relPath, maxBytes), }, // ===== 工具管理 ===== diff --git a/electron/services/__tests__/database-migration.test.ts b/electron/services/__tests__/database-migration.test.ts index bf0c7be..a06e55e 100644 --- a/electron/services/__tests__/database-migration.test.ts +++ b/electron/services/__tests__/database-migration.test.ts @@ -281,15 +281,21 @@ describe.skipIf(!dbAvailable)('DatabaseService — 多版本迁移矩阵(user_ rmSync(dir, { recursive: true, force: true }); }); - it('user_version=2 的库:快速路径跳过迁移,版本号保持不变', async () => { + it('user_version=2 的库:增量迁移至 SCHEMA_VERSION(v0.8.0 补 sessions.deleted_at)', async () => { const { DatabaseService } = await import('../database.service'); const dir = mkdtempSync(join(tmpdir(), 'metona-migv2-')); const legacy = openLegacyDb(dir); + // v0.8.0 注: 真实 v2 库必含 sessions 表 —— fixture 补齐以覆盖迁移 10 legacy.exec(` CREATE TABLE messages ( id TEXT PRIMARY KEY, session_id TEXT NOT NULL, role TEXT NOT NULL, content TEXT, created_at INTEGER NOT NULL ); + CREATE TABLE sessions ( + id TEXT PRIMARY KEY, title TEXT DEFAULT '新会话', created_at INTEGER NOT NULL, + updated_at INTEGER NOT NULL, message_count INTEGER DEFAULT 0, + pinned INTEGER DEFAULT 0, archived INTEGER DEFAULT 0, metadata TEXT DEFAULT '{}' + ); CREATE VIRTUAL TABLE messages_fts USING fts5(content, content=messages, content_rowid=rowid, tokenize='trigram'); `); legacy.pragma('user_version = 2'); @@ -298,7 +304,14 @@ describe.skipIf(!dbAvailable)('DatabaseService — 多版本迁移矩阵(user_ const svc = new DatabaseService(dir); svc.initialize(); const db = svc.getDB(); - expect(db.pragma('user_version', { simple: true })).toBe(2); + // v0.8.0 P2-1 契约更新: SCHEMA_VERSION 升至 3(迁移 10 sessions.deleted_at), + // user_version=2 的库不再是最新版 —— 会执行增量迁移并盖章到 3; + // FTS 已是 trigram(迁移 9 幂等跳过),增量迁移只补 sessions.deleted_at 列。 + expect(db.pragma('user_version', { simple: true })).toBe(DatabaseService.SCHEMA_VERSION); + const delCol = db + .prepare("SELECT COUNT(*) AS c FROM pragma_table_info('sessions') WHERE name='deleted_at'") + .get() as { c: number }; + expect(delCol.c).toBe(1); // 快速路径不重建 FTS(保持原有 trigram 表) const ftsSql = db .prepare(`SELECT sql FROM sqlite_master WHERE type='table' AND name='messages_fts'`) diff --git a/electron/services/__tests__/session-search.test.ts b/electron/services/__tests__/session-search.test.ts index 7b4766c..2cb0b3a 100644 --- a/electron/services/__tests__/session-search.test.ts +++ b/electron/services/__tests__/session-search.test.ts @@ -50,6 +50,7 @@ describe.skipIf(!dbAvailable)('FTS5 会话内容搜索', () => { message_count INTEGER DEFAULT 0, pinned INTEGER DEFAULT 0, archived INTEGER DEFAULT 0, + deleted_at INTEGER, metadata TEXT DEFAULT '{}' ); CREATE TABLE messages ( @@ -281,6 +282,7 @@ describe.skipIf(!dbAvailable)('SessionService — 会话 CRUD 与消息边界', total_tokens INTEGER DEFAULT 0, pinned INTEGER DEFAULT 0, archived INTEGER DEFAULT 0, + deleted_at INTEGER, metadata TEXT DEFAULT '{}' ); CREATE TABLE messages ( @@ -618,6 +620,7 @@ describe.skipIf(!dbAvailable)('searchMessages — 更多查询形态与边界', message_count INTEGER DEFAULT 0, pinned INTEGER DEFAULT 0, archived INTEGER DEFAULT 0, + deleted_at INTEGER, metadata TEXT DEFAULT '{}' ); CREATE TABLE messages ( @@ -787,11 +790,26 @@ describe.skipIf(!dbAvailable)('searchMessages — 更多查询形态与边界', expect(hits.some((r: any) => r.sessionId === 'q8')).toBe(true); }); - it('已删除会话的标题回退为占位文案((已删除会话))', () => { - // 消息引用已不存在的会话 id(真实库 FK 会阻止,此处验证 title 补全兜底) + it('回收站会话(deleted_at 非空)从搜索结果整体剔除(v0.8.0 P2-1 契约)', () => { + // v0.8.0 P2-1 契约变更: 旧实现把缺标题的会话显示为"(已删除会话)"占位; + // 软删除引入后,回收站会话的搜索命中整体剔除(恢复后重新可见), + // 硬删除会话的消息已级联清除,占位分支不再可达。 insertMessage('q9_m1', 'q9_ghost', 'user', 'ghostkeyword 消息', 1); - const hits = sessionService.searchMessages('ghostkeyword'); - expect(hits.some((r: any) => r.sessionTitle === '(已删除会话)')).toBe(true); + db.prepare( + `INSERT INTO sessions (id, title, created_at, updated_at) VALUES ('q9_ghost', 'ghost', 1, 1)`, + ).run(); + const before = sessionService.searchMessages('ghostkeyword'); + expect(before.some((r: any) => r.sessionId === 'q9_ghost')).toBe(true); + + // 软删除后 → 搜索不再命中 + db.prepare(`UPDATE sessions SET deleted_at = ? WHERE id = 'q9_ghost'`).run(Date.now()); + const after = sessionService.searchMessages('ghostkeyword'); + expect(after.some((r: any) => r.sessionId === 'q9_ghost')).toBe(false); + + // 恢复后 → 重新命中 + db.prepare(`UPDATE sessions SET deleted_at = NULL WHERE id = 'q9_ghost'`).run(); + const restored = sessionService.searchMessages('ghostkeyword'); + expect(restored.some((r: any) => r.sessionId === 'q9_ghost')).toBe(true); }); it('单字符 CJK 查询走 LIKE 回退命中子串', () => { @@ -945,6 +963,7 @@ describe.skipIf(!dbAvailable)('saveMessage id 透传与 updateMessageContent(R message_count INTEGER DEFAULT 0, pinned INTEGER DEFAULT 0, archived INTEGER DEFAULT 0, + deleted_at INTEGER, metadata TEXT DEFAULT '{}' ); CREATE TABLE messages ( diff --git a/electron/services/__tests__/session-summary.test.ts b/electron/services/__tests__/session-summary.test.ts index f0f4ae3..eb94d98 100644 --- a/electron/services/__tests__/session-summary.test.ts +++ b/electron/services/__tests__/session-summary.test.ts @@ -56,6 +56,7 @@ describe.skipIf(!dbAvailable)('SessionSummary 分层上下文 × 截断交互', message_count INTEGER DEFAULT 0, pinned INTEGER DEFAULT 0, archived INTEGER DEFAULT 0, + deleted_at INTEGER, metadata TEXT DEFAULT '{}' ); CREATE TABLE messages ( @@ -284,6 +285,7 @@ describe.skipIf(!dbAvailable)('clearMessages × 摘要游标交互(B-2)', () message_count INTEGER DEFAULT 0, pinned INTEGER DEFAULT 0, archived INTEGER DEFAULT 0, + deleted_at INTEGER, metadata TEXT DEFAULT '{}' ); CREATE TABLE messages ( @@ -392,6 +394,7 @@ describe.skipIf(!dbAvailable)('session_summaries 级联删除(F-3)', () => { message_count INTEGER DEFAULT 0, pinned INTEGER DEFAULT 0, archived INTEGER DEFAULT 0, + deleted_at INTEGER, metadata TEXT DEFAULT '{}' ); CREATE TABLE session_summaries ( @@ -472,6 +475,7 @@ describe.skipIf(!dbAvailable)('maybeSummarize — 触发阈值与增量', () => message_count INTEGER DEFAULT 0, pinned INTEGER DEFAULT 0, archived INTEGER DEFAULT 0, + deleted_at INTEGER, metadata TEXT DEFAULT '{}' ); CREATE TABLE messages ( @@ -652,6 +656,7 @@ describe.skipIf(!dbAvailable)('buildHistoryMessages — 无摘要与 role 过滤 message_count INTEGER DEFAULT 0, pinned INTEGER DEFAULT 0, archived INTEGER DEFAULT 0, + deleted_at INTEGER, metadata TEXT DEFAULT '{}' ); CREATE TABLE messages ( @@ -746,6 +751,7 @@ describe.skipIf(!dbAvailable)('restoreHistoryImages — 更多边界', () => { message_count INTEGER DEFAULT 0, pinned INTEGER DEFAULT 0, archived INTEGER DEFAULT 0, + deleted_at INTEGER, metadata TEXT DEFAULT '{}' ); CREATE TABLE messages ( diff --git a/electron/services/__tests__/session-trash.test.ts b/electron/services/__tests__/session-trash.test.ts new file mode 100644 index 0000000..a999131 --- /dev/null +++ b/electron/services/__tests__/session-trash.test.ts @@ -0,0 +1,157 @@ +/** + * v0.8.0 P2-1: 会话回收站(软删除 30 天保留)回归测试。 + * + * 钉住契约: + * - delete → 软删除(deleted_at 时间戳),list/getSession 不再返回,listDeleted 返回; + * - restore → 恢复到主列表; + * - purge → 硬删除(不可恢复); + * - purgeExpiredTrash(30d) → 仅清理超期条目; + * - searchMessages 聚合剔除回收站会话。 + * + * 运行要求:better-sqlite3 为 Electron ABI 构建(test:electron),系统 Node 下自动跳过。 + */ + +import { describe, it, expect, beforeAll, afterAll, vi } from 'vitest'; + +vi.mock('electron-log', () => ({ + default: { info: vi.fn(), warn: vi.fn(), error: vi.fn(), debug: vi.fn() }, +})); + +let dbAvailable = true; +let Database: typeof import('better-sqlite3'); +try { + // eslint-disable-next-line @typescript-eslint/no-require-imports + Database = require('better-sqlite3'); + const probe = new Database(':memory:'); + probe.close(); +} catch { + dbAvailable = false; +} + +describe.skipIf(!dbAvailable)('会话回收站(v0.8.0 P2-1)', () => { + let sessionService: any; + let db: InstanceType; + + beforeAll(async () => { + const { SessionService } = await import('../session.service'); + db = new Database(':memory:'); + db.pragma('foreign_keys = ON'); + // 最小 schema(与 DatabaseService.createTables 对齐的关键列:deleted_at + 级联删除) + db.exec(` + CREATE TABLE sessions ( + id TEXT PRIMARY KEY, + title TEXT DEFAULT '新会话', + created_at INTEGER NOT NULL, + updated_at INTEGER NOT NULL, + message_count INTEGER DEFAULT 0, + total_tokens INTEGER DEFAULT 0, + pinned INTEGER DEFAULT 0, + archived INTEGER DEFAULT 0, + deleted_at INTEGER, + metadata TEXT DEFAULT '{}' + ); + CREATE TABLE messages ( + id TEXT PRIMARY KEY, + session_id TEXT NOT NULL, + role TEXT NOT NULL, + content TEXT, + reasoning_content TEXT, + tool_calls TEXT, + tool_result TEXT, + attachments TEXT, + iteration INTEGER, + created_at INTEGER NOT NULL, + FOREIGN KEY (session_id) REFERENCES sessions(id) ON DELETE CASCADE + ); + CREATE TABLE session_summaries ( + session_id TEXT PRIMARY KEY, + summary TEXT NOT NULL, + summarized_until_rowid INTEGER NOT NULL, + updated_at INTEGER NOT NULL, + FOREIGN KEY (session_id) REFERENCES sessions(id) ON DELETE CASCADE + ); + `); + sessionService = new SessionService(() => db); + }); + + afterAll(() => { + try { + db.close(); + } catch { + /* ignore */ + } + }); + + it('delete → 软删除:list 不返回、listDeleted 返回、getSession 排除', () => { + const a = sessionService.create('A'); + const b = sessionService.create('B'); + expect(sessionService.delete(a.id)).toBe(true); + + const list = sessionService.list().map((s: { id: string }) => s.id); + expect(list).toContain(b.id); + expect(list).not.toContain(a.id); + + const trash = sessionService.listDeleted().map((s: { id: string }) => s.id); + expect(trash).toEqual([a.id]); + + expect(sessionService.getSession(a.id)).toBeNull(); + expect(sessionService.getSession(b.id)).not.toBeNull(); + }); + + it('restore → 恢复到主列表', () => { + const a = sessionService.listDeleted()[0]; + expect(sessionService.restore(a.id)).toBe(true); + expect(sessionService.list().map((s: { id: string }) => s.id)).toContain(a.id); + expect(sessionService.listDeleted()).toHaveLength(0); + }); + + it('purge → 硬删除(仅对回收站条目生效)', () => { + const s = sessionService.create('temp'); + sessionService.delete(s.id); + sessionService.saveMessage({ sessionId: s.id, role: 'user', content: 'x' }); + expect(sessionService.purge(s.id)).toBe(true); + + const count = db + .prepare('SELECT COUNT(*) AS c FROM messages WHERE session_id = ?') + .get(s.id) as { + c: number; + }; + expect(count.c).toBe(0); + // 未删除的会话不可 purge(幂等保护) + const live = sessionService.create('live'); + expect(sessionService.purge(live.id)).toBe(false); + }); + + it('purgeExpiredTrash(30d) → 仅清理超期条目', () => { + const old = sessionService.create('old'); + sessionService.delete(old.id); + db.prepare('UPDATE sessions SET deleted_at = ? WHERE id = ?').run( + Date.now() - 31 * 24 * 60 * 60 * 1000, + old.id, + ); + + const fresh = sessionService.create('fresh'); + sessionService.delete(fresh.id); // deleted_at = now + + const purged = sessionService.purgeExpiredTrash(30); + expect(purged).toBe(1); + const remaining = db.prepare('SELECT COUNT(*) AS c FROM sessions WHERE id = ?').get(old.id) as { + c: number; + }; + expect(remaining.c).toBe(0); + expect(sessionService.listDeleted().map((s: { id: string }) => s.id)).toContain(fresh.id); + }); + + it('searchMessages 聚合剔除回收站会话(≤2 字符 LIKE 回退路径)', () => { + const s1 = sessionService.create('搜索目标会话'); + sessionService.saveMessage({ sessionId: s1.id, role: 'user', content: '量子计算机的独特内容' }); + // '量子' ≤2 字符走 LIKE 回退(最小 schema 无 FTS 表,与生产 trigram 路径 + // 共用同一个 aggregateSearchResults 聚合出口,剔除逻辑同源) + const r1 = sessionService.searchMessages('量子'); + expect(r1.length).toBe(1); + + sessionService.delete(s1.id); + const r2 = sessionService.searchMessages('量子'); + expect(r2.length).toBe(0); + }); +}); diff --git a/electron/services/__tests__/session-truncate-trace.test.ts b/electron/services/__tests__/session-truncate-trace.test.ts index bfcb3fc..47acbd5 100644 --- a/electron/services/__tests__/session-truncate-trace.test.ts +++ b/electron/services/__tests__/session-truncate-trace.test.ts @@ -45,6 +45,7 @@ describe.skipIf(!dbAvailable)('truncateMessagesAfter × TRACE metadata 过滤( message_count INTEGER DEFAULT 0, pinned INTEGER DEFAULT 0, archived INTEGER DEFAULT 0, + deleted_at INTEGER, metadata TEXT DEFAULT '{}' ); CREATE TABLE messages ( diff --git a/electron/services/database.service.ts b/electron/services/database.service.ts index 6cc1628..a9f7922 100644 --- a/electron/services/database.service.ts +++ b/electron/services/database.service.ts @@ -113,8 +113,9 @@ export class DatabaseService { * 版本号相同的库将跳过整个探测式迁移批次。 * v0.7.4 P4-4: 1 → 2 —— 迁移 9(messages_fts trigram)纳入版本化, * 失败中断批次不盖章 → 下次启动重试(根治"失败被吞 + 永久跳过")。 + * v0.8.0 P2-1: 2 → 3 —— 迁移 10(sessions.deleted_at 回收站软删除列)。 */ - static readonly SCHEMA_VERSION = 2; + static readonly SCHEMA_VERSION = 3; constructor(workspacePath?: string) { const baseDir = workspacePath ?? join(app.getPath('userData'), 'MetonaWorkspaces', 'default'); @@ -180,6 +181,7 @@ export class DatabaseService { db.exec(` -- ===== 会话表 ===== + -- v0.8.0 P2-1: deleted_at 回收站软删除(NULL = 正常;时间戳 = 已删除待清理) CREATE TABLE IF NOT EXISTS sessions ( id TEXT PRIMARY KEY, title TEXT NOT NULL DEFAULT '新会话', @@ -189,6 +191,7 @@ export class DatabaseService { total_tokens INTEGER NOT NULL DEFAULT 0, pinned INTEGER NOT NULL DEFAULT 0, archived INTEGER NOT NULL DEFAULT 0, + deleted_at INTEGER, metadata TEXT DEFAULT '{}' ); @@ -321,6 +324,8 @@ export class DatabaseService { CREATE INDEX IF NOT EXISTS idx_messages_role ON messages(role); CREATE INDEX IF NOT EXISTS idx_sessions_updated ON sessions(updated_at DESC); CREATE INDEX IF NOT EXISTS idx_sessions_pinned ON sessions(pinned DESC, updated_at DESC); + -- v0.8.0 P2-1: idx_sessions_deleted 由迁移 10 统一创建(createTables 的静态 + -- CREATE INDEX 在"表存在但缺 deleted_at 列"的遗留库上会 no-such-column 失败) CREATE INDEX IF NOT EXISTS idx_audit_session ON audit_logs(session_id); CREATE INDEX IF NOT EXISTS idx_audit_type ON audit_logs(event_type); CREATE INDEX IF NOT EXISTS idx_audit_created ON audit_logs(created_at); @@ -639,6 +644,19 @@ export class DatabaseService { // 非致命错误 — 如果迁移失败,NOT NULL 约束仍生效,saveMessage 会保存空字符串 } + // v0.8.0 P2-1 迁移 10: sessions.deleted_at 回收站软删除列 + 索引 + // 存在性守卫:极简遗留库(如 v2 盖章的早期测试库)可能没有 sessions 表, + // 迁移必须可重试且不因缺表中断整个批次 + { + const hasSessions = db + .prepare("SELECT name FROM sqlite_master WHERE type='table' AND name='sessions'") + .get(); + if (hasSessions) { + tryAddColumn('sessions', 'deleted_at', 'INTEGER'); + db.exec('CREATE INDEX IF NOT EXISTS idx_sessions_deleted ON sessions(deleted_at);'); + } + } + // v0.7.4 P4-4 迁移 9: messages_fts 升级 trigram tokenizer // 存量库的 messages_fts 建表语句不含 trigram —— 直接 DROP + 重建 + rebuild, // 使中文非连续子串搜索(trigram ≥3 字符)可用。检测方式:读 sqlite_master 的 diff --git a/electron/services/mcp-manager.service.ts b/electron/services/mcp-manager.service.ts index 925a1aa..cda6969 100644 --- a/electron/services/mcp-manager.service.ts +++ b/electron/services/mcp-manager.service.ts @@ -234,6 +234,20 @@ export interface MCPServerConfig { enabled: boolean; } +/** v0.8.0 P2-5: MCP Resource / Prompt 发现条目(渲染端展示用宽松形态) */ +export interface MCPResourceInfo { + uri: string; + name: string; + description?: string; + mimeType?: string; +} + +export interface MCPPromptInfo { + name: string; + description?: string; + arguments?: Array<{ name: string; description?: string; required?: boolean }>; +} + export interface MCPServerState { config: MCPServerConfig; status: MCPServerStatus; @@ -241,6 +255,9 @@ export interface MCPServerState { tools: Tool[]; error?: string; connectedAt?: number; + /** v0.8.0 P2-5: 发现的 Resources / Prompts(可选能力,未声明/失败时为空数组) */ + resources: MCPResourceInfo[]; + prompts: MCPPromptInfo[]; } // ===== MCP Tool Adapter ===== @@ -506,6 +523,8 @@ export class MCPManager { status: 'connecting', client: null, tools: [], + resources: [], + prompts: [], }); try { @@ -558,6 +577,42 @@ export class MCPManager { const toolsResult = await client.listTools(); const tools = toolsResult.tools ?? []; + // v0.8.0 P2-5: Resources / Prompts 发现(可选能力 —— server 未声明或 + // 请求失败一律置空数组,不阻断连接与工具注册) + const discoveredResources: MCPResourceInfo[] = []; + const discoveredPrompts: MCPPromptInfo[] = []; + try { + const res = await client.listResources(); + for (const r of res.resources ?? []) { + discoveredResources.push({ + uri: String(r.uri ?? ''), + name: String(r.name ?? ''), + description: typeof r.description === 'string' ? r.description : undefined, + mimeType: typeof r.mimeType === 'string' ? r.mimeType : undefined, + }); + } + } catch { + /* server 未声明 resources 能力 */ + } + try { + const res = await client.listPrompts(); + for (const p of res.prompts ?? []) { + discoveredPrompts.push({ + name: String(p.name ?? ''), + description: typeof p.description === 'string' ? p.description : undefined, + arguments: Array.isArray(p.arguments) + ? p.arguments.map((a) => ({ + name: String(a.name ?? ''), + description: typeof a.description === 'string' ? a.description : undefined, + required: a.required === true, + })) + : undefined, + }); + } + } catch { + /* server 未声明 prompts 能力 */ + } + // 注册到 ToolRegistry(v0.6.4: registerMCP 对重名冲突返回 false,此处聚合上报) let registeredCount = 0; let skippedCount = 0; @@ -580,6 +635,8 @@ export class MCPManager { state.tools = tools; state.connectedAt = Date.now(); state.error = undefined; + state.resources = discoveredResources; + state.prompts = discoveredPrompts; // v0.7.3 P4-2: 连接成功 —— 清零重连计数并取消排程 this.reconnectAttempts.delete(name); @@ -646,6 +703,8 @@ export class MCPManager { } state.client = null; state.tools = []; + state.resources = []; + state.prompts = []; this.notifyToolsChanged(); } @@ -801,6 +860,17 @@ export class MCPManager { }; } + /** + * v0.8.0 P2-5: 获取 server 发现的 Resources / Prompts(未连接返回空集合)。 + */ + getServerContents(name: string): { resources: MCPResourceInfo[]; prompts: MCPPromptInfo[] } { + const state = this.servers.get(name); + if (!state || state.status !== 'connected') { + return { resources: [], prompts: [] }; + } + return { resources: state.resources ?? [], prompts: state.prompts ?? [] }; + } + /** * 关闭所有连接 */ diff --git a/electron/services/session.service.ts b/electron/services/session.service.ts index f035c60..d48708f 100644 --- a/electron/services/session.service.ts +++ b/electron/services/session.service.ts @@ -81,6 +81,9 @@ export class SessionService { /** * 列出所有会话 + * + * v0.8.0 P2-1: 排除回收站(deleted_at 非空)会话 —— 删除为软删除, + * 列表/查询面不再出现;回收站内容经 listDeleted 查看。 */ list(options?: { archived?: boolean }): SessionInfo[] { const db = this.getDBFn(); @@ -90,7 +93,7 @@ export class SessionService { .prepare( ` SELECT * FROM sessions - WHERE archived = ? + WHERE archived = ? AND deleted_at IS NULL ORDER BY pinned DESC, updated_at DESC `, ) @@ -99,16 +102,74 @@ export class SessionService { return rows.map((row) => this.toSessionInfo(row)); } + /** + * v0.8.0 P2-1: 列出回收站会话(deleted_at 非空,按删除时间倒序) + */ + listDeleted(): SessionInfo[] { + const db = this.getDBFn(); + const rows = db + .prepare( + ` + SELECT * FROM sessions + WHERE deleted_at IS NOT NULL + ORDER BY deleted_at DESC + `, + ) + .all() as SessionRow[]; + return rows.map((row) => this.toSessionInfo(row)); + } + + /** + * v0.8.0 P2-1: 从回收站恢复会话(清除 deleted_at) + */ + restore(sessionId: string): boolean { + const db = this.getDBFn(); + const result = db + .prepare('UPDATE sessions SET deleted_at = NULL WHERE id = ? AND deleted_at IS NOT NULL') + .run(sessionId); + if (result.changes > 0) log.info(`Session restored from trash: ${sessionId}`); + return result.changes > 0; + } + + /** + * v0.8.0 P2-1: 彻底删除(硬删除,级联清消息/摘要/任务) + */ + purge(sessionId: string): boolean { + const db = this.getDBFn(); + const result = db + .prepare('DELETE FROM sessions WHERE id = ? AND deleted_at IS NOT NULL') + .run(sessionId); + if (result.changes > 0) log.info(`Session purged from trash: ${sessionId}`); + return result.changes > 0; + } + + /** + * v0.8.0 P2-1: 清理回收站中超过 maxAgeDays 的会话(硬删除,级联清理) + * 启动时与每 24h 定时调用(main.ts)。 + */ + purgeExpiredTrash(maxAgeDays = 30): number { + const db = this.getDBFn(); + const cutoff = Date.now() - maxAgeDays * 24 * 60 * 60 * 1000; + const rows = db + .prepare('SELECT id FROM sessions WHERE deleted_at IS NOT NULL AND deleted_at < ?') + .all(cutoff) as Array<{ id: string }>; + for (const row of rows) { + db.prepare('DELETE FROM sessions WHERE id = ?').run(row.id); + } + return rows.length; + } + /** * v0.7.4 P1-4: 按 ID 查询会话(含归档会话)。 * 供 agent:sendMessage 前置存在性预检 —— 指向已删除会话的消息会在 saveMessage - * 因 FK 约束抛错,提前校验可走统一的 ERROR+DONE 收尾路径而非 invoke reject。 + * 因 FK 约束(foreign_keys=ON)抛错,提前校验可走统一的 ERROR+DONE 收尾路径而非 invoke reject。 + * v0.8.0 P2-1: 排除回收站会话(deleted_at 非空 = 已删除,消息不可继续发送)。 */ getSession(sessionId: string): SessionInfo | null { const db = this.getDBFn(); - const row = db.prepare('SELECT * FROM sessions WHERE id = ?').get(sessionId) as - | SessionRow - | undefined; + const row = db + .prepare('SELECT * FROM sessions WHERE id = ? AND deleted_at IS NULL') + .get(sessionId) as SessionRow | undefined; return row ? this.toSessionInfo(row) : null; } @@ -163,14 +224,20 @@ export class SessionService { } /** - * 删除会话(级联删除消息) + * 删除会话 + * + * v0.8.0 P2-1 根治: 改为**软删除**(置 deleted_at)—— 旧实现直接硬删除, + * 误删无法恢复。现进入回收站(listDeleted 查看,30 天后 purgeExpiredTrash + * 自动清理),彻底删除由 purge 显式触发(级联清消息)。 */ delete(sessionId: string): boolean { const db = this.getDBFn(); - const result = db.prepare('DELETE FROM sessions WHERE id = ?').run(sessionId); + const result = db + .prepare('UPDATE sessions SET deleted_at = ? WHERE id = ? AND deleted_at IS NULL') + .run(Date.now(), sessionId); if (result.changes > 0) { - log.info(`Session deleted: ${sessionId}`); + log.info(`Session moved to trash: ${sessionId}`); return true; } return false; @@ -606,14 +673,20 @@ export class SessionService { if (aggregated.size === 0) return []; // 补全会话标题(archived 一起搜——搜索不应因归档而缺结果) + // v0.8.0 P2-1: 回收站会话(deleted_at 非空)从搜索结果剔除 const sessionIds = Array.from(aggregated.keys()); const placeholders = sessionIds.map(() => '?').join(','); const titleRows = db - .prepare(`SELECT id, title FROM sessions WHERE id IN (${placeholders})`) - .all(...sessionIds) as Array<{ id: string; title: string }>; - const titleMap = new Map(titleRows.map((r) => [r.id, r.title])); + .prepare(`SELECT id, title, deleted_at FROM sessions WHERE id IN (${placeholders})`) + .all(...sessionIds) as Array<{ id: string; title: string; deleted_at: number | null }>; + const titleMap = new Map( + titleRows.filter((r) => r.deleted_at == null).map((r) => [r.id, r.title]), + ); + const filteredAggregated = Array.from(aggregated.entries()).filter(([sid]) => + titleMap.has(sid), + ); - return Array.from(aggregated.entries()) + return filteredAggregated .map(([sessionId, info]) => ({ sessionId, sessionTitle: titleMap.get(sessionId) ?? '(已删除会话)', diff --git a/electron/services/update.service.ts b/electron/services/update.service.ts index 1a87d86..e4cd0b1 100644 --- a/electron/services/update.service.ts +++ b/electron/services/update.service.ts @@ -1,15 +1,17 @@ /** - * UpdateService — 轻量自动更新检查(v0.6.4 P4-2) + * UpdateService — 轻量自动更新检查(v0.6.4 P4-2 引入,v0.8.0 P2-3 接入 electron-updater) * - * 选型说明:完整方案为 electron-updater(需签名/发布通道配套);本轮依赖冻结 - * 且尚无发布基础设施,落地"feed 比对"最小闭环: - * - 配置 app.updateFeedUrl 指向版本清单 JSON: - * { "version": "0.7.0", "url": "https://…/MetonaAI-Setup.exe", "notes": "…" } - * - 用户点击"检查更新"→ 主进程拉取清单 → 语义化版本比较 → 结果经 toast/UI 反馈, - * available 时提供下载直链(shell.openExternal 由既有 app:openExternal 白名单承接)。 - * 未来接入 electron-updater 时,本服务的 IPC 面与 UI 不变,仅替换 check() 实现。 + * 双轨设计: + * 1. 手动检查(check()):feed 比对式 —— 拉取版本清单 JSON 与当前版本比较, + * 结果经 IPC 直接返回(不依赖 electron-updater,dev 模式同样可用)。 + * 2. 自动更新(startAutoUpdater,v0.8.0 P2-3):生产环境接入 electron-updater + * (generic provider),启动 5s 后静默检查;available → 广播 update:status + + * 系统通知;用户触发下载安装(app:updateInstall)→ downloadAndInstall 后重启。 + * 未配置 feed / dev 模式 / electron-updater 不可用时静默禁用(可用性优先)。 */ +import log from 'electron-log'; + export interface UpdateFeedInfo { version: string; url?: string; @@ -22,6 +24,15 @@ export type UpdateCheckResult = | { status: 'up-to-date'; latestVersion: string } | { status: 'available'; latestVersion: string; downloadUrl?: string; notes?: string }; +/** v0.8.0 P2-3: 广播给渲染层的更新状态事件(update:status 通道载荷) */ +export type UpdateStatusEvent = + | { status: 'checking' } + | { status: 'available'; latestVersion: string; notes?: string } + | { status: 'not-available'; latestVersion: string } + | { status: 'downloading'; percent: number } + | { status: 'downloaded' } + | { status: 'error'; message: string }; + /** * 语义化版本比较(仅支持 Major.Minor.Patch 与可选的预发布后缀忽略策略: * 带预发布标签视为小于同号正式版)。非法输入返回 false(宁可不提示升级)。 @@ -91,3 +102,98 @@ export class UpdateService { } } } + +// ===== v0.8.0 P2-3: electron-updater 集成(生产环境自动更新) ===== + +/** AutoUpdater 句柄(IPC 层消费;null = 未启用/dev 模式/未配置 feed) */ +export interface AutoUpdaterHandle { + checkForUpdates: () => Promise; + downloadAndInstall: () => Promise; +} + +let activeHandle: AutoUpdaterHandle | null = null; + +/** 获取已启动的 AutoUpdater 句柄(app:updateInstall 消费;未启用时返回 null) */ +export function getAutoUpdaterHandle(): AutoUpdaterHandle | null { + return activeHandle; +} + +/** + * 启动 electron-updater 自动更新(生产环境专用)。 + * + * @param feedUrl 更新源根目录(generic provider;app.updateFeedUrl 配置) + * @param onEvent 状态事件回调(主进程广播 update:status 给渲染层) + * @param autoCheckDelayMs 启动后延迟检查时间(默认 5s,让窗口先就绪) + * @returns 句柄(未启用时返回 null —— dev / electron-updater 加载失败) + */ +export async function startAutoUpdater( + feedUrl: string, + onEvent: (event: UpdateStatusEvent) => void, + autoCheckDelayMs = 5_000, +): Promise { + if (!feedUrl) { + log.info('[Update] auto-updater disabled: no feed URL configured'); + return null; + } + try { + // 动态导入:dev 模式(未打包)下 electron-updater 无法工作(无 app-update.yml), + // 加载失败时静默禁用,保持可用性 + const { autoUpdater } = await import('electron-updater'); + autoUpdater.autoDownload = false; // 下载需用户触发(app:updateInstall) + autoUpdater.autoInstallOnAppQuit = true; + autoUpdater.logger = null; // 事件统一经 onEvent 广播,避免双通道日志噪声 + + autoUpdater.on('checking-for-update', () => onEvent({ status: 'checking' })); + autoUpdater.on('update-available', (info) => { + onEvent({ status: 'available', latestVersion: String(info.version ?? ''), notes: undefined }); + }); + autoUpdater.on('update-not-available', (info) => { + onEvent({ status: 'not-available', latestVersion: String(info.version ?? '') }); + }); + autoUpdater.on('download-progress', (progress) => { + onEvent({ status: 'downloading', percent: Math.round(progress.percent ?? 0) }); + }); + autoUpdater.on('update-downloaded', () => onEvent({ status: 'downloaded' })); + autoUpdater.on('error', (err) => { + onEvent({ status: 'error', message: (err as Error).message }); + }); + + // generic provider feed(app-update.yml 缺失时显式 setFeedURL 兜底) + autoUpdater.setFeedURL({ provider: 'generic', url: feedUrl }); + + const handle: AutoUpdaterHandle = { + checkForUpdates: async () => { + try { + await autoUpdater.checkForUpdates(); + } catch (err) { + onEvent({ status: 'error', message: (err as Error).message }); + } + }, + downloadAndInstall: async () => { + try { + await autoUpdater.downloadUpdate(); + onEvent({ status: 'downloaded' }); + autoUpdater.quitAndInstall(); + } catch (err) { + onEvent({ status: 'error', message: (err as Error).message }); + } + }, + }; + activeHandle = handle; + + // 启动 5s 后静默检查(不阻塞窗口就绪) + const timer = setTimeout(() => { + void handle.checkForUpdates(); + }, autoCheckDelayMs); + timer.unref?.(); + + log.info(`[Update] auto-updater started (feed=${feedUrl})`); + return handle; + } catch (err) { + log.warn( + '[Update] electron-updater unavailable, auto-update disabled:', + (err as Error).message, + ); + return null; + } +} diff --git a/electron/services/window-manager.service.ts b/electron/services/window-manager.service.ts index 7195955..85532b8 100644 --- a/electron/services/window-manager.service.ts +++ b/electron/services/window-manager.service.ts @@ -27,6 +27,18 @@ export interface WindowState { export class WindowManager { private windows = new Map(); private activeWindowId: string | null = null; + /** + * v0.8.0 P1-3.7: 渲染进程崩溃自愈退避 —— 按 window id 记录 { count, windowStart }。 + * 60s 窗口内连续崩溃 ≥3 次不再自动 reload(避免 crash→reload→crash 死循环), + * 改弹系统错误对话框;自上次 reload 起 5 分钟无崩溃则计数复位。 + */ + private crashReloadAttempts = new Map< + number, + { count: number; windowStart: number; lastAt: number } + >(); + private static readonly CRASH_RELOAD_MAX = 3; + private static readonly CRASH_RELOAD_WINDOW_MS = 60_000; + private static readonly CRASH_RELOAD_RESET_MS = 5 * 60_000; /** * 创建新窗口 @@ -92,11 +104,51 @@ export class WindowManager { // 崩溃可观测性 + 自愈: 渲染进程崩溃(OOM/原生崩溃)时记录归因日志并自动重载, // 替代静默白屏(用户此前感知为"应用崩溃"且无从恢复) + // v0.8.0 P1-3.7: 增加退避 —— 60s 内连续 ≥3 次崩溃停止自动 reload 并弹窗 + //(根因持久存在时旧实现会 crash→reload→crash 死循环);稳定 5 分钟后计数复位。 win.webContents.on('render-process-gone', (_event, details) => { log.error( `[WindowManager] render-process-gone (id=${win.id}): reason=${details.reason} exitCode=${details.exitCode}`, ); if (!win.isDestroyed() && details.reason !== 'clean-exit') { + const now = Date.now(); + const record = this.crashReloadAttempts.get(win.id) ?? { + count: 0, + windowStart: now, + lastAt: 0, + }; + // 稳定 5 分钟 → 计数复位 + if (record.lastAt > 0 && now - record.lastAt > WindowManager.CRASH_RELOAD_RESET_MS) { + record.count = 0; + record.windowStart = now; + } + // 60s 滑动窗口外 → 重新起算 + if (now - record.windowStart > WindowManager.CRASH_RELOAD_WINDOW_MS) { + record.count = 0; + record.windowStart = now; + } + record.count += 1; + record.lastAt = now; + this.crashReloadAttempts.set(win.id, record); + + if (record.count > WindowManager.CRASH_RELOAD_MAX) { + log.error( + `[WindowManager] Crash reload threshold exceeded (${record.count} in ${WindowManager.CRASH_RELOAD_WINDOW_MS / 1000}s) — NOT reloading automatically (crash loop protection)`, + ); + try { + // showErrorBox 为同步 API(void 返回),崩溃循环保护路径直接弹窗 + // eslint-disable-next-line @typescript-eslint/no-require-imports + const { dialog } = require('electron') as typeof import('electron'); + dialog.showErrorBox( + 'MetonaAI Desktop 反复崩溃', + `渲染进程在 ${WindowManager.CRASH_RELOAD_WINDOW_MS / 1000} 秒内连续崩溃 ${record.count} 次,已停止自动恢复。\n\n请重启应用;若反复出现,请携带日志(设置 → 日志与数据 → 打开日志目录)反馈。`, + ); + } catch { + /* dialog 不可用时仅保留日志 */ + } + return; + } + try { win.webContents.reload(); log.info(`[WindowManager] Window ${id} reloaded after renderer crash`); diff --git a/electron/services/workspace.service.ts b/electron/services/workspace.service.ts index c86d2bf..b489a64 100644 --- a/electron/services/workspace.service.ts +++ b/electron/services/workspace.service.ts @@ -12,10 +12,12 @@ * @see standard/开发规范.md — 使用 fs/path 内置模块(非第三方库) */ -import { join } from 'path'; -import { existsSync, mkdirSync, readFileSync, writeFileSync } from 'fs'; +import { join, resolve, sep } from 'path'; +import { existsSync, mkdirSync, readFileSync, readdirSync, writeFileSync, statSync } from 'fs'; import { app } from 'electron'; import log from 'electron-log'; +// v0.8.0 P2-4: @ 提及文本片段的编码检测(与 read_file 同源) +import { decodeBufferWithDetection } from '../harness/tools/built-in/file-guard'; // ===== 类型定义 ===== @@ -138,6 +140,112 @@ export class WorkspaceService { return { ...this.files }; } + // ===== v0.8.0 P2-4: @ 文件提及数据源 ===== + + /** 列出目录树时的噪声目录(依赖产物/版本库/运行时数据,永不进入提及索引) */ + private static readonly MENTION_SKIP_DIRS = new Set([ + 'node_modules', + '.git', + 'logs', + '.metona', + 'dist', + 'dist-electron', + 'out', + 'build', + 'release', + ]); + + /** + * 工作空间文件名索引(@ 文件提及联想数据源)。 + * + * 同步递归收集相对路径;跳过 MENTION_SKIP_DIRS 与隐藏目录;深度上限 8; + * 数量上限 maxFiles(防超大工作空间拖慢主进程)。FileGuard 的 MEMORY.md + * 保护不适用于"文件名索引"(只列名字不读内容)。 + */ + listFiles(maxFiles = 2000): string[] { + const out: string[] = []; + const walk = (dir: string, rel: string, depth: number): void => { + if (out.length >= maxFiles || depth > 8) return; + let entries: import('fs').Dirent[]; + try { + entries = readdirSync(dir, { withFileTypes: true }); + } catch { + return; + } + for (const entry of entries) { + if (out.length >= maxFiles) return; + if (entry.name.startsWith('.')) continue; + if (entry.isDirectory() && WorkspaceService.MENTION_SKIP_DIRS.has(entry.name)) continue; + const relName = rel ? `${rel}/${entry.name}` : entry.name; + if (entry.isFile()) { + out.push(relName); + } else if (entry.isDirectory()) { + walk(join(dir, entry.name), relName, depth + 1); + } + } + }; + walk(this.workspacePath, '', 0); + return out; + } + + /** + * 读取 @ 提及文件的文本片段(≤ maxBytes,默认 512KB,二进制拒绝)。 + * + * 安全:路径 resolve 后必须位于工作空间内;根目录 MEMORY.md 与工具读取 + * 同口径保护(isProtectedWorkspaceFile);文本判定复用 file-guard 的 + * decodeBufferWithDetection(NUL 启发式在 BOM 之后仍命中则拒绝)。 + */ + readFileClip( + relPath: string, + maxBytes = 512 * 1024, + ): { + success: boolean; + error?: string; + name?: string; + content?: string; + truncated?: boolean; + size?: number; + } { + try { + const resolved = resolve(this.workspacePath, relPath); + if (resolved !== this.workspacePath && !resolved.startsWith(this.workspacePath + sep)) { + return { success: false, error: 'Path escapes workspace' }; + } + if (!existsSync(resolved) || !statSync(resolved).isFile()) { + return { success: false, error: 'File not found' }; + } + // 根目录 MEMORY.md 保护(与 read_file 工具同口径) + if ( + resolve(resolved).toLowerCase() === resolve(this.workspacePath, 'MEMORY.md').toLowerCase() + ) { + return { success: false, error: 'Access to workspace root MEMORY.md is protected' }; + } + const size = statSync(resolved).size; + const buf = readFileSync(resolved); + const head = buf.subarray(0, Math.min(buf.length, 8192)); + // 二进制判定:非 UTF-16 BOM 前缀下含 NUL 字节 → 拒绝(文本附件语义) + const isUtf16Bom = + (head[0] === 0xff && head[1] === 0xfe) || (head[0] === 0xfe && head[1] === 0xff); + if (!isUtf16Bom) { + for (const b of head) { + if (b === 0) + return { success: false, error: 'Binary file is not supported for @ mention' }; + } + } + const clipped = buf.length > maxBytes ? buf.subarray(0, maxBytes) : buf; + const { content } = decodeBufferWithDetection(clipped); + return { + success: true, + name: relPath, + content, + truncated: buf.length > maxBytes, + size, + }; + } catch (err) { + return { success: false, error: (err as Error).message }; + } + } + /** * 更新 MEMORY.md 的最后更新时间戳 */ diff --git a/electron/utils/__tests__/pinned-proxy.test.ts b/electron/utils/__tests__/pinned-proxy.test.ts new file mode 100644 index 0000000..b466c66 --- /dev/null +++ b/electron/utils/__tests__/pinned-proxy.test.ts @@ -0,0 +1,194 @@ +/** + * v0.8.0 P1-4/P1-5: Pinned CONNECT 代理 + 配置 URL 深校验测试。 + * + * - pinned-proxy:公网/私网目标的 CONNECT 校验分流(resolver 注入 mock, + * 不触网);端口分配与幂等启动。 + * - assertSafeConfigTargetDeep:域名解析到云元数据 IP → 拒绝;DNS 失败 → + * DeepCheckSoftFailure(软失败放行);IP 字面量高危段 → 拒绝。 + */ + +import { describe, it, expect, afterAll } from 'vitest'; +import * as net from 'net'; +import { + createPinnedProxy, + ensurePinnedProxy, + isPinnedProxyActive, +} from '../../utils/pinned-proxy'; +import { + assertSafeConfigTargetDeep, + DeepCheckSoftFailure, + __dnsLookup, +} from '../../harness/tools/built-in/ssrf-guard'; + +// ===== Pinned CONNECT 代理 ===== + +describe('pinned CONNECT proxy (P1-4)', () => { + const keepAlive: Array = []; + + it('public target → CONNECT tunnel established (200), traffic pinned to resolved IP', async () => { + // 本地 echo 服务器充当"上游":mock resolver 把域名 pin 到 127.0.0.1, + // 验证隧道建立后字节双向透传(不触公网) + const echo = net.createServer((socket) => { + socket.pipe(socket); + }); + await new Promise((r) => echo.listen(0, '127.0.0.1', r)); + const echoPort = (echo.address() as net.AddressInfo).port; + keepAlive.push(echo); + + const proxy = await createPinnedProxy(async () => { + // 校验通过 → 返回校验期锁定的 IP(pinning 语义:连接用同一批 IP) + return ['127.0.0.1']; + }); + keepAlive.push(proxy as unknown as net.Server); + + const target = net.connect({ host: '127.0.0.1', port: proxy.port }); + const result = await new Promise((resolve, reject) => { + const timer = setTimeout(() => reject(new Error('timeout')), 3_000); + let buf = ''; + target.on('data', (d) => { + buf += d.toString(); + if (buf.includes('\r\n\r\n')) { + // 200 已到 → 隧道建立,写 payload 验证透传 + target.write(`ping-to-${echoPort}`); + } + if (buf.includes(`ping-to-${echoPort}`)) { + clearTimeout(timer); + resolve(buf.split('\r\n')[0]); + } + }); + target.on('error', (e) => { + clearTimeout(timer); + reject(e); + }); + target.write(`CONNECT example.com:${echoPort} HTTP/1.1\r\nHost: example.com\r\n\r\n`); + }); + expect(result).toContain('200'); + target.destroy(); + echo.close(); + await proxy.close(); + }); + + it('private/metadata target → 403 rejected (fail-closed)', async () => { + const proxy = await createPinnedProxy(async () => { + throw new Error('Blocked SSRF: resolves to private IP'); + }); + const target = net.connect({ host: '127.0.0.1', port: proxy.port }); + const response = await new Promise((resolve, reject) => { + const timer = setTimeout(() => reject(new Error('timeout')), 2_000); + let buf = ''; + target.on('data', (d) => { + buf += d.toString(); + if (buf.includes('\r\n\r\n')) { + clearTimeout(timer); + resolve(buf.split('\r\n')[0]); + } + }); + target.on('error', (e) => { + clearTimeout(timer); + reject(e); + }); + target.write('CONNECT 169.254.169.254:80 HTTP/1.1\r\nHost: 169.254.169.254\r\n\r\n'); + }); + expect(response).toContain('403'); + target.destroy(); + await proxy.close(); + }); + + it('invalid CONNECT target → 400', async () => { + const proxy = await createPinnedProxy(async () => ['1.2.3.4']); + const target = net.connect({ host: '127.0.0.1', port: proxy.port }); + const response = await new Promise((resolve, reject) => { + const timer = setTimeout(() => reject(new Error('timeout')), 2_000); + let buf = ''; + target.on('data', (d) => { + buf += d.toString(); + if (buf.includes('\r\n\r\n')) { + clearTimeout(timer); + resolve(buf.split('\r\n')[0]); + } + }); + target.on('error', (e) => { + clearTimeout(timer); + reject(e); + }); + target.write('CONNECT no-port-target HTTP/1.1\r\n\r\n'); + }); + expect(response).toContain('400'); + target.destroy(); + await proxy.close(); + }); + + it('ensurePinnedProxy: idempotent start + active flag', async () => { + const port1 = await ensurePinnedProxy(); + const port2 = await ensurePinnedProxy(); + expect(port1).not.toBeNull(); + expect(port2).toBe(port1); + expect(isPinnedProxyActive()).toBe(true); + }, 10_000); + + afterAll(() => { + for (const s of keepAlive) { + try { + s.close(); + } catch { + /* ignore */ + } + } + }); +}); + +// ===== 配置 URL 深校验 ===== + +describe('assertSafeConfigTargetDeep (P1-5)', () => { + it('hostname resolving to metadata IP → rejected', async () => { + const original = __dnsLookup.current; + __dnsLookup.current = (async () => [ + { address: '169.254.169.254', family: 4 }, + ]) as unknown as typeof __dnsLookup.current; + try { + await expect(assertSafeConfigTargetDeep('https://attacker.example/search')).rejects.toThrow( + /high-risk/, + ); + } finally { + __dnsLookup.current = original; + } + }); + + it('hostname resolving to public IP → allowed', async () => { + const original = __dnsLookup.current; + __dnsLookup.current = (async () => [ + { address: '93.184.216.34', family: 4 }, + ]) as unknown as typeof __dnsLookup.current; + try { + await expect(assertSafeConfigTargetDeep('https://example.com')).resolves.toBeUndefined(); + } finally { + __dnsLookup.current = original; + } + }); + + it('DNS failure → DeepCheckSoftFailure (soft-fail allow, caller logs)', async () => { + const original = __dnsLookup.current; + __dnsLookup.current = (async () => { + throw new Error('ENOTFOUND'); + }) as unknown as typeof __dnsLookup.current; + try { + await expect(assertSafeConfigTargetDeep('https://offline.example')).rejects.toBeInstanceOf( + DeepCheckSoftFailure, + ); + } finally { + __dnsLookup.current = original; + } + }); + + it('IP-literal link-local target → rejected without DNS', async () => { + await expect(assertSafeConfigTargetDeep('http://169.254.169.254/latest')).rejects.toThrow( + /link-local|metadata/, + ); + }); + + it('loopback target → allowed (local SearXNG/MCP legit)', async () => { + await expect( + assertSafeConfigTargetDeep('http://127.0.0.1:8080/search'), + ).resolves.toBeUndefined(); + }); +}); diff --git a/electron/utils/network-proxy.ts b/electron/utils/network-proxy.ts index ea0fce0..3908d53 100644 --- a/electron/utils/network-proxy.ts +++ b/electron/utils/network-proxy.ts @@ -24,6 +24,9 @@ import { session } from 'electron'; import log from 'electron-log'; +// v0.8.0 P1-4: Agent 浏览器分区的 Pinned 代理状态探测(避免覆盖浏览器通道 +// 的 SSRF pinning 代理设置) +import { isPinnedProxyActive } from './pinned-proxy'; const AGENT_BROWSER_PARTITION = 'persist:metona-agent-browser'; @@ -50,9 +53,23 @@ export async function applySessionProxy(proxyUrl: string | null | undefined): Pr try { const config = rules !== '' ? { proxyRules: rules } : { mode: 'direct' as const }; await session.defaultSession.setProxy(config); - await session.fromPartition(AGENT_BROWSER_PARTITION).setProxy(config); + // v0.8.0 P1-4: agent-browser 分区的代理优先级 —— + // ① 用户显式配置了代理(rules 非空)→ 用户代理语义优先,覆盖 Pinned 代理 + // (M7 已知限制口径:该模式下浏览器通道无 pinning); + // ② 未配置代理但 Pinned 代理已激活 → 不动该分区(保留浏览器通道 SSRF pinning); + // ③ 其余 → 显式直连。 + if (rules !== '' || !isPinnedProxyActive()) { + await session.fromPartition(AGENT_BROWSER_PARTITION).setProxy(config); + log.info( + `[Network] agent-browser partition proxy ${rules === '' ? 'set to direct' : `rules: ${rules}`}`, + ); + } else { + log.info( + '[Network] agent-browser partition left on SSRF pinned proxy (no user proxy configured)', + ); + } log.info( - `[Network] Chromium sessions proxy ${rules === '' ? 'set to direct (no proxy)' : `rules: ${rules}`}`, + `[Network] Chromium default session proxy ${rules === '' ? 'set to direct (no proxy)' : `rules: ${rules}`}`, ); } catch (err) { log.warn( diff --git a/electron/utils/pinned-proxy.ts b/electron/utils/pinned-proxy.ts new file mode 100644 index 0000000..260f0e2 --- /dev/null +++ b/electron/utils/pinned-proxy.ts @@ -0,0 +1,267 @@ +/** + * Pinned CONNECT Proxy — Agent 浏览器通道的 SSRF 根治层(v0.8.0 P1-4) + * + * 背景:web_fetch 的浏览器回退通道与 web_browser 走 Chromium 自有网络栈, + * 此前只有工具入口一次 SSRF 校验 —— Chromium 自己解析 DNS,入口校验通过到 + * 实际连接之间若攻击者切换 DNS(TTL=0),可借隐藏浏览器触达内网(DNS + * rebinding 残余窗口,比 HTTP 通道的 DNS pinning 防线弱一档)。 + * + * 根治方案:主进程启动本地 HTTP 代理(127.0.0.1 随机端口),Agent 浏览器 + * 分区(persist:metona-agent-browser)的全部流量经此代理: + * - CONNECT(https):解析目标 host → resolvePublicAddresses 校验(与工具 + * 入口同策略:仅公网)→ **直连校验阶段锁定的 IP** 建立隧道。TLS/SNI/证书 + * 校验仍由 Chromium 端到端完成(隧道内字节透传,不做 MITM),既闭 + * rebinding 又不破坏 HTTPS。 + * - 绝对形式 HTTP:同校验后按 pinned IP 转发。 + * - 校验失败 → 403 拒绝 + WARN 日志(fail-closed)。 + * + * 边界(显式记录): + * - 用户已配置 network.proxyUrl(isProxyActive())时,调用方不启用本代理, + * 保持用户代理语义 —— 代理模式下 DNS 在代理端解析,pinning 不可实现 + * (与 M7 结论一致,该场景的已知限制维持原文档口径)。 + * - 代理启动失败时降级为"无 pinned 代理"(可用性优先),仅 WARN,不阻断 + * 浏览器工具。 + */ + +import * as http from 'http'; +import * as net from 'net'; +import log from 'electron-log'; +import { resolvePublicAddresses } from '../harness/tools/built-in/ssrf-guard'; + +/** 校验函数签名(可注入以便测试;默认为生产实现 resolvePublicAddresses) */ +export type AddressResolver = (url: string) => Promise; + +export interface PinnedProxy { + /** 本地监听端口(127.0.0.1) */ + port: number; + /** 停止代理服务器 */ + close: () => Promise; +} + +/** 解析 "host:port" 形态(CONNECT 请求行目标) */ +function splitHostPort(target: string): { host: string; port: number } | null { + const lastColon = target.lastIndexOf(':'); + if (lastColon <= 0) return null; + let host = target.slice(0, lastColon); + const portStr = target.slice(lastColon + 1); + // IPv6 字面量目标:去方括号(resolvePublicAddresses 侧已处理 [host] 形态) + if (host.startsWith('[') && host.endsWith(']')) { + host = host.slice(1, -1); + } + const port = Number(portStr); + if (!Number.isInteger(port) || port <= 0 || port > 65535) return null; + return { host, port }; +} + +/** 400/403 之类的简易响应后断开 */ +function rejectRequest(socket: net.Socket, status: string, body: string): void { + if (socket.writable) { + socket.write( + `HTTP/1.1 ${status}\r\nConnection: close\r\nContent-Length: ${body.length}\r\n\r\n${body}`, + ); + } + socket.destroy(); +} + +/** + * 创建 Pinned 代理服务器(工厂形式,测试可注入 resolver)。 + * + * @param resolve 地址校验/解析实现(默认 resolvePublicAddresses —— 仅公网放行) + */ +export function createPinnedProxy( + resolve: AddressResolver = resolvePublicAddresses, +): Promise { + const server = http.createServer((req, res) => { + // 绝对形式 HTTP(Chromium 经代理的明文请求):同校验后按 pinned IP 转发 + void handlePlainHttp(req, res, resolve); + }); + server.on('connect', (req, clientSocket, head) => { + void handleConnect(req, clientSocket as net.Socket, head, resolve); + }); + server.on('clientError', (_err, socket) => { + try { + socket.destroy(); + } catch { + /* ignore */ + } + }); + + return new Promise((resolvePromise, rejectPromise) => { + const onError = (err: Error): void => { + server.removeListener('listening', onListening); + rejectPromise(err); + }; + const onListening = (): void => { + server.removeListener('error', onError); + const addr = server.address(); + const port = typeof addr === 'object' && addr ? addr.port : 0; + resolvePromise({ + port, + close: () => + new Promise((res) => { + server.close(() => res()); + }), + }); + }; + server.once('error', onError); + server.once('listening', onListening); + server.listen(0, '127.0.0.1'); + }); +} + +/** CONNECT 隧道:校验目标 → 直连 pinned IP → 双向管道 */ +async function handleConnect( + req: http.IncomingMessage, + clientSocket: net.Socket, + head: Buffer, + resolve: AddressResolver, +): Promise { + const target = req.url ?? ''; + const parsed = splitHostPort(target); + if (!parsed) { + rejectRequest(clientSocket, '400 Bad Request', 'Invalid CONNECT target'); + return; + } + const { host, port } = parsed; + + let ips: string[]; + try { + // 以 http(s) 形式构造 URL 供 resolvePublicAddresses 校验(其只关心协议与 host; + // 443 用 https 仅为语义准确,校验逻辑对两者一致) + const scheme = port === 443 ? 'https' : 'http'; + ips = await resolve(`${scheme}://${host}:${port}`); + } catch (err) { + log.warn(`[PinnedProxy] CONNECT ${host}:${port} blocked: ${(err as Error).message}`); + rejectRequest(clientSocket, '403 Forbidden', 'SSRF validation failed'); + return; + } + + const upstream = net.connect({ host: ips[0] ?? host, port }); + let established = false; + + upstream.once('connect', () => { + established = true; + if (clientSocket.writable) { + clientSocket.write('HTTP/1.1 200 Connection Established\r\n\r\n'); + if (head && head.length > 0) upstream.write(head); + clientSocket.pipe(upstream); + upstream.pipe(clientSocket); + } else { + upstream.destroy(); + } + }); + + const cleanup = (): void => { + try { + clientSocket.destroy(); + } catch { + /* ignore */ + } + try { + upstream.destroy(); + } catch { + /* ignore */ + } + }; + clientSocket.once('error', cleanup); + clientSocket.once('close', () => { + if (established) cleanup(); + }); + upstream.once('error', (err) => { + log.warn(`[PinnedProxy] upstream error for ${host}:${port}: ${(err as Error).message}`); + if (!established) { + rejectRequest(clientSocket, '502 Bad Gateway', 'Upstream connect failed'); + } else { + cleanup(); + } + }); +} + +/** 绝对形式 HTTP 转发:校验目标 → 按 pinned IP 发起请求 → 回写响应 */ +async function handlePlainHttp( + req: http.IncomingMessage, + res: http.ServerResponse, + resolve: AddressResolver, +): Promise { + const rawUrl = req.url ?? ''; + let parsed: URL; + try { + parsed = new URL(rawUrl); + } catch { + res.writeHead(400).end('Invalid absolute-form URL'); + return; + } + if (parsed.protocol !== 'http:') { + // https 走 CONNECT,明文通道只处理 http 绝对形式 + res.writeHead(400).end('Only http absolute-form supported (https uses CONNECT)'); + return; + } + + let ips: string[]; + try { + ips = await resolve(rawUrl); + } catch (err) { + log.warn(`[PinnedProxy] HTTP ${parsed.host} blocked: ${(err as Error).message}`); + res.writeHead(403).end('SSRF validation failed'); + return; + } + + const port = parsed.port ? Number(parsed.port) : 80; + const upstreamReq = http.request( + { + host: ips[0] ?? parsed.hostname, + port, + method: req.method, + path: parsed.pathname + parsed.search, + headers: { ...req.headers, host: parsed.host }, + }, + (upstreamRes) => { + res.writeHead(upstreamRes.statusCode ?? 502, upstreamRes.headers); + upstreamRes.pipe(res); + }, + ); + upstreamReq.on('error', (err) => { + log.warn(`[PinnedProxy] upstream http error for ${parsed.host}: ${(err as Error).message}`); + if (!res.headersSent) res.writeHead(502); + res.end('Upstream request failed'); + }); + req.pipe(upstreamReq); +} + +// ===== 模块级单例(BrowserWindowManager / network-proxy 消费) ===== + +let singleton: PinnedProxy | null = null; +let starting: Promise | null = null; + +/** 是否已有 Pinned 代理在运行(applySessionProxy 据此决定是否覆盖分区代理) */ +export function isPinnedProxyActive(): boolean { + return singleton !== null; +} + +/** + * 确保 Pinned 代理已启动,返回端口;失败返回 null(调用方降级为无 pinned 代理)。 + * 幂等:并发调用共享同一次启动。 + */ +export async function ensurePinnedProxy(): Promise { + if (singleton) return singleton.port; + if (!starting) { + starting = (async (): Promise => { + try { + singleton = await createPinnedProxy(); + log.info(`[PinnedProxy] SSRF-pinned proxy listening on 127.0.0.1:${singleton.port}`); + return singleton; + } catch (err) { + log.warn( + '[PinnedProxy] failed to start pinned proxy (browser channel falls back to unpinned):', + err, + ); + singleton = null; + return null; + } finally { + starting = null; + } + })(); + } + const result = await starting; + return result?.port ?? null; +} diff --git a/package-lock.json b/package-lock.json index 1b05688..4d0cfe9 100644 --- a/package-lock.json +++ b/package-lock.json @@ -1,12 +1,12 @@ { "name": "metona-ai-desktop", - "version": "0.7.4", + "version": "0.8.0", "lockfileVersion": 3, "requires": true, "packages": { "": { "name": "metona-ai-desktop", - "version": "0.7.4", + "version": "0.8.0", "license": "MIT", "dependencies": { "@emotion/react": "^11.14.0", @@ -19,6 +19,7 @@ "date-fns": "^4.1.0", "dotenv": "^17.4.2", "electron-log": "^5.3.3", + "electron-updater": "^6.8.9", "fuse.js": "^7.1.0", "i18next": "^26.4.0", "lru-cache": "^11.1.0", @@ -3572,9 +3573,9 @@ } }, "node_modules/@typescript-eslint/eslint-plugin/node_modules/ignore": { - "version": "7.0.6", - "resolved": "https://registry.npmmirror.com/ignore/-/ignore-7.0.6.tgz", - "integrity": "sha512-BAg6QkE8W+TuQLrrw0Ugr7HegXduRuuj8/ti2kSOc+jz1dmx8/WNcjr6XGnq5YpDWxFwwaavqD0+jIUOKelTsw==", + "version": "7.0.8", + "resolved": "https://registry.npmmirror.com/ignore/-/ignore-7.0.8.tgz", + "integrity": "sha512-YYNsSlXBjMk92SKnkwvB5LOVSa6OznlFUGcsvrFgNJbJCd0M1XKeFVRc8ZByeCqz32FivYNHJVooLmdqrmvp/Q==", "dev": true, "license": "MIT", "engines": { @@ -4263,7 +4264,6 @@ "version": "2.0.1", "resolved": "https://registry.npmmirror.com/argparse/-/argparse-2.0.1.tgz", "integrity": "sha512-8+9WqebbFzpX9OR+Wa6O29asIogeRMzcGtAINdpMHHyAg10f05aSFVBbcEqGf/PXw1EjAZ+q2/bEBg3DvurK3Q==", - "dev": true, "license": "Python-2.0" }, "node_modules/aria-query": { @@ -4665,7 +4665,6 @@ "version": "9.7.0", "resolved": "https://registry.npmmirror.com/builder-util-runtime/-/builder-util-runtime-9.7.0.tgz", "integrity": "sha512-g/kR520giAFYkSXTzcmF3kqQq7wi8F6N6SzeDgZrqTBN+VHdmgWOyTdD1yD7AATDId/yXLvuP34CxW46/BwCdw==", - "dev": true, "license": "MIT", "dependencies": { "debug": "^4.3.4", @@ -5953,6 +5952,69 @@ "dev": true, "license": "ISC" }, + "node_modules/electron-updater": { + "version": "6.8.9", + "resolved": "https://registry.npmmirror.com/electron-updater/-/electron-updater-6.8.9.tgz", + "integrity": "sha512-ZhVxM9iGONUpZGI1FxdMRgJjUFXi7AYGVa5PwKlO1tV1/4zDxQmfKpXOHVztKrd6L9rLcFjERvi1Mf2vxyTkig==", + "license": "MIT", + "dependencies": { + "builder-util-runtime": "9.7.0", + "fs-extra": "^10.1.0", + "js-yaml": "^4.1.0", + "lazy-val": "^1.0.5", + "lodash.escaperegexp": "^4.1.2", + "lodash.isequal": "^4.5.0", + "semver": "~7.7.3", + "tiny-typed-emitter": "^2.1.0" + } + }, + "node_modules/electron-updater/node_modules/fs-extra": { + "version": "10.1.0", + "resolved": "https://registry.npmmirror.com/fs-extra/-/fs-extra-10.1.0.tgz", + "integrity": "sha512-oRXApq54ETRj4eMiFzGnHWGy+zo5raudjuxN0b8H7s/RU2oW0Wvsx9O0ACRN/kRq9E8Vu/ReskGB5o3ji+FzHQ==", + "license": "MIT", + "dependencies": { + "graceful-fs": "^4.2.0", + "jsonfile": "^6.0.1", + "universalify": "^2.0.0" + }, + "engines": { + "node": ">=12" + } + }, + "node_modules/electron-updater/node_modules/jsonfile": { + "version": "6.2.1", + "resolved": "https://registry.npmmirror.com/jsonfile/-/jsonfile-6.2.1.tgz", + "integrity": "sha512-zwOTdL3rFQ/lRdBnntKVOX6k5cKJwEc1HdilT71BWEu7J41gXIB2MRp+vxduPSwZJPWBxEzv4yH1wYLJGUHX4Q==", + "license": "MIT", + "dependencies": { + "universalify": "^2.0.0" + }, + "optionalDependencies": { + "graceful-fs": "^4.1.6" + } + }, + "node_modules/electron-updater/node_modules/semver": { + "version": "7.7.4", + "resolved": "https://registry.npmmirror.com/semver/-/semver-7.7.4.tgz", + "integrity": "sha512-vFKC2IEtQnVhpT78h1Yp8wzwrf8CM+MzKMHGJZfBtzhZNycRFnXsHk6E5TxIkkMsgNS7mdX3AGB7x2QM2di4lA==", + "license": "ISC", + "bin": { + "semver": "bin/semver.js" + }, + "engines": { + "node": ">=10" + } + }, + "node_modules/electron-updater/node_modules/universalify": { + "version": "2.0.1", + "resolved": "https://registry.npmmirror.com/universalify/-/universalify-2.0.1.tgz", + "integrity": "sha512-gptHNQghINnc/vTGIk0SOFGFNXw7JVrlRUtConJRlvaw6DuX0wO5Jeko9sWrMBhh+PsYAZ7oXAiOnf/UKogyiw==", + "license": "MIT", + "engines": { + "node": ">= 10.0.0" + } + }, "node_modules/electron-vite": { "version": "3.1.0", "resolved": "https://registry.npmmirror.com/electron-vite/-/electron-vite-3.1.0.tgz", @@ -7192,7 +7254,6 @@ "version": "4.2.11", "resolved": "https://registry.npmmirror.com/graceful-fs/-/graceful-fs-4.2.11.tgz", "integrity": "sha512-RbJ5/jmFcNNCcDV5o9eTnBLJ/HszWV0P73bc+Ff4nS/rJj+YaS6IGyiOL0VoBYX+l1Wrl3k63h/KrH+nhJ0XvQ==", - "dev": true, "license": "ISC" }, "node_modules/has-flag": { @@ -7874,7 +7935,6 @@ "version": "4.2.0", "resolved": "https://registry.npmmirror.com/js-yaml/-/js-yaml-4.2.0.tgz", "integrity": "sha512-ePWsvanv0DWuDRsW8dnt+R4jQ31SCRCQ7hhNcPXZPsoBZiemuZNYGf7adZdqX2D86j6rvKp3RpCxVTSb8WQlOw==", - "dev": true, "funding": [ { "type": "github", @@ -8023,7 +8083,6 @@ "version": "1.0.5", "resolved": "https://registry.npmmirror.com/lazy-val/-/lazy-val-1.0.5.tgz", "integrity": "sha512-0/BnGCCfyUMkBpeDgWihanIAF9JmZhHBgUhEqzvf+adhNGLoP6TaiI5oF8oyb3I45P+PcnrqihSf01M0l0G5+Q==", - "dev": true, "license": "MIT" }, "node_modules/levn": { @@ -8344,9 +8403,9 @@ } }, "node_modules/lint-staged/node_modules/tinyexec": { - "version": "1.3.0", - "resolved": "https://registry.npmmirror.com/tinyexec/-/tinyexec-1.3.0.tgz", - "integrity": "sha512-QKAl9m8gWWGHV8jZcPeym6j+XULi6tOf1mT83WYJ4Lk2ytW/uwAWkrP0uFsdoYMdueVJ0qs26wZ+23xeB4ibNQ==", + "version": "1.3.1", + "resolved": "https://registry.npmmirror.com/tinyexec/-/tinyexec-1.3.1.tgz", + "integrity": "sha512-GCvB3aoys96IuDFBMcTB46JOR6mdMtAToqwiW8JlWhsoh1mhHi/xn9ss/Dg7N555GiJyEt2qzoG/NHCwM6h1EA==", "dev": true, "license": "MIT", "engines": { @@ -8376,6 +8435,19 @@ "dev": true, "license": "MIT" }, + "node_modules/lodash.escaperegexp": { + "version": "4.1.2", + "resolved": "https://registry.npmmirror.com/lodash.escaperegexp/-/lodash.escaperegexp-4.1.2.tgz", + "integrity": "sha512-TM9YBvyC84ZxE3rgfefxUWiQKLilstD6k7PTGt6wfbtXF8ixIJLOL3VYyV/z+ZiPLsVxAsKAFVwWlWeb2Y8Yyw==", + "license": "MIT" + }, + "node_modules/lodash.isequal": { + "version": "4.5.0", + "resolved": "https://registry.npmmirror.com/lodash.isequal/-/lodash.isequal-4.5.0.tgz", + "integrity": "sha512-pDo3lu8Jhfjqls6GkMgpahsF9kCyayhgykjyLMNFTKWrpVdAQtYyB4muAMWozBB4ig/dtWAmsMxLEI8wuz+DYQ==", + "deprecated": "This package is deprecated. Use require('node:util').isDeepStrictEqual instead.", + "license": "MIT" + }, "node_modules/lodash.merge": { "version": "4.6.2", "resolved": "https://registry.npmmirror.com/lodash.merge/-/lodash.merge-4.6.2.tgz", @@ -9653,9 +9725,9 @@ } }, "node_modules/node-gyp/node_modules/undici": { - "version": "6.28.0", - "resolved": "https://registry.npmmirror.com/undici/-/undici-6.28.0.tgz", - "integrity": "sha512-LIY910g9TI13YS95lrMFrs8Rm/u/irgHeTWoKCoteeJ04CUJ92eEfj0rVn+7VKMPBpUPiUoBKfhNyLI23EE/KA==", + "version": "6.28.1", + "resolved": "https://registry.npmmirror.com/undici/-/undici-6.28.1.tgz", + "integrity": "sha512-zWpdTVD54H48CIybL0rWQ3ukpb9d23wM7eH5RtfdmeP70cWHNjtfo7P4vZX+5CoDcO53J4Pu5uXp7lNfjc6DRA==", "dev": true, "license": "MIT", "engines": { @@ -10989,7 +11061,6 @@ "version": "1.6.0", "resolved": "https://registry.npmmirror.com/sax/-/sax-1.6.0.tgz", "integrity": "sha512-6R3J5M4AcbtLUdZmRv2SygeVaM7IhrLXu9BmnOGmmACak8fiUtOsYNWUS4uK7upbmHIBbLBeFeI//477BKLBzA==", - "dev": true, "license": "BlueOak-1.0.0", "engines": { "node": ">=11.0.0" @@ -11718,6 +11789,12 @@ "semver": "bin/semver" } }, + "node_modules/tiny-typed-emitter": { + "version": "2.1.0", + "resolved": "https://registry.npmmirror.com/tiny-typed-emitter/-/tiny-typed-emitter-2.1.0.tgz", + "integrity": "sha512-qVtvMxeXbVej0cQWKqVSSAHmKZEHAvxdF8HEUBFWts8h+xEo5m/lEiPakuyZ3BnCBjOD8i24kzNOiOLLgsSxhA==", + "license": "MIT" + }, "node_modules/tinybench": { "version": "2.9.0", "resolved": "https://registry.npmmirror.com/tinybench/-/tinybench-2.9.0.tgz", @@ -12012,9 +12089,9 @@ } }, "node_modules/undici": { - "version": "8.10.0", - "resolved": "https://registry.npmmirror.com/undici/-/undici-8.10.0.tgz", - "integrity": "sha512-HvltHd7avK13QIw/oLe4qoOLyoVSoafqJ2jYOrtMRBkbYT31eiBQ8O0ehRKZiEZCMEyLFQNIADpgCWC5fALvYQ==", + "version": "8.10.2", + "resolved": "https://registry.npmmirror.com/undici/-/undici-8.10.2.tgz", + "integrity": "sha512-/y4/bH9YNU5hi9NIrpOuvGXFcxrj3CMrV+/AYpowAYTpHn8gX/XPFjNy766FPoYY0miQhdW977JFWKGNhBdwyQ==", "license": "MIT", "engines": { "node": ">=22.19.0" diff --git a/package.json b/package.json index 448c672..e8d045a 100644 --- a/package.json +++ b/package.json @@ -1,6 +1,6 @@ { "name": "metona-ai-desktop", - "version": "0.7.4", + "version": "0.8.0", "description": "MetonaAI Desktop — 生产级通用 AI Agent 智能体桌面应用", "main": "dist-electron/main/main.js", "author": "Metona Team", @@ -47,6 +47,7 @@ "date-fns": "^4.1.0", "dotenv": "^17.4.2", "electron-log": "^5.3.3", + "electron-updater": "^6.8.9", "fuse.js": "^7.1.0", "i18next": "^26.4.0", "lru-cache": "^11.1.0", diff --git a/src/__tests__/use-agent-stream.test.tsx b/src/__tests__/use-agent-stream.test.tsx new file mode 100644 index 0000000..601d162 --- /dev/null +++ b/src/__tests__/use-agent-stream.test.tsx @@ -0,0 +1,129 @@ +// @vitest-environment jsdom +/** + * v0.8.0 P0-4/P3-3: useAgentStream hook 级测试。 + * + * 以前是流式链路最大零覆盖区(仅 stream-event-pipeline 纯函数有表测)。 + * 本文件钉住 hook 的事件分派核心行为: + * - done(completed) → isStreaming=false / agentStatus=idle / runId 清空; + * - done(completed) + finishReason=length → 截断提示 system 消息(P0-4); + * - error(output_length_exceeded) → 友好文案 system 消息(P0-4); + * - reasoning_delta → 消息卡片思考内容累积; + * - text_delta → 正文增量累积。 + * + * 事件总线(P1-1a)使 IPC 与回放共用同一入口:本测试直接经 + * emitStreamBus 注入事件,无需真实 IPC。 + */ + +import { describe, it, expect, beforeEach, vi } from 'vitest'; +import { renderHook } from '@testing-library/react'; +import { useAgentStream } from '../hooks/useAgentStream'; +import { useAgentStore } from '../stores/agent-store'; +import { emitStreamBus } from '../lib/stream-event-bus'; +import '@renderer/lib/i18n-strings'; + +vi.mock('electron-log', () => ({ + default: { info: vi.fn(), warn: vi.fn(), error: vi.fn(), debug: vi.fn() }, +})); + +function makeEvent(partial: Record): Record { + return { + type: 'text_delta', + requestId: 'r', + sessionId: 'sess_1', + iteration: 1, + seq: 0, + timestamp: Date.now(), + runId: 'run_x', + ...partial, + }; +} + +describe('useAgentStream(v0.8.0 P3-3 hook 级)', () => { + beforeEach(() => { + // 重置 store 到"流式中"的典型状态 + useAgentStore.setState({ + currentSessionId: 'sess_1', + isStreaming: true, + currentRunId: 'run_x', + agentStatus: 'thinking', + currentIteration: 1, + messages: [], + traceSteps: [ + { + id: 'trace_1', + iteration: 1, + state: 'THINKING', + states: ['THINKING'], + startedAt: Date.now(), + runId: 'run_x', + }, + ], + tokenUsage: { + inputTokens: 0, + outputTokens: 0, + totalTokens: 0, + lastInputTokens: 0, + lastCompressedSaved: 0, + }, + }); + }); + + it('done(completed) → 收尾:isStreaming=false、runId 清空、agentStatus=idle', () => { + renderHook(() => useAgentStream()); + emitStreamBus('streamEvent', makeEvent({ type: 'done', terminationReason: 'completed' })); + const s = useAgentStore.getState(); + expect(s.isStreaming).toBe(false); + expect(s.currentRunId).toBeNull(); + expect(s.agentStatus).toBe('idle'); + }); + + it('done(completed) + finishReason=length → 插入输出截断提示(P0-4)', () => { + renderHook(() => useAgentStream()); + // 先产出一段正文(模拟截断前的部分回答) + emitStreamBus('streamEvent', makeEvent({ type: 'text_delta', delta: '部分回答' })); + emitStreamBus( + 'streamEvent', + makeEvent({ type: 'done', terminationReason: 'completed', finishReason: 'length' }), + ); + const s = useAgentStore.getState(); + const notice = s.messages.find((m) => m.role === 'system'); + expect(notice).toBeDefined(); + expect(notice?.content).toContain('输出上限'); + expect(s.isStreaming).toBe(false); + }); + + it("error(output_length_exceeded) → 友好文案(含'思考内容耗尽'提示)", () => { + renderHook(() => useAgentStream()); + emitStreamBus( + 'streamEvent', + makeEvent({ + type: 'error', + error: { code: 'output_length_exceeded', message: 'raw message', retryable: false }, + }), + ); + const s = useAgentStore.getState(); + expect(s.agentStatus).toBe('error'); + const err = s.messages.find((m) => m.role === 'system'); + expect(err?.content).toContain('思考内容耗尽'); + }); + + it('reasoning_delta → 思考内容累积进 assistant 卡片', () => { + renderHook(() => useAgentStream()); + emitStreamBus('streamEvent', makeEvent({ type: 'reasoning_delta', delta: '思考A' })); + emitStreamBus('streamEvent', makeEvent({ type: 'reasoning_delta', delta: '思考B', seq: 1 })); + const s = useAgentStore.getState(); + const assistant = s.messages.find((m) => m.role === 'assistant'); + expect(assistant?.reasoningContent).toBe('思考A思考B'); + }); + + it('text_delta → 正文累积进 assistant 卡片(rAF 批处理 flush 后)', async () => { + renderHook(() => useAgentStream()); + emitStreamBus('streamEvent', makeEvent({ type: 'text_delta', delta: '你好' })); + emitStreamBus('streamEvent', makeEvent({ type: 'text_delta', delta: '世界', seq: 1 })); + // text_delta 走 rAF 批处理 —— 等 2 帧让缓冲 flush + await new Promise((r) => setTimeout(r, 40)); + const s = useAgentStore.getState(); + const assistant = s.messages.find((m) => m.role === 'assistant'); + expect(assistant?.content).toBe('你好世界'); + }); +}); diff --git a/src/components/chat/ChatInput.tsx b/src/components/chat/ChatInput.tsx index d774f72..a1ff0ad 100644 --- a/src/components/chat/ChatInput.tsx +++ b/src/components/chat/ChatInput.tsx @@ -9,7 +9,7 @@ * @see docs/MetonaAI-Desktop UI UX 设计集成方案.html — 输入框智能特性 */ -import { useState, useCallback, useRef, useEffect } from 'react'; +import { useState, useCallback, useRef, useEffect, useMemo } from 'react'; import { Box, Typography, @@ -22,7 +22,8 @@ import { List, ListItemButton, } from '@mui/material'; -import { Send, Paperclip, Square, X, FileText, Image as ImageIcon } from 'lucide-react'; +import { Send, Paperclip, Square, X, FileText, Image as ImageIcon, AtSign } from 'lucide-react'; +import Fuse from 'fuse.js'; import { nanoid } from 'nanoid'; import { useAgentStore } from '@renderer/stores/agent-store'; import { useSessionStore } from '@renderer/stores/session-store'; @@ -102,6 +103,12 @@ export function ChatInput(): React.JSX.Element { const [attachments, setAttachments] = useState([]); const [showSlashMenu, setShowSlashMenu] = useState(false); const [slashFilter, setSlashFilter] = useState(''); + // v0.8.0 P2-4: @ 文件提及 —— 联想弹层状态 + 工作空间文件索引(懒加载) + const [mentionOpen, setMentionOpen] = useState(false); + const [mentionQuery, setMentionQuery] = useState(''); + const [mentionIndex, setMentionIndex] = useState(0); + const [workspaceFiles, setWorkspaceFiles] = useState([]); + const workspaceFilesLoadedRef = useRef(false); const textareaRef = useRef(null); const fileInputRef = useRef(null); const sendMessage = useAgentStore((s) => s.sendMessage); @@ -302,6 +309,29 @@ export function ChatInput(): React.JSX.Element { // ===== 发送消息 ===== + /** + * v0.8.0 P2-4: 从消息文本提取 @ 文件提及路径。 + * 规则:'@' 前为单词字符(邮箱场景)跳过;路径字符集 [A-Za-z0-9._\-/]; + * 尾部 '.' 剥离(句末标点);至少包含 '.' 或 '/'(排除纯代词式 @someone)。 + */ + const extractMentions = (text: string): string[] => { + const out: string[] = []; + for (let i = 0; i < text.length; i++) { + if (text[i] !== '@') continue; + if (i > 0 && /\w/.test(text[i - 1])) continue; + let j = i + 1; + let path = ''; + while (j < text.length && /[A-Za-z0-9._\-/]/.test(text[j])) { + path += text[j]; + j++; + } + path = path.replace(/\.+$/, ''); + if (path && (path.includes('/') || path.includes('.'))) out.push(path); + i = j - 1; + } + return [...new Set(out)]; + }; + const handleSend = useCallback(async () => { const trimmed = input.trim(); if (!trimmed && attachments.length === 0) return; @@ -387,10 +417,63 @@ export function ChatInput(): React.JSX.Element { } } + // v0.8.0 P2-4: @ 文件提及注入 —— 解析 @path token,读取文本片段并作为 + // 文本附件注入(复用附件管线:UI 卡片 + 主进程"内容已内联/截断"提示)。 + // 主进程 agent-store.sendMessage 会把 textContent 结构化内联进 LLM 消息。 + const mentionInfos: Array<{ + id: string; + name: string; + type: 'text'; + size: number; + textContent: string; + truncated?: boolean; + }> = []; + const mentionedPaths = extractMentions(messageContent).slice(0, 5); + for (const p of mentionedPaths) { + try { + const clip = await window.metona?.workspace?.readFileClip?.(p); + if (clip?.success && typeof clip.content === 'string') { + let textContent = clip.content; + if (clip.truncated) { + textContent += t('input.text.truncatedNote', { + size: formatFileSize(clip.size ?? 0), + limit: formatFileSize(MAX_TEXT_ATTACHMENT_BYTES), + }); + } + mentionInfos.push({ + id: `mnt_${nanoid(6)}`, + name: p, + type: 'text', + size: clip.size ?? clip.content.length, + textContent, + truncated: clip.truncated, + }); + } else { + import('@metona-team/metona-toast') + .then((mod) => + mod.default.warning( + t('input.mention.failed', { path: p, reason: clip?.error ?? '' }), + ), + ) + .catch(() => {}); + } + } catch (err) { + import('@metona-team/metona-toast') + .then((mod) => + mod.default.warning( + t('input.mention.failed', { path: p, reason: (err as Error).message }), + ), + ) + .catch(() => {}); + } + } + + const allAttachmentInfos = [...attachmentInfos, ...mentionInfos]; + sendMessage( messageContent, images.length > 0 ? images : undefined, - attachmentInfos.length > 0 ? attachmentInfos : undefined, + allAttachmentInfos.length > 0 ? allAttachmentInfos : undefined, ); setInput(''); setAttachments([]); @@ -403,8 +486,60 @@ export function ChatInput(): React.JSX.Element { abort(); }, [abort]); + // v0.8.0 P2-4: @ 联想候选(Fuse 模糊匹配;query 为空时按路径字典序取前 8 条) + const mentionCandidates = useMemo(() => { + if (workspaceFiles.length === 0) return []; + if (!mentionQuery) return [...workspaceFiles].sort().slice(0, 8); + const fuse = new Fuse(workspaceFiles, { includeScore: true, threshold: 0.4 }); + return fuse + .search(mentionQuery) + .map((r) => r.item) + .slice(0, 8); + }, [workspaceFiles, mentionQuery]); + + // v0.8.0 P2-4: 选中候选 —— 把 "@query" 片段替换为 "@path" + const applyMention = useCallback( + (path: string): void => { + setInput((prev) => { + const pos = prev.lastIndexOf(`@${mentionQuery}`); + if (pos < 0) return `${prev}${path}`; + return `${prev.slice(0, pos)}@${path}${prev.slice(pos + mentionQuery.length + 1)}`; + }); + setMentionOpen(false); + setMentionQuery(''); + requestAnimationFrame(() => textareaRef.current?.focus()); + }, + [mentionQuery], + ); + const handleKeyDown = useCallback( (e: React.KeyboardEvent) => { + // v0.8.0 P2-4: @ 联想弹层的键盘导航(优先级高于发送/斜杠菜单) + if (mentionOpen && mentionCandidates.length > 0) { + if (e.key === 'ArrowDown') { + e.preventDefault(); + setMentionIndex((i) => (i + 1) % mentionCandidates.length); + return; + } + if (e.key === 'ArrowUp') { + e.preventDefault(); + setMentionIndex((i) => (i - 1 + mentionCandidates.length) % mentionCandidates.length); + return; + } + if (e.key === 'Tab' || e.key === 'Enter') { + if (e.key === 'Enter' && (e.nativeEvent.isComposing || e.keyCode === 229)) return; + e.preventDefault(); + applyMention(mentionCandidates[Math.min(mentionIndex, mentionCandidates.length - 1)]); + return; + } + if (e.key === 'Escape') { + e.preventDefault(); + setMentionOpen(false); + setMentionQuery(''); + return; + } + } + // H-9 修复: 快捷键对齐规范 // @see docs/MetonaAI-Desktop UI UX 设计集成方案.html — 快捷键规范 // 规范要求: Cmd/Ctrl+Enter = 发送消息, Cmd/Ctrl+Shift+Enter = 换行 @@ -456,7 +591,7 @@ export function ChatInput(): React.JSX.Element { return; } }, - [handleSend, showSlashMenu], + [handleSend, showSlashMenu, mentionOpen, mentionCandidates, mentionIndex, applyMention], ); // H-8 修复: 使用 InputBase 后,onChange 类型需兼容 HTMLInputElement | HTMLTextAreaElement @@ -473,6 +608,30 @@ export function ChatInput(): React.JSX.Element { } else { setShowSlashMenu(false); } + + // v0.8.0 P2-4: @ 文件提及触发 —— 光标前最后一个 '@' 到光标之间无空白时 + // 视为提及输入中;空白/删除 '@' 即关闭。首次触发懒加载文件索引。 + const pos = (e.target as HTMLTextAreaElement | HTMLInputElement).selectionStart ?? v.length; + const before = v.slice(0, pos); + const atIdx = before.lastIndexOf('@'); + if (atIdx >= 0 && !/[\s]/.test(before.slice(atIdx + 1))) { + const q = before.slice(atIdx + 1); + setMentionOpen(true); + setMentionQuery(q); + setMentionIndex(0); + if (!workspaceFilesLoadedRef.current) { + workspaceFilesLoadedRef.current = true; + void window.metona?.workspace + ?.listFiles?.() + .then((r) => { + if (r?.success && r.data) setWorkspaceFiles(r.data); + }) + .catch((err) => console.error('[ChatInput] listFiles failed:', err)); + } + } else { + setMentionOpen(false); + setMentionQuery(''); + } }, [], ); @@ -560,6 +719,51 @@ export function ChatInput(): React.JSX.Element { )} + {/* v0.8.0 P2-4: @ 文件提及联想弹层(MUI Paper+List,与斜杠菜单同形态) */} + {mentionOpen && mentionCandidates.length > 0 && ( + + + {mentionCandidates.map((path, idx) => ( + setMentionIndex(idx)} + onClick={() => applyMention(path)} + sx={{ gap: 1, py: 0.75 }} + > + + + {path} + + + ))} + + + )} + s.detailTab); const setTab = useUIStore((s) => s.setDetailTab); + // v0.8.0 P2-2: 当前会话 ID(回放按钮的会话标识) + const currentSessionId = useAgentStore((s) => s.currentSessionId); // v0.3.0 修复:验证 tab 值有效性,非预期值回退到 'trace' const validTabs: DetailTab[] = ['trace', 'memory', 'tasks', 'workspace']; @@ -97,6 +102,8 @@ export function DetailPanel(): React.JSX.Element { + {/* v0.8.0 P2-2: 会话回放(TRACE JSONL 播放器) */} + )} diff --git a/src/components/layout/Sidebar.tsx b/src/components/layout/Sidebar.tsx index 70abe77..6f6172f 100644 --- a/src/components/layout/Sidebar.tsx +++ b/src/components/layout/Sidebar.tsx @@ -272,6 +272,8 @@ export function Sidebar(): React.JSX.Element { {/* F-2 修复: 归档会话视图 —— 此前归档入口激活后(会话右键菜单), 归档会话从列表永久消失且无恢复路径(数据在 DB 但 UI 不可达) */} + {/* v0.8.0 P2-1: 回收站 —— 软删除会话恢复/彻底删除 */} + @@ -379,6 +381,173 @@ function ArchivedSessionsPanel(): React.JSX.Element | null { ); } +/** + * v0.8.0 P2-1: 回收站折叠面板 —— 软删除会话的恢复 / 彻底删除入口。 + * 自主拉取 sessions:listTrash(回收站数据不在 session-store 中); + * 展开时刷新,恢复/彻底删除后同步刷新。彻底删除需确认(级联清消息,不可逆)。 + */ +function TrashSessionsPanel(): React.JSX.Element | null { + const [expanded, setExpanded] = useState(false); + const [trash, setTrash] = useState< + Array<{ id: string; title: string; deletedAt?: number; updatedAt?: number }> + >([]); + const [purgeTarget, setPurgeTarget] = useState<{ id: string; title: string } | null>(null); + + const refresh = async (): Promise => { + try { + const rows = (await window.metona?.sessions?.listTrash()) ?? []; + setTrash(rows); + } catch (err) { + console.error('[Sidebar] listTrash failed:', err); + } + }; + + const handleExpand = (): void => { + const next = !expanded; + setExpanded(next); + if (next) void refresh(); + }; + + const handleRestore = async (sessionId: string): Promise => { + try { + const r = await window.metona?.sessions?.restore(sessionId); + if (r?.success) { + void refresh(); + } else { + import('@metona-team/metona-toast') + .then((mod) => mod.default.error(r?.error ?? t('sidebar.toast.restoreFailed'))) + .catch(() => {}); + } + } catch (err) { + console.error('[Sidebar] Restore from trash failed:', err); + } + }; + + const handlePurge = async (): Promise => { + if (!purgeTarget) return; + try { + const r = await window.metona?.sessions?.purge(purgeTarget.id); + if (r?.success) { + void refresh(); + } else { + import('@metona-team/metona-toast') + .then((mod) => mod.default.error(r?.error ?? t('sidebar.toast.deleteFailed'))) + .catch(() => {}); + } + } catch (err) { + console.error('[Sidebar] Purge failed:', err); + } finally { + setPurgeTarget(null); + } + }; + + if (trash.length === 0 && !expanded) return null; + + return ( + + + + {expanded ? : } + + + + {t('sidebar.trash')} + + } + /> + {trash.length > 0 && ( + + {t('sidebar.count', { count: trash.length })} + + )} + + + + {trash.length === 0 ? ( + + {t('sidebar.trash.empty')} + + ) : ( + trash.map((s) => ( + + + {s.title} + + + + + )) + )} + + + {/* 彻底删除确认(级联清消息,不可逆) */} + setPurgeTarget(null)}> + {t('sidebar.trash.purgeTitle')} + + + {t('sidebar.trash.purgeBody', { title: purgeTarget?.title ?? '' })} + + + + + + + + + ); +} + function SessionItem({ session, isActive, @@ -624,6 +793,8 @@ function ToolManagerPanel() { low: 'LOW', medium: 'MEDIUM', high: 'HIGH', + // v0.8.0 P1-3.8: 补 critical —— 旧映射缺失导致 CRITICAL 工具显示原始字符串 + critical: 'CRITICAL', }; return ( diff --git a/src/components/memory/MemoryViewer.tsx b/src/components/memory/MemoryViewer.tsx index ad947c0..1faceb1 100644 --- a/src/components/memory/MemoryViewer.tsx +++ b/src/components/memory/MemoryViewer.tsx @@ -374,7 +374,7 @@ export function MemoryViewer(): React.JSX.Element { variant="caption" sx={{ textAlign: 'center', py: 4, display: 'block', color: 'text.disabled' }} > - 暂无记忆数据 + {t('memory.empty')} ) : ( MEMORY_TYPES.map((type) => { diff --git a/src/components/settings/LLMSettings.tsx b/src/components/settings/LLMSettings.tsx index 0996d11..1804c87 100644 --- a/src/components/settings/LLMSettings.tsx +++ b/src/components/settings/LLMSettings.tsx @@ -32,6 +32,8 @@ import { PROVIDER_LABELS } from '@renderer/lib/constants'; import { t } from '@renderer/lib/i18n'; import '@renderer/lib/i18n-strings'; import { PROVIDER_URLS } from './useConfig'; +// v0.8.0 FEAT-1: 输出上限配置可见性门控(Provider 支持矩阵) +import { supportsOutputLimitConfig } from '@renderer/lib/model-capabilities'; /** v0.7.2 P3-9: 动态模型条目(渲染端子集,与 global.d.ts MetonaModelInfoLite 对齐) */ interface ModelOption { @@ -39,6 +41,8 @@ interface ModelOption { name?: string; supportsToolCalling?: boolean; supportsThinking?: boolean; + /** v0.8.0 FEAT-1: 模型输出上限(llm:listModels 元数据,用于钳制提示) */ + maxOutputTokens?: number; description?: string; } @@ -52,6 +56,10 @@ export function LLMSettings() { const [apiKey, setApiKey] = useState(''); const [baseURL, setBaseURL] = useState(''); const [numCtx, setNumCtx] = useState(null); + // v0.8.0 FEAT-1: 最大输出上限(llm.maxTokens,按请求传给 API;null = 使用引擎默认) + const [maxOutputTokens, setMaxOutputTokens] = useState(null); + // v0.8.0 P0-3: agent.enableThinking(模型不支持思考时的联动提示) + const [agentThinkingEnabled, setAgentThinkingEnabled] = useState(true); // v0.3.1: DeepSeek/Agnes/MiMo contextWindow 可配置(不再写死) const [dsCtxWindow, setDsCtxWindow] = useState(1000000); const [agnesCtxWindow, setAgnesCtxWindow] = useState(1000000); @@ -235,14 +243,18 @@ export function LLMSettings() { window.metona.config.get('llm.fallbackBaseURL'), // v0.5.4: 多模态开关 window.metona.config.get('llm.multimodalEnabled'), + // v0.8.0 FEAT-1: 最大输出上限 + 思考开关(P0-3 模型能力联动提示) + window.metona.config.get('llm.maxTokens'), + window.metona.config.get('agent.enableThinking'), ]); if (cancelled) return; - const [p, m, k, u, nc, ds, ag, mi, oa, an, fbp, fbm, fbk, fbu, mm] = results; + const [p, m, k, u, nc, ds, ag, mi, oa, an, fbp, fbm, fbk, fbu, mm, mt, et] = results; setProvider((p as string) ?? ''); setModel((m as string) ?? ''); setApiKey((k as string) ?? ''); setBaseURL((u as string) ?? ''); setNumCtx((nc as number | null) ?? null); + if (typeof mt === 'number' && mt > 0) setMaxOutputTokens(mt); if (typeof ds === 'number' && ds > 0) setDsCtxWindow(ds); if (typeof ag === 'number' && ag > 0) setAgnesCtxWindow(ag); if (typeof mi === 'number' && mi > 0) setMimoCtxWindow(mi); @@ -253,6 +265,7 @@ export function LLMSettings() { setFbApiKey((fbk as string) ?? ''); setFbBaseURL((fbu as string) ?? ''); setMultimodalEnabled(mm === true); + setAgentThinkingEnabled(et !== false); } catch (err) { console.error('[LLMSettings]', err); } finally { @@ -313,12 +326,27 @@ export function LLMSettings() { const mimoCtxError = !Number.isFinite(mimoCtxWindow) || mimoCtxWindow < 4096; const oaCtxError = !Number.isFinite(oaCtxWindow) || oaCtxWindow < 4096; const anthropicCtxError = !Number.isFinite(anthropicCtxWindow) || anthropicCtxWindow < 4096; + // v0.8.0 FEAT-1: 最大输出上限校验(≥256;上限钳制由 adapter 按模型元信息执行) + const maxTokensError = + maxOutputTokens != null && (!Number.isFinite(maxOutputTokens) || maxOutputTokens < 256); + + // v0.8.0 FEAT-1: 当前模型的输出上限元信息(模型列表已加载时展示钳制提示) + const modelMeta = model ? modelOptions.find((o) => o.id === model) : undefined; + const modelMaxOutput = modelMeta?.maxOutputTokens; + const exceedsModelCap = + maxOutputTokens != null && + typeof modelMaxOutput === 'number' && + maxOutputTokens > modelMaxOutput; + // v0.8.0 P0-3: 模型不支持思考 + 全局思考开关开启 → 联动提示 + const thinkingUnsupportedByModel = + model != null && modelMeta?.supportsThinking === false && agentThinkingEnabled; // 是否存在阻断保存的错误(API Key 为空只警告,不阻断 — 允许先填其他字段再回来填 key) const hasBlockingError = urlError || modelHasSpace || numCtxError || + maxTokensError || (provider === 'deepseek' && dsCtxError) || (provider === 'agnes' && agnesCtxError) || (provider === 'mimo' && mimoCtxError) || @@ -370,6 +398,11 @@ export function LLMSettings() { { key: 'llm.baseURL', value: baseURL }, // v0.5.4: 多模态总开关 { key: 'llm.multimodalEnabled', value: multimodalEnabled }, + // v0.8.0 FEAT-1: 最大输出上限(按请求传给 API;清空 = 引擎默认 63488) + { + key: 'llm.maxTokens', + value: maxOutputTokens != null && maxOutputTokens >= 256 ? maxOutputTokens : 63488, + }, { key: 'ollama.numCtx', value: numCtx }, { key: 'deepseek.contextWindow', value: dsCtxWindow }, { key: 'agnes.contextWindow', value: agnesCtxWindow }, @@ -390,6 +423,16 @@ export function LLMSettings() { } else { // v0.5.4: 保存成功后同步多模态开关到 Agent Store(立即生效,控制上传入口) useAgentStore.getState().setMultimodalEnabled(multimodalEnabled); + // v0.8.0 FEAT-1: 配置值超过模型上限 → 明示"将按模型上限生效" + //(adapter 侧按 MODEL_INFO.maxOutputTokens 钳制,llm.maxTokens 经 + // applyEngineConfigKey 热更新引擎,无需重启) + if (exceedsModelCap && typeof modelMaxOutput === 'number') { + import('@metona-team/metona-toast') + .then((mod) => + mod.default.warning(t('llm.maxTokens.exceedCap', { cap: modelMaxOutput })), + ) + .catch(() => {}); + } import('@metona-team/metona-toast') .then((mod) => mod.default.success(t('llm.save.success'))) .catch(() => {}); @@ -530,6 +573,37 @@ export function LLMSettings() { {t('llm.models.hint')} + {/* v0.8.0 P0-3: 模型不支持思考 × 全局思考开关 → 联动提示(adapter 侧已硬门控) */} + {thinkingUnsupportedByModel && ( + + {t('llm.thinking.unsupported')} + + )} + {/* v0.8.0 FEAT-1: 最大输出上限 —— 仅对支持输出上限参数的 Provider 显示 + (PROVIDER_OUTPUT_LIMIT_SUPPORT 矩阵门控;经 llm.maxTokens 热生效) */} + {supportsOutputLimitConfig(provider) && ( + { + const v = e.target.value; + setMaxOutputTokens(v === '' ? null : Number(v)); + }} + slotProps={{ htmlInput: { min: 256, step: 256 } }} + error={maxTokensError} + helperText={ + maxTokensError + ? t('llm.maxTokens.minError') + : exceedsModelCap && typeof modelMaxOutput === 'number' + ? t('llm.maxTokens.exceedCap', { cap: modelMaxOutput }) + : typeof modelMaxOutput === 'number' + ? `${t('llm.maxTokens.helper')} · ${t('llm.maxTokens.capInfo', { cap: modelMaxOutput })}` + : t('llm.maxTokens.helper') + } + /> + )} {/* v0.5.4: 多模态总开关 — 未开启时即使模型支持也不能上传图片 */} + {/* v0.8.0 P2-3: 检查更新 / 下载安装(electron-updater 状态实时显示) */} + + {/* v0.7.3 P3-4: SLO / 健康快照可视化(只读 + 手动刷新) */} @@ -495,6 +499,107 @@ function TraceFilesPanel(): React.JSX.Element { ); } +// ===== v0.8.0 P2-3: 检查更新 / 下载安装(electron-updater) ===== + +function UpdatePanel(): React.JSX.Element { + const [checking, setChecking] = useState(false); + const [installing, setInstalling] = useState(false); + const [result, setResult] = useState(null); + const [progress, setProgress] = useState(null); + + useEffect(() => { + if (!window.metona?.app?.onUpdateStatus) return; + const off = window.metona.app.onUpdateStatus((event) => { + if (event.status === 'downloading') { + setProgress(event.percent ?? 0); + } else if (event.status === 'downloaded') { + setProgress(100); + setResult(t('settings.logs.update.downloaded')); + } else if (event.status === 'available') { + setResult(t('settings.logs.update.available', { version: event.latestVersion ?? '' })); + } else if (event.status === 'not-available') { + setResult(t('settings.logs.update.upToDate', { version: event.latestVersion ?? '' })); + } else if (event.status === 'error') { + setResult(t('settings.logs.update.error', { message: event.message ?? '' })); + } + }); + return off; + }, []); + + const handleCheck = async (): Promise => { + setChecking(true); + setResult(null); + setProgress(null); + try { + const r = await window.metona?.app?.updateCheck(); + if (r?.status === 'available') { + setResult(t('settings.logs.update.available', { version: r.latestVersion })); + } else if (r?.status === 'up-to-date') { + setResult(t('settings.logs.update.upToDate', { version: r.latestVersion })); + } else if (r?.status === 'disabled') { + setResult(t('settings.logs.update.disabled')); + } else if (r?.status === 'error') { + setResult(t('settings.logs.update.error', { message: r.message })); + } + } finally { + setChecking(false); + } + }; + + const handleInstall = async (): Promise => { + setInstalling(true); + try { + const r = await window.metona?.app?.updateInstall(); + if (r && r.success === false) { + setResult(t('settings.logs.update.error', { message: r.error ?? '' })); + } + } finally { + setInstalling(false); + } + }; + + return ( + <> + + + {t('settings.logs.update.title')} + + + + + + {progress != null && ( + + )} + {result && ( + + {result} + + )} + + ); +} + // ===== v0.7.3 P3-4: SLO / 健康快照可视化 ===== function HealthSnapshotPanel(): React.JSX.Element { diff --git a/src/components/settings/MCPSettings.tsx b/src/components/settings/MCPSettings.tsx index 70fb13e..232ab41 100644 --- a/src/components/settings/MCPSettings.tsx +++ b/src/components/settings/MCPSettings.tsx @@ -164,6 +164,33 @@ export function MCPSettings() { } }; + // v0.8.0 P2-5: Resources / Prompts 发现视图(按 server 名懒加载展开) + const [contentsFor, setContentsFor] = useState(null); + const [contents, setContents] = useState<{ + resources: Array<{ uri: string; name: string; description?: string; mimeType?: string }>; + prompts: Array<{ + name: string; + description?: string; + arguments?: Array<{ name: string; description?: string; required?: boolean }>; + }>; + } | null>(null); + const handleToggleContents = async (name: string): Promise => { + if (contentsFor === name) { + setContentsFor(null); + setContents(null); + return; + } + try { + const r = await window.metona?.mcp?.listServerContents?.(name); + if (r?.success && r.data) { + setContentsFor(name); + setContents(r.data); + } + } catch (err) { + console.error('[MCPSettings] listServerContents failed:', err); + } + }; + return ( @@ -175,80 +202,124 @@ export function MCPSettings() { ) : ( servers.map((s) => ( - - - - - {s.name} - - - {statusLabel(s)} - - {s.toolCount > 0 && ( - - {t('settings.mcp.toolCount', { count: s.toolCount })} + + + + + + {s.name} - )} - - - - {/* L-11 修复: 点击移除打开 MUI Dialog 二次确认,而非原生 confirm() */} + }} + > + {s.status === 'connected' + ? t('settings.mcp.disconnect') + : t('settings.mcp.connect')} + + {/* L-11 修复: 点击移除打开 MUI Dialog 二次确认,而非原生 confirm() */} + + + + {s.status === 'connected' && ( - - + )} + {contentsFor === s.name && contents && ( + + {contents.resources.length === 0 && contents.prompts.length === 0 && ( + + {t('settings.mcp.contents.empty')} + + )} + {contents.resources.map((r) => ( + + 📄 {r.name} · {r.uri} + + ))} + {contents.prompts.map((p) => ( + + 💬 {p.name} + {p.arguments?.length ? ` (${p.arguments.map((a) => a.name).join(', ')})` : ''} + + ))} + + )} + )) )} diff --git a/src/components/trace/SessionReplayPlayer.tsx b/src/components/trace/SessionReplayPlayer.tsx new file mode 100644 index 0000000..63f0ccd --- /dev/null +++ b/src/components/trace/SessionReplayPlayer.tsx @@ -0,0 +1,353 @@ +/** + * SessionReplayPlayer — 会话回放播放器(v0.8.0 P2-2) + * + * 基于 TRACE 层的 JSONL 录制文件(session_{id}_*.jsonl,9 种事件)做完整回放: + * 文件选择 → 时间轴(iteration_start 定锚点)→ 播放/暂停/步进/速度控制 → + * 事件流渲染(思考/正文/工具调用/结果)。 + * + * 数据来源:sessions:listRecordings(列出该会话录制)+ sessions:readRecording + * (严格文件名白名单 + 目录边界校验,主进程侧实施)。 + */ + +import { useCallback, useEffect, useMemo, useRef, useState } from 'react'; +import { + Box, + Button, + Dialog, + DialogContent, + DialogTitle, + IconButton, + LinearProgress, + Slider, + Stack, + Typography, + Chip, + Select, + MenuItem, + FormControl, + InputLabel, +} from '@mui/material'; +import { Play, Pause, ChevronLeft, ChevronRight, X, Film } from 'lucide-react'; +import { formatDuration, formatFileSize } from '@renderer/lib/formatters'; +import { t } from '@renderer/lib/i18n'; +import '@renderer/lib/i18n-strings'; + +/** 录制事件(session-recorder TraceEvent 的宽松形态) */ +interface ReplayEvent { + seq?: number; + ts?: string; + event?: string; + sessionId?: string; + iteration?: number; + [key: string]: unknown; +} + +const SPEEDS = [0.5, 1, 2, 4]; + +export function SessionReplayPlayer({ + open, + onClose, + sessionId, +}: { + open: boolean; + onClose: () => void; + sessionId: string | null; +}): React.JSX.Element { + const [files, setFiles] = useState>([]); + const [selected, setSelected] = useState(''); + const [events, setEvents] = useState([]); + const [skipped, setSkipped] = useState(0); + const [loading, setLoading] = useState(false); + const [error, setError] = useState(null); + const [cursor, setCursor] = useState(0); + const [playing, setPlaying] = useState(false); + const [speed, setSpeed] = useState(1); + const timerRef = useRef | null>(null); + + // 打开时拉取录制文件列表 + useEffect(() => { + if (!open || !sessionId) return; + setSelected(''); + setEvents([]); + setSkipped(0); + setCursor(0); + setPlaying(false); + setError(null); + void (async () => { + try { + const r = await window.metona?.sessions?.listRecordings(sessionId); + if (r?.success && r.data) setFiles(r.data); + else setError(r?.error ?? t('replay.loadFailed')); + } catch (err) { + setError((err as Error).message); + } + })(); + }, [open, sessionId]); + + const loadFile = useCallback(async (name: string): Promise => { + setSelected(name); + setEvents([]); + setCursor(0); + setPlaying(false); + setLoading(true); + setError(null); + try { + const r = await window.metona?.sessions?.readRecording(name); + if (r?.success && r.data) { + setEvents((r.data.events as ReplayEvent[]) ?? []); + setSkipped(r.data.skipped ?? 0); + } else { + setError(r?.error ?? t('replay.loadFailed')); + } + } catch (err) { + setError((err as Error).message); + } finally { + setLoading(false); + } + }, []); + + // 播放循环:按 speed 定时推进游标,到尾部自动暂停 + useEffect(() => { + if (!playing || events.length === 0) return; + if (cursor >= events.length - 1) { + setPlaying(false); + return; + } + timerRef.current = setTimeout(() => { + setCursor((c) => Math.min(c + 1, events.length - 1)); + }, 60 / speed); + return () => { + if (timerRef.current) clearTimeout(timerRef.current); + }; + }, [playing, cursor, speed, events.length]); + + /** 渲染单条事件为可读行 */ + const renderEvent = useMemo(() => { + return (ev: ReplayEvent): React.JSX.Element => { + const type = String(ev.event ?? ''); + let detail = ''; + switch (type) { + case 'iteration_start': + detail = t('replay.event.iterationStart', { iteration: ev.iteration ?? 0 }); + break; + case 'iteration_end': + detail = t('replay.event.iterationEnd', { + iteration: ev.iteration ?? 0, + duration: formatDuration(Number(ev.durationMs ?? 0)), + }); + break; + case 'llm_request': + detail = `${ev.provider ?? ''}/${ev.model ?? ''}`; + break; + case 'llm_response': + detail = String((ev.contentPreview as string) ?? ''); + break; + case 'tool_call': + detail = `${String(ev.tool ?? '')} ${JSON.stringify(ev.args ?? {})}`; + break; + case 'tool_result': + detail = `${String(ev.tool ?? '')} ${ev.success ? '✓' : '✗'} ${formatDuration( + Number(ev.durationMs ?? 0), + )}${ev.error ? ` — ${String(ev.error).slice(0, 120)}` : ''}`; + break; + case 'session_end': + detail = t('replay.event.sessionEnd', { + reason: String(ev.terminationReason ?? ''), + iterations: Number(ev.totalIterations ?? 0), + }); + break; + default: + detail = JSON.stringify(ev).slice(0, 160); + } + const isTool = type === 'tool_call' || type === 'tool_result'; + const isThought = type === 'llm_response' || type === 'llm_request'; + return ( + + + + + {detail || '—'} + + {ev.ts && ( + + {String(ev.ts).slice(11, 19)} + + )} + + + ); + }; + }, []); + + return ( + + + {t('replay.title')} + + + + + + + {/* 文件选择 */} + + {t('replay.file')} + + + + {loading && } + {error && ( + + {error} + + )} + + {events.length > 0 && ( + <> + {/* 时间轴 + 控制 */} + + setCursor((c) => Math.max(0, c - 1))} + disabled={cursor === 0} + aria-label={t('replay.stepBack')} + > + + + setPlaying((p) => !p)} + disabled={cursor >= events.length - 1} + aria-label={playing ? t('replay.pause') : t('replay.play')} + > + {playing ? : } + + setCursor((c) => Math.min(events.length - 1, c + 1))} + disabled={cursor >= events.length - 1} + aria-label={t('replay.stepForward')} + > + + + { + setPlaying(false); + setCursor(Number(v)); + }} + sx={{ flex: 1 }} + /> + + + + + {cursor + 1}/{events.length} + + + + {/* 事件流(只渲染到当前游标,最近 200 条防 DOM 爆炸) */} + + {events.slice(Math.max(0, cursor - 199), cursor + 1).map(renderEvent)} + + {skipped > 0 && ( + + {t('replay.skippedLines', { count: skipped })} + + )} + + )} + + {files.length === 0 && !loading && !error && ( + + {t('replay.noRecordings')} + + )} + + + + ); +} + +/** Trace 面板的回放入口按钮 + 播放器挂载 */ +export function SessionReplayButton({ + sessionId, +}: { + sessionId: string | null; +}): React.JSX.Element { + const [open, setOpen] = useState(false); + return ( + <> + + setOpen(false)} sessionId={sessionId} /> + + ); +} diff --git a/src/hooks/useAgentStream.ts b/src/hooks/useAgentStream.ts index a45c025..4898fb5 100644 --- a/src/hooks/useAgentStream.ts +++ b/src/hooks/useAgentStream.ts @@ -27,6 +27,8 @@ import { needsNewAssistantCard, findTraceStepIndexForToolCall, } from '@renderer/lib/stream-event-pipeline'; +// v0.8.0 P1-1a: 事件总线 —— IPC 与后台会话回放统一入口 +import { subscribeStreamBus, emitStreamBus } from '@renderer/lib/stream-event-bus'; // v0.7.2 A4: traceSteps 入 store 时即剥离工具结果中的 base64/超长字段 import { toDisplayResult } from '@renderer/lib/tool-result-display'; @@ -119,6 +121,32 @@ export function useAgentStream(): void { flushTraceThought(); }, [flushTextDelta, flushTraceThought]); + /** + * v0.8.0 P1-1d: 清扫流结束后仍处于 executing/pending 的工具调用卡片。 + * abort 尾巴过滤会把中断后到达的 tool_result 丢弃(设计意图:停止后聊天区 + * 不得长出内容),副作用是卡片永久转圈。done/error 收尾时统一置 error 态。 + */ + const sweepInterruptedToolCalls = useCallback((): void => { + const store = useAgentStore.getState(); + const stamp = t('agent.system.tool_interrupted'); + const sweep = (tcs?: ToolCallInfo[]): ToolCallInfo[] | undefined => + tcs?.some((tc) => tc.status === 'executing' || tc.status === 'pending') + ? tcs.map((tc) => + tc.status === 'executing' || tc.status === 'pending' + ? { ...tc, status: 'error' as const, error: stamp } + : tc, + ) + : undefined; + for (const m of store.messages) { + const updated = sweep(m.toolCalls); + if (updated) store.updateMessage(m.id, { toolCalls: updated }); + } + for (const step of store.traceSteps) { + const updated = sweep(step.toolCalls); + if (updated && step.id) store.updateTraceStepById(step.id, { toolCalls: updated }); + } + }, []); + // ===== 流式内容事件 ===== useEffect(() => { if (!window.metona?.agent?.onStreamEvent) return; @@ -134,7 +162,13 @@ export function useAgentStream(): void { if (textRafIdRef.current != null) return; textRafIdRef.current = requestAnimationFrame(flushTextDelta); }; - const unsubscribe = window.metona.agent.onStreamEvent((event: unknown) => { + // v0.8.0 P1-1a: IPC 监听器只负责把事件投递进总线;处理逻辑订阅总线。 + // 后台会话回放(agent-store.setCurrentSession)经同一总线灌入缓冲事件, + // 与实时事件无缝衔接。 + const unsubscribeIpc = window.metona.agent.onStreamEvent((event: unknown) => { + emitStreamBus('streamEvent', event); + }); + const unsubscribe = subscribeStreamBus('streamEvent', (event: unknown) => { const data = event as { type?: string; requestId?: string; @@ -170,6 +204,8 @@ export function useAgentStream(): void { }; /** DONE 事件的终止原因(completed / max_iterations / timeout / user_interrupt / dead_loop / error) */ terminationReason?: string; + /** v0.8.0 P0-1: DONE 事件携带的 Provider 原生停止原因(length 触发截断提示) */ + finishReason?: string; error?: { code: string; message: string }; state?: string; }; @@ -529,6 +565,35 @@ export function useAgentStream(): void { traceRafIdRef.current = null; } flushTraceThought(); + // v0.8.0 P1-1d: 清扫悬空的 executing/pending 工具卡片 —— abort 尾巴 + // 过滤曾让中断后到达的 tool_result 被丢弃,卡片永久转圈 + sweepInterruptedToolCalls(); + // v0.8.0 P0-4: 输出截断 / 空完成可见性(引擎空响应守卫的 UI 侧配合) + if (reason === 'completed' && data.finishReason === 'length') { + getStore().addMessage({ + id: genMsgId('system'), + role: 'system', + content: t('agent.terminated.output_truncated'), + timestamp: Date.now(), + }); + } else if (reason === 'completed') { + const targetId = currentRunAssistantIdRef.current; + const card = targetId ? getStore().messages.find((m) => m.id === targetId) : undefined; + if ( + card && + card.role === 'assistant' && + !card.content && + !card.reasoningContent && + !(card.toolCalls && card.toolCalls.length > 0) + ) { + getStore().addMessage({ + id: genMsgId('system'), + role: 'system', + content: t('agent.system.empty_completed'), + timestamp: Date.now(), + }); + } + } getStore().setStreaming(false); getStore().setCurrentRunId(null); // 不覆盖 error 状态 — error handler 已设置 agentStatus='error' @@ -559,23 +624,35 @@ export function useAgentStream(): void { getStore().setCurrentRunId(null); getStore().setAgentStatus('error'); getStore().updateLastTraceStep({ completedAt: Date.now() }); + // v0.8.0 P1-1d: 清扫悬空的 executing/pending 工具卡片 + sweepInterruptedToolCalls(); // v0.3.17: 对 content_filtered 错误码显示更友好的提示 + // v0.8.0 P0-4: output_length_exceeded / invalid_response(空响应重试耗尽) + // 映射针对性文案(引擎 P0-2 空响应守卫的结构化错误码) const errorCode = data.error?.code; const errorMessage = data.error?.message ?? '未知错误'; + const friendlyContent = + errorCode === 'content_filtered' + ? t('agent.error.content_filtered', { message: errorMessage }) + : errorCode === 'output_length_exceeded' + ? t('agent.error.output_length_exceeded') + : errorCode === 'invalid_response' + ? t('agent.error.empty_response') + : t('agent.error.generic', { message: errorMessage }); getStore().addMessage({ id: genMsgId('error'), role: 'system', - content: - errorCode === 'content_filtered' - ? t('agent.error.content_filtered', { message: errorMessage }) - : t('agent.error.generic', { message: errorMessage }), + content: friendlyContent, timestamp: Date.now(), }); break; } }); - cleanupRef.current = unsubscribe; + cleanupRef.current = () => { + unsubscribe(); + unsubscribeIpc(); + }; return () => { // F5: 组件卸载时清理挂起的 rAF,并 flush 残留 delta(保留已接收内容) @@ -592,14 +669,18 @@ export function useAgentStream(): void { flushTraceThought(); cleanupRef.current?.(); }; - }, [flushTextDelta, flushTraceThought]); + }, [flushTextDelta, flushTraceThought, sweepInterruptedToolCalls]); // ===== 状态变化事件(迭代追踪 + Trace 步骤 + 消息卡片) ===== useEffect(() => { if (!window.metona?.agent?.onStateChange) return; - const unsubscribe = window.metona.agent.onStateChange((state: unknown) => { - const data = state as { + // v0.8.0 P1-1a: IPC → 总线,处理逻辑订阅总线(回放复用同一管线) + const unsubscribeIpc = window.metona.agent.onStateChange((state: unknown) => { + emitStreamBus('stateChange', state); + }); + const unsubscribe = subscribeStreamBus('stateChange', (raw: unknown) => { + const data = raw as { sessionId?: string; iteration?: number; state?: string; @@ -635,6 +716,21 @@ export function useAgentStream(): void { if (decision.action === 'reject') return; if (decision.adoptRunId) store.setCurrentRunId(decision.adoptRunId); + // v0.8.0 P1-1b: abort 失败自愈 —— 用户中断后若引擎仍在运行(IPC 失败或 + // waitForAbort 超时),其后续状态事件持续到达;本地 isStreaming 已被 + // abort() 置 false,此处按 runId 匹配恢复流式显示,避免"UI 显示 idle + // 但引擎静默运行、内容全部被准入层丢弃"。 + if ( + !store.isStreaming && + data.runId && + data.runId === store.currentRunId && + ['THINKING', 'PARSING', 'EXECUTING', 'OBSERVING', 'REFLECTING', 'COMPRESSING'].includes( + data.state ?? '', + ) + ) { + store.setStreaming(true); + } + // --- 迭代号更新 + 新消息卡片创建 --- if (data.iteration != null && data.iteration !== store.currentIteration) { const prevIteration = store.currentIteration; @@ -757,7 +853,10 @@ export function useAgentStream(): void { } }); - return () => unsubscribe(); + return () => { + unsubscribe(); + unsubscribeIpc(); + }; }, [flushPendingBuffersBeforeNewIteration]); // 监听 Provider 切换通知 diff --git a/src/lib/i18n-strings.ts b/src/lib/i18n-strings.ts index 629e4a6..c16550f 100644 --- a/src/lib/i18n-strings.ts +++ b/src/lib/i18n-strings.ts @@ -21,11 +21,42 @@ registerTranslations('zh-CN', { 'agent.system.stopped': '⏹ 会话已停止:{{reason}}', 'agent.error.generic': '错误: {{message}}', 'agent.error.content_filtered': '⚠️ {{message}}', + // v0.8.0 P2-4: @ 文件提及 + 'input.mention.failed': '@{{path}} 读取失败:{{reason}}', + + // ===== v0.8.0 P0-2/P0-4: 流结束契约提示 ===== + 'agent.terminated.output_truncated': + '⚠️ 回答可能因输出上限被截断(finish_reason=length)。可在 设置 → LLM 配置 中提高"最大输出上限",或降低思考强度', + 'agent.error.output_length_exceeded': + '⚠️ 模型的思考内容耗尽了输出 token 预算,未能产出回答(finish_reason=length)。建议:提高"最大输出上限" / 关闭思考模式或降低思考强度 / 更换输出预算更大的模型', + 'agent.error.empty_response': + '⚠️ 模型未返回任何内容(含重试)。可能为上游中断或输出上限截断,请重试或调整输出上限/思考设置', + 'agent.system.empty_completed': '⚠️ 模型未返回内容。请重试,或检查输出上限与思考模式设置', + 'agent.system.finalize_missing': + '⚠️ 会话已结束但未收到正常收尾事件,已恢复输入(结果以数据库内容为准)', + 'agent.system.abort_failed': '⚠️ 中断请求失败,Agent 可能仍在运行;已恢复实时显示', + 'agent.system.replay_truncated': '(提示:后台运行内容较多,仅恢复了最近部分事件)', + 'agent.system.tool_interrupted': '已中断', // ===== 通用 ===== 'common.cancel': '取消', 'common.delete': '删除', 'common.refresh': '刷新', + 'common.close': '关闭', + // v0.8.0 P2-2: 会话回放 + 'replay.title': '会话回放', + 'replay.file': '选择录制文件', + 'replay.play': '播放', + 'replay.pause': '暂停', + 'replay.stepBack': '上一步', + 'replay.stepForward': '下一步', + 'replay.speed': '播放速度', + 'replay.loadFailed': '加载录制失败', + 'replay.noRecordings': '该会话还没有 TRACE 录制文件(logs/session_*.jsonl)', + 'replay.skippedLines': '{{count}} 行损坏/截断被跳过', + 'replay.event.iterationStart': '▶ 第 {{iteration}} 轮开始', + 'replay.event.iterationEnd': '■ 第 {{iteration}} 轮结束({{duration}})', + 'replay.event.sessionEnd': '会话结束:{{reason}}({{iterations}} 轮)', // ===== 工具确认弹框(ConfirmationDialog)===== 'confirm.title': '工具执行确认', @@ -68,6 +99,12 @@ registerTranslations('zh-CN', { 'sidebar.noMatch': '无匹配结果', 'sidebar.empty': '暂无会话', 'sidebar.archived': '已归档', + // v0.8.0 P2-1: 回收站 + 'sidebar.trash': '回收站(30 天后自动清理)', + 'sidebar.trash.empty': '回收站为空', + 'sidebar.trash.purge': '彻底删除', + 'sidebar.trash.purgeTitle': '彻底删除会话?', + 'sidebar.trash.purgeBody': '「{{title}}」及其全部消息将被永久删除,此操作不可恢复。', 'sidebar.count': '{{count}} 个', 'sidebar.restore': '恢复', 'sidebar.deleteTitle': '删除会话', @@ -159,6 +196,13 @@ registerTranslations('zh-CN', { 'llm.ctxWindow.mimoHelper': '默认 1000000(1M),用于上下文压缩判断', 'llm.ctxWindow.oaHelper': 'gpt-4o 默认 128K,gpt-4.1 默认 1M', 'llm.ctxWindow.anthropicHelper': 'Claude 默认 200K', + // v0.8.0 FEAT-1: 最大输出上限(llm.maxTokens,按请求传给 API) + 'llm.maxTokens.label': '最大输出上限 (tokens)', + 'llm.maxTokens.helper': '单次回复(含思考内容)的最大 token 数;思考会占用此预算', + 'llm.maxTokens.minError': '最小值为 256', + 'llm.maxTokens.exceedCap': '当前模型输出上限为 {{cap}},保存后将按模型上限生效(配置值被钳制)', + 'llm.maxTokens.capInfo': '当前模型输出上限:{{cap}} tokens', + 'llm.thinking.unsupported': '当前模型不支持思考模式:请求将自动关闭思考(避免思考耗尽输出预算)', 'llm.pull.title': '下载 Ollama 模型', 'llm.pull.placeholder': '模型名,如 qwen3:8b', 'llm.pull.download': '下载', @@ -397,6 +441,17 @@ registerTranslations('zh-CN', { 'settings.logs.data.memories': '所有记忆', 'settings.logs.data.auditLogs': '审计日志', 'settings.logs.traceFilesTitle': '会话录制文件(TRACE JSONL)', + // v0.8.0 P2-3: 检查更新 / 下载安装 + 'settings.logs.update.title': '应用更新', + 'settings.logs.update.check': '检查更新', + 'settings.logs.update.checking': '检查中…', + 'settings.logs.update.downloadInstall': '下载并安装', + 'settings.logs.update.installing': '准备安装…', + 'settings.logs.update.available': '新版本 {{version}} 可用,点击"下载并安装"升级', + 'settings.logs.update.upToDate': '已是最新版本({{version}})', + 'settings.logs.update.disabled': '未配置更新源(app.updateFeedUrl)', + 'settings.logs.update.error': '更新失败:{{message}}', + 'settings.logs.update.downloaded': '下载完成,即将重启安装…', 'settings.logs.traceStats': '{{count}} 个文件 · 共 {{size}}(保留策略:最近 200 个,启动时自动清理)', 'settings.logs.statsLoading': '统计中...', @@ -442,6 +497,10 @@ registerTranslations('zh-CN', { 'settings.mcp.addFailed': '添加 MCP 服务失败', 'settings.mcp.addFailedMsg': '添加 MCP 服务失败:{{message}}', 'settings.mcp.removeFailed': '移除失败', + // v0.8.0 P2-5: MCP Resources / Prompts 发现 + 'settings.mcp.contents.show': '查看资源与提示词', + 'settings.mcp.contents.hide': '收起资源与提示词', + 'settings.mcp.contents.empty': '该 Server 未声明 Resources / Prompts 能力', 'settings.mcp.removeFailedMsg': '移除失败:{{message}}', 'settings.mcp.toggleFailed': '操作失败', 'settings.mcp.toggleFailedMsg': '操作失败:{{message}}', @@ -666,10 +725,44 @@ registerTranslations('en-US', { 'agent.system.stopped': '⏹ Session stopped: {{reason}}', 'agent.error.generic': 'Error: {{message}}', 'agent.error.content_filtered': '⚠️ {{message}}', + // v0.8.0 P2-4: @ file mention + 'input.mention.failed': '@{{path}} read failed: {{reason}}', + + // ===== v0.8.0 P0-2/P0-4: stream-end contract notices ===== + 'agent.terminated.output_truncated': + '⚠️ The answer may have been truncated by the output token limit (finish_reason=length). Raise "Max Output Tokens" in Settings → LLM, or lower thinking effort', + 'agent.error.output_length_exceeded': + '⚠️ The model consumed its entire output token budget on reasoning and produced no answer (finish_reason=length). Try: raise "Max Output Tokens" / disable or lower thinking / switch to a model with a larger output budget', + 'agent.error.empty_response': + '⚠️ The model returned no content (after retries). This may be an upstream drop or output-limit cutoff — retry, or adjust output limit / thinking settings', + 'agent.system.empty_completed': + '⚠️ The model returned no content. Please retry, or check the output limit / thinking settings', + 'agent.system.finalize_missing': + '⚠️ The run ended without a normal completion event — input restored (results reflect the database)', + 'agent.system.abort_failed': + '⚠️ Abort request failed — the agent may still be running; live display restored', + 'agent.system.replay_truncated': + ' (Note: many events occurred in the background; only the most recent were restored)', + 'agent.system.tool_interrupted': 'Interrupted', 'common.cancel': 'Cancel', 'common.delete': 'Delete', 'common.refresh': 'Refresh', + 'common.close': 'Close', + // v0.8.0 P2-2: Session replay + 'replay.title': 'Session replay', + 'replay.file': 'Select recording', + 'replay.play': 'Play', + 'replay.pause': 'Pause', + 'replay.stepBack': 'Step back', + 'replay.stepForward': 'Step forward', + 'replay.speed': 'Playback speed', + 'replay.loadFailed': 'Failed to load recording', + 'replay.noRecordings': 'No TRACE recordings for this session yet (logs/session_*.jsonl)', + 'replay.skippedLines': '{{count}} corrupt/truncated line(s) skipped', + 'replay.event.iterationStart': '▶ Iteration {{iteration}} start', + 'replay.event.iterationEnd': '■ Iteration {{iteration}} end ({{duration}})', + 'replay.event.sessionEnd': 'Session ended: {{reason}} ({{iterations}} iterations)', 'confirm.title': 'Tool Execution Confirmation', 'confirm.parallelSuffix': ' ({{count}} parallel requests)', @@ -712,6 +805,13 @@ registerTranslations('en-US', { 'sidebar.noMatch': 'No matches', 'sidebar.empty': 'No sessions yet', 'sidebar.archived': 'Archived', + // v0.8.0 P2-1: Trash + 'sidebar.trash': 'Trash (auto-purged after 30 days)', + 'sidebar.trash.empty': 'Trash is empty', + 'sidebar.trash.purge': 'Delete forever', + 'sidebar.trash.purgeTitle': 'Delete session forever?', + 'sidebar.trash.purgeBody': + '"{{title}}" and all of its messages will be permanently deleted. This cannot be undone.', 'sidebar.count': '{{count}}', 'sidebar.restore': 'Restore', 'sidebar.deleteTitle': 'Delete Session', @@ -806,6 +906,16 @@ registerTranslations('en-US', { 'llm.ctxWindow.mimoHelper': 'Default 1000000 (1M), used for compression decisions', 'llm.ctxWindow.oaHelper': 'gpt-4o defaults to 128K, gpt-4.1 to 1M', 'llm.ctxWindow.anthropicHelper': 'Claude defaults to 200K', + // v0.8.0 FEAT-1: max output tokens (llm.maxTokens, sent per request) + 'llm.maxTokens.label': 'Max output tokens', + 'llm.maxTokens.helper': + 'Maximum tokens per response (reasoning included); thinking consumes this budget', + 'llm.maxTokens.minError': 'Minimum is 256', + 'llm.maxTokens.exceedCap': + 'This model caps output at {{cap}} — the saved value will be clamped to the model limit', + 'llm.maxTokens.capInfo': 'Model output cap: {{cap}} tokens', + 'llm.thinking.unsupported': + 'This model does not support thinking: requests will disable thinking automatically (to avoid burning the output budget on reasoning)', 'llm.pull.title': 'Download Ollama model', 'llm.pull.placeholder': 'Model name, e.g. qwen3:8b', 'llm.pull.download': 'Download', @@ -1048,6 +1158,18 @@ registerTranslations('en-US', { 'settings.logs.data.memories': 'all memories', 'settings.logs.data.auditLogs': 'audit logs', 'settings.logs.traceFilesTitle': 'Session recording files (TRACE JSONL)', + // v0.8.0 P2-3: App update + 'settings.logs.update.title': 'App update', + 'settings.logs.update.check': 'Check for updates', + 'settings.logs.update.checking': 'Checking…', + 'settings.logs.update.downloadInstall': 'Download & install', + 'settings.logs.update.installing': 'Preparing to install…', + 'settings.logs.update.available': + 'Version {{version}} is available — click "Download & install" to upgrade', + 'settings.logs.update.upToDate': 'Up to date ({{version}})', + 'settings.logs.update.disabled': 'No update feed configured (app.updateFeedUrl)', + 'settings.logs.update.error': 'Update failed: {{message}}', + 'settings.logs.update.downloaded': 'Download complete — restarting to install…', 'settings.logs.traceStats': '{{count}} file(s) · {{size}} total (retention: last 200, auto-cleaned on startup)', 'settings.logs.statsLoading': 'Computing...', @@ -1093,6 +1215,10 @@ registerTranslations('en-US', { 'settings.mcp.addFailed': 'Failed to add MCP server', 'settings.mcp.addFailedMsg': 'Failed to add MCP server: {{message}}', 'settings.mcp.removeFailed': 'Remove failed', + // v0.8.0 P2-5: MCP discovery + 'settings.mcp.contents.show': 'Show resources & prompts', + 'settings.mcp.contents.hide': 'Hide resources & prompts', + 'settings.mcp.contents.empty': 'This server does not declare Resources / Prompts capabilities', 'settings.mcp.removeFailedMsg': 'Remove failed: {{message}}', 'settings.mcp.toggleFailed': 'Operation failed', 'settings.mcp.toggleFailedMsg': 'Operation failed: {{message}}', diff --git a/src/lib/model-capabilities.ts b/src/lib/model-capabilities.ts index e835d38..1471229 100644 --- a/src/lib/model-capabilities.ts +++ b/src/lib/model-capabilities.ts @@ -60,3 +60,34 @@ export function supportsImageUpload(input: ImageUploadGateInput): ImageUploadGat return { allowed: true }; } + +// ===== v0.8.0 FEAT-1: Provider 输出上限参数支持矩阵 ===== + +/** + * 各 Provider 是否支持"最大输出上限"请求参数(LLM 设置页「最大输出上限」 + * 配置项的可见性门控 —— 仅支持的 Provider 显示该配置)。 + * + * 数据来源(各厂商 API 文档 + adapter 实现): + * - deepseek: max_tokens(上限按模型钳制,384K / vision 8192) + * - agnes: max_tokens(上限 65.5K) + * - mimo: max_completion_tokens(上限 131072 / 32768,含推理 token) + * - openai: max_tokens(chat 系列)/ max_completion_tokens(o 系列) + * - anthropic: max_tokens(必填参数) + * - ollama: options.num_predict + * + * 六家当前均支持;此表为唯一数据驱动点,未来接入不支持的 Provider 时 + * 置 false 即可自动隐藏配置项(无需改 UI 代码)。 + */ +export const PROVIDER_OUTPUT_LIMIT_SUPPORT: Record = { + deepseek: true, + agnes: true, + mimo: true, + openai: true, + anthropic: true, + ollama: true, +}; + +/** 查询当前 Provider 是否支持输出上限配置(未知 Provider 保守返回 false → 隐藏) */ +export function supportsOutputLimitConfig(provider: string): boolean { + return PROVIDER_OUTPUT_LIMIT_SUPPORT[provider] === true; +} diff --git a/src/lib/stream-event-bus.ts b/src/lib/stream-event-bus.ts new file mode 100644 index 0000000..23aaacb --- /dev/null +++ b/src/lib/stream-event-bus.ts @@ -0,0 +1,46 @@ +/** + * Stream Event Bus — 渲染层流式事件总线(v0.8.0 P1-1a) + * + * 背景:useAgentStream 的事件消费逻辑此前只与 IPC 直连(window.metona.agent. + * onStreamEvent / onStateChange)—— 会话切换期间同 run 的后续事件被准入层 + * 拒绝(non-streaming-content / terminal-without-run),引擎跑完但 UI 永不 + * 更新("回复到一半消失")。 + * + * 现契约:IPC 监听器与「后台会话回放」统一经本总线分发到 useAgentStream 的 + * 处理函数 —— 切回运行中会话时,agent:getReplayState 返回的缓冲事件按序 + * emit 进总线,与实时事件无缝衔接。 + * + * 契约:零业务逻辑的极简 pub-sub;每通道单消费者(useAgentStream 在 App 根 + * 挂载一次),重复订阅由调用方保证先退订。 + */ + +export type StreamEventChannel = 'streamEvent' | 'stateChange'; + +type Listener = (payload: unknown) => void; + +const listeners = new Map>(); + +export function subscribeStreamBus(channel: StreamEventChannel, fn: Listener): () => void { + let set = listeners.get(channel); + if (!set) { + set = new Set(); + listeners.set(channel, set); + } + set.add(fn); + return () => { + set?.delete(fn); + }; +} + +export function emitStreamBus(channel: StreamEventChannel, payload: unknown): void { + const set = listeners.get(channel); + if (!set) return; + for (const fn of set) { + try { + fn(payload); + } catch (err) { + // 单个消费者异常不阻断其他消费者(当前单消费者,防御性) + console.error('[stream-event-bus] listener error:', err); + } + } +} diff --git a/src/stores/agent-store.ts b/src/stores/agent-store.ts index b6d9ddf..895c620 100644 --- a/src/stores/agent-store.ts +++ b/src/stores/agent-store.ts @@ -11,6 +11,11 @@ import { useSessionStore } from './session-store'; import { toDisplayResult } from '../lib/tool-result-display'; // v0.7.3 P3-3: Trace 生命周期 —— metadata 只保留最近 N 个 run 的步骤 import { keepRecentRuns } from '../lib/trace-lifecycle'; +// v0.8.0 P1-1a: 后台会话回放经事件总线灌入 useAgentStream 管线 +import { emitStreamBus } from '../lib/stream-event-bus'; +// v0.8.0 P1-1b/c: 文案出层 +import { t } from '../lib/i18n'; +import '../lib/i18n-strings'; // L-1: 消息 ID 防碰撞计数器 let _msgIdCounter = 0; @@ -380,6 +385,46 @@ export const useAgentStore = create((set, get) => ({ console.error('[AgentStore]', err); }); } + + // v0.8.0 P1-1a: 后台会话运行内容恢复 —— 该会话正在后台运行(或刚结束但 + // 缓冲未清)时,拉取主进程回放缓冲并按序灌入事件总线,重建消息/Trace 状态; + // 实时事件随后经同一总线无缝衔接(runId 对齐由缓冲记录提供)。 + if (id && window.metona?.agent?.getReplayState) { + window.metona.agent + .getReplayState(id) + .then((r) => { + // 竞态保护:用户已切走则丢弃 + if (get().currentSessionId !== id) return; + if (!r?.success || !r.data || r.data.events.length === 0) return; + const { events, truncated, runId, isRunning } = r.data; + // 恢复流式态与 run 归属:回放的内容事件需要 isStreaming=true 才能通过 + // 准入层(non-streaming-content 过滤);终止事件(done/error)会正常 + // 收尾并把 isStreaming 复位为 false。 + if (runId) { + set({ currentRunId: runId, isStreaming: true }); + } + for (const ev of events) { + emitStreamBus(ev.channel, ev.payload); + } + if (truncated) { + get().addMessage({ + id: genMsgId('system'), + role: 'system', + content: t('agent.system.replay_truncated'), + timestamp: Date.now(), + }); + } + // 已运行中的会话:回放后实时事件继续;stateChange THINKING 自愈 + //(P1-1b)会在流式态意外复位时恢复。此处对 isRunning 会话兜底确保 + // isStreaming 不为 false(done 未到达时 bus 回放不会复位它)。 + if (isRunning && runId && !get().isStreaming) { + set({ isStreaming: true }); + } + }) + .catch((err) => { + console.error('[AgentStore] getReplayState failed:', err); + }); + } }, setMessages: (messages) => set({ messages }), @@ -544,20 +589,54 @@ export const useAgentStore = create((set, get) => ({ content: llmContent, images: llmImages.length > 0 ? llmImages : undefined, }; - window.metona.agent.sendMessage(messageWithImages, sessionId).catch((err) => { - console.error('[AgentStore]', 'IPC sendMessage failed:', err); - set({ agentStatus: 'error', isStreaming: false }); - get().addMessage({ - id: genMsgId('error'), - role: 'system', - content: `错误: 消息发送失败 — ${(err as Error).message}`, - timestamp: Date.now(), - }); - // 额外弹 toast 作为通知,确保用户感知(system message 仅在聊天流内可见) - import('@metona-team/metona-toast') - .then((mod) => mod.default.error(`消息发送失败:${(err as Error).message}`)) - .catch(() => {}); - }); + window.metona.agent.sendMessage(messageWithImages, sessionId).then( + (r) => { + // v0.8.0 P1-1c: 收尾兜底 —— agent:sendMessage 的 invoke 在整个 run 结束后 + // 才 resolve。resolve 时若本地仍处于流式态,说明 DONE/ERROR 收尾事件 + // 未到达(事件投递异常/主进程异常收尾),渲染层将永久卡在流式状态。 + const st = get(); + if (st.currentSessionId !== sessionId || !st.isStreaming) return; + if (r && r.success === false) { + // 主进程显式失败(注入拦截/防重入/adapter 加载失败等)——通常已广播 + // ERROR 事件;若事件未达,本地以返回的 error 消息兜底收尾 + set({ agentStatus: 'error', isStreaming: false, currentRunId: null }); + get().addMessage({ + id: genMsgId('error'), + role: 'system', + content: t('agent.error.generic', { message: r.error ?? '消息发送失败' }), + timestamp: Date.now(), + }); + return; + } + // 成功 resolve:给事件投递留 2s 宽限,仍卡流式则强制收尾 + setTimeout(() => { + const s2 = get(); + if (s2.currentSessionId !== sessionId || !s2.isStreaming) return; + set({ agentStatus: 'idle', isStreaming: false, currentRunId: null }); + s2.addMessage({ + id: genMsgId('system'), + role: 'system', + content: t('agent.system.finalize_missing'), + timestamp: Date.now(), + }); + s2.saveTraceData(); + }, 2_000); + }, + (err) => { + console.error('[AgentStore]', 'IPC sendMessage failed:', err); + set({ agentStatus: 'error', isStreaming: false }); + get().addMessage({ + id: genMsgId('error'), + role: 'system', + content: `错误: 消息发送失败 — ${(err as Error).message}`, + timestamp: Date.now(), + }); + // 额外弹 toast 作为通知,确保用户感知(system message 仅在聊天流内可见) + import('@metona-team/metona-toast') + .then((mod) => mod.default.error(`消息发送失败:${(err as Error).message}`)) + .catch(() => {}); + }, + ); } }, @@ -810,11 +889,23 @@ export const useAgentStore = create((set, get) => ({ // 不清除 currentRunId — 保留用于过滤旧 run 的延迟终止事件(DONE/TERMINATED) // 旧 run 的 DONE 到达时会匹配 currentRunId,处理后清除;新 run 的 INIT 会覆盖 set({ agentStatus: 'idle', isStreaming: false }); - // v0.7.2 P3-11: 立即熄灭运行指示点;若中断未生效,后续 THINKING 状态事件会重新点亮(自愈) + // v0.7.2 P3-11: 立即熄灭运行指示点。 + // v0.8.0 P1-1b 修正注释与实现:此前注释称"中断未生效时 THINKING 事件会重新 + // 点亮(自愈)",但 isStreaming=false 下内容事件仍被准入层丢弃。现两层自愈: + // ① 本处 IPC 失败 → 回滚本地状态并恢复运行指示(下方 catch); + // ② IPC 成功但引擎未真正停止 → useAgentStream 的 stateChange THINKING + // 自愈(runId 匹配时恢复 isStreaming=true)。 if (sessionId) get().updateSessionRunState(sessionId, null); if (sessionId && window.metona?.agent?.abortSession) { window.metona.agent.abortSession(sessionId).catch((err) => { console.error('[AgentStore]', err); + // v0.8.0 P1-1b: 中断请求失败 —— 引擎很可能仍在运行,回滚本地状态 + // 并恢复运行指示与实时显示(内容事件不再被准入层丢弃) + set({ agentStatus: 'thinking', isStreaming: true }); + if (sessionId) get().updateSessionRunState(sessionId, 'thinking'); + import('@metona-team/metona-toast') + .then((mod) => mod.default.warning(t('agent.system.abort_failed'))) + .catch(() => {}); }); } }, diff --git a/src/types/global.d.ts b/src/types/global.d.ts index 478eb1a..a385f57 100644 --- a/src/types/global.d.ts +++ b/src/types/global.d.ts @@ -53,6 +53,10 @@ interface MetonaStreamEventData { retryable?: boolean; }; state?: string; + /** v0.8.0 P0-1: DONE 事件携带的 Provider 原生停止原因(stop/length/tool_calls/content_filter) */ + finishReason?: string; + /** v0.8.0 P1-1a: DONE 事件终止原因(useAgentStream 消费;与 terminationReason 字段对齐) */ + terminationReason?: string; } /** v0.5.0: SubAgent 生命周期事件(AgentMonitor 的 SubAgent 状态区展示) */ @@ -69,7 +73,10 @@ interface MetonaSubAgentEvent { interface MetonaAgentAPI { /** 发送 MetonaMessage(IR 类型)+ sessionId 到主进程 */ - sendMessage: (message: MetonaMessageInput, sessionId: string) => Promise<{ success: boolean }>; + sendMessage: ( + message: MetonaMessageInput, + sessionId: string, + ) => Promise<{ success: boolean; error?: string }>; onStreamEvent: (callback: (event: MetonaStreamEventData) => void) => () => void; onStateChange: ( callback: (state: { @@ -81,6 +88,20 @@ interface MetonaAgentAPI { }) => void, ) => () => void; abortSession: (sessionId: string) => Promise<{ success: boolean }>; + /** + * v0.8.0 P1-1a: 拉取后台会话的流式回放缓冲(切回运行中会话时恢复内容)。 + * events 为按序保存的 { channel: 'streamEvent'|'stateChange', payload } 条目。 + */ + getReplayState: (sessionId: string) => Promise<{ + success: boolean; + error?: string; + data?: { + isRunning: boolean; + runId: string | null; + truncated: boolean; + events: Array<{ channel: 'streamEvent' | 'stateChange'; payload: unknown; ts: number }>; + }; + }>; onProviderSwitched: ( callback: (data: { from?: string; to?: string; reason?: string }) => void, ) => () => void; @@ -142,6 +163,32 @@ interface MetonaSessionsAPI { messageId: string, inclusive?: boolean, ) => Promise<{ success: boolean; truncated?: number; error?: string }>; + // ===== v0.8.0 P2-1: 回收站(软删除 30 天保留) ===== + listTrash: () => Promise< + Array<{ + id: string; + title: string; + createdAt: number; + updatedAt: number; + messageCount: number; + totalTokens: number; + pinned: boolean; + archived: boolean; + }> + >; + restore: (sessionId: string) => Promise<{ success: boolean; error?: string }>; + purge: (sessionId: string) => Promise<{ success: boolean; error?: string }>; + // ===== v0.8.0 P2-2: 会话回放 ===== + listRecordings: (sessionId: string) => Promise<{ + success: boolean; + error?: string; + data?: Array<{ name: string; size: number; mtime: number }>; + }>; + readRecording: (name: string) => Promise<{ + success: boolean; + error?: string; + data?: { name: string; events: Array>; skipped: number }; + }>; /** v0.5.0: 全文搜索会话内容(FTS5,跨会话,按会话聚合) */ searchContent: (query: string) => Promise<{ success: boolean; @@ -192,6 +239,19 @@ interface MetonaMCPAPI { addServer: (config: MetonaMCPServerConfig) => Promise<{ success: boolean; error?: string }>; removeServer: (name: string) => Promise<{ success: boolean; error?: string }>; toggleServer: (name: string, enabled: boolean) => Promise<{ success: boolean; error?: string }>; + // v0.8.0 P2-5: Resources / Prompts 发现 + listServerContents: (name: string) => Promise<{ + success: boolean; + error?: string; + data?: { + resources: Array<{ uri: string; name: string; description?: string; mimeType?: string }>; + prompts: Array<{ + name: string; + description?: string; + arguments?: Array<{ name: string; description?: string; required?: boolean }>; + }>; + }; + }>; } // ===== Memory API ===== @@ -252,6 +312,16 @@ interface MetonaAppAPI { | { status: 'up-to-date'; latestVersion: string } | { status: 'available'; latestVersion: string; downloadUrl?: string; notes?: string } >; + // ===== v0.8.0 P2-3: electron-updater 自动更新 ===== + updateInstall: () => Promise<{ success: boolean; error?: string }>; + onUpdateStatus: ( + callback: (event: { + status: 'checking' | 'available' | 'not-available' | 'downloading' | 'downloaded' | 'error'; + latestVersion?: string; + percent?: number; + message?: string; + }) => void, + ) => () => void; /** v0.7.3 P3-4: 健康快照(SLO 指标 + 最近健康检查报告) */ getHealthSnapshot: () => Promise<{ success: boolean; @@ -355,6 +425,19 @@ interface MetonaWorkspaceAPI { detail?: string; error?: string; }>; + // v0.8.0 P2-4: @ 文件提及 + listFiles: (limit?: number) => Promise<{ success: boolean; error?: string; data?: string[] }>; + readFileClip: ( + relPath: string, + maxBytes?: number, + ) => Promise<{ + success: boolean; + error?: string; + name?: string; + content?: string; + truncated?: boolean; + size?: number; + }>; } // ===== LLM 运行时查询 API =====