重构: 巨函数拆分 + 清理历史标记注释 + custom_prompt/停止按钮/tunnel 改进

2026-07-31 21:36:12 +08:00
parent 365af554da
commit bd9031d35d
73 changed files with 1421 additions and 1064 deletions
@@ -20,7 +20,7 @@ fn cfg(max_tokens: u32) -> ContextConfig {

 #[test]
 fn estimate_message_counts_parts_tokens() {
-    // F-260614-05 多模态回归：含图消息的大段 base64 必须计入 token 预算，
+    // 多模态回归：含图消息的大段 base64 必须计入 token 预算，
    // 否则 history_tokens 严重低估 → build_for_request 不裁剪 → provider 超限。
    let est = TokenEstimator::default();

@@ -263,7 +263,7 @@ fn system_over_budget_trims_to_protect_zone() {
    );
 }

-// ── F-15 阶段1 辅助方法单测 ──
+// ── 辅助方法单测 ──

 #[test]
 fn compress_old_messages_marks_compressed_and_returns_refs() {
@@ -430,7 +430,7 @@ fn build_eviction_units_keeps_triplet_atomic_public() {
    );
 }

-// ── F-260619-04 P1 消息级溯源:last_assistant/last_user message_id ──
+// ── 消息级溯源:last_assistant/last_user message_id ──

 #[test]
 fn last_assistant_message_id_returns_latest() {
@@ -165,7 +165,7 @@ impl ContextManager {
        // 未超预算 → 直接返回全量（仍做畸形配对自愈，防历史中毒触发 provider 500 死循环）
        if self.history_tokens <= available {
            let sanitized = Self::sanitize_messages(self.all_messages_clone());
-            // 阶段2 出口断言:占位配对完整性,失败降级 TOOL_MISSING_PREFIX 自愈(防 400 orphan)
+            // 出口断言:占位配对完整性,失败降级 TOOL_MISSING_PREFIX 自愈(防 400 orphan)
            return (
                Self::assert_placeholder_pairing(sanitized, PLACEHOLDER_INTEGRITY_ENABLED),
                false,
@@ -191,7 +191,7 @@ impl ContextManager {
                "history (~{} tokens) 超预算 ({}) 但无可淘汰单元(全在保护区 {} 条),发送兜底可能触发 provider 超限",
                self.history_tokens, available, PROTECT_COUNT
            );
-            // B-260626-01: 兜底全量也过 sanitize(对齐分支 1/3),防绕过序列修复直送 provider
+            // 兜底全量也过 sanitize(对齐分支 1/3),防绕过序列修复直送 provider
            // 触发"首条 assistant 非法"/orphan/连续 role。原裸返 all_messages_clone 不过滤
            // truncated/中毒三元组/首条非法——是主 loop 唯一的 sanitize 漏洞(大体量 tool_result
            // 致超预算且保护区满时命中)。异常会话(开头连续 assistant/tool 无 user)经
@@ -214,7 +214,7 @@ impl ContextManager {
            trim_end, removed
        );
        let sanitized = Self::sanitize_messages(msgs);
-        // 阶段2 出口断言:占位配对完整性,失败降级 TOOL_MISSING_PREFIX 自愈(防 400 orphan)
+        // 出口断言:占位配对完整性,失败降级 TOOL_MISSING_PREFIX 自愈(防 400 orphan)
        (
            Self::assert_placeholder_pairing(sanitized, PLACEHOLDER_INTEGRITY_ENABLED),
            true,
@@ -350,7 +350,7 @@ impl ContextManager {
        self.messages.iter().map(|t| &t.message)
    }

-    /// F-260619-04 P1 消息级溯源:取末条指定 role 消息的 id(ULID)。
+    /// 消息级溯源:取末条指定 role 消息的 id(ULID)。
    ///
    /// 从尾部反向扫描(末条消息命中即停,避免全量 O(n) 正扫累积),返回最近一条
    /// `role` 匹配且 `id` 非空消息的 id。无匹配或老消息无 id → None(向前兼容:
@@ -417,7 +417,7 @@ impl ContextManager {
        &self.config
    }

-    /// 可变消息切片（供阶段2 标记 status="compressed"/"archived_segment" + 调整 token）
+    /// 可变消息切片（供标记 status="compressed"/"archived_segment" + 调整 token）
    ///
    /// 调用方约定:仅改 `message.status` / `message.content`,不增删条目(增删走
    /// [`push`] / [`insert_at`]),否则 `history_tokens` 会与实际脱钩。
@@ -427,7 +427,7 @@ impl ContextManager {

    /// 在给定位置插入一条消息(其余向后移),并把它计入 token 预算(active 才计)。
    ///
-    /// 供阶段2 在压缩点插入摘要 system 消息。`index` 越界则 panic(对齐 Vec::insert 语义,
+    /// 供压缩点插入摘要 system 消息。`index` 越界则 panic(对齐 Vec::insert 语义,
    /// 调用方负责算合法 index,如 `compress_end` 已由 `compress_old_messages` 校验)。
    pub fn insert_at(&mut self, index: usize, message: ChatMessage) {
        let tokens = self.estimator.estimate_message(&message);
@@ -449,7 +449,7 @@ impl ContextManager {
    /// - 工具调用三元组(Head + Tail* + 紧随的 Standalone Assistant)在同一单元
    /// - 保护区 `[protect_start, len)` 内的消息不纳入任何单元
    ///
-    /// 公开供阶段2 会话分段(`archived_segment` 按组原子标记)与压缩定位共用。
+    /// 公开会话分段(`archived_segment` 按组原子标记)与压缩定位共用。
    pub fn build_eviction_units(&self, protect_start: usize) -> Vec<EvictionUnit> {
        let mut units = Vec::new();
        let mut i = 0usize;
@@ -488,7 +488,7 @@ impl ContextManager {
    /// 不参与二次压缩,幂等)。`protect_start` 为保护区起点(如 `len - PROTECT_COUNT`)。
    pub fn has_compressible_messages(&self, protect_start: usize) -> bool {
        let end = protect_start.min(self.messages.len());
-        // BUG-260624-05:排除 system 角色(压缩摘要 / 话题切换锚点)。这些是上下文锚点非压缩目标——
+        // 排除 system 角色(压缩摘要 / 话题切换锚点)。这些是上下文锚点非压缩目标——
        // 若计入,压缩摘要 insert_at(0) 落在可压缩区 [0..protect_start) 且 is_active(status=None),
        // 致每轮 has_compressible 恒 true → 无限循环压缩(用户报"压缩后每轮提示已压缩并停止")。
        // compress_old_messages 不改:被调用时仍标旧 system 摘要 compressed(被新摘要替代,防堆积)。
@@ -498,7 +498,7 @@ impl ContextManager {
    }

    /// 把保护区 `[0, compress_end)` 范围内的 active 消息标记为 `status="compressed"`,
-    /// 同步从 `history_tokens` 扣除其 token,返回被压缩消息的克隆(供阶段2 喂 LLM 摘要)。
+    /// 同步从 `history_tokens` 扣除其 token,返回被压缩消息的克隆(供喂 LLM 摘要)。
    ///
    /// **幂等**:已 compressed(或任何 !active)的消息跳过,不会被二次压缩;`history_tokens`
    /// 也只扣首次标记的 token。返回的 Vec 仅含**本次新标记**的消息(已 compressed 的不返)。
@@ -151,7 +151,7 @@ pub fn sanitize_messages(messages: Vec<ChatMessage>) -> Vec<ChatMessage> {
        sanitized
    };

-    // step 3.5(阶段2 占位配对完整性):反向 orphan 检测 —— tool_result 无对应 tool_call 头 → 丢。
+    // step 3.5(占位配对完整性):反向 orphan 检测 —— tool_result 无对应 tool_call 头 → 丢。
    //
    // 根因(解 400 orphan):审批挂起占位 tool_result(内容 audit/cache.rs:PENDING_APPROVAL_PLACEHOLDER)
    // 经 step3(正向 orphan:头无 result→丢头 + 其 result)或 build_eviction_units(预算裁剪从三元组
@@ -227,7 +227,7 @@ pub fn drop_reverse_orphans(messages: Vec<ChatMessage>) -> Vec<ChatMessage> {
    filtered
 }

-/// 发送视图出口断言(阶段2 占位配对完整性):确保所有 tool_result(含审批占位)都有
+/// 发送视图出口断言(占位配对完整性):确保所有 tool_result(含审批占位)都有
 /// 对应 tool_call 头,失败降级 TOOL_MISSING_PREFIX 自愈。
 ///
 /// **职责**:在消息即将发给 LLM 前(`build_for_request` 出口)最后一道防线:若仍有
@@ -470,7 +470,7 @@ mod tests {
        assert_eq!(msgs.len(), 3, "正常三元组不应被 sanitize 剔除");
    }

-    // ── 阶段2 占位配对完整性(解 400 orphan):反向 orphan 检测 + 出口自愈 ──
+    // ── 占位配对完整性(解 400 orphan):反向 orphan 检测 + 出口自愈 ──

    #[test]
    fn sanitize_drops_reverse_orphan_tool_result() {