thzxx
9b45c445bf
硬性契约：删除代码中一切写死的上下文窗口与最大输出上限（含六家模型元信息
钳制与全部兜底值）——唯一合法来源是设置面板「上下文长度」（llm.contextWindow）
与「最大输出上限」（llm.maxTokens），跨 Provider/模型原样透传。
P0 正确性收口：
- 迁移 11/12（SCHEMA_VERSION 5）：记忆表 embedding 列 + 分 Provider 窗口键清理
- 记忆生命周期接线：会话终态清理 working memory / episodic 90 天 TTL / access_count 回写
- 回放缓冲模块化 + 会话终态清理（杜绝 4MB/会话内存滞留）
- i18n 收口：主进程 main-locale（zh/en，ui.locale 热切换）+ 渲染层 17 处出层
P1 能力演进：
- 本地向量混合检索：0.6×向量余弦 + 0.4×TF-IDF，Ollama embeddings 首次投产，
存量记忆惰性回填，嵌入不可用自动回退 TF-IDF
- MEMORY.md 维护闭环：固化去重消除截断盲区；两阶段维护（AI 建议 → 用户确认 →
原子改写 + 语义记忆双轨同步 + 审计）；>50KB 告警
- 可观测闭环：cacheTokens 引擎→前端透传（Token 面板命中率/成本行）+ 输入框
上下文占用指示条
- MCP Prompts/Resources 对话可用：/mcp:{server}:{prompt} 与 @mcp:{server}:{uri}
P2 体验补全：
- 工具自定义策略（正则白/黑名单 + 频率 + 强制确认，热生效）
- 连续 ≥3 同类工具确认聚合为单弹框
- 会话消息游标分页（首屏 200 条向上翻页）
- 开机自启；Playwright + Electron E2E 冒烟（本地 mock LLM 零外联）
Review 回归修复：MCP 大小写失配 / 分页状态复位 / 清空=未配置语义（Number(null)=0
隐患）/ MEMORY.md 告警位置 / working_memories FK（迁移 13）/ 全局配置层废键清理；
附带根治权限加固启动时序、代理回环放行、safeStorage 降级、悬空 symlink 逃逸。
验证：typecheck/lint 0 问题；test:electron 2478/2478（0 跳过）；E2E 2/2；
docs/v0.8.1-迭代实施清单.md 全项留档。
100 lines
3.5 KiB
TypeScript
100 lines
3.5 KiB
TypeScript
/**
|
||
* Post-Tool Hooks — 工具执行后钩子
|
||
*
|
||
* 用途：结果验证、审计日志写入、记忆触发、错误恢复建议。
|
||
*
|
||
* @see docs/生产级通用 AI Agent 智能体桌面应用：完整设计与构建指南.html — 第五章
|
||
*/
|
||
|
||
import type { MetonaToolCall, MetonaToolResult } from '../types';
|
||
import type { AuditService } from '../../services/audit.service';
|
||
import type { MemoryManager } from '../memory/manager';
|
||
import log from 'electron-log';
|
||
|
||
export interface PostToolHook {
|
||
/**
|
||
* 工具执行后钩子
|
||
*
|
||
* P0-2: 返回修改后的 MetonaToolResult 可替换原始结果（如 SecurityScanHook 对
|
||
* 网页内容脱敏）；返回 void / undefined 表示保持原结果不变。
|
||
*/
|
||
afterExecute(
|
||
toolCall: MetonaToolCall,
|
||
result: MetonaToolResult,
|
||
sessionId: string,
|
||
): Promise<MetonaToolResult | void>;
|
||
}
|
||
|
||
/** 审计日志钩子 */
|
||
export class AuditLogHook implements PostToolHook {
|
||
constructor(private auditService: AuditService) {}
|
||
|
||
async afterExecute(
|
||
toolCall: MetonaToolCall,
|
||
result: MetonaToolResult,
|
||
sessionId: string,
|
||
): Promise<void> {
|
||
// #17 修复: AuditHook 应为 "fire and forget"，hook 失败不应影响工具执行链
|
||
// 虽然 AuditService.log() 内部已 try-catch，但 hook 层再加一层防御，
|
||
// 确保任何意外异常（如 getDB 抛错、JSON.stringify 失败）都不会冒泡到 ToolRegistry
|
||
try {
|
||
this.auditService.logToolCall({
|
||
sessionId,
|
||
iteration: toolCall.iteration,
|
||
toolName: toolCall.name,
|
||
args: toolCall.args,
|
||
outcome: result.success ? 'success' : 'error',
|
||
result: result.result,
|
||
error: result.error,
|
||
durationMs: result.durationMs,
|
||
});
|
||
} catch (err) {
|
||
log.error('[AuditLogHook] Failed to log audit:', err);
|
||
// 不抛出，让工具执行链继续
|
||
}
|
||
}
|
||
}
|
||
|
||
/** 记忆触发钩子 */
|
||
export class MemoryTriggerHook implements PostToolHook {
|
||
/** 仅对搜索类工具触发记忆，避免每次 read_file 都产生低价值记忆条目 */
|
||
private memorableTools = ['web_search', 'memory_search'];
|
||
|
||
/** 单次工具结果存储上限（字符），防止过大内容淹没记忆系统 */
|
||
private readonly MAX_MEMORY_CONTENT = 500;
|
||
|
||
/**
|
||
* v0.8.1 P0-2: 工具结果类情节记忆的 TTL（90 天）。
|
||
* 此前 episodic_memories.expires_at 全链路无写入方 —— cleanupExpired（健康检查
|
||
* 周期调用）空转，情节记忆只增不减。现为此类低价值记忆写入过期时间，90 天后
|
||
* 由周期清理回收；用户偏好等高价值记忆（consolidator 写入 semantic 表）不受影响。
|
||
*/
|
||
private readonly EPISODIC_TTL_MS = 90 * 24 * 60 * 60 * 1000;
|
||
|
||
constructor(private memoryManager: MemoryManager) {}
|
||
|
||
async afterExecute(
|
||
toolCall: MetonaToolCall,
|
||
result: MetonaToolResult,
|
||
sessionId: string,
|
||
): Promise<void> {
|
||
if (this.memorableTools.includes(toolCall.name) && result.success) {
|
||
const content =
|
||
typeof result.result === 'string' ? result.result : JSON.stringify(result.result);
|
||
try {
|
||
this.memoryManager.store({
|
||
type: 'episodic',
|
||
content: `Tool ${toolCall.name} returned: ${content.slice(0, this.MAX_MEMORY_CONTENT)}`,
|
||
source: 'tool_result',
|
||
sessionId,
|
||
importance: 0.6,
|
||
expiresAt: Date.now() + this.EPISODIC_TTL_MS,
|
||
});
|
||
} catch (error) {
|
||
// 记忆存储失败不应影响工具执行结果
|
||
log.error('[MemoryTriggerHook] Failed to store episodic memory:', error);
|
||
}
|
||
}
|
||
}
|
||
}
|