feat: 实现上下文使用量监控和会话压缩功能

- sseHandler: 新增 onContextUsage 回调处理上下文使用量事件
- dialogService: 集成上下文使用量回调,追踪 AI 消息用于后端重启恢复
- apiClient: 新增 compactDialog API 支持手动压缩对话历史
- messageHandler: 新增 lastTaskId 管理机制,支持会话恢复后的压缩操作,转发上下文使用量到 WebView
This commit is contained in:
XiaoFeng
2025-12-31 18:50:20 +08:00
parent 259310a29d
commit b794d1ceb0
4 changed files with 64 additions and 2 deletions

View File

@ -73,6 +73,8 @@ export interface DialogCallbacks {
onError?: (message: string) => void;
/** 通知消息 */
onNotification?: (message: string) => void;
/** 上下文使用量更新 */
onContextUsage?: (data: { currentTokens: number; maxTokens: number; percentage: number }) => void;
}
/**
@ -553,6 +555,12 @@ export class DialogSession {
onComplete: (data) => {
this.isActive = false;
this.finalizeTextSegment();
// 追踪 AI 消息(用于后端重启后恢复)
if (this.accumulatedText) {
historyManager.trackAiMessage(this.accumulatedText);
}
// 发送所有段落
callbacks.onComplete?.(this.segments);
},
@ -639,6 +647,11 @@ export class DialogSession {
await historyManager.saveCompactedData(data.compactedData);
},
onContextUsage: (data) => {
console.log('[DialogSession] onContextUsage:', data.currentTokens, '/', data.maxTokens);
callbacks.onContextUsage?.(data);
},
onOpen: () => {
console.log('[DialogSession] SSE 连接已建立');
},