@lyogvce/dsh-codex-subscription 0.2.9 → 0.2.10

package/README.md

@@ -22,7 +22,7 @@
    - 底层复用 DSH 内置的 `PiAiAdapter` 与 `@earendil-works/pi-ai` 的 `openaiCodexProvider()`，固定使用 `sse` 传输（含 `zstd` 请求体压缩、规范化纯 UUID `session-id` / `x-client-request-id` 缓存头、`reasoning.encrypted_content` 签名回放与图片自动缩放卸载）。
 5. **Codex 订阅限额内联展示与 ⚡ Fast 极速模式一键切换（`conversation.input.right`）**
    - 当会话选中 `openai-codex` 模型时，在输入框右下角紧邻模型选择器左侧展示限额胶囊（根据 `wham/usage` 返回的 `limit_window_seconds` 自适应展示 Plus 的 `5h X% · 7d Y%` 双窗口或 Pro 的 `7d X%` 单窗口，内置 60 秒缓存、单飞合并与每轮回复结束自动刷新），点击可展开查看对应窗口进度条、重置倒计时与手动刷新按钮；切换至其他 Provider 模型时自动隐藏。
-   - **⚡ 会话级 Fast 极速模式（`service_tier: "priority"`，最高 2x 推理速度）**：每个会话可在限额弹窗底部独立开关 Fast 模式（互不影响，新会话默认继承插件设置中的 `fastMode`，未配置时默认 `false` 关闭），状态自动持久化至 `~/.codex/cache/dsh-codex-subscription/runtime-state.json`。开启后当前会话胶囊显示闪电标识 `⚡`，并在该会话请求体中自动注入 `"service_tier": "priority"`。
+   - **⚡ 会话树级 Fast 极速模式（`service_tier: "priority"`，最高 2x 推理速度，主会话与 Subagent / 智能体团队同开同关）**：每个主会话可在限额弹窗底部独立开关 Fast 模式（不同主会话互不影响，新会话默认继承插件设置中的 `fastMode`，未配置时默认 `false` 关闭），该主会话派生的所有 `subagent`、`subagent_fork`、`workflow` 子任务以及智能体团队成员（`spawn_teammate`）均自动绑定至同一主会话开关，实时联动、同开同关；状态自动持久化至 `~/.codex/cache/dsh-codex-subscription/runtime-state.json`。开启后胶囊显示闪电标识 `⚡`，并在请求体中自动注入 `"service_tier": "priority"`。
 
 ---
 

package/lib/client.js

@@ -18,6 +18,26 @@ window.__ModuleLoader__.load({
 			const stripped = sessionId.trim().replace(/^session-/iu, "").trim();
 			return stripped.length > 0 ? stripped : void 0;
 		}
+		function resolveRootSessionId(sessionId, sessionParents, directParentSessionId) {
+			const startKey = normalizeCodexSessionId(sessionId);
+			const directParentKey = normalizeCodexSessionId(directParentSessionId);
+			if (startKey === void 0) return directParentKey ?? "__default__";
+			const readParent = (key) => {
+				if (key === startKey && directParentKey !== void 0 && directParentKey !== startKey) return directParentKey;
+				if (!sessionParents) return void 0;
+				const raw = sessionParents instanceof Map ? sessionParents.get(key) : sessionParents[key];
+				return typeof raw === "string" ? normalizeCodexSessionId(raw) : void 0;
+			};
+			let current = startKey;
+			const visited = new Set([current]);
+			while (visited.size <= 32) {
+				const parent = readParent(current);
+				if (parent === void 0 || visited.has(parent)) break;
+				visited.add(parent);
+				current = parent;
+			}
+			return current;
+		}
 		//#endregion
 		//#region src/usage.ts
 		function safeErrorMessage(error) {
@@ -30,14 +50,16 @@ window.__ModuleLoader__.load({
 		//#endregion
 		//#region src/client/quota.ts
 		const RING_CIRCUMFERENCE = 2 * Math.PI * 5.5;
-		const DEFAULT_SESSION_KEY = "__default__";
 		const DEFAULT_PRIMARY_WINDOW_SECONDS = 18e3;
 		const DEFAULT_SECONDARY_WINDOW_SECONDS = 604800;
-		function resolveSessionFastMode(snapshot, sessionId) {
+		function resolveSessionFastMode(snapshot, sessionId, parentSessionId) {
 			if (!snapshot) return false;
-			const key = normalizeCodexSessionId(sessionId) ?? DEFAULT_SESSION_KEY;
-			const sessionVal = snapshot.sessionFastModes?.[key];
-			if (typeof sessionVal === "boolean") return sessionVal;
+			const rootKey = resolveRootSessionId(sessionId, snapshot.sessionParents, parentSessionId);
+			const rootVal = snapshot.sessionFastModes?.[rootKey];
+			if (typeof rootVal === "boolean") return rootVal;
+			const selfKey = normalizeCodexSessionId(sessionId) ?? "__default__";
+			const selfVal = snapshot.sessionFastModes?.[selfKey];
+			if (typeof selfVal === "boolean") return selfVal;
 			return snapshot.defaultFastMode === true;
 		}
 		function resolveActiveProvider(directorySnapshot, projectedSelection) {
@@ -154,13 +176,14 @@ window.__ModuleLoader__.load({
 				quota: null,
 				defaultFastMode: false,
 				sessionFastModes: {},
+				sessionParents: {},
 				loading: false,
 				error: null
 			};
 			let confirmedSessionFastModes = {};
-			let lastConfirmedFastSeq = 0;
-			const latestSeqBySession = new Map();
-			const pendingBySession = new Map();
+			let knownSessionParents = {};
+			const latestSeqByNode = new Map();
+			const pendingBySeq = new Map();
 			let inflight = null;
 			let inflightForce = false;
 			let requestSeq = 0;
@@ -171,10 +194,63 @@ window.__ModuleLoader__.load({
 				state = next;
 				for (const listener of listeners) listener();
 			}
+			function recordClientSessionParent(childKey, parentKey) {
+				if (childKey === "" || parentKey === "" || childKey === parentKey) return;
+				let cursor = parentKey;
+				let depth = 0;
+				while (cursor !== void 0 && depth < 32) {
+					if (cursor === childKey) return;
+					cursor = knownSessionParents[cursor];
+					depth += 1;
+				}
+				if (knownSessionParents[childKey] === parentKey) return;
+				knownSessionParents = {
+					...knownSessionParents,
+					[childKey]: parentKey
+				};
+			}
+			function mergeServerSessionParents(raw) {
+				if (!raw || typeof raw !== "object" || Array.isArray(raw)) return;
+				for (const [child, parent] of Object.entries(raw)) {
+					const childKey = normalizeCodexSessionId(child);
+					const parentKey = typeof parent === "string" ? normalizeCodexSessionId(parent) : void 0;
+					if (childKey !== void 0 && parentKey !== void 0) recordClientSessionParent(childKey, parentKey);
+				}
+			}
+			function isTreeSeqSuperseded(seq, selfKey, parentKey) {
+				const myRoot = resolveRootSessionId(selfKey, knownSessionParents, parentKey);
+				for (const [nodeKey, nodeSeq] of latestSeqByNode.entries()) if (nodeSeq > seq && resolveRootSessionId(nodeKey, knownSessionParents) === myRoot) return true;
+				return false;
+			}
+			function collectTreeKeysForRoot(rootKey, extraKeys = []) {
+				const keys = new Set([rootKey, ...extraKeys]);
+				for (const k of Object.keys(knownSessionParents)) if (resolveRootSessionId(k, knownSessionParents) === rootKey) keys.add(k);
+				for (const k of Object.keys(confirmedSessionFastModes)) if (resolveRootSessionId(k, knownSessionParents) === rootKey) keys.add(k);
+				for (const k of latestSeqByNode.keys()) if (resolveRootSessionId(k, knownSessionParents) === rootKey) keys.add(k);
+				return [...keys];
+			}
 			function deriveSessionFastModes() {
-				if (pendingBySession.size === 0) return confirmedSessionFastModes;
-				const merged = { ...confirmedSessionFastModes };
-				for (const [k, pending] of pendingBySession) merged[k] = pending.enabled;
+				const merged = {};
+				for (const [k, v] of Object.entries(confirmedSessionFastModes)) {
+					const rootKey = resolveRootSessionId(k, knownSessionParents);
+					const rootVal = confirmedSessionFastModes[rootKey] ?? v;
+					merged[rootKey] = rootVal;
+					merged[k] = rootVal;
+				}
+				for (const childKey of Object.keys(knownSessionParents)) {
+					const rootKey = resolveRootSessionId(childKey, knownSessionParents);
+					if (typeof merged[rootKey] === "boolean") merged[childKey] = merged[rootKey];
+				}
+				for (const pending of pendingBySeq.values()) {
+					if (isTreeSeqSuperseded(pending.seq, pending.selfKey, pending.parentKey)) continue;
+					const rootKey = resolveRootSessionId(pending.selfKey, knownSessionParents, pending.parentKey);
+					const extra = pending.parentKey !== void 0 ? [
+						rootKey,
+						pending.parentKey,
+						pending.selfKey
+					] : [rootKey, pending.selfKey];
+					for (const k of collectTreeKeysForRoot(rootKey, extra)) merged[k] = pending.enabled;
+				}
 				return merged;
 			}
 			function parseServerSessionFastModes(raw) {
@@ -185,20 +261,22 @@ window.__ModuleLoader__.load({
 			}
 			function mergeFastStateFromPayload(payload, allowSessionReplace) {
 				const defaultFastMode = typeof payload?.defaultFastMode === "boolean" ? payload.defaultFastMode : state.defaultFastMode;
+				mergeServerSessionParents(payload?.sessionParents);
 				if (allowSessionReplace) {
 					const parsedMap = parseServerSessionFastModes(payload?.sessionFastModes);
 					if (parsedMap !== void 0) confirmedSessionFastModes = parsedMap;
 				}
 				return {
 					defaultFastMode,
-					sessionFastModes: deriveSessionFastModes()
+					sessionFastModes: deriveSessionFastModes(),
+					sessionParents: knownSessionParents
 				};
 			}
 			async function refresh(force = false) {
 				if (!force && state.quota !== null && nowFn() - state.quota.fetchedAt < ttlMs) return;
 				if (inflight !== null && (!force || inflightForce)) return inflight;
 				const seq = ++requestSeq;
-				const canReplaceFastAtStart = pendingBySession.size === 0;
+				const canReplaceFastAtStart = pendingBySeq.size === 0;
 				const fastEpochAtStart = fastMutationEpoch;
 				inflightForce = force;
 				setState({
@@ -224,7 +302,7 @@ window.__ModuleLoader__.load({
 						}
 						const payload = await response.json();
 						if (seq !== requestSeq) return;
-						const fastState = mergeFastStateFromPayload(payload, canReplaceFastAtStart && pendingBySession.size === 0 && fastEpochAtStart === fastMutationEpoch);
+						const fastState = mergeFastStateFromPayload(payload, canReplaceFastAtStart && pendingBySeq.size === 0 && fastEpochAtStart === fastMutationEpoch);
 						if (payload && payload.ok === true && payload.quota) setState({
 							quota: payload.quota,
 							...fastState,
@@ -258,29 +336,38 @@ window.__ModuleLoader__.load({
 				inflight = task;
 				return task;
 			}
-			async function setFastMode(sessionId, enabled) {
+			async function setFastMode(sessionId, enabled, parentSessionId) {
 				const nextFastMode = Boolean(enabled);
-				const key = normalizeCodexSessionId(sessionId) ?? DEFAULT_SESSION_KEY;
+				const selfKey = normalizeCodexSessionId(sessionId) ?? "__default__";
+				const parentKey = normalizeCodexSessionId(parentSessionId);
+				if (parentKey !== void 0 && parentKey !== selfKey) recordClientSessionParent(selfKey, parentKey);
 				const seq = ++fastModeSeq;
 				fastMutationEpoch += 1;
-				latestSeqBySession.set(key, seq);
-				pendingBySession.set(key, {
+				latestSeqByNode.set(selfKey, seq);
+				if (parentKey !== void 0) latestSeqByNode.set(parentKey, seq);
+				const initialRoot = resolveRootSessionId(selfKey, knownSessionParents, parentKey);
+				latestSeqByNode.set(initialRoot, seq);
+				pendingBySeq.set(seq, {
 					seq,
+					selfKey,
+					parentKey,
 					enabled: nextFastMode
 				});
 				const rollback = (errorMsg) => {
-					if (latestSeqBySession.get(key) !== seq) return;
+					pendingBySeq.delete(seq);
+					if (isTreeSeqSuperseded(seq, selfKey, parentKey)) return;
 					fastMutationEpoch += 1;
-					pendingBySession.delete(key);
 					setState({
 						...state,
 						sessionFastModes: deriveSessionFastModes(),
+						sessionParents: knownSessionParents,
 						error: errorMsg
 					});
 				};
 				setState({
 					...state,
 					sessionFastModes: deriveSessionFastModes(),
+					sessionParents: knownSessionParents,
 					error: null
 				});
 				try {
@@ -291,41 +378,55 @@ window.__ModuleLoader__.load({
 							Accept: "application/json"
 						},
 						body: JSON.stringify({
-							sessionId: key,
+							sessionId: selfKey,
 							fastMode: nextFastMode
 						})
 					});
-					if (latestSeqBySession.get(key) !== seq) return;
 					if (!response.ok) {
 						const errPayload = await response.json().catch(() => null);
+						mergeServerSessionParents(errPayload?.sessionParents);
 						rollback(typeof errPayload?.error === "string" && errPayload.error !== "" ? errPayload.error : `HTTP ${response.status}`);
 						return;
 					}
 					const payload = await response.json();
-					if (latestSeqBySession.get(key) !== seq) return;
+					mergeServerSessionParents(payload?.sessionParents);
+					if (isTreeSeqSuperseded(seq, selfKey, parentKey)) {
+						pendingBySeq.delete(seq);
+						return;
+					}
 					if (!payload || payload.ok === false) {
 						rollback(typeof payload?.error === "string" && payload.error !== "" ? payload.error : "Invalid response");
 						return;
 					}
 					fastMutationEpoch += 1;
-					pendingBySession.delete(key);
+					pendingBySeq.delete(seq);
+					const actualRootKey = resolveRootSessionId(selfKey, knownSessionParents, parentKey);
+					latestSeqByNode.set(actualRootKey, Math.max(latestSeqByNode.get(actualRootKey) ?? 0, seq));
 					const parsedMap = parseServerSessionFastModes(payload.sessionFastModes);
-					if (seq > lastConfirmedFastSeq && parsedMap !== void 0) {
-						lastConfirmedFastSeq = seq;
-						confirmedSessionFastModes = parsedMap;
-					} else confirmedSessionFastModes = {
-						...confirmedSessionFastModes,
-						[key]: parsedMap?.[key] ?? nextFastMode
-					};
+					const nextConfirmed = { ...confirmedSessionFastModes };
+					if (parsedMap !== void 0) {
+						for (const [k, v] of Object.entries(parsedMap)) if (!isTreeSeqSuperseded(seq, k, void 0)) nextConfirmed[k] = v;
+					}
+					const effectiveVal = parsedMap?.[selfKey] ?? parsedMap?.[actualRootKey] ?? nextFastMode;
+					const extraKeys = parentKey !== void 0 && parentKey !== selfKey ? [
+						actualRootKey,
+						parentKey,
+						selfKey
+					] : [actualRootKey, selfKey];
+					for (const k of collectTreeKeysForRoot(actualRootKey, extraKeys)) nextConfirmed[k] = effectiveVal;
+					confirmedSessionFastModes = nextConfirmed;
 					const defaultFastMode = typeof payload.defaultFastMode === "boolean" ? payload.defaultFastMode : state.defaultFastMode;
 					setState({
 						...state,
 						defaultFastMode,
 						sessionFastModes: deriveSessionFastModes(),
+						sessionParents: knownSessionParents,
 						error: null
 					});
 				} catch (error) {
 					rollback(safeErrorMessage(error));
+				} finally {
+					pendingBySeq.delete(seq);
 				}
 			}
 			return {
@@ -430,6 +531,8 @@ window.__ModuleLoader__.load({
 			const directorySnap = React.useSyncExternalStore(modelStore.subscribe, modelStore.getSnapshot);
 			const projectedSelection = typeof useProjection === "function" ? useProjection("modelSelection") : void 0;
 			const running = typeof useSession === "function" ? Boolean(useSession((s) => s?.running)) : false;
+			const rawParentSessionId = typeof useSession === "function" ? useSession((s) => s?.subagent?.address?.parentSessionId) : void 0;
+			const parentSessionId = typeof rawParentSessionId === "string" ? rawParentSessionId : void 0;
 			const isCodex = isCodexProvider(resolveActiveProvider(directorySnap, projectedSelection));
 			const quotaSnap = React.useSyncExternalStore(quotaStore.subscribe, quotaStore.getSnapshot);
 			const [open, setOpen] = React.useState(false);
@@ -489,7 +592,7 @@ window.__ModuleLoader__.load({
 				})), h("div", { className: "dsh-codex-quota-meta" }, h("span", null, `${t("quota.remainPrefix")}${remainPercent}%`), countdown !== "" ? h("span", null, countdown) : null));
 			};
 			const quota = quotaSnap.quota;
-			const isFastMode = resolveSessionFastMode(quotaSnap, sessionId);
+			const isFastMode = resolveSessionFastMode(quotaSnap, sessionId, parentSessionId);
 			const secondaryReached = Boolean(quota?.secondary && quota.secondary.usedPercent >= 100);
 			const primaryReached = Boolean(quota?.limitReached && !secondaryReached);
 			return h("span", {
@@ -549,7 +652,7 @@ window.__ModuleLoader__.load({
 				"aria-label": t("quota.fastTitle"),
 				"data-checked": isFastMode ? "true" : "false",
 				onClick: () => {
-					quotaStore.setFastMode(sessionId, !isFastMode);
+					quotaStore.setFastMode(sessionId, !isFastMode, parentSessionId);
 				}
 			}, h("span", { className: "dsh-codex-fast-switch-thumb" }))), quotaSnap.error !== null ? h("div", { className: "dsh-codex-quota-error" }, quotaSnap.error) : null) : null);
 		}

package/lib/index.d.ts

@@ -17,7 +17,7 @@ interface PluginContextLike {
   llm: LlmServiceLike;
   logger?: LoggerLike | undefined;
   get?: (name: string) => unknown;
-  on?: (event: string, listener: () => void) => () => void;
+  on?: (event: string, listener: (...args: unknown[]) => void) => () => void;
   effect?: (factory: () => void | (() => void), label?: string) => () => void;
   inject?: (services: readonly string[], callback: (ctx: PluginContextLike) => void) => () => void;
 }
@@ -134,12 +134,14 @@ type CodexQuotaResponse = {
   readonly quota: CodexQuotaSnapshot;
   readonly defaultFastMode?: boolean | undefined;
   readonly sessionFastModes?: Readonly<Record<string, boolean>> | undefined;
+  readonly sessionParents?: Readonly<Record<string, string>> | undefined;
 } | {
   readonly ok: false;
   readonly error: string;
   readonly quota?: CodexQuotaSnapshot | undefined;
   readonly defaultFastMode?: boolean | undefined;
   readonly sessionFastModes?: Readonly<Record<string, boolean>> | undefined;
+  readonly sessionParents?: Readonly<Record<string, string>> | undefined;
 };
 interface UsageFetcherOptions {
   readonly resolveCredentials: () => Promise<CodexResolvedCredentials>;

package/lib/index.js

@@ -16,6 +16,7 @@ const DEFAULT_CONTEXT_WINDOW = 872e3;
 const PLUGIN_CACHE_DIR_NAME = "cache/dsh-codex-subscription";
 const BACKUP_CACHE_FILE_NAME = "models-cache-backup.json";
 const RUNTIME_STATE_FILE_NAME = "runtime-state.json";
+const DEFAULT_SESSION_KEY = "__default__";
 const THINKING_LEVELS = [
 	"off",
 	"minimal",
@@ -44,6 +45,26 @@ function normalizeCodexSessionId(sessionId) {
 	const stripped = sessionId.trim().replace(/^session-/iu, "").trim();
 	return stripped.length > 0 ? stripped : void 0;
 }
+function resolveRootSessionId(sessionId, sessionParents, directParentSessionId) {
+	const startKey = normalizeCodexSessionId(sessionId);
+	const directParentKey = normalizeCodexSessionId(directParentSessionId);
+	if (startKey === void 0) return directParentKey ?? "__default__";
+	const readParent = (key) => {
+		if (key === startKey && directParentKey !== void 0 && directParentKey !== startKey) return directParentKey;
+		if (!sessionParents) return void 0;
+		const raw = sessionParents instanceof Map ? sessionParents.get(key) : sessionParents[key];
+		return typeof raw === "string" ? normalizeCodexSessionId(raw) : void 0;
+	};
+	let current = startKey;
+	const visited = new Set([current]);
+	while (visited.size <= 32) {
+		const parent = readParent(current);
+		if (parent === void 0 || visited.has(parent)) break;
+		visited.add(parent);
+		current = parent;
+	}
+	return current;
+}
 function isSameModelsCatalogPayload(a, b) {
 	if (!a || !b) return false;
 	const { fetched_at: _fa, ...restA } = a;
@@ -589,7 +610,7 @@ const REQUEST_IMAGE_PIXEL_BUDGET = 4194304;
 const REQUEST_IMAGE_MAX_BYTES = 1048576;
 const MAX_LOG_FILE_BYTES = 524288;
 const MAX_PERSISTED_SESSIONS = 200;
-const DEFAULT_SESSION_KEY = "__default__";
+const MAX_PERSISTED_PARENTS = 1e3;
 function resolvePluginCacheDir(codexHome) {
 	return join(codexHome, PLUGIN_CACHE_DIR_NAME);
 }
@@ -649,19 +670,34 @@ function writeBackupCache(codexHome, ctx, raw) {
 		appendPluginLog(codexHome, ctx, "WARN", "catalog.backup_write_error", { error: safeErrorMessage(error) });
 	}
 }
-function readPersistedSessionFastModes(codexHome) {
-	const result = new Map();
+function readPersistedRuntimeState(codexHome) {
+	const sessionFastModes = new Map();
+	const sessionParents = new Map();
 	try {
-		const entries = JSON.parse(readFileSync(join(resolvePluginCacheDir(codexHome), RUNTIME_STATE_FILE_NAME), "utf8"))?.sessionFastModes;
-		if (entries && typeof entries === "object" && !Array.isArray(entries)) {
-			for (const [k, v] of Object.entries(entries)) if (typeof k === "string" && k !== "" && typeof v === "boolean") result.set(k, v);
+		const raw = JSON.parse(readFileSync(join(resolvePluginCacheDir(codexHome), RUNTIME_STATE_FILE_NAME), "utf8"));
+		const entries = raw?.sessionFastModes;
+		if (entries && typeof entries === "object" && !Array.isArray(entries)) for (const [k, v] of Object.entries(entries)) {
+			const normKey = normalizeCodexSessionId(k) ?? (k === "__default__" ? "__default__" : void 0);
+			if (normKey !== void 0 && typeof v === "boolean") sessionFastModes.set(normKey, v);
+		}
+		const parents = raw?.sessionParents;
+		if (parents && typeof parents === "object" && !Array.isArray(parents)) for (const [child, parent] of Object.entries(parents)) {
+			const childKey = normalizeCodexSessionId(child);
+			const parentKey = typeof parent === "string" ? normalizeCodexSessionId(parent) : void 0;
+			if (childKey !== void 0 && parentKey !== void 0 && childKey !== parentKey) sessionParents.set(childKey, parentKey);
 		}
 	} catch {}
-	return result;
+	return {
+		sessionFastModes,
+		sessionParents
+	};
 }
-function writePersistedSessionFastModes(codexHome, ctx, sessionFastModes) {
+function writePersistedRuntimeState(codexHome, ctx, sessionFastModes, sessionParents) {
 	try {
-		writeFileAtomicSync(join(resolvePluginCacheDir(codexHome), RUNTIME_STATE_FILE_NAME), `${JSON.stringify({ sessionFastModes: Object.fromEntries(sessionFastModes) }, null, 2)}\n`);
+		writeFileAtomicSync(join(resolvePluginCacheDir(codexHome), RUNTIME_STATE_FILE_NAME), `${JSON.stringify({
+			sessionFastModes: Object.fromEntries(sessionFastModes),
+			sessionParents: Object.fromEntries(sessionParents)
+		}, null, 2)}\n`);
 	} catch (error) {
 		appendPluginLog(codexHome, ctx, "WARN", "fast_mode.persist_error", { error: safeErrorMessage(error) });
 	}
@@ -708,29 +744,142 @@ function apply(ctx, config = {}) {
 	const getConfigFastMode = () => readVolatile(config.fastMode, false) === true;
 	const getDisabledModels = () => readVolatile(config.disabledModels, DEFAULT_DISABLED_MODELS);
 	const getContextWindows = () => readVolatile(config.contextWindows, {});
-	const sessionFastModes = readPersistedSessionFastModes(getCodexHome());
+	const { sessionFastModes, sessionParents } = readPersistedRuntimeState(getCodexHome());
+	const recordSessionParent = (childKey, parentKey) => {
+		if (childKey === "" || parentKey === "" || childKey === parentKey) return false;
+		let cursor = parentKey;
+		let depth = 0;
+		while (cursor !== void 0 && depth < 32) {
+			if (cursor === childKey) return false;
+			cursor = sessionParents.get(cursor);
+			depth += 1;
+		}
+		if (sessionParents.get(childKey) === parentKey) return false;
+		sessionParents.delete(childKey);
+		sessionParents.set(childKey, parentKey);
+		while (sessionParents.size > MAX_PERSISTED_PARENTS) {
+			let victim;
+			for (const candidate of sessionParents.keys()) {
+				const root = resolveRootSessionId(candidate, sessionParents);
+				if (!sessionFastModes.has(root)) {
+					victim = candidate;
+					break;
+				}
+			}
+			const keyToDelete = victim ?? sessionParents.keys().next().value;
+			if (keyToDelete === void 0) break;
+			sessionParents.delete(keyToDelete);
+		}
+		return true;
+	};
+	const findLiveSession = (key) => {
+		const sessions = ctx.get?.("sessions");
+		if (typeof sessions?.get !== "function") return void 0;
+		return sessions.get(key) ?? sessions.get(`session-${key}`);
+	};
+	const findLiveAgent = (key) => {
+		const agents = ctx.get?.("agents");
+		if (typeof agents?.get !== "function") return void 0;
+		return agents.get(key) ?? agents.get(`session-${key}`);
+	};
+	const inspectSessionParentFromHost = (sessionLike, agentLike) => {
+		const childKey = normalizeCodexSessionId(sessionLike?.id ?? sessionLike?.header?.id ?? agentLike?.id ?? agentLike?.session?.id);
+		if (childKey === void 0) return false;
+		const header = sessionLike?.header ?? agentLike?.session?.header;
+		if (header?.origin === "subagent" && typeof header.parentSession === "string") {
+			const parentKey = normalizeCodexSessionId(header.parentSession);
+			if (parentKey !== void 0) return recordSessionParent(childKey, parentKey);
+		}
+		const agentTeams = ctx.get?.("agentTeams");
+		if (typeof agentTeams?.tryMembership === "function") {
+			const agent = agentLike ?? findLiveAgent(childKey);
+			if (agent !== void 0) {
+				const membership = agentTeams.tryMembership(agent);
+				if (membership?.role === "teammate" && typeof membership.root?.id === "string") {
+					const rootKey = normalizeCodexSessionId(membership.root.id);
+					if (rootKey !== void 0) return recordSessionParent(childKey, rootKey);
+				}
+			}
+		}
+		return false;
+	};
+	const normalizeFastModesToRoots = () => {
+		let migrated = false;
+		for (const [k, v] of [...sessionFastModes.entries()]) {
+			const rootKey = resolveRootSessionId(k, sessionParents);
+			if (rootKey !== k) {
+				if (!sessionFastModes.has(rootKey)) sessionFastModes.set(rootKey, v);
+				sessionFastModes.delete(k);
+				migrated = true;
+			}
+		}
+		return migrated;
+	};
+	const syncAllLiveSessionsFromHost = () => {
+		let changed = false;
+		const sessions = ctx.get?.("sessions");
+		if (typeof sessions?.list === "function") {
+			for (const s of sessions.list()) if (inspectSessionParentFromHost(s)) changed = true;
+		}
+		const agents = ctx.get?.("agents");
+		if (typeof agents?.list === "function") {
+			for (const a of agents.list()) if (inspectSessionParentFromHost(a?.session, a)) changed = true;
+		}
+		if (normalizeFastModesToRoots()) changed = true;
+		return changed;
+	};
+	const resolveRootSessionKey = (sessionId) => {
+		const initiatorId = (ctx.get?.("agents"))?.currentInitiator?.()?.id;
+		const startKey = normalizeCodexSessionId(sessionId) ?? normalizeCodexSessionId(initiatorId);
+		if (startKey === void 0) return DEFAULT_SESSION_KEY;
+		let current = startKey;
+		const visited = new Set([current]);
+		while (visited.size <= 32) {
+			const liveSession = findLiveSession(current);
+			const liveAgent = findLiveAgent(current);
+			if (liveSession !== void 0 || liveAgent !== void 0) inspectSessionParentFromHost(liveSession, liveAgent);
+			const parent = sessionParents.get(current);
+			if (parent === void 0 || visited.has(parent)) break;
+			visited.add(parent);
+			current = parent;
+		}
+		normalizeFastModesToRoots();
+		return current;
+	};
 	const getEffectiveFastMode = (sessionId) => {
-		const key = normalizeCodexSessionId(sessionId);
-		if (key !== void 0 && sessionFastModes.has(key)) return sessionFastModes.get(key) === true;
-		if (key === void 0 && sessionFastModes.has(DEFAULT_SESSION_KEY)) return sessionFastModes.get(DEFAULT_SESSION_KEY) === true;
+		const rootKey = resolveRootSessionKey(sessionId);
+		if (sessionFastModes.has(rootKey)) return sessionFastModes.get(rootKey) === true;
 		return getConfigFastMode();
 	};
 	const setSessionFastMode = (sessionId, nextFastMode) => {
 		const codexHome = getCodexHome();
-		const key = normalizeCodexSessionId(sessionId) ?? DEFAULT_SESSION_KEY;
-		sessionFastModes.delete(key);
-		sessionFastModes.set(key, nextFastMode);
+		syncAllLiveSessionsFromHost();
+		const rootKey = resolveRootSessionKey(sessionId);
+		sessionFastModes.delete(rootKey);
+		sessionFastModes.set(rootKey, nextFastMode);
+		normalizeFastModesToRoots();
 		while (sessionFastModes.size > MAX_PERSISTED_SESSIONS) {
 			const oldest = sessionFastModes.keys().next().value;
 			if (oldest === void 0) break;
 			sessionFastModes.delete(oldest);
 		}
-		writePersistedSessionFastModes(codexHome, ctx, sessionFastModes);
+		writePersistedRuntimeState(codexHome, ctx, sessionFastModes, sessionParents);
 		appendPluginLog(codexHome, ctx, "INFO", "fast_mode.updated", {
-			sessionId: key,
+			sessionId: normalizeCodexSessionId(sessionId) ?? "__default__",
+			rootSessionId: rootKey,
 			fastMode: nextFastMode
 		});
 	};
+	const buildProjectedSessionFastModes = () => {
+		syncAllLiveSessionsFromHost();
+		const projected = {};
+		for (const [rootKey, v] of sessionFastModes) projected[rootKey] = v;
+		for (const childKey of [...sessionParents.keys()]) {
+			const rootKey = resolveRootSessionId(childKey, sessionParents);
+			if (sessionFastModes.has(rootKey)) projected[childKey] = sessionFastModes.get(rootKey) === true;
+		}
+		return projected;
+	};
 	const authManager = createAuthManager({
 		getCodexHome,
 		getClientVersion: () => ensureSnapshot().catalog.clientVersion,
@@ -762,7 +911,9 @@ function apply(ctx, config = {}) {
 		const { catalog } = ensureSnapshot();
 		const modelMeta = catalog.metadataById.get(model?.id);
 		const { temperature: _temperature, sessionId, onPayload, ...rest } = options;
-		const normalizedSessionId = normalizeCodexSessionId(sessionId);
+		const initiatorId = (ctx.get?.("agents"))?.currentInitiator?.()?.id;
+		const normalizedSessionId = normalizeCodexSessionId(sessionId) ?? normalizeCodexSessionId(initiatorId);
+		resolveRootSessionKey(normalizedSessionId);
 		return {
 			...rest,
 			...normalizedSessionId !== void 0 ? { sessionId: normalizedSessionId } : {},
@@ -960,7 +1111,8 @@ function apply(ctx, config = {}) {
 					return new Response(JSON.stringify({
 						ok: true,
 						defaultFastMode: getConfigFastMode(),
-						sessionFastModes: Object.fromEntries(sessionFastModes)
+						sessionFastModes: buildProjectedSessionFastModes(),
+						sessionParents: Object.fromEntries(sessionParents)
 					}), {
 						status: 200,
 						headers: { "content-type": "application/json" }
@@ -971,7 +1123,8 @@ function apply(ctx, config = {}) {
 				return new Response(JSON.stringify({
 					...result,
 					defaultFastMode: getConfigFastMode(),
-					sessionFastModes: Object.fromEntries(sessionFastModes)
+					sessionFastModes: buildProjectedSessionFastModes(),
+					sessionParents: Object.fromEntries(sessionParents)
 				}), {
 					status: 200,
 					headers: { "content-type": "application/json" }
@@ -983,6 +1136,17 @@ function apply(ctx, config = {}) {
 	};
 	if (typeof ctx.inject === "function") ctx.inject(["connection"], mountUsageRoute);
 	else mountUsageRoute(ctx);
+	ctx.on?.("session/created", (session) => {
+		if (session && typeof session === "object") {
+			if (inspectSessionParentFromHost(session)) writePersistedRuntimeState(getCodexHome(), ctx, sessionFastModes, sessionParents);
+		}
+	});
+	ctx.on?.("agent/created", (payload) => {
+		if (payload && typeof payload === "object") {
+			const agent = payload.agent;
+			if (inspectSessionParentFromHost(agent?.session, agent)) writePersistedRuntimeState(getCodexHome(), ctx, sessionFastModes, sessionParents);
+		}
+	});
 	ctx.on?.("loader/volatile-update", () => {
 		ensureSnapshot();
 		ctx.llm.emitAdaptersUpdated?.();

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@lyogvce/dsh-codex-subscription",
-  "version": "0.2.9",
+  "version": "0.2.10",
   "description": "复用本机 ~/.codex 订阅凭证与 models_cache.json 模型缓存，在 DSH 中通过 SSE 直连调用 Codex 模型。",
   "type": "module",
   "main": "./lib/index.js",