feat: v0.7.3 成本收口 · 状态一致 · 死账清理 — Prompt Cache 根治 + SSRF DNS Pinning + 87 用例扩充全量回归
P1 修复面收口: Prompt Cache 根治(日期/记忆/附件三类易变内容出 system 入用户消息 前置块 user-context.ts, system 跨 run 字节级稳定; Anthropic system 块数组化 + cache_control ephemeral 断言, DeepSeek 自动缓存前缀命中 — 多轮对话输入 token 成本降数量级); 编辑重发/重新生成幽灵 Trace 双侧根治(DB truncateMessagesAfter 同步过滤 metadata.traceSteps + 前端 trimTraceStepsByAnchor 镜像, 严格小于锚点 时间戳, 同毫秒等值判废); sessions:deleteMessage 死通道全链路删除(渲染层零调用 + message_count 漂移面); Ollama vision 能力门控全链路(MetonaModelInfo .supportsVision 贯穿 adapter/IPC/store/UI, model-capabilities.ts 三道判定纯函数, 未知保守放行); 记忆固化节流(consolidation-policy 纯函数: 总开关 + 内容门控 [回答>=200字符或存在成功工具调用] + 会话级 10 分钟频率窗口, 三 memory.* 配置键) P2 安全纵深: SSRF DNS Pinning 关闭 rebinding 窗口(ssrf-guard 重构 resolvePublicAddresses 单源; ssrf-dispatcher 以 undici Agent.connect.lookup 钉死校验 IP, TLS SNI 保持原域名, 一次性 dispatcher 用后即毁; 代理激活显式 退化为仅入口校验); web_fetch 重写手动逐跳重定向循环(每跳先校验后连接, 替代 redirect:follow 内核跟跳的中间跳裸奔, 上限 5 跳); http_request 换用 pinned fetch; web_search 可达性预检加固(私有 URL 零请求 + 不跟跳, 3xx 视为 可达); Agent 浏览器 CORS 通配收紧为 Origin 回显 + Vary: Origin; ConfirmationHook.forgetSession 会话终态清理(会话删除/abort 联动/SubAgent 终结三处接线, 根治 rememberedDecisions 泄漏) P3 架构还债: agent.enableReflection 死配置全链路接线(main→shared→引擎→ Orchestrator→设置开关, REFLECTING 状态真实可达); AgentLoopConfig.timeoutMs 死字段删除; MemoryManager.cleanupExpired 挂入健康检查周期(expires_at 回收 管道真实化); buildSafeEnv 收敛 utils/safe-env.ts 单源(run_command 与 MCP stdio 共用, 终结双实现漂移); Trace 生命周期治理(metadata 只保留最近 20 个 run — keepRecentRuns 纯函数; JSONL 录制启动自动清理保留 200 个 + 设置页 手动清理); SLO/健康快照可视化(app:healthSnapshot IPC + 设置页只读卡片 + 审计链一键校验) P4 能力演进: 会话标题 LLM 自动生成(TitleGenerator — 每会话幂等/并发重入复用 同一 Promise/自定义标题不覆盖/失败静默回退, Sidebar 经 config:changed 实时 刷新); MCP 自动重连(5s/15s/60s 退避最多 3 次, reconnecting 状态机, teardownConnection 内部拆除保留簿记 — 用户断开/开关关闭即时取消, 设置页 显示第 N/3 次); 死循环检测 ABAB 乒乓模式(最近4轮 A→B→A→B 交替判定, 补齐 docs 第五章"两状态反复切换"检测契约); i18n 第三阶段(ChatInput/LLMSettings/ OnboardingWizard/MemoryViewer 主链路文案出层, zh-CN + en-US 双字典补齐) 测试: 737 → 824 用例(+87, 新增 8 个测试文件 + 扩展 3 个)。新覆盖: user-context 分组/空值收缩/拼接契约、context-builder 字节级稳定性、Anthropic cache_control 四态、consolidation-policy 九路判定矩阵、ssrf-dispatcher(pinned lookup/重定向 解析/IP 校验)、forget-session 会话隔离、trace-lifecycle run 淘汰、 trace-trim 严格小于边界、safe-env 净化矩阵、mcp-reconnect 退避状态机 (fake timers)、title-generator 并发重入、SQLite 侧 truncate×TRACE 联动 (Electron ABI)。测试驱动修复: GIT_*/ 注释终止块注释、重连计数被自身重试 前置断开重置(拆 teardownConnection 保留簿记)、TitleGenerator 幂等占位与 并发去重的检查顺序竞态(去重先于幂等) 版本: 0.7.3; README 同步(配置表新增 agent.enableReflection/memory.*/mcp.autoReconnect) 回归: typecheck 双端 0 错误; ESLint 0/0; 系统 Node 771 通过 53 跳过 (better-sqlite3 ABI); Electron ABI 全量 824/824 零跳过
This commit is contained in:
@@ -0,0 +1,157 @@
|
||||
/**
|
||||
* TitleGenerator 测试（v0.7.3 P4-1）
|
||||
*
|
||||
* T1 sanitizeTitle 清洗规则矩阵（围栏/引号/自述前缀/空白/截断/空值）；
|
||||
* T2 maybeGenerateTitle 幂等（每会话仅一次）；
|
||||
* T3 已有自定义标题的会话不覆盖；
|
||||
* T4 LLM 失败/超时静默回退（不抛错，返回 null）；
|
||||
* T5 空输入门控。
|
||||
*/
|
||||
|
||||
import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest';
|
||||
|
||||
vi.mock('electron-log', () => ({
|
||||
default: { info: vi.fn(), warn: vi.fn(), error: vi.fn(), debug: vi.fn() },
|
||||
}));
|
||||
|
||||
import { TitleGenerator, sanitizeTitle } from '../title-generator.service';
|
||||
import type { IMetonaProviderAdapter, MetonaResponse } from '../../harness/types';
|
||||
|
||||
describe('sanitizeTitle', () => {
|
||||
it('T1a: 干净文本原样通过', () => {
|
||||
expect(sanitizeTitle('修复登录超时')).toBe('修复登录超时');
|
||||
});
|
||||
|
||||
it('T1b: markdown 围栏与列表标记剥离', () => {
|
||||
expect(sanitizeTitle('```json\n"标题"\n```')).toBe('标题');
|
||||
expect(sanitizeTitle('- 修复登录超时')).toBe('修复登录超时');
|
||||
expect(sanitizeTitle('# 修复登录超时')).toBe('修复登录超时');
|
||||
});
|
||||
|
||||
it('T1c: 成对包裹引号剥离', () => {
|
||||
expect(sanitizeTitle('"修复登录超时"')).toBe('修复登录超时');
|
||||
expect(sanitizeTitle('“修复登录超时”')).toBe('修复登录超时');
|
||||
});
|
||||
|
||||
it('T1d: 自述前缀剥离（标题:/Title:）', () => {
|
||||
expect(sanitizeTitle('标题: 修复登录超时')).toBe('修复登录超时');
|
||||
expect(sanitizeTitle('Title: Fix login timeout')).toBe('Fix login timeout');
|
||||
});
|
||||
|
||||
it('T1e: 换行折叠为空格、首尾空白去除', () => {
|
||||
expect(sanitizeTitle('修复\n登录 超时\n')).toBe('修复 登录 超时');
|
||||
});
|
||||
|
||||
it('T1f: 超长截断到 maxLen', () => {
|
||||
const out = sanitizeTitle('a'.repeat(100), 40);
|
||||
expect(out?.length).toBe(40);
|
||||
});
|
||||
|
||||
it('T1g: 空值/纯符号 → null', () => {
|
||||
expect(sanitizeTitle('')).toBeNull();
|
||||
expect(sanitizeTitle(' ')).toBeNull();
|
||||
expect(sanitizeTitle('"""')).toBeNull();
|
||||
expect(sanitizeTitle('###')).toBeNull();
|
||||
});
|
||||
});
|
||||
|
||||
function makeAdapter(content: string | Error): IMetonaProviderAdapter {
|
||||
return {
|
||||
providerId: 'mock',
|
||||
supportedModels: ['m'],
|
||||
supportsToolCalling: true,
|
||||
supportsThinking: false,
|
||||
getContextWindow: () => 128_000,
|
||||
send: vi.fn(async (): Promise<MetonaResponse> => {
|
||||
if (content instanceof Error) throw content;
|
||||
return {
|
||||
meta: { requestId: 'r', provider: 'mock', model: 'm', latencyMs: 1, timestamp: Date.now() },
|
||||
content,
|
||||
usage: { inputTokens: 1, outputTokens: 1, totalTokens: 2 },
|
||||
finishReason: 'stop' as never,
|
||||
};
|
||||
}),
|
||||
sendStream: vi.fn(),
|
||||
setAbortSignal: vi.fn(),
|
||||
healthCheck: async () => true,
|
||||
} as unknown as IMetonaProviderAdapter;
|
||||
}
|
||||
|
||||
function makeSessionService(existing: Array<{ id: string; title: string }> = []) {
|
||||
// 注意：真实 SessionService.list() 是同步方法（better-sqlite3），
|
||||
// mock 必须保持同步返回，否则 generate 内的 .find() 拿到 Promise
|
||||
return {
|
||||
list: vi.fn(() => existing) as unknown as () => Array<{ id: string; title: string }>,
|
||||
rename: vi.fn(() => true),
|
||||
};
|
||||
}
|
||||
|
||||
describe('TitleGenerator.maybeGenerateTitle', () => {
|
||||
beforeEach(() => {
|
||||
vi.useFakeTimers();
|
||||
});
|
||||
afterEach(() => {
|
||||
vi.useRealTimers();
|
||||
});
|
||||
|
||||
it('T2: 首个 run 生成标题并调用 rename；同会话幂等', async () => {
|
||||
const svc = makeSessionService([{ id: 's1', title: '新会话' }]);
|
||||
const gen = new TitleGenerator(() => makeAdapter('修复登录超时'), svc as never);
|
||||
const title = await gen.maybeGenerateTitle(
|
||||
's1',
|
||||
'帮我看看为什么登录会超时',
|
||||
'已定位为 token 过期…',
|
||||
);
|
||||
expect(title).toBe('修复登录超时');
|
||||
expect(svc.rename).toHaveBeenCalledWith('s1', '修复登录超时');
|
||||
|
||||
// 第二次调用（同会话）幂等 —— 不再调用 LLM
|
||||
const second = await gen.maybeGenerateTitle('s1', '再问一次', '再答一次');
|
||||
expect(second).toBeNull();
|
||||
});
|
||||
|
||||
it('T3: 已有自定义标题的会话不覆盖', async () => {
|
||||
const svc = makeSessionService([{ id: 's1', title: '用户手动命名' }]);
|
||||
const adapterSend = vi.fn();
|
||||
const adapter = makeAdapter('不应被使用');
|
||||
(adapter.send as ReturnType<typeof vi.fn>).mockImplementation(async (...args: unknown[]) => {
|
||||
adapterSend(...args);
|
||||
throw new Error('should not be called');
|
||||
});
|
||||
const gen = new TitleGenerator(() => adapter, svc as never);
|
||||
const title = await gen.maybeGenerateTitle('s1', '内容', '回答');
|
||||
expect(title).toBeNull();
|
||||
expect(adapterSend).not.toHaveBeenCalled();
|
||||
});
|
||||
|
||||
it('T4: LLM 失败静默回退（返回 null，不抛错），且同会话不重试', async () => {
|
||||
const svc = makeSessionService([{ id: 's1', title: '新会话' }]);
|
||||
const gen = new TitleGenerator(() => makeAdapter(new Error('network down')), svc as never);
|
||||
await expect(gen.maybeGenerateTitle('s1', 'q', 'a')).resolves.toBeNull();
|
||||
await expect(gen.maybeGenerateTitle('s1', 'q2', 'a2')).resolves.toBeNull();
|
||||
expect(svc.rename).not.toHaveBeenCalled();
|
||||
});
|
||||
|
||||
it('T5: 空 sessionId / 双向空内容 → 不生成', async () => {
|
||||
const svc = makeSessionService();
|
||||
const adapter = makeAdapter('x');
|
||||
const gen = new TitleGenerator(() => adapter, svc as never);
|
||||
expect(await gen.maybeGenerateTitle('', 'q', 'a')).toBeNull();
|
||||
expect(await gen.maybeGenerateTitle('s1', ' ', ' ')).toBeNull();
|
||||
});
|
||||
|
||||
it('T6: 并发重入复用同一 Promise（不重复调用 LLM）', async () => {
|
||||
const svc = makeSessionService([{ id: 's1', title: '新会话' }]);
|
||||
const gen = new TitleGenerator(() => makeAdapter('并发标题'), svc as never);
|
||||
// 使用真实定时器场景下并发触发
|
||||
vi.useRealTimers();
|
||||
const [a, b] = await Promise.all([
|
||||
gen.maybeGenerateTitle('s2', 'q', 'a'),
|
||||
gen.maybeGenerateTitle('s2', 'q', 'a'),
|
||||
]);
|
||||
expect(a).toBe('并发标题');
|
||||
expect(b).toBe('并发标题');
|
||||
// rename 仅一次（并发重入复用）
|
||||
expect(svc.rename).toHaveBeenCalledTimes(1);
|
||||
});
|
||||
});
|
||||
Reference in New Issue
Block a user