release: v3.3.4 continuity and state endpoints

Author: LegnaOS

RELEASE_NOTES_v3.3.4.md

@@ -0,0 +1,54 @@
+# Augment Proxy Manager v3.3.4 - Continuity 修复与协议状态承接
+
+## 核心结论
+
+v3.3.4 的改动集中在代理层的 continuation handling、history projection、state endpoint compatibility 和 per-conversation serialization：
+
+- continuation `"..."` 仅作为 continuation signal 处理，不再改写用户消息内容
+- 原始 `chat_history` 保留，压缩结果单独写入 `compressed_chat_history`
+- `save-chat`、`record-*`、`context-canvas/list` 新增最小状态写入与读取逻辑
+- 同一 `conversation_id` 的请求通过 `state.conversationQueues` 串行化处理
+
+## 本次修复
+
+### 1. Continuity 主链修复
+- continuation `"..."` 按 continuation signal 处理，不再通过 prompt 改写参与消息编译
+- 原始 `chat_history` 保留为会话原始记录
+- 压缩结果单独写入 `compressed_chat_history`
+- provider 统一从 canonical / normalized timeline 编译消息
+
+### 2. 状态端点最小实现
+以下端点已从固定成功响应调整为最小状态实现：
+
+- `/save-chat`
+- `/record-session-events`
+- `/record-user-events`
+- `/record-request-events`
+- `/context-canvas/list`
+- `/generate-conversation-title`
+- `/notifications/mark-read`（兼容处理）
+
+代理层新增以下内存状态容器，用于保存最小会话状态：
+
+- `conversationStates`
+- `canvasStates`
+- `sessionEventStore`
+- `userEventStore`
+- `requestEventStore`
+
+### 3. 会话级请求串行化
+- 请求队列统一挂到全局 `state.conversationQueues`
+- 同一 `conversation_id` 的请求按顺序执行，避免并发交错影响工具调用链
+
+## 验证结果
+
+已完成本地 runtime smoke：
+
+- `save-chat`
+- `record-session-events`
+- `record-user-events`
+- `record-request-events`
+- `context-canvas/list`
+- `generate-conversation-title`
+
+以上端点均返回与会话状态相关的结构化结果，而不是固定 `success: true` 响应。

package-lock.json

@@ -2,7 +2,7 @@
   "name": "augment-proxy-manager",
   "displayName": "Augment Proxy Manager",
   "description": "管理 Augment API 代理服务器，支持自定义 API 端点和多种 AI 供应商",
-  "version": "3.1.4",
+  "version": "3.3.4",
   "publisher": "legna",
   "repository": {
     "type": "git",
@@ -24,7 +24,7 @@
         {
           "id": "augmentProxy",
           "title": "Augment Proxy",
-          "icon": "$(radio-tower)"
+          "icon": "icon.svg"
         }
       ]
     },
@@ -33,7 +33,8 @@
         {
           "type": "webview",
           "id": "augmentProxy.sidebar",
-          "name": "控制面板"
+          "name": "控制面板",
+          "icon": "icon.svg"
         }
       ]
     },

package.json

@@ -4,11 +4,11 @@
 
 # Augment Proxy Manager
 
-**用任意 AI 供应商驱动 Augment 的强大编码 Agent。**
+**Augment 前端 + 多模型后端的本地协议代理。**
 
 零注入 · 零登录 · 零配置
 
-[![Version](https://img.shields.io/badge/version-3.1.4-blue.svg)](https://github.com/LegnaOS/VSC-Augment-Proxy-Manager)
+[![Version](https://img.shields.io/badge/version-3.3.4-blue.svg)](https://github.com/LegnaOS/VSC-Augment-Proxy-Manager)
 [![Platform](https://img.shields.io/badge/platform-macOS%20%7C%20Windows%20%7C%20Linux-lightgrey.svg)]()
 
 </div>
@@ -19,6 +19,8 @@
 
 Augment Proxy Manager 运行一个本地 HTTP 代理服务器，拦截 Augment 扩展的 API 请求并转发到你选择的 AI 供应商。
 
+代理层负责承接 Augment 的请求格式、上下文投影、工具调用和状态端点，再转换到不同模型后端。
+
 ```
 Augment 扩展  →  本地代理 (:8765)  →  你的 AI 供应商 API
                   ↑ 自动配置          ↑ Viking 上下文增强
@@ -66,6 +68,9 @@ Augment 扩展  →  本地代理 (:8765)  →  你的 AI 供应商 API
 - **零注入绕过** — 自动配置 Augment 使用代理，无需修改任何代码
 - **流式响应** — 聊天、补全、指令全程实时 SSE 流式传输
 - **完整 Agent 模式** — 工具调用、文件编辑、代码库检索全部正常工作
+- **协议转换层** — 代理统一承接 Augment 请求格式，并转换到 Anthropic / OpenAI / Google 等不同后端协议
+- **continuity 修复** — 保留原始 `chat_history`，压缩结果写入 `compressed_chat_history`，避免多步任务上下文被压坏
+- **状态端点最小实现** — `/save-chat`、`/record-session-events`、`/record-user-events`、`/record-request-events`、`/context-canvas/list` 已从固定成功响应调整为最小状态实现
 - **配置热更新** — 切换供应商或模型无需重启代理，实时生效
 
 ### 🔍 RAG 语义搜索
@@ -157,6 +162,14 @@ src/
 
 ## 更新日志
 
+### v3.3.4 — Continuity 修复 + 协议状态承接
+
+- **continuation handling** — `"..."` 按 continuation signal 处理，不再改写用户消息内容
+- **history projection** — 原始 `chat_history` 保留，压缩结果写入 `compressed_chat_history`
+- **state endpoints** — `save-chat` / `record-*` / `context-canvas/list` 新增最小状态写入与读取逻辑
+- **request serialization** — 同一 `conversation_id` 的请求统一进入 `state.conversationQueues`
+- **runtime smoke** — 已验证 `save-chat`、`record-session-events`、`record-user-events`、`record-request-events`、`context-canvas/list`、`generate-conversation-title`
+
 ### v3.1.4 — Agent 循环修复 + 任务系统生效
 
 **🔴 致命修复**

readme.md

@@ -8,6 +8,8 @@ export async function applyContextCompression(augmentReq: any, providerName: str
     const compressionThresholdPercent = config.get('compressionThreshold', 60) as number; // 降低到 60%
     const compressionThreshold = compressionThresholdPercent / 100;
 
+    delete augmentReq.compressed_chat_history;
+
     if (!enableCompression || !augmentReq.chat_history || augmentReq.chat_history.length === 0) return;
 
     const modelName = state.currentConfig.model || 'unknown';
@@ -38,7 +40,7 @@ export async function applyContextCompression(augmentReq: any, providerName: str
     if (contextStats.needs_compression || contextStats.usage_percentage > (preemptiveThreshold * 100)) {
         const compressionResult = await compressChatHistoryByTokens(augmentReq.chat_history, tokenLimit, 0.3, compressionThreshold); // 目标 30%
         if (compressionResult.compressed_count < compressionResult.original_count) {
-            augmentReq.chat_history = compressionResult.compressed_exchanges;
+            augmentReq.compressed_chat_history = compressionResult.compressed_exchanges;
             log(`[CONTEXT] ✂️ 压缩: ${compressionResult.original_count} → ${compressionResult.compressed_count} 次交互`);
             log(`[CONTEXT] 📉 Token: ${compressionResult.estimated_tokens_before} → ${compressionResult.estimated_tokens_after} (${(compressionResult.compression_ratio * 100).toFixed(1)}%)`);
             if (compressionResult.summary) log(`[CONTEXT] 📝 摘要: ${compressionResult.summary.slice(0, 80)}...`);

src/context-compression.ts

@@ -158,6 +158,10 @@ function estimateExchangesTokens(exchanges: Exchange[]): number {
     return totalTokens;
 }
 
+function hasToolUse(exchange?: Exchange): boolean {
+    return !!exchange?.response_nodes?.some((node: any) => node.type === 5 && node.tool_use);
+}
+
 /**
  * 获取上下文统计信息
  */
@@ -238,8 +242,14 @@ export async function compressChatHistoryByTokens(
     keepCount = Math.max(keepCount, Math.min(3, chatHistory.length));
 
     // 分离最近的和需要压缩的历史
-    const recentExchanges = chatHistory.slice(-keepCount);
-    const oldExchanges = chatHistory.slice(0, -keepCount);
+    // 不要在 tool_use / tool_result 的边界中间切断，否则会破坏多步任务连续性
+    let splitIndex = Math.max(0, chatHistory.length - keepCount);
+    while (splitIndex > 0 && hasToolUse(chatHistory[splitIndex - 1])) {
+        splitIndex--;
+    }
+
+    const recentExchanges = chatHistory.slice(splitIndex);
+    const oldExchanges = chatHistory.slice(0, splitIndex);
 
     if (oldExchanges.length === 0) {
         return {
@@ -255,6 +265,7 @@ export async function compressChatHistoryByTokens(
     // 生成旧历史的摘要
     const summary = generateHistorySummary(oldExchanges);
     const summaryTokens = estimateTokens(summary);
+    const recentTokens = estimateExchangesTokens(recentExchanges);
 
     // 创建一个摘要交互
     const summaryExchange: Exchange = {
@@ -269,7 +280,7 @@ export async function compressChatHistoryByTokens(
     };
 
     const compressedExchanges = [summaryExchange, ...recentExchanges];
-    const tokensAfter = summaryTokens + accumulatedTokens;
+    const tokensAfter = summaryTokens + recentTokens;
 
     return {
         compressed_exchanges: compressedExchanges,
@@ -278,7 +289,7 @@ export async function compressChatHistoryByTokens(
         compressed_count: compressedExchanges.length,
         estimated_tokens_before: tokensBefore,
         estimated_tokens_after: tokensAfter,
-        compression_ratio: tokensAfter / tokensBefore
+        compression_ratio: tokensBefore > 0 ? tokensAfter / tokensBefore : 1.0
     };
 }
 

src/context-manager.ts

@@ -23,6 +23,8 @@ export const AUGMENT_ENDPOINTS = [
     // 订阅和用户
     '/subscription-banner',
     '/save-chat',
+    '/context-canvas/list',
+    '/generate-conversation-title',
     // 用户密钥
     '/user-secrets/list',
     '/user-secrets/upsert',
@@ -33,6 +35,7 @@ export const AUGMENT_ENDPOINTS = [
     // 遥测和事件
     '/client-completion-timelines',
     '/record-session-events',
+    '/record-user-events',
     '/record-request-events',
     // 其他
     '/next-edit-stream',

src/endpoints.ts

@@ -4,6 +4,42 @@ import { CurrentConfig } from './types';
 import type { VikingContextStore } from './rag/viking-context';
 import type { SessionMemory } from './rag/session-memory';
 
+export interface RecordedEvent {
+    id: string;
+    source: 'session' | 'user' | 'request';
+    type: string;
+    recordedAt: string;
+    conversationId?: string;
+    sessionId?: string;
+    requestId?: string;
+    canvasId?: string;
+    userId?: string;
+    payload: any;
+}
+
+export interface ConversationState {
+    conversationId: string;
+    canvasId?: string;
+    title?: string;
+    createdAt: string;
+    updatedAt: string;
+    lastMessage?: string;
+    lastRequestId?: string;
+    chatHistory: any[];
+    compressedChatHistory?: any[];
+    nodes: any[];
+    metadata?: Record<string, any>;
+}
+
+export interface CanvasState {
+    canvasId: string;
+    conversationId?: string;
+    title: string;
+    createdAt: string;
+    updatedAt: string;
+    metadata?: Record<string, any>;
+}
+
 // ===== 全局共享状态 =====
 // 所有模块通过 state 对象访问共享状态
 
@@ -22,7 +58,13 @@ export const state = {
 
     // 会话级请求队列
     conversationQueues: new Map<string, Promise<void>>(),
-    conversationUserMessages: new Map<string, string>(),
+
+    // 最小状态承接
+    conversationStates: new Map<string, ConversationState>(),
+    canvasStates: new Map<string, CanvasState>(),
+    sessionEventStore: new Map<string, RecordedEvent[]>(),
+    userEventStore: new Map<string, RecordedEvent[]>(),
+    requestEventStore: new Map<string, RecordedEvent[]>(),
 
     // 当前配置
     currentConfig: {

src/globals.ts

@@ -92,10 +92,10 @@ function generateInjectionCode(proxyUrl: string): string {
         if (urlStr.includes('/chat-stream')) ep = '/chat-stream'; else if (urlStr.includes('/chat-input-completion')) ep = '/chat-input-completion'; else if (urlStr.includes('/chat')) ep = '/chat'; else if (urlStr.includes('/instruction-stream')) ep = '/instruction-stream'; else if (urlStr.includes('/smart-paste-stream')) ep = '/smart-paste-stream'; else if (urlStr.includes('/completion')) ep = '/completion';
         else if (urlStr.includes('/getPluginState')) ep = '/getPluginState'; else if (urlStr.includes('/get-model-config')) ep = '/get-model-config'; else if (urlStr.includes('/get-models')) ep = '/get-models';
         else if (urlStr.includes('/agents/codebase-retrieval')) ep = '/agents/codebase-retrieval'; else if (urlStr.includes('/agents/edit-file')) ep = '/agents/edit-file'; else if (urlStr.includes('/agents/list-remote-tools')) ep = '/agents/list-remote-tools'; else if (urlStr.includes('/agents/run-remote-tool')) ep = '/agents/run-remote-tool';
-        else if (urlStr.includes('/remote-agents/list-stream')) ep = '/remote-agents/list-stream'; else if (urlStr.includes('/subscription-banner')) ep = '/subscription-banner'; else if (urlStr.includes('/save-chat')) ep = '/save-chat';
+        else if (urlStr.includes('/remote-agents/list-stream')) ep = '/remote-agents/list-stream'; else if (urlStr.includes('/subscription-banner')) ep = '/subscription-banner'; else if (urlStr.includes('/save-chat')) ep = '/save-chat'; else if (urlStr.includes('/context-canvas/list')) ep = '/context-canvas/list'; else if (urlStr.includes('/generate-conversation-title')) ep = '/generate-conversation-title';
         else if (urlStr.includes('/user-secrets/list')) ep = '/user-secrets/list'; else if (urlStr.includes('/user-secrets/upsert')) ep = '/user-secrets/upsert'; else if (urlStr.includes('/user-secrets/delete')) ep = '/user-secrets/delete';
         else if (urlStr.includes('/notifications/mark-read')) ep = '/notifications/mark-read'; else if (urlStr.includes('/notifications')) ep = '/notifications';
-        else if (urlStr.includes('/client-completion-timelines')) ep = '/client-completion-timelines'; else if (urlStr.includes('/record-session-events')) ep = '/record-session-events'; else if (urlStr.includes('/record-request-events')) ep = '/record-request-events';
+        else if (urlStr.includes('/client-completion-timelines')) ep = '/client-completion-timelines'; else if (urlStr.includes('/record-session-events')) ep = '/record-session-events'; else if (urlStr.includes('/record-user-events')) ep = '/record-user-events'; else if (urlStr.includes('/record-request-events')) ep = '/record-request-events';
         else if (urlStr.includes('/next-edit-stream')) ep = '/next-edit-stream'; else if (urlStr.includes('/find-missing')) ep = '/find-missing'; else if (urlStr.includes('/client-metrics')) ep = '/client-metrics'; else if (urlStr.includes('/batch-upload')) ep = '/batch-upload'; else if (urlStr.includes('/report-feature-vector')) ep = '/report-feature-vector'; else if (urlStr.includes('/report-error')) ep = '/report-error';
         if (!ep) { log('Passing through (no matching endpoint):', urlStr); return originalFetch.call(this, url, options); }
         const target = CONFIG.proxyUrl + ep; log('Intercepted:', urlStr, '->', target);
@@ -105,7 +105,7 @@ function generateInjectionCode(proxyUrl: string): string {
     // Node.js http/https 拦截
     try {
         const https = require('https'); const http = require('http'); const oHttps = https.request; const oHttp = http.request;
-        const getEp = (u) => { if(u.includes('/chat-stream'))return'/chat-stream';if(u.includes('/chat-input-completion'))return'/chat-input-completion';if(u.includes('/chat'))return'/chat';if(u.includes('/instruction-stream'))return'/instruction-stream';if(u.includes('/smart-paste-stream'))return'/smart-paste-stream';if(u.includes('/completion'))return'/completion';if(u.includes('/getPluginState'))return'/getPluginState';if(u.includes('/get-model-config'))return'/get-model-config';if(u.includes('/get-models'))return'/get-models';if(u.includes('/agents/codebase-retrieval'))return'/agents/codebase-retrieval';if(u.includes('/agents/edit-file'))return'/agents/edit-file';if(u.includes('/agents/list-remote-tools'))return'/agents/list-remote-tools';if(u.includes('/agents/run-remote-tool'))return'/agents/run-remote-tool';if(u.includes('/remote-agents/list-stream'))return'/remote-agents/list-stream';if(u.includes('/next-edit-stream'))return'/next-edit-stream';if(u.includes('/find-missing'))return'/find-missing';if(u.includes('/client-metrics'))return'/client-metrics';if(u.includes('/batch-upload'))return'/batch-upload';if(u.includes('/report-feature-vector'))return'/report-feature-vector';if(u.includes('/report-error'))return'/report-error';return null; };
+        const getEp = (u) => { if(u.includes('/chat-stream'))return'/chat-stream';if(u.includes('/chat-input-completion'))return'/chat-input-completion';if(u.includes('/chat'))return'/chat';if(u.includes('/instruction-stream'))return'/instruction-stream';if(u.includes('/smart-paste-stream'))return'/smart-paste-stream';if(u.includes('/completion'))return'/completion';if(u.includes('/getPluginState'))return'/getPluginState';if(u.includes('/get-model-config'))return'/get-model-config';if(u.includes('/get-models'))return'/get-models';if(u.includes('/agents/codebase-retrieval'))return'/agents/codebase-retrieval';if(u.includes('/agents/edit-file'))return'/agents/edit-file';if(u.includes('/agents/list-remote-tools'))return'/agents/list-remote-tools';if(u.includes('/agents/run-remote-tool'))return'/agents/run-remote-tool';if(u.includes('/remote-agents/list-stream'))return'/remote-agents/list-stream';if(u.includes('/subscription-banner'))return'/subscription-banner';if(u.includes('/save-chat'))return'/save-chat';if(u.includes('/context-canvas/list'))return'/context-canvas/list';if(u.includes('/generate-conversation-title'))return'/generate-conversation-title';if(u.includes('/record-session-events'))return'/record-session-events';if(u.includes('/record-user-events'))return'/record-user-events';if(u.includes('/record-request-events'))return'/record-request-events';if(u.includes('/next-edit-stream'))return'/next-edit-stream';if(u.includes('/find-missing'))return'/find-missing';if(u.includes('/client-metrics'))return'/client-metrics';if(u.includes('/batch-upload'))return'/batch-upload';if(u.includes('/report-feature-vector'))return'/report-feature-vector';if(u.includes('/report-error'))return'/report-error';return null; };
         const wrap = (orig, proto) => function(urlOrOpts, opts, cb) { let t=''; if(typeof urlOrOpts==='string')t=urlOrOpts; else if(urlOrOpts&&urlOrOpts.hostname)t=proto+'://'+urlOrOpts.hostname+(urlOrOpts.path||''); if(!CONFIG.enabled||!t.includes('augmentcode.com')||!CONFIG.proxyAvailable)return orig.apply(this,arguments); const ep=getEp(t); if(!ep){log('HTTP: pass through:',t.substring(0,80));return orig.apply(this,arguments);} log('HTTP: Intercepting '+ep); const po={hostname:'localhost',port:8765,path:ep,method:(typeof urlOrOpts==='object'?urlOrOpts.method:'GET')||'GET',headers:typeof urlOrOpts==='object'?urlOrOpts.headers:{}}; po.headers['Content-Type']='application/json'; return oHttp.call(http,po,typeof opts==='function'?opts:cb); };
         https.request = wrap(oHttps, 'https'); http.request = wrap(oHttp, 'http'); log('Node.js http intercepted');
     } catch (e) { log('Failed to intercept http:', e.message); }