/** * InputArea - 输入区域组件 */ import { state, KEYS } from '../state/state.js'; import type { MetonaDesktopAPI } from '../types.js'; import { detectLanguage, truncate, escapeHtml, formatSize } from '../utils/utils.js'; import { getSelectedModel, isThinkEnabled, isVisionAvailable, isToolCallingSupported } from './model-bar.js'; import { renderMessages, clearMessagesDOM, appendAssistantPlaceholder, appendSystemMessage, updateLastAssistantMessage, clearMessages, safeMarkdown, enableAutoScroll, resetAutoScroll, appendToolCallCardToPlaceholder, updateToolCallCardInPlaceholder, resetCurrentPlaceholder, removeCurrentPlaceholder } from './chat-area.js'; import { showToast } from './toast.js'; import { addToolCard, startToolCard, updateToolCard, clearToolCardsExternal, clearTerminalExternal, getWorkspaceDirPath, hasActiveCards, showWorkingHint, clearWorkingHint } from './workspace-panel.js'; import { ChatDB } from '../db/chat-db.js'; import { OllamaAPI } from '../api/ollama.js'; import { runAgentLoop } from '../services/agent-engine.js'; import { estimateTokens } from '../services/context-manager.js'; import { showToolConfirm } from './tool-confirm-modal.js'; import { logInfo, logStream, logError, logSuccess, logWarn, resetVideoProgress, updateVideoProgress } from '../services/log-service.js'; import type { ChatSession, ChatMessage, OllamaStreamChunk, OllamaMessage, FileContent, ChatFile, ToolCallRecord, AgentMode } from '../types.js'; let chatInputEl: HTMLTextAreaElement; let btnSendEl: HTMLButtonElement; let imagePreviewEl: HTMLElement; let filePreviewEl: HTMLElement; let videoPreviewEl: HTMLElement; let planToggleCheckbox: HTMLInputElement; let pendingImages: Array<{ name: string; base64: string }> = []; let pendingFiles: Array<{ name: string; content: string; language: string; size: number }> = []; let pendingVideoFrames: Array<{ name: string; base64: string; timestampSeconds: number }> = []; let pendingVideoMeta: { fileName: string; duration: number; width: number; height: number } | null = null; let completedVideos: Array<{ fileName: string; duration: number; frames: Array<{ name: string; base64: string; timestampSeconds: number }> }> = []; const TEXT_EXTENSIONS = new Set([ 'txt','md','markdown','rst','log','csv','tsv', 'py','pyw','pyi','js','mjs','cjs','ts','tsx','jsx', 'java','c','h','cpp','cc','cxx','hpp','go','rs','rb','php', 'sh','bash','zsh','fish','sql','html','htm','css', 'json','jsonl','xml','svg','yaml','yml','toml','ini','cfg','conf', 'swift','kt','kts','scala','lua','r','m','mm', 'cs','fs','vb','ex','exs','erl','hs','clj','lisp', 'diff','patch','pl','dockerfile','makefile' ]); const MAX_FILE_SIZE = 500 * 1024; const VIDEO_EXTENSIONS = new Set(['mp4', 'avi', 'mov', 'mkv', 'webm', 'flv', 'wmv', 'm4v']); function getFileIcon(filename: string): string { const name = filename.toLowerCase(); if (name === 'dockerfile') return '🐳'; if (name === 'makefile' || name === 'gnumakefile') return '🔨'; const ext = filename.split('.').pop()?.toLowerCase() || ''; const map: Record = { py: '🐍', js: '💛', ts: '🔷', java: '☕', go: '🐹', rs: '🦀', cpp: '⚙️', c: '⚙️', sh: '🐚', html: '🌐', css: '🎨', json: '📋', md: '📝', txt: '📄', sql: '🗃️', csv: '📊' }; return map[ext] || '📄'; } function isTextFile(name: string): boolean { const lower = name.toLowerCase(); if (lower === 'dockerfile' || lower === 'makefile') return true; const ext = lower.split('.').pop() || ''; return TEXT_EXTENSIONS.has(ext); } function getBridge(): MetonaDesktopAPI | undefined { return window.metonaDesktop; } export function initInputArea(): void { chatInputEl = document.querySelector('#chatInput')!; btnSendEl = document.querySelector('#btnSend')!; imagePreviewEl = document.querySelector('#imagePreview')!; filePreviewEl = document.querySelector('#filePreview')!; videoPreviewEl = document.querySelector('#videoPreview')!; btnSendEl.addEventListener('click', () => { if (state.get(KEYS.IS_STREAMING)) { stopGeneration(); } else { sendMessage(); } }); chatInputEl.addEventListener('keydown', (e) => { if (e.key === 'Enter' && !e.shiftKey) { e.preventDefault(); sendMessage(); } }); // R39: 全局键盘快捷键 document.addEventListener('keydown', (e) => { // Ctrl+K: 清空聊天（需要确认） if ((e.ctrlKey || e.metaKey) && e.key === 'k') { e.preventDefault(); if (confirm('确定要清空当前对话吗？')) { clearMessages(); } } // Escape: 停止生成（当正在流式输出时） if (e.key === 'Escape' && state.get(KEYS.IS_STREAMING)) { e.preventDefault(); stopGeneration(); } // Ctrl+/: 聚焦输入框 if ((e.ctrlKey || e.metaKey) && e.key === '/') { e.preventDefault(); chatInputEl.focus(); } // Ctrl+Shift+C: 复制最后一条 AI 消息 if ((e.ctrlKey || e.metaKey) && e.shiftKey && e.key === 'C') { e.preventDefault(); const session = state.get(KEYS.CURRENT_SESSION); if (session && session.messages.length > 0) { const lastAssistant = [...session.messages].reverse().find(m => m.role === 'assistant' && m.content); if (lastAssistant) { navigator.clipboard.writeText(lastAssistant.content).then(() => { showToast('已复制最后一条 AI 回复', 'success'); }).catch(() => {}); } } } }); chatInputEl.addEventListener('input', autoResizeTextarea); // ── 图片上传：原生文件对话框 ── document.querySelector('#btnAttachImg')!.addEventListener('click', async () => { if (!isVisionAvailable()) { showToast('当前模型不支持图片分析', 'warning'); return; } const bridge = getBridge(); if (!bridge) return; const paths = await bridge.dialog.openFile({ filters: [{ name: '图片', extensions: ['png', 'jpg', 'jpeg', 'gif', 'webp', 'bmp', 'svg'] }] }); if (!paths || paths.length === 0) return; await handleImagePaths(paths); }); // ── 文件上传：原生文件对话框 ── document.querySelector('#btnAttachFile')!.addEventListener('click', async () => { const bridge = getBridge(); if (!bridge) return; const paths = await bridge.dialog.openFile({ filters: [{ name: '文本/代码', extensions: Array.from(TEXT_EXTENSIONS) }] }); if (!paths || paths.length === 0) return; await handleTextFilePaths(paths); }); // ── 视频上传：提取帧作为图片 ── document.querySelector('#btnAttachVideo')!.addEventListener('click', async () => { if (!isVisionAvailable()) { showToast('当前模型不支持图片/视频分析', 'warning'); return; } const bridge = getBridge(); if (!bridge) return; const paths = await bridge.dialog.openFile({ filters: [{ name: '视频', extensions: Array.from(VIDEO_EXTENSIONS) }] }); if (!paths || paths.length === 0) return; for (const p of paths) { await handleVideoPath(p); } }); imagePreviewEl.addEventListener('click', (e) => { const target = e.target as HTMLElement; if (target.classList.contains('remove-img')) { pendingImages.splice(parseInt((target as HTMLElement).dataset.index!), 1); renderImagePreviews(); } }); filePreviewEl.addEventListener('click', (e) => { const target = e.target as HTMLElement; if (target.classList.contains('remove-file')) { pendingFiles.splice(parseInt((target as HTMLElement).dataset.index!), 1); renderFilePreviews(); } }); // ── Plan Mode 切换 ── const planToggleEl = document.querySelector('#togglePlan') as HTMLInputElement; if (planToggleEl) { planToggleCheckbox = planToggleEl; planToggleEl.addEventListener('change', () => { const mode: AgentMode = planToggleEl.checked ? 'plan' : 'auto'; state.set('agentMode', mode); // Plan Mode 切换时注册/移除 plan_track 工具 import('../services/tool-registry.js').then(({ setPlanModeActive }) => { setPlanModeActive(mode === 'plan'); }); logInfo(`Agent 模式切换: ${mode === 'plan' ? 'Plan（先规划后执行）' : 'Auto（直接执行）'}`); }); } } function autoResizeTextarea(): void { chatInputEl.style.height = 'auto'; chatInputEl.style.height = Math.min(chatInputEl.scrollHeight, 120) + 'px'; } // ── 图片压缩管线 ── const IMG_MAX_WIDTH = 1280; const IMG_MAX_HEIGHT = 1280; const IMG_QUALITY = 0.8; const IMG_MAX_SIZE_BEFORE_COMPRESS = 300 * 1024; // 300KB以下不压缩 /** * Canvas 前端压缩图片：等比缩放 + JPEG 转换 * base64 → Image → Canvas 缩放绘制 → toBlob → 压缩后 base64 */ function compressImage(base64: string): Promise<{ compressed: string; origBytes: number; newBytes: number }> { return new Promise((resolve, reject) => { // 先计算原始 base64 的大小（字节数 ≈ base64长度 * 0.75） const origBytes = Math.ceil(base64.length * 0.75); // 小图跳过压缩 if (origBytes <= IMG_MAX_SIZE_BEFORE_COMPRESS) { resolve({ compressed: base64, origBytes, newBytes: origBytes }); return; } const img = new Image(); img.onload = () => { let { width, height } = img; // 等比缩放 if (width > IMG_MAX_WIDTH || height > IMG_MAX_HEIGHT) { const ratio = Math.min(IMG_MAX_WIDTH / width, IMG_MAX_HEIGHT / height); width = Math.round(width * ratio); height = Math.round(height * ratio); } const canvas = document.createElement('canvas'); canvas.width = width; canvas.height = height; const ctx = canvas.getContext('2d')!; ctx.drawImage(img, 0, 0, width, height); canvas.toBlob((blob) => { if (!blob) { resolve({ compressed: base64, origBytes, newBytes: origBytes }); return; } const reader = new FileReader(); let newBytes = 0; reader.onloadend = () => { const compressed = (reader.result as string).split(',')[1] || base64; newBytes = blob.size; resolve({ compressed, origBytes, newBytes }); }; reader.onerror = () => { resolve({ compressed: base64, origBytes, newBytes: origBytes }); }; reader.readAsDataURL(blob); }, 'image/jpeg', IMG_QUALITY); }; img.onerror = () => resolve({ compressed: base64, origBytes, newBytes: origBytes }); img.src = `data:image/png;base64,${base64}`; }); } async function handleImagePaths(paths: string[]): Promise { const bridge = getBridge(); if (!bridge) return; const maxSize = 20 * 1024 * 1024; // 并行读取所有图片 const results = await Promise.all(paths.map(async (filePath) => { const name = filePath.split(/[/\\]/).pop() || filePath; try { const result = await bridge.fs.readFileBase64(filePath); if (!result.success) return { name, error: result.error }; const size = result.size ?? 0; if (size > maxSize) return { name, error: `超过 20MB 限制` }; return { name, base64: result.content!, size }; } catch (err) { return { name, error: (err as Error).message }; } })); for (const r of results) { if ('error' in r && r.error) { appendSystemMessage(`❌ 读取 ${r.name} 失败: ${r.error}`); continue; } if ('base64' in r) { const imgResult = r as { name: string; base64: string; size: number }; const { compressed, origBytes, newBytes } = await compressImage(imgResult.base64); pendingImages.push({ name: imgResult.name, base64: compressed }); if (origBytes !== newBytes) { logInfo(`图片已压缩: ${r.name}`, `${formatSize(origBytes)} → ${formatSize(newBytes)} (${Math.round((1 - newBytes / origBytes) * 100)}%)`); } else { logInfo(`图片已添加: ${r.name}`, formatSize(origBytes)); } } } renderImagePreviews(); } async function handleVideoPath(filePath: string): Promise { const bridge = getBridge(); if (!bridge?.video) { showToast('视频功能需要桌面版', 'warning'); return; } const name = filePath.split(/[/\\]/).pop() || filePath; // 聊天区：开始消息（带 class 标记，后续原地更新进度） logInfo('🎬 正在提取视频帧', name); appendSystemMessage(`🎬 正在提取视频帧 (1fps): ${name}...`, 'sysmsg-video-progress'); pendingVideoMeta = { fileName: name, duration: 0, width: 0, height: 0 }; renderVideoPreview(); resetVideoProgress(name); // 监听提取进度：实时更新日志条目 + 视频芯片 + 聊天区系统消息 let progressEl: HTMLElement | null = null; bridge.video.onProgress(({ current }) => { updateVideoProgress(current); if (!progressEl) progressEl = videoPreviewEl.querySelector('.video-meta'); if (progressEl) progressEl.textContent = `提取中 · ${current}帧`; // 聊天区系统消息原地更新 const sysMsg = document.querySelector('.sysmsg-video-progress'); if (sysMsg) sysMsg.textContent = `🎬 正在提取视频帧 (1fps): ${name}... ${current} 帧`; }); try { const result = await bridge.video.extractFrames(filePath, { maxFrames: 600, maxWidth: 512 }); bridge.video.removeProgressListener(); if (!result.success) { const sysMsg = document.querySelector('.sysmsg-video-progress'); if (sysMsg) { sysMsg.classList.remove('sysmsg-video-progress'); sysMsg.textContent = `❌ 视频帧提取失败: ${result.error || ''}`; } pendingVideoFrames = []; pendingVideoMeta = null; renderVideoPreview(); logError('视频帧提取失败', result.error || ''); return; } const frameCount = result.frames!.length; const duration = result.videoInfo!.duration; // 移入 completedVideos，清除 pending 状态（避免预览重复） completedVideos.push({ fileName: name, duration, frames: [...result.frames!] }); pendingVideoFrames = []; pendingVideoMeta = null; // 完成：聊天区系统消息原地更新为完成 const sysMsg = document.querySelector('.sysmsg-video-progress'); if (sysMsg) { sysMsg.classList.remove('sysmsg-video-progress'); sysMsg.textContent = `✅ ${name}: 已提取 ${frameCount} 帧 (1fps 采样)`; } logSuccess('✅ 视频帧提取完成', `${frameCount} 帧 / ${duration.toFixed(1)}s (1fps)`); renderVideoPreview(); } catch (err) { bridge.video.removeProgressListener(); const sysMsg = document.querySelector('.sysmsg-video-progress'); if (sysMsg) { sysMsg.classList.remove('sysmsg-video-progress'); sysMsg.textContent = `❌ 视频处理失败: ${(err as Error).message}`; } pendingVideoFrames = []; pendingVideoMeta = null; renderVideoPreview(); logError('视频帧提取失败', (err as Error).message); } } function renderVideoPreview(): void { const all: Array<{ fileName: string; frameCount: number; duration: number; isPending: boolean }> = []; if (pendingVideoMeta) { all.push({ fileName: pendingVideoMeta.fileName, frameCount: pendingVideoFrames.length, duration: pendingVideoMeta.duration, isPending: true }); } for (const v of completedVideos) { all.push({ fileName: v.fileName, frameCount: v.frames.length, duration: v.duration, isPending: false }); } if (all.length === 0) { videoPreviewEl.style.display = 'none'; videoPreviewEl.innerHTML = ''; return; } videoPreviewEl.style.display = ''; videoPreviewEl.innerHTML = all.map((v, i) => { const meta = v.isPending && v.duration <= 0 ? '提取中...' : `${v.frameCount}帧 · ${v.duration.toFixed(0)}s`; return `
🎬 ${escapeHtml(v.fileName)} ${meta}
`; }).join(''); videoPreviewEl.querySelectorAll('.remove-video').forEach(btn => { btn.addEventListener('click', (e) => { const idx = parseInt((e.target as HTMLElement).dataset.idx!); // 从 all 数组反查是 pending 还是 completed if (idx === 0 && pendingVideoMeta) { pendingVideoFrames = []; pendingVideoMeta = null; } else { const completedIdx = pendingVideoMeta ? idx - 1 : idx; completedVideos.splice(completedIdx, 1); } renderVideoPreview(); }); }); } function renderImagePreviews(): void { if (pendingImages.length === 0) { imagePreviewEl.style.display = 'none'; imagePreviewEl.innerHTML = ''; return; } imagePreviewEl.style.display = ''; imagePreviewEl.innerHTML = pendingImages.map((img, i) => `
${escapeHtml(img.name)}
`).join(''); } async function handleTextFilePaths(paths: string[]): Promise { const bridge = getBridge(); if (!bridge) return; // 并行读取所有文件 const results = await Promise.all(paths.map(async (filePath) => { const name = filePath.split(/[/\\]/).pop() || filePath; if (!isTextFile(name)) return { name, error: `不是支持的文本/代码文件` }; try { const result = await bridge.fs.readFile(filePath); if (!result.success) return { name, error: result.error }; const content = result.content as string; const size = new TextEncoder().encode(content).length; if (size > MAX_FILE_SIZE) return { name, error: `超过 500KB 限制（${formatSize(size)}）` }; if (pendingFiles.some(f => f.name === name)) return { name, skipped: true }; return { name, content, language: detectLanguage(name), size }; } catch (err) { return { name, error: (err as Error).message }; } })); for (const r of results) { if ('skipped' in r) { appendSystemMessage(`ℹ️ ${r.name} 已添加`); continue; } if ('error' in r && r.error) { appendSystemMessage(`❌ 读取 ${r.name} 失败: ${r.error}`); continue; } if ('content' in r) { const fileResult = r as { name: string; content: string; language: string; size: number }; pendingFiles.push({ name: fileResult.name, content: fileResult.content, language: fileResult.language, size: fileResult.size }); logInfo(`文件已添加: ${fileResult.name}`, formatSize(fileResult.size)); } } renderFilePreviews(); } function renderFilePreviews(): void { if (pendingFiles.length === 0) { filePreviewEl.style.display = 'none'; filePreviewEl.innerHTML = ''; return; } filePreviewEl.style.display = ''; filePreviewEl.innerHTML = pendingFiles.map((f, i) => `
${getFileIcon(f.name)} ${escapeHtml(f.name)} ${formatSize(f.size)}
`).join(''); } // ══════════════════════════════════════════════ // v4.1 会话管理命令：/retry, /undo, /compress // ══════════════════════════════════════════════ async function handleRetry(): Promise { const currentSession = state.get(KEYS.CURRENT_SESSION); if (!currentSession || currentSession.messages.length < 2) { showToast('没有可重试的消息', 'warning'); return; } if (state.get(KEYS.IS_STREAMING)) { showToast('请先停止当前生成', 'warning'); return; } // 找到最后一条 user 消息 let lastUserIdx = -1; for (let i = currentSession.messages.length - 1; i >= 0; i--) { if (currentSession.messages[i].role === 'user') { lastUserIdx = i; break; } } if (lastUserIdx < 0) { showToast('没有找到用户消息', 'warning'); return; } // 删除该 user 消息之后的所有消息（assistant 回复 + 可能的 tool 调用链） const removedCount = currentSession.messages.length - lastUserIdx - 1; const userMsg = currentSession.messages[lastUserIdx]; currentSession.messages = currentSession.messages.slice(0, lastUserIdx + 1); logInfo(`重试: 删除了 ${removedCount} 条消息，重新发送`); const db = state.get(KEYS.DB); if (db) await db.saveSession(currentSession); clearMessagesDOM(); renderMessages(); // 重新触发 Agent Loop const model = getSelectedModel() || currentSession.model; // 新一轮开始：清理上一轮的工作空间状态 clearToolCardsExternal(); clearTerminalExternal(); appendAssistantPlaceholder(); state.set(KEYS.IS_STREAMING, true); updateSendButton(true); // 始终走 Agent Loop // ── 构建历史消息（不含最后一条用户消息，因为它是重试目标）── const historyMessages = buildHistoryMessages( currentSession.messages .filter(m => m.role === 'user' || m.role === 'assistant') .slice(0, -1), // 不包含刚保留的最后一条 user 消息 30 ); let retryMonitor: ReturnType | null = null; try { let retryContent = ''; let retryThinkContent = ''; let retryIterations = 0; // P1 修复：追踪当前迭代的工具记录（与 send 路径保持一致） let retryIterationToolRecords: ToolCallRecord[] = []; // 构建重试用的完整内容和 images（含文件内容和视频帧） let retryUserContent = userMsg.content || ''; const retryImages: string[] = [...(userMsg.images || [])]; if ((userMsg as any)._fileContents?.length) { const fileParts = buildFileContentParts((userMsg as any)._fileContents); retryUserContent = retryUserContent ? retryUserContent + '\n\n---\n' + fileParts.join('\n\n---\n') : fileParts.join('\n\n---\n'); } if ((userMsg as any)._videoFrames?.length) { retryImages.push(...(userMsg as any)._videoFrames.map((f: any) => f.base64)); } // ── 监控定时器 ── retryMonitor = setInterval(() => { if (!hasActiveCards()) { showWorkingHint('⏳ AI 正在处理中...'); } }, 2000); await runAgentLoop(retryUserContent, retryImages, historyMessages, { onToolCallPrepare: (call) => { addToolCard({ name: call.function.name, arguments: call.function.arguments, result: null, status: 'pending', timestamp: Date.now() }); }, onNewIteration: (toolCalls, stats) => { removeCurrentPlaceholder(); const hasContent = !!retryContent?.trim(); if (hasContent || retryIterations === 0) { const now = Date.now(); const prevMsg: ChatMessage = { role: 'assistant', content: retryContent || '', model: getSelectedModel(), timestamp: now, ...(retryThinkContent && { think: retryThinkContent }), ...(retryIterationToolRecords.length > 0 && { toolCalls: [...retryIterationToolRecords] }), // P0 修复：中间迭代消息携带本轮独立 token 统计 ...(stats?.eval_count && { eval_count: stats.eval_count }), ...(stats?.prompt_eval_count && { prompt_eval_count: stats.prompt_eval_count }), ...(stats?.total_duration && { total_duration: stats.total_duration }), }; retryIterationToolRecords = []; state.update(KEYS.CURRENT_SESSION, (s: any) => ({ ...s, messages: [...s.messages, prevMsg], updatedAt: Date.now() })); renderMessages(); } appendAssistantPlaceholder(); retryContent = ''; retryIterations++; }, onThinking: (thinking) => { retryThinkContent = thinking; updateLastAssistantMessage(retryContent, thinking, null, getSelectedModel()); }, onContent: (content) => { retryContent = content; updateLastAssistantMessage(content, retryThinkContent || null, null, getSelectedModel()); }, onToolCallStart: (call) => { startToolCard({ name: call.function.name, arguments: call.function.arguments, result: null, status: 'running', timestamp: Date.now() }); }, onToolCallResult: (name, result, call) => { updateToolCard({ name, arguments: call.function.arguments, result, status: result.success ? 'success' : 'error', timestamp: Date.now() }); retryIterationToolRecords.push({ name, arguments: call.function.arguments, result, status: result.success ? 'success' : 'error', timestamp: Date.now() }); }, onToolCallError: (name, error, call) => { updateToolCard({ name, arguments: call.function.arguments, result: { success: false, error }, status: 'error', timestamp: Date.now() }); retryIterationToolRecords.push({ name, arguments: call.function.arguments, result: { success: false, error }, status: 'error', timestamp: Date.now() }); }, onConfirmTool: async (call) => showToolConfirm(call), onPlanReady: async (plan: string, steps: string[]) => { // 检查是否启用了自动确认计划 const autoConfirm = state.get('planAutoConfirm', false); if (autoConfirm) { logInfo('Plan Mode: 自动确认计划（用户已启用自动确认）'); return true; } try { const { showHtmlConfirm } = await import('./prompt-modal.js'); const html = buildPlanConfirmHtml(plan, steps); return showHtmlConfirm(html, '📋 执行计划确认', true); } catch (err) { logError('Plan Mode: 弹窗加载失败，拒绝执行以保护用户', (err as Error).message); return false; // 弹窗失败时拒绝执行，避免绕过用户审批 } }, onDone: async (finalContent, _toolRecords, loopStats) => { retryContent = finalContent; // P1 修复：使用本地追踪的当前迭代工具记录，而非引擎的 allToolRecords（含所有迭代，会与中间消息重复） const finalToolRecords = retryIterationToolRecords.length > 0 ? retryIterationToolRecords : undefined; if (retryIterations > 0) { if (finalContent) { const lastMsg: ChatMessage = { role: 'assistant', content: finalContent, model: getSelectedModel(), timestamp: Date.now(), ...(retryThinkContent && { think: retryThinkContent }), ...(finalToolRecords?.length && { toolCalls: finalToolRecords }), ...(loopStats?.eval_count && { eval_count: loopStats.eval_count }), ...(loopStats?.prompt_eval_count && { prompt_eval_count: loopStats.prompt_eval_count }), ...(loopStats?.total_duration && { total_duration: loopStats.total_duration }), }; state.update(KEYS.CURRENT_SESSION, (s: any) => ({ ...s, messages: [...s.messages, lastMsg], updatedAt: Date.now() })); } } else { const assistantMsg: ChatMessage = { role: 'assistant', content: finalContent || '', model: getSelectedModel(), timestamp: Date.now(), ...(retryThinkContent && { think: retryThinkContent }), ...(finalToolRecords?.length && { toolCalls: finalToolRecords }), ...(loopStats?.eval_count && { eval_count: loopStats.eval_count }), ...(loopStats?.prompt_eval_count && { prompt_eval_count: loopStats.prompt_eval_count }), ...(loopStats?.total_duration && { total_duration: loopStats.total_duration }), }; state.update(KEYS.CURRENT_SESSION, (s: any) => ({ ...s, messages: [...s.messages, assistantMsg], updatedAt: Date.now() })); } // renderMessages 统一渲染 session 数据，先清掉流式 placeholder 避免 DOM 重复 clearMessagesDOM(); renderMessages(); // ── 消费 _lastPlanStatus（不再保存为 session 消息，避免 system 消息渲染到聊天界面）── state.set('_lastPlanStatus', ''); await saveCurrentSession(); } }); } catch (err) { logError('重试失败', (err as Error).message); appendSystemMessage(`❌ 重试失败: ${(err as Error).message}`); } finally { if (retryMonitor) clearInterval(retryMonitor as unknown as number); clearWorkingHint(); } state.set(KEYS.IS_STREAMING, false); updateSendButton(false); } async function handleUndo(): Promise { const currentSession = state.get(KEYS.CURRENT_SESSION); if (!currentSession || currentSession.messages.length < 2) { showToast('没有可撤销的消息', 'warning'); return; } if (state.get(KEYS.IS_STREAMING)) { showToast('请先停止当前生成', 'warning'); return; } // 从末尾删除：先删 assistant（含可能的 tool 消息），再删 user const msgs = currentSession.messages; let removeEnd = msgs.length; let removeStart = msgs.length - 1; // 找到最后一条 user 消息的位置 while (removeStart >= 0 && msgs[removeStart].role !== 'user') { removeStart--; } if (removeStart < 0) { showToast('没有找到用户消息', 'warning'); return; } // 删除 user 消息及之后的所有消息 const removedCount = removeEnd - removeStart; currentSession.messages = msgs.slice(0, removeStart); logInfo(`撤销: 删除了 ${removedCount} 条消息`); const db = state.get(KEYS.DB); if (db) await db.saveSession(currentSession); clearMessagesDOM(); renderMessages(); showToast('已撤销上一轮对话', 'success'); } async function handleCompress(): Promise { const currentSession = state.get(KEYS.CURRENT_SESSION); if (!currentSession || currentSession.messages.length < 6) { showToast('消息太少，无需压缩', 'warning'); return; } if (state.get(KEYS.IS_STREAMING)) { showToast('请先停止当前生成', 'warning'); return; } const api = state.get(KEYS.API); const model = getSelectedModel() || currentSession.model; if (!api || !model) { showToast('请先选择模型', 'warning'); return; } logInfo('上下文压缩: 开始'); // 取中间的消息做摘要（保留首尾各 2 条，跳过已压缩的） const messages = currentSession.messages; const keepStart = 2; const keepEnd = 2; const head = messages.slice(0, keepStart); const tail = messages.slice(-keepEnd); const middle = messages.slice(keepStart, messages.length - keepEnd); // 过滤掉已压缩的消息，避免重复压缩 const uncompressedMiddle = middle.filter(m => !m.compressed); if (uncompressedMiddle.length === 0) { showToast('中间消息已全部压缩，无需再次压缩', 'info'); return; } const conversationText = uncompressedMiddle.map(m => { const role = m.role === 'user' ? '用户' : 'AI'; let content = m.content || ''; if (content.length > 500) content = content.slice(0, 500) + '...'; if (m.toolCalls?.length) content += ` [工具调用: ${m.toolCalls.map(t => t.name).join(', ')}]`; return `${role}: ${content}`; }).join('\n'); showToast('正在压缩上下文...', 'info'); try { let summary = ''; await api.chatStream({ model, messages: [{ role: 'user', content: `请将以下对话摘要为一段简洁的上下文总结。保留关键信息：讨论的主题、得出的结论、用户的偏好、未完成的任务、关键工具调用结果。用中文输出，不超过 500 字。\n\n对话记录：\n${conversationText}` }], stream: true, think: false, options: { num_ctx: 8192, temperature: 0.3 } } as any, (chunk: OllamaStreamChunk) => { if (chunk.message?.content) { summary += chunk.message.content; } }); if (!summary.trim()) { showToast('压缩失败：模型未返回内容', 'error'); return; } // 构建压缩后的摘要消息（标记 compressed） const summaryMsg: ChatMessage = { role: 'system', content: `📋 以下是对之前对话的摘要（已压缩 ${uncompressedMiddle.length} 条消息）：\n\n${summary}`, timestamp: Date.now(), compressed: true }; // 保留已压缩的中间消息 + 新摘要 const alreadyCompressed = middle.filter(m => m.compressed); currentSession.messages = [...head, ...alreadyCompressed, summaryMsg, ...tail]; const db = state.get(KEYS.DB); if (db) await db.saveSession(currentSession); clearMessagesDOM(); renderMessages(); logInfo(`上下文压缩完成: ${uncompressedMiddle.length} 条 → 1 条摘要`); showToast(`已压缩 ${uncompressedMiddle.length} 条消息为摘要`, 'success'); } catch (err) { logError('上下文压缩失败', (err as Error).message); showToast(`压缩失败: ${(err as Error).message}`, 'error'); } } function updateSendButton(streaming: boolean): void { if (streaming) { btnSendEl.innerHTML = ``; btnSendEl.classList.add('stop-btn'); btnSendEl.classList.remove('disabled'); btnSendEl.title = '停止生成'; } else { btnSendEl.innerHTML = ``; btnSendEl.classList.remove('stop-btn'); btnSendEl.classList.remove('disabled'); btnSendEl.title = '发送'; } } function stopGeneration(): void { const abortController = state.get(KEYS.ABORT_CONTROLLER); if (abortController) abortController.abort(); } /** Token 预算比例：文件内容最多占用上下文窗口的比例 */ const FILE_TOKEN_BUDGET_RATIO = 0.3; // ── 注释剥离 ── interface CommentRule { single: RegExp | null; multi: [RegExp, string] | null } const COMMENT_RULES: Record = { javascript: { single: /\/\/.*$/gm, multi: [/\/\*[\s\S]*?\*\//g, ''] }, typescript: { single: /\/\/.*$/gm, multi: [/\/\*[\s\S]*?\*\//g, ''] }, tsx: { single: /\/\/.*$/gm, multi: [/\/\*[\s\S]*?\*\//g, ''] }, jsx: { single: /\/\/.*$/gm, multi: [/\/\*[\s\S]*?\*\//g, ''] }, java: { single: /\/\/.*$/gm, multi: [/\/\*[\s\S]*?\*\//g, ''] }, c: { single: /\/\/.*$/gm, multi: [/\/\*[\s\S]*?\*\//g, ''] }, cpp: { single: /\/\/.*$/gm, multi: [/\/\*[\s\S]*?\*\//g, ''] }, csharp: { single: /\/\/.*$/gm, multi: [/\/\*[\s\S]*?\*\//g, ''] }, go: { single: /\/\/.*$/gm, multi: [/\/\*[\s\S]*?\*\//g, ''] }, rust: { single: /\/\/.*$/gm, multi: [/\/\*[\s\S]*?\*\//g, ''] }, swift: { single: /\/\/.*$/gm, multi: [/\/\*[\s\S]*?\*\//g, ''] }, kotlin: { single: /\/\/.*$/gm, multi: [/\/\*[\s\S]*?\*\//g, ''] }, scala: { single: /\/\/.*$/gm, multi: [/\/\*[\s\S]*?\*\//g, ''] }, css: { single: null, multi: [/\/\*[\s\S]*?\*\//g, ''] }, python: { single: /#.*$/gm, multi: [/('''[\s\S]*?'''|\"\"\"[\s\S]*?\"\"\")/g, ''] }, ruby: { single: /#.*$/gm, multi: null }, perl: { single: /#.*$/gm, multi: null }, shell: { single: /#.*$/gm, multi: null }, bash: { single: /#.*$/gm, multi: null }, r: { single: /#.*$/gm, multi: null }, yaml: { single: /#.*$/gm, multi: null }, sql: { single: /--.*$/gm, multi: [/\/\*[\s\S]*?\*\//g, ''] }, lua: { single: /--.*$/gm, multi: [/--\[\[[\s\S]*?\]\]/g, ''] }, haskell: { single: /--.*$/gm, multi: [/\{-[\s\S]*?-\}/g, ''] }, html: { single: null, multi: [//g, ''] }, xml: { single: null, multi: [//g, ''] }, markdown: { single: null, multi: [//g, ''] }, }; function stripComments(content: string, language: string): string { const rule = COMMENT_RULES[language]; if (!rule) return content; let result = content; if (rule.single) result = result.replace(rule.single, ''); if (rule.multi) result = result.replace(rule.multi[0], rule.multi[1]); // 压缩连续空白行：>2个连续换行 → 2个换行 result = result.replace(/\n{3,}/g, '\n\n'); return result; } /** * 构建文件内容 Parts，带 token 预算感知 * - 小文件直接嵌入，大文件按预算截断 * - 代码文件自动剥离注释 */ /** 构建 Plan Mode 确认弹窗 HTML */ function buildPlanConfirmHtml(plan: string, steps: string[]): string { const renderedPlan = safeMarkdown(plan.slice(0, 2000)); const truncated = plan.length > 2000; const stepsHtml = steps.length > 0 ? `
📌 执行步骤（共 ${steps.length} 步）
${steps.map((s, i) => `
${i + 1} ${escapeHtml(s)}
`).join('')}
` : ''; return `
🤖 AI 执行计划 批准后 AI 将按步骤依次执行，每完成一步自动标记进度
${renderedPlan} ${truncated ? '
…（计划内容已截断）
' : ''}
${stepsHtml}
`; } function buildFileContentParts(fileContents: Array<{ name: string; language: string; content: string }>): string[] { const numCtx = state.get(KEYS.NUM_CTX, 24576); const fileBudget = Math.floor(numCtx * FILE_TOKEN_BUDGET_RATIO); let usedTokens = 0; const maxSingleFileTokens = Math.floor(fileBudget * 0.5); return fileContents.map((f, i) => { const lang = f.language || 'text'; let text = f.content; const rawTokens = estimateTokens(text); // 小文件直接使用，大文件剥离注释后截断 if (rawTokens > 500) { const stripped = stripComments(text, lang); const strippedTokens = estimateTokens(stripped); if (strippedTokens > maxSingleFileTokens) { const ratio = Math.min(1, maxSingleFileTokens / strippedTokens); text = stripped.slice(0, Math.floor(stripped.length * ratio)); } else { text = stripped; } } const finalTokens = estimateTokens(text); if (usedTokens + finalTokens > fileBudget) return ''; usedTokens += finalTokens; return JSON.stringify({ file_name: f.name, file_type: lang, context: text, }); }).filter(Boolean); } function buildApiMessages(messages: ChatMessage[]): Array<{ role: string; content: string; images?: string[] }> { return messages.map(m => { let content = m.content || ''; if (m._fileContents && m._fileContents.length > 0) { const fileParts = buildFileContentParts(m._fileContents); if (content) { content += '\n\n---\n' + fileParts.join('\n\n---\n'); } else { const count = m._fileContents.length; content = `请分析以下 ${count > 1 ? count + ' 个' : ''}文件：\n\n${fileParts.join('\n\n---\n')}`; } } return { role: m.role, content, ...(m.images && { images: m.images }) }; }); } /** * 从会话消息构建 Ollama 格式的历史消息列表。 * 注入 assistant + user（含 _apiContent）+ tool_calls + role:'tool' 结果。 * 不注入 system 消息——保证传给 API 时始终只有 1 条 system 且排在首位。 */ function buildHistoryMessages(msgs: ChatMessage[], maxCount = 20): OllamaMessage[] { const result: OllamaMessage[] = []; for (const msg of msgs) { if (msg.role !== 'assistant' && msg.role !== 'user') continue; if (result.length >= maxCount) break; if (msg.role === 'user') { // 用 _apiContent（含附件 JSON 结构化数据），没有则回退 content const content = (msg as any)._apiContent || msg.content || ''; result.push({ role: 'user', content, ...(msg.images?.length && { images: msg.images }) }); continue; } if (msg.role === 'assistant') { const content = msg.content || ''; const assistantMsg: OllamaMessage = { role: 'assistant', content, ...(msg.think && { thinking: msg.think }), ...(msg.images?.length && { images: msg.images }), }; // ── 注入 tool_calls（Ollama 格式）── if (msg.toolCalls?.length) { assistantMsg.tool_calls = msg.toolCalls.map(tc => ({ type: 'function' as const, function: { name: tc.name, arguments: tc.arguments, }, })); } result.push(assistantMsg); // ── 注入 tool 结果消息（role: 'tool'）── if (msg.toolCalls?.length) { for (const tc of msg.toolCalls) { if (!tc.result) continue; if (result.length >= maxCount) break; const resultContent = tc.status === 'success' ? JSON.stringify(tc.result) : JSON.stringify({ success: false, error: tc.result.error || '工具执行失败' }); result.push({ role: 'tool', tool_name: tc.name, content: resultContent, }); } } } } // 超过 maxCount 从尾部截取 if (result.length > maxCount) { return result.slice(-maxCount); } return result; } export async function sendMessage(): Promise { const text = chatInputEl.value.trim(); // ── v4.1 会话管理命令 ── if (text === '/retry') { chatInputEl.value = ''; await handleRetry(); return; } if (text === '/undo') { chatInputEl.value = ''; await handleUndo(); return; } if (text === '/compress') { chatInputEl.value = ''; await handleCompress(); return; } if (!text && pendingImages.length === 0 && pendingFiles.length === 0) return; if (state.get(KEYS.IS_STREAMING)) return; const model = getSelectedModel(); if (!model) { appendSystemMessage('⚠️ 请先选择一个模型'); return; } logInfo(`发送消息: ${model}`, text ? text.slice(0, 60) + (text.length > 60 ? '...' : '') : '[附件消息]'); resetAutoScroll(); // 发送新消息时强制恢复自动滚动 if (pendingImages.length > 0 && !isVisionAvailable()) { appendSystemMessage('⚠️ 当前模型不支持图片分析，请移除图片后重试'); return; } const currentSession = state.get(KEYS.CURRENT_SESSION); if (!currentSession) return; // 所有消息统一走 Agent Loop await sendMessageWithAgentLoop(text, currentSession, model); return; } /** 更新当前会话消息中最近一个 'running' 状态的同名工具记录 */ function updateMessageToolRecord(toolName: string, status: 'success' | 'error', result: Record): void { // 工具卡片统一由 onDone 挂载，不在执行中更新消息记录 } async function sendMessageWithAgentLoop(text: string, currentSession: ChatSession, model: string): Promise { const now = Date.now(); const msgsToAdd: ChatMessage[] = []; const userFiles: ChatFile[] = pendingFiles.map(f => ({ name: f.name, language: f.language, size: f.size })); // 合并图片和视频帧为统一的 images 数组 const images: string[] = [ ...pendingImages.map(img => img.base64), ...pendingVideoFrames.map(f => f.base64), ...completedVideos.flatMap(v => v.frames.map(f => f.base64)) ]; let apiContentForModel = text || ''; const hasVideos = pendingVideoFrames.length > 0 || completedVideos.length > 0; const hasAttachments = pendingImages.length > 0 || hasVideos || userFiles.length > 0; if (hasAttachments) { // ── 聊天卡片：只显示用户自己的文字 + 附件视觉预览（缩略图/文件卡片/视频帧），不含代码生成的文本标记 ── const displayContent = text || ''; // ── Ollama API 用：用户文字消息在最前，附件 JSON 结构化数据在后 ── const apiParts: string[] = []; // 用户文字始终放最前 if (text) apiParts.push(text); if (userFiles.length > 0) { const fileContents = pendingFiles.map(f => ({ name: f.name, language: f.language, content: f.content })); const fileParts = buildFileContentParts(fileContents); apiParts.push(...fileParts); logInfo('📄 文件内容已注入', `${fileParts.length} 段, 共 ${fileContents.reduce((s, f) => s + f.content.length, 0)} 字符`); } // 视频：帧图片通过 images[] 传递，元数据用 JSON 结构化（与文件格式一致） let allVids: Array<{ name: string; count: number; dur: number; frames: Array<{ timestampSeconds: number; name: string; base64: string }> }> = []; if (hasVideos) { if (pendingVideoMeta && pendingVideoFrames.length > 0) { allVids.push({ name: pendingVideoMeta.fileName, count: pendingVideoFrames.length, dur: pendingVideoMeta.duration, frames: pendingVideoFrames }); } for (const v of completedVideos) { allVids.push({ name: v.fileName, count: v.frames.length, dur: v.duration, frames: v.frames }); } // 计算每个视频在全局 images[] 中的起始索引（跳过用户上传的图片） let imageStart = pendingImages.length; for (const v of allVids) { apiParts.push(JSON.stringify({ file_name: v.name, file_type: v.name.split('.').pop() || 'video', context: JSON.stringify({ frame_count: v.count, fps: 1, duration: Math.round(v.dur), image_start: imageStart, image_end: imageStart + v.count - 1, frames: v.frames.map((f, fi) => ({ timestamp: f.timestampSeconds, image_index: imageStart + fi, })), }), })); imageStart += v.count; } } apiContentForModel = apiParts.join('\n\n'); const msg: ChatMessage = { role: 'user', content: displayContent, images: [...pendingImages.map(img => img.base64)], _apiContent: apiContentForModel, // 给模型的完整内容，不显示给用户 timestamp: now }; if (allVids.length > 0) { const totalFrames = allVids.flatMap(v => v.frames); msg._videoFrames = totalFrames; msg._videos = allVids.map(v => ({ fileName: v.name, frameCount: v.count, duration: v.dur })); } if (userFiles.length > 0) { msg.files = userFiles; msg._fileContents = pendingFiles.map(f => ({ name: f.name, language: f.language, content: f.content })); } msgsToAdd.push(msg); } else if (text || pendingFiles.length > 0) { // 纯文本/纯文件消息（无图片、视频附件） const msg: ChatMessage = { role: 'user', content: text || '', timestamp: now }; if (userFiles.length > 0) { msg.files = userFiles; msg._fileContents = pendingFiles.map(f => ({ name: f.name, language: f.language, content: f.content })); } msgsToAdd.push(msg); } const isFirstMsg = currentSession.messages.length === 0; state.update(KEYS.CURRENT_SESSION, (session: any) => ({ ...session, ...(isFirstMsg && { title: truncate(text || (pendingFiles.length > 0 ? `[文件: ${pendingFiles.map(f => f.name).join(', ')}]` : pendingVideoMeta ? '[视频消息]' : '[图片消息]'), 30), model }), messages: [...session.messages, ...msgsToAdd], updatedAt: Date.now() })); const freshSession = state.get(KEYS.CURRENT_SESSION); enableAutoScroll(); renderMessages(); await saveCurrentSession(); chatInputEl.value = ''; pendingImages = []; pendingFiles = []; pendingVideoFrames = []; pendingVideoMeta = null; completedVideos = []; imagePreviewEl.style.display = 'none'; imagePreviewEl.innerHTML = ''; filePreviewEl.style.display = 'none'; filePreviewEl.innerHTML = ''; videoPreviewEl.style.display = 'none'; videoPreviewEl.innerHTML = ''; autoResizeTextarea(); // 新一轮开始：清理上一轮的工作空间状态 clearToolCardsExternal(); clearTerminalExternal(); enableAutoScroll(); // renderMessages 的 DOM 操作可能触发 scroll 事件导致 autoScroll=false appendAssistantPlaceholder(); state.set(KEYS.IS_STREAMING, true); updateSendButton(true); // 构建历史消息 // ── 构建历史消息（含工具调用和结果，消除跨 Loop 上下文断裂）── const historyMessages = buildHistoryMessages( freshSession.messages.filter(m => m.role === 'user' || m.role === 'assistant'), 30 // 预留空间给 tool 结果消息，实际有效轮次仍约 20 轮 ); let assistantContent = ''; let thinkContent = ''; // P1 修复：追踪当前迭代的工具记录，onNewIteration 时保存到消息中 let currentIterationToolRecords: ToolCallRecord[] = []; state.set('_currentEvalCount', 0); // ── 监控定时器：流式输出期间持续显示工作提示 ── // 模型生成大参数 tool_call 可能需要几分钟，期间工作空间显示提示 const monitorInterval = setInterval(() => { if (!hasActiveCards()) { showWorkingHint('⏳ AI 正在处理中...'); } }, 2000); try { await runAgentLoop(apiContentForModel || (pendingFiles.length > 0 ? `请分析 ${pendingFiles.map(f => f.name).join(', ')}` : ''), images, historyMessages, { /** 模型正在生成 tool_call 参数中 — 提前在工作空间显示"准备中"卡片 */ onToolCallPrepare: (call) => { addToolCard({ name: call.function.name, arguments: call.function.arguments, result: null, status: 'pending', timestamp: Date.now() }); }, onNewIteration: (toolCalls, stats) => { // 保存上一轮的卡片（含工具记录） const prevMsg: ChatMessage = { role: 'assistant', content: assistantContent || '', model: getSelectedModel(), timestamp: Date.now(), ...(thinkContent && { think: thinkContent }), ...(currentIterationToolRecords.length > 0 && { toolCalls: [...currentIterationToolRecords] }), // P0 修复：中间迭代消息携带本轮独立 token 统计 ...(stats?.eval_count && { eval_count: stats.eval_count }), ...(stats?.prompt_eval_count && { prompt_eval_count: stats.prompt_eval_count }), ...(stats?.total_duration && { total_duration: stats.total_duration }), }; currentIterationToolRecords = []; state.update(KEYS.CURRENT_SESSION, (session: any) => ({ ...session, messages: [...session.messages, prevMsg], updatedAt: Date.now() })); removeCurrentPlaceholder(); renderMessages(); appendAssistantPlaceholder(); assistantContent = ''; thinkContent = ''; }, onThinking: (thinking) => { thinkContent = thinking; updateLastAssistantMessage(assistantContent, thinkContent || null, null, getSelectedModel()); }, onContent: (content) => { assistantContent = content; updateLastAssistantMessage(assistantContent, thinkContent || null, null, getSelectedModel()); }, onToolCallStart: (call) => { startToolCard({ name: call.function.name, arguments: call.function.arguments, result: null, status: 'running', timestamp: Date.now() }); }, onToolCallResult: (name, result, call) => { updateToolCard({ name, arguments: call.function.arguments, result, status: result.success ? 'success' : 'error', timestamp: Date.now() }); currentIterationToolRecords.push({ name, arguments: call.function.arguments, result, status: result.success ? 'success' : 'error', timestamp: Date.now() }); }, onToolCallError: (name, error, call) => { updateToolCard({ name, arguments: call.function.arguments, result: { success: false, error }, status: 'error', timestamp: Date.now() }); currentIterationToolRecords.push({ name, arguments: call.function.arguments, result: { success: false, error }, status: 'error', timestamp: Date.now() }); }, onConfirmTool: async (call) => { return showToolConfirm(call); }, /** Plan Mode — 计划生成后等待用户确认（Markdown 渲染） */ onPlanReady: async (plan: string, steps: string[]) => { // 检查是否启用了自动确认计划 const autoConfirm = state.get('planAutoConfirm', false); if (autoConfirm) { logInfo('Plan Mode: 自动确认计划（用户已启用自动确认）'); return true; } try { const { showHtmlConfirm } = await import('./prompt-modal.js'); const html = buildPlanConfirmHtml(plan, steps); const approved = await showHtmlConfirm(html, '📋 执行计划确认', true); if (approved) { logInfo('Plan Mode: 用户批准计划'); } return approved; } catch (err) { logError('Plan Mode: 弹窗加载失败，拒绝执行以保护用户', (err as Error).message); return false; // 弹窗失败时拒绝执行，避免绕过用户审批 } }, onDone: async (finalContent, _toolRecords, loopStats) => { // P1 修复：使用本地追踪的当前迭代工具记录，而非引擎的 allToolRecords（含所有迭代，会与中间消息重复） const finalToolRecords = currentIterationToolRecords.length > 0 ? currentIterationToolRecords : undefined; const assistantMsg: ChatMessage = { role: 'assistant', content: finalContent || '', model: getSelectedModel(), timestamp: Date.now(), ...(thinkContent && { think: thinkContent }), ...(finalToolRecords?.length && { toolCalls: finalToolRecords }), ...(loopStats?.eval_count && { eval_count: loopStats.eval_count }), ...(loopStats?.prompt_eval_count && { prompt_eval_count: loopStats.prompt_eval_count }), ...(loopStats?.total_duration && { total_duration: loopStats.total_duration }), }; state.update(KEYS.CURRENT_SESSION, (session: any) => ({ ...session, messages: [...session.messages, assistantMsg], updatedAt: Date.now() })); // 先移除流式 placeholder 再 renderMessages，避免未渲染(纯文本)和已渲染(markdown)两个卡片同时存在 removeCurrentPlaceholder(); renderMessages(); // ── 消费 _lastPlanStatus（不再保存为 session 消息，避免 system 消息渲染到聊天界面）── state.set('_lastPlanStatus', ''); await saveCurrentSession(); } }); } catch (err) { logError('Agent Loop 错误', (err as Error).message); if ((err as Error).name === 'AbortError') { const placeholder = document.querySelector('#messagesContainer .message.assistant.loading') as HTMLElement; if (placeholder) { placeholder.classList.remove('loading'); placeholder.querySelector('.loading-dots')?.remove(); placeholder.querySelector('.loading-text')?.remove(); const contentDiv = placeholder.querySelector('.msg-content'); if (contentDiv) { let html = assistantContent ? safeMarkdown(assistantContent) : ''; html += '

[已停止]

'; contentDiv.innerHTML = html; } } // 获取中止时已执行的工具记录 const abortToolRecords = state.get('_abortToolRecords', null); if (abortToolRecords) state.set('_abortToolRecords', null); const partialMsg: ChatMessage = { role: 'assistant', content: assistantContent, timestamp: Date.now(), ...(thinkContent && { think: thinkContent }), ...(abortToolRecords?.length && { toolCalls: abortToolRecords }), stopped: true }; state.update(KEYS.CURRENT_SESSION, (session: any) => ({ ...session, messages: [...session.messages, partialMsg], updatedAt: Date.now() })); appendSystemMessage('⏹ 已停止生成'); await saveCurrentSession(); } else { const placeholder = document.querySelector('#messagesContainer .message.assistant.loading') as HTMLElement; if (placeholder) { placeholder.classList.remove('loading'); placeholder.querySelector('.loading-dots')?.remove(); placeholder.querySelector('.loading-text')?.remove(); if (!assistantContent) { placeholder.remove(); resetCurrentPlaceholder(); } else { const contentDiv = placeholder.querySelector('.msg-content'); if (contentDiv) contentDiv.innerHTML = safeMarkdown(assistantContent) + '

[已中断]

'; } } let errMsg = `❌ 错误: ${(err as Error).message}`; if ((err as Error).message.includes('Failed to fetch') || (err as Error).message.includes('NetworkError')) { errMsg = '❌ 连接失败。请检查 Ollama 是否正在运行'; } appendSystemMessage(errMsg); } } finally { clearInterval(monitorInterval); clearWorkingHint(); state.set(KEYS.IS_STREAMING, false); updateSendButton(false); state.set(KEYS.ABORT_CONTROLLER, null); } } async function saveCurrentSession(): Promise { const db = state.get(KEYS.DB); const currentSession = state.get(KEYS.CURRENT_SESSION); const isHistoryView = state.get(KEYS.IS_HISTORY_VIEW); if (!currentSession || !db || isHistoryView) return; currentSession.updatedAt = Date.now(); try { await db.saveSession(currentSession); } catch (err) { logError('保存会话失败', (err as Error).message); } }