@rexvane/piweb 0.3.9 → 0.3.11

package/src/components/SettingsPanel.tsx

@@ -35,7 +35,8 @@ import { useI18n } from "@/i18n";
 import { applyPebrelTheme, loadPebrelTheme, loadThemeMode, type PebrelTheme, type ThemeMode } from "@/lib/theme";
 import { modelDraftFromConfig, serializeProviderDraft, validateModelDrafts, type ModelDraft } from "@/lib/model-draft";
 import { customApiOptions } from "@/lib/provider-display";
-import type { ProviderUsage, ProviderView } from "@/lib/models-service";
+import { quotaWindowKind, supportsUsageProbe, type BalanceAmount, type BalanceDetailKey, type ProviderUsage, type QuotaWindow, type XaiUsage } from "@/lib/provider-usage";
+import type { ProviderView } from "@/lib/models-service";
 import type { ToolPreset } from "@/lib/types";
 
 type Section = "general" | "models" | "tools" | "skills" | "plugins";
@@ -886,7 +887,7 @@ function ModelsSection() {
 	}, [oauthProvider, load]);
 
 	// xAI 订阅配额（x-ratelimit-* 响应头；后端 5 分钟缓存，force 跳过）
-	const [xaiUsage, setXaiUsage] = useState<ProviderUsage | null>(null);
+	const [xaiUsage, setXaiUsage] = useState<XaiUsage | null>(null);
 	const [usageBusy, setUsageBusy] = useState(false);
 	const [usageError, setUsageError] = useState(false);
 	const fetchUsage = useCallback(async (force: boolean) => {
@@ -904,6 +905,106 @@ function ModelsSection() {
 		}
 	}, []);
 
+	// 窗口制/余额制配额：各提供商走各自的只读端点（后端 5 分钟缓存，force 跳过）
+	const [providerUsages, setProviderUsages] = useState<Record<string, ProviderUsage | null>>({});
+	const [usageBusyId, setUsageBusyId] = useState<string | null>(null);
+	const [usageFailedIds, setUsageFailedIds] = useState<Record<string, boolean>>({});
+	const fetchProviderUsage = useCallback(async (providerId: string, force: boolean) => {
+		setUsageBusyId(providerId);
+		setUsageFailedIds((prev) => ({ ...prev, [providerId]: false }));
+		try {
+			const r = await fetch("/api/models", { method: "POST", headers: { "Content-Type": "application/json" }, body: JSON.stringify({ action: "providerUsage", providerId, force }) });
+			const j = await r.json();
+			if (!r.ok || !j.success) throw new Error(j.error || "usage probe failed");
+			setProviderUsages((prev) => ({ ...prev, [providerId]: j.data.usage ?? null }));
+		} catch {
+			setUsageFailedIds((prev) => ({ ...prev, [providerId]: true }));
+		} finally {
+			setUsageBusyId((current) => (current === providerId ? null : current));
+		}
+	}, []);
+
+	// 这些端点都是只读探测、不消耗额度，登录后自动取一次；xAI 那条会发一次可能计费的请求，仍然只手动触发
+	const requestedUsage = useRef(new Set<string>());
+	useEffect(() => {
+		for (const p of providers) {
+			if (!p.authReady || !supportsUsageProbe(p.id) || p.id === "xai" || requestedUsage.current.has(p.id)) continue;
+			requestedUsage.current.add(p.id);
+			void fetchProviderUsage(p.id, false);
+		}
+	}, [providers, fetchProviderUsage]);
+
+	// 窗口时长 → 标签：7 天叫「周限」、30 天叫「月限」，更短的按小时/天如实标注
+	const quotaWindowLabel = (seconds: number): string => {
+		const kind = quotaWindowKind(seconds);
+		if (kind === "weekly") return t.usageWindowWeekly;
+		if (kind === "monthly") return t.usageWindowMonthly;
+		return (kind === "hours" ? t.usageWindowHours : t.usageWindowDays).replace("{n}", String(Math.round(seconds / (kind === "hours" ? 3600 : 86400))));
+	};
+	// 服务端不给周期时（Copilot 的额度类别、OpenCode 的 rolling）按名字标注
+	const quotaWindowName = (name: string): string => {
+		if (name === "premium") return t.usageWindowPremium;
+		if (name === "chat") return t.usageWindowChat;
+		if (name === "completions") return t.usageWindowCompletions;
+		if (name === "rolling") return t.usageWindowRolling;
+		return titleCase(name);
+	};
+	const quotaVariantLabel = (variant: string): string => {
+		if (variant === "opus") return t.usageVariantOpus;
+		if (variant === "code") return t.usageVariantCode;
+		return titleCase(variant);
+	};
+	const quotaWindowHeading = (w: QuotaWindow): string => {
+		if (w.limitWindowSeconds !== undefined) return `${quotaWindowLabel(w.limitWindowSeconds)}${w.variant ? ` (${quotaVariantLabel(w.variant)})` : ""}`;
+		return w.name ? quotaWindowName(w.name) : "";
+	};
+
+	// 余额明细项的标签由界面翻译，服务端只给语义 key
+	const balanceDetailLabel = (key: BalanceDetailKey): string => {
+		if (key === "granted") return t.usageBalanceGranted;
+		if (key === "toppedUp") return t.usageBalanceToppedUp;
+		if (key === "voucher") return t.usageBalanceVoucher;
+		if (key === "cash") return t.usageBalanceCash;
+		if (key === "used") return t.usageBalanceUsed;
+		return t.usageBalanceLimit;
+	};
+	const fmtAmount = (value: BalanceAmount): string => `${value.currency === "USD" ? "$" : value.currency === "CNY" ? "¥" : ""}${value.amount}${value.currency && value.currency !== "USD" && value.currency !== "CNY" ? ` ${value.currency}` : ""}`;
+
+	// 一个提供商的配额行：窗口制显示每个窗口的剩余与重置，余额制显示主余额与明细
+	const renderProviderUsage = (providerId: string) => {
+		const usage = providerUsages[providerId];
+		const busy = usageBusyId === providerId;
+		const failed = usageFailedIds[providerId];
+		return (
+			<>
+				<div className="flex items-center gap-2">
+					{!usage && <span>{failed ? t.usageProbeFailed : busy ? t.queryingQuota : providerId === "xai" ? t.usageXaiDisclosure : t.usageReadOnlyDisclosure}</span>}
+					{usage?.kind === "quota" && usage.planType && <span>{t.usagePlan} {titleCase(usage.planType)}</span>}
+					{usage?.kind === "balance" && usage.primary && (
+						<span>{usage.primaryKey === "used" ? t.usageBalanceUsed : t.usageQuotaRemaining} <span style={{ color: "var(--dsw-label-secondary)" }}>{fmtAmount(usage.primary)}</span></span>
+					)}
+					<button type="button" className="pw-chip" disabled={busy} title={providerId === "xai" ? t.usageXaiDisclosure : t.usageReadOnlyDisclosure} onClick={() => void fetchProviderUsage(providerId, true)}>
+						{busy ? t.usageProbing : usage ? t.usageRefresh : t.usageProbe}
+					</button>
+				</div>
+				{usage?.kind === "balance" && usage.details.length > 0 && (
+					<span>{usage.details.map((detail) => `${balanceDetailLabel(detail.key)} ${fmtAmount(detail)}`).join(" · ")}</span>
+				)}
+				{usage?.kind === "quota" && usage.windows.length === 0 && <span>{t.usageQuotaRemaining} —</span>}
+				{usage?.kind === "quota" && usage.windows.map((w, index) => (
+					<div key={`${w.limitWindowSeconds ?? w.name ?? "window"}-${w.variant ?? ""}-${index}`} className="flex items-center gap-2">
+						<span style={{ width: 84, flex: "none" }}>{quotaWindowHeading(w)}</span>
+						<span style={{ width: 92, flex: "none" }}>{t.usageQuotaRemaining} {w.remainingPercent}%</span>
+						<span>{t.usageReset} {fmtQuotaReset(w.resetAt)}</span>
+						<div className="h-1 overflow-hidden rounded-full" style={{ width: 120, flex: "none", background: "var(--dsw-border-l2)" }}>
+							<div className="h-full rounded-full" style={{ width: `${w.remainingPercent}%`, background: quotaToneColor(w.remainingPercent) }} />
+						</div>
+					</div>
+				))}
+			</>
+		);
+	};
+
 	const notify = (ok: boolean, msg?: string) => {
 		setToast({ ok, msg: msg ?? (ok ? t.toastSaved : t.toastError) });
 		setTimeout(() => setToast(null), 2400);
@@ -1201,6 +1302,12 @@ function ModelsSection() {
 								</div>
 							)}
 
+							{supportsUsageProbe(p.id) && p.id !== "xai" && p.authReady && (
+								<div className="mt-1.5 flex flex-col gap-1.5 px-4" style={{ fontSize: 11.5, color: "var(--dsw-label-tertiary)" }}>
+									{renderProviderUsage(p.id)}
+								</div>
+							)}
+
 							{editingThis && (
 								<div className="mt-2 rounded-2xl p-4" style={{ border: "0.5px solid var(--dsw-border-l2)" }}>
 									{isCustom ? (
@@ -1501,6 +1608,24 @@ function fmtQuotaTokens(value: number): string {
 	return value.toLocaleString();
 }
 
+/** 套餐值来自服务端声明（如 "pro"/"plus"），首字母大写后直接跟在标签后显示。 */
+function titleCase(value: string): string {
+	return value.length === 0 ? value : value[0].toUpperCase() + value.slice(1);
+}
+
+function fmtQuotaReset(value: string | undefined): string {
+	if (!value) return "—";
+	const date = new Date(value);
+	return Number.isNaN(date.getTime()) ? "—" : date.toLocaleString();
+}
+
+/** 剩余越少越警示：<30% 红、≤60% 黄、其余常规。 */
+function quotaToneColor(percent: number): string {
+	if (percent < 30) return "var(--dsw-danger)";
+	if (percent <= 60) return "var(--dsw-warning)";
+	return "var(--dsw-success)";
+}
+
 function Field({ label, children }: { label: string; children: React.ReactNode }) {
 	return (
 		<div>

package/src/hooks/usePiWeb.ts

@@ -4,6 +4,7 @@
  * 中央状态：会话列表轮询、当前会话 SSE 订阅、事件折叠（快照→增量）、命令发送。
  */
 import { useCallback, useEffect, useMemo, useRef, useState } from "react";
+import { createEventBatcher, type EventBatcher } from "@/lib/event-batcher";
 import type {
 	GrowthStep,
 	SessionSummary,
@@ -242,9 +243,15 @@ export const foldPiWebEvent = (state: PiWebState, evt: WebEvent): PiWebState =>
 		case "growth_error":
 			s.growth = { ...s.growth, pending: [], error: evt.message };
 			return s;
+		case "retry":
+			// 只保留最后一次尝试：重试过程在界面上是「重试 3/5」这一条，而不是五行报错
+			s.retryNotice = evt.maxAttempts
+				? `重试 ${evt.attempt}/${evt.maxAttempts}${evt.message ? `：${evt.message}` : ""}`
+				: `重试 ${evt.attempt}${evt.message ? `：${evt.message}` : ""}`;
+			return s;
 		case "error":
-			// 自动重试属流程内通知：随消息流显示、流结束清除，不走 6 秒错误条
-			if (/^自动重试/.test(evt.message)) s.retryNotice = evt.message;
+			// 旧服务端把自动重试塞在 error 里：同样按通知处理，不占错误条
+			if (/^(自动重试|重试)/.test(evt.message)) s.retryNotice = evt.message;
 			else s.error = evt.message;
 			return s;
 		default:
@@ -328,6 +335,8 @@ export function usePiWeb() {
 	const [currentId, setCurrentId] = useState<string | null>(null);
 	const [currentPath, setCurrentPath] = useState<string | null>(null);
 	const [state, setState] = useState<PiWebState>(emptyState);
+	/** 当前 SSE 连接的增量批量器：按帧合并事件，避免每个 token 一次整树渲染 */
+	const batcherRef = useRef<EventBatcher<WebEvent> | null>(null);
 	const [models, setModels] = useState<{ providers: any[]; models: any[] } | null>(null);
 	const [addedWorkspaces, setAddedWorkspaces] = useState<string[]>([]);
 	const [removedWorkspaces, setRemovedWorkspaces] = useState<string[]>([]);
@@ -631,6 +640,11 @@ export function usePiWeb() {
 					if (esRef.current === null) connect();
 				}, delay);
 			};
+			// 流式事件按帧合并：token 速率再高也只每 16ms 渲染一次（见 event-batcher）
+			const batcher = createEventBatcher<WebEvent>({
+				flush: (events) => setState((s) => events.reduce((acc, evt) => foldPiWebEvent(acc, evt), s)),
+			});
+			batcherRef.current = batcher;
 			es.onmessage = (e) => {
 				let parsed: any;
 				try {
@@ -640,6 +654,8 @@ export function usePiWeb() {
 					return;
 				}
 				if (parsed.type === "snapshot" && parsed.snapshot) {
+					// 快照是权威状态：之前缓冲的增量已经过时，直接丢弃
+					batcher.cancel();
 					retryDelayMs = 2000;
 					const snap = parsed.snapshot as WebSnapshot;
 					setSnapshotEpoch((epoch) => epoch + 1);
@@ -661,13 +677,15 @@ export function usePiWeb() {
 						growth: { ...prev.growth, error: snap.growthError ?? null },
 					}));
 				} else {
-					setState((s) => foldPiWebEvent(s, parsed as WebEvent));
+					batcher.push(parsed as WebEvent);
 				}
 			};
 		};
 		let retryTimer: ReturnType<typeof setTimeout> | undefined;
 		connect();
 		return () => {
+			batcherRef.current?.cancel();
+			batcherRef.current = null;
 			if (retryTimer) clearTimeout(retryTimer);
 			esRef.current?.close();
 			esRef.current = null;
@@ -847,18 +865,16 @@ export function usePiWeb() {
 		presetRestorePromiseRef.current = restore;
 	}, [currentId, snapshotEpoch, state.connected, state.snapshot, sendCommand]);
 
-	// 会话可见性规则：当前会话豁免归档过滤——归档当前会话不关闭、
-	// 主列表保持可见可聊，已归档区也不显示它（取消归档前菜单按真实归档态切换）。
+	// 会话可见性规则：归档即离开工作区列表——包括当前会话（归档后切到相邻会话或空态），
+	// 归档区显示全部已归档会话。
 	const visibleSessions = useMemo(() => {
 		const set = new Set(archivedSessionPaths.map(pathKey));
-		const currentKey = currentPath ? pathKey(currentPath) : null;
-		return sessions.filter((s) => !set.has(pathKey(s.path)) || (currentKey != null && pathKey(s.path) === currentKey));
-	}, [sessions, archivedSessionPaths, currentPath]);
+		return sessions.filter((s) => !set.has(pathKey(s.path)));
+	}, [sessions, archivedSessionPaths]);
 	const archivedSessions = useMemo(() => {
 		const set = new Set(archivedSessionPaths.map(pathKey));
-		const currentKey = currentPath ? pathKey(currentPath) : null;
-		return sessions.filter((s) => set.has(pathKey(s.path)) && !(currentKey != null && pathKey(s.path) === currentKey));
-	}, [sessions, archivedSessionPaths, currentPath]);
+		return sessions.filter((s) => set.has(pathKey(s.path)));
+	}, [sessions, archivedSessionPaths]);
 
 	return {
 		sessions: visibleSessions,

package/src/i18n.tsx

@@ -19,6 +19,10 @@ const zh = {
 	error: "出错",
 	copy: "复制",
 	copyFailed: "复制失败",
+	editMessage: "编辑这条消息",
+	sendEdit: "重新发送",
+	editWhileRunning: "会话正在运行，先停止再编辑",
+	editMessageFailed: "编辑重发失败",
 	export: "导出",
 	settings: "设置",
 	// 侧栏
@@ -206,6 +210,28 @@ const zh = {
 	usageProbing: "检测中…",
 	usageProbeDisclosure: "检测会向 xAI 发送一次可能计费的最小请求。",
 	usageProbeFailed: "配额检测失败，未获得用量信息。",
+	usagePlan: "套餐",
+	usageQuotaRemaining: "额度剩余",
+	queryingQuota: "正在查询额度…",
+	usageReadOnlyDisclosure: "读取提供商的只读用量端点，不消耗额度。",
+	usageXaiDisclosure: "会向 xAI 发出一次 1 token 的请求以读取限流头，会计入用量。",
+	usageReset: "刷新时间",
+	usageWindowWeekly: "周限",
+	usageWindowMonthly: "月限",
+	usageWindowHours: "{n} 小时",
+	usageWindowDays: "{n} 天",
+	usageWindowRolling: "滚动窗口",
+	usageWindowChat: "对话",
+	usageWindowCompletions: "补全",
+	usageWindowPremium: "高级请求",
+	usageVariantOpus: "Opus",
+	usageVariantCode: "代码",
+	usageBalanceGranted: "赠金",
+	usageBalanceToppedUp: "充值",
+	usageBalanceVoucher: "代金券",
+	usageBalanceCash: "现金",
+	usageBalanceUsed: "已用",
+	usageBalanceLimit: "上限",
 	oauthWait: "等待授权…",
 	oauthCancel: "取消登录",
 	noSessions: "暂无会话",
@@ -551,6 +577,10 @@ const en: Dict = {
 	error: "Error",
 	copy: "Copy",
 	copyFailed: "Copy failed",
+	editMessage: "Edit this message",
+	sendEdit: "Send again",
+	editWhileRunning: "This session is running; stop it before editing",
+	editMessageFailed: "Edit and resend failed",
 	export: "Export",
 	settings: "Settings",
 	newChat: "New session",
@@ -737,6 +767,28 @@ const en: Dict = {
 	usageProbing: "Checking…",
 	usageProbeDisclosure: "Checking sends one minimal request to xAI that may be billed.",
 	usageProbeFailed: "Quota check failed; no usage data was returned.",
+	usagePlan: "Plan",
+	usageQuotaRemaining: "Remaining quota",
+	queryingQuota: "Querying quota…",
+	usageReadOnlyDisclosure: "Reads the provider's read-only usage endpoint; consumes no quota.",
+	usageXaiDisclosure: "Sends one 1-token request to xAI to read the rate-limit headers; counts against your usage.",
+	usageReset: "Resets",
+	usageWindowWeekly: "Weekly",
+	usageWindowMonthly: "Monthly",
+	usageWindowHours: "{n}h",
+	usageWindowDays: "{n}d",
+	usageWindowRolling: "Rolling",
+	usageWindowChat: "Chat",
+	usageWindowCompletions: "Completions",
+	usageWindowPremium: "Premium requests",
+	usageVariantOpus: "Opus",
+	usageVariantCode: "Code",
+	usageBalanceGranted: "Granted",
+	usageBalanceToppedUp: "Topped up",
+	usageBalanceVoucher: "Voucher",
+	usageBalanceCash: "Cash",
+	usageBalanceUsed: "Used",
+	usageBalanceLimit: "Limit",
 	oauthWait: "Waiting for authorization…",
 	oauthCancel: "Cancel login",
 	noSessions: "No sessions yet",

package/src/lib/agent-manager.ts

@@ -2,9 +2,11 @@
  * agent-manager：AgentSession 池 + pi 事件 → Web 事件翻译 + SSE 订阅管理。
  * 打开会话即惰性冷启动（createAgentSession），空闲回收，事件先快照后增量。
  */
-import { existsSync } from "node:fs";
+import { existsSync, readFileSync, statSync } from "node:fs";
 import { randomUUID } from "node:crypto";
+import path from "node:path";
 import type { AgentSession, AgentSessionEvent } from "@earendil-works/pi-coding-agent";
+import { generateUnifiedPatch } from "@earendil-works/pi-coding-agent";
 import {
 	getResourceLoader,
 	createSessionResourceLoader,
@@ -28,6 +30,7 @@ import { createExtensionUiBridge, type ExtensionUiBridge } from "./extension-ui"
 import { createGrowthTracker, GROWTH_TRACKER_VERSION, type GrowthTracker } from "./growth-tracker";
 import { userTurnsFromEntries } from "./growth-turns";
 import { TrajLedger, buildTrajectoryFromEntries, toTrajTokens } from "./trajectory";
+import { startArchivedSessionRetention } from "./session-retention";
 import { sanitizeToolOutput } from "./text-sanitize";
 import { BoundaryError } from "./path-security";
 import { estimateTokensOf } from "./process-format";
@@ -182,6 +185,51 @@ function patchOf(details: unknown): string | undefined {
 	return typeof patch === "string" && patch.trim() ? patch.slice(0, 60_000) : undefined;
 }
 
+/**
+ * write 工具不产 patch（SDK 的 write.js 里没有 patch 字段），覆盖已有文件时前端只能把
+ * 全部内容当新增行——删掉的行永远不会标红。这里在工具开始执行（文件还没被写）时读一份
+ * 旧内容，结束时用 SDK 的 generateUnifiedPatch 生成和 edit 同格式的 patch，
+ * 于是「新增绿 / 删除红」在两条写入路径上一致。新文件没有旧内容，仍按新增行展示。
+ */
+const WRITE_BASE_MAX_BYTES = 512 * 1024;
+const writeBases = new Map<string, string>();
+
+function writeArgsOf(args: unknown): { target: string; content: string | undefined } {
+	const a = args as { path?: unknown; file_path?: unknown; content?: unknown } | undefined;
+	const target = typeof a?.path === "string" ? a.path : typeof a?.file_path === "string" ? a.file_path : "";
+	return { target, content: typeof a?.content === "string" ? a.content : undefined };
+}
+
+export function captureWriteBase(cwd: string, sessionPath: string, toolCallId: string, args: unknown): void {
+	const { target } = writeArgsOf(args);
+	if (!target) return;
+	try {
+		const abs = path.isAbsolute(target) ? target : path.resolve(cwd, target);
+		const stat = statSync(abs);
+		if (!stat.isFile() || stat.size > WRITE_BASE_MAX_BYTES) return;
+		const previous = readFileSync(abs, "utf8");
+		if (previous.includes("\u0000")) return; // 二进制不做行级 diff
+		if (writeBases.size > 256) writeBases.clear();
+		writeBases.set(`${sessionPath}\u0000${toolCallId}`, previous);
+	} catch {
+		/* 新文件或读不到：没有旧内容，走全绿兜底 */
+	}
+}
+
+export function takeWritePatch(cwd: string, sessionPath: string, toolCallId: string, args: unknown): string | undefined {
+	const key = `${sessionPath}\u0000${toolCallId}`;
+	const previous = writeBases.get(key);
+	writeBases.delete(key);
+	if (previous === undefined) return undefined;
+	const { target, content } = writeArgsOf(args);
+	if (content === undefined || content === previous) return undefined;
+	try {
+		return generateUnifiedPatch(target, previous, content).slice(0, 60_000);
+	} catch {
+		return undefined;
+	}
+}
+
 function coercePartial(p: unknown): string | undefined {
 	if (p == null) return undefined;
 	if (typeof p === "string") return p.slice(-4000);
@@ -426,6 +474,8 @@ function translate(m: Managed, evt: AgentSessionEvent): void {
 			break;
 		}
 		case "tool_execution_start":
+			// write 覆盖前先留一份旧内容，结束时才能给出真实的新增/删除 diff
+			if (evt.toolName?.toLowerCase() === "write") captureWriteBase(m.cwd, m.sessionPath, evt.toolCallId, evt.args);
 			publish(m, {
 				type: "tool",
 				id: evt.toolCallId,
@@ -460,7 +510,7 @@ function translate(m: Managed, evt: AgentSessionEvent): void {
 				result: result.text || undefined,
 				isError: evt.isError,
 				encodingLoss: result.encodingLoss || undefined,
-				patch: patchOf((evt as any).result?.details),
+				patch: patchOf((evt as any).result?.details) ?? (evt.toolName?.toLowerCase() === "write" ? takeWritePatch(m.cwd, m.sessionPath, evt.toolCallId, (evt as any).args) : undefined),
 				ts: now,
 			});
 			publishUsage(m);
@@ -514,7 +564,15 @@ function translate(m: Managed, evt: AgentSessionEvent): void {
 			else publish(m, { type: "model", thinkingLevel: (evt as any).level, ts: now });
 			break;
 		case "auto_retry_start":
-			publish(m, { type: "error", message: `自动重试 ${(evt as any).attempt}/${(evt as any).maxAttempts}：${(evt as any).errorMessage ?? ""}`, ts: now });
+			// 重试是流程内通知：发一条结构化 retry 事件（界面聚合成「重试 n/max」一行），
+			// 不再走 error 通道，免得每次尝试都多出一条报错。
+			publish(m, {
+				type: "retry",
+				attempt: Number((evt as any).attempt) || 0,
+				maxAttempts: Number((evt as any).maxAttempts) || 0,
+				message: String((evt as any).errorMessage ?? ""),
+				ts: now,
+			});
 			break;
 		default:
 			break;
@@ -881,6 +939,11 @@ export function disposeSession(m: Managed): void {
 	sessions.delete(m.sessionPath);
 }
 
+/**
+ * 让某个会话立刻收工：中断正在跑的回合并回收 agent。
+ * 归档用它——归档后会话从工作区消失，若它的回合还在后台跑，用户看不见却仍在消耗 token。
+ * 只影响这一个会话，其它会话的运行不受影响（LRU/空闲回收同样跳过 busy 的会话）。
+ */
 export async function disposeSessionPath(sessionPath: string): Promise<void> {
 	const m = sessions.get(sessionPath);
 	if (!m) return;
@@ -907,6 +970,17 @@ export function reap(): void {
 
 setInterval(reap, 60_000).unref?.();
 
+/** 归档会话的保留期清理：只删「已归档且长期无活动」的会话；正在使用的会话一律跳过 */
+function isSessionInUse(sessionPath: string): boolean {
+	const key = process.platform === "win32" ? sessionPath.toLowerCase() : sessionPath;
+	for (const managed of sessions.keys()) {
+		if ((process.platform === "win32" ? managed.toLowerCase() : managed) === key) return true;
+	}
+	return false;
+}
+
+startArchivedSessionRetention({ isActive: isSessionInUse, log: (message) => console.log(message) });
+
 // ---------- 资源清单（技能 / 提示模板 / 扩展命令 / 信任 / 诊断） ----------
 
 type ResourceBundle = Pick<WebSnapshot, "skills" | "promptTemplates" | "extensionCommands" | "projectTrust" | "resourceDiagnostics">;

package/src/lib/event-batcher.ts

@@ -0,0 +1,62 @@
+/**
+ * 事件批量器：把高频流式事件合并成"每帧一次"的批量更新。
+ *
+ * 服务端按 token 推 delta，客户端原本每个事件一次 setState → 一次整树渲染。
+ * 合并后一次渲染里按顺序折叠整批事件，token 速率再高也只有每 16ms 一次渲染，
+ * 主线程不再被切碎（点击、滚动因此变顺）。
+ */
+export interface EventBatcher<T> {
+	/** 入队；达到上限立刻冲刷，避免极端情况下无限堆积 */
+	push(event: T): void;
+	flush(): void;
+	/** 丢弃未处理的事件（例如快照到达：旧事件已经过时） */
+	cancel(): void;
+	size(): number;
+}
+
+export function createEventBatcher<T>({
+	flush,
+	schedule = (run) => {
+		const id = setTimeout(run, 16);
+		return () => clearTimeout(id);
+	},
+	limit = 400,
+}: {
+	flush: (events: T[]) => void;
+	schedule?: (run: () => void) => () => void;
+	limit?: number;
+}): EventBatcher<T> {
+	let pending: T[] = [];
+	let cancelScheduled: (() => void) | null = null;
+
+	const stopScheduled = () => {
+		if (!cancelScheduled) return;
+		cancelScheduled();
+		cancelScheduled = null;
+	};
+
+	const flushNow = () => {
+		stopScheduled();
+		if (!pending.length) return;
+		const batch = pending;
+		pending = [];
+		flush(batch);
+	};
+
+	return {
+		push(event) {
+			pending.push(event);
+			if (pending.length >= limit) {
+				flushNow();
+				return;
+			}
+			cancelScheduled ??= schedule(flushNow);
+		},
+		flush: flushNow,
+		cancel() {
+			stopScheduled();
+			pending = [];
+		},
+		size: () => pending.length,
+	};
+}