@lyogvce/dsh-codex-subscription 0.2.9 → 0.2.11

package/README.md

@@ -15,14 +15,15 @@
    - 直接读取 `~/.codex/models_cache.json`（每次按文件 `mtime` 与大小做微秒级 `stat` 校验，仅在文件变化时重读解析），提取 `visibility === "list"` 的模型及其 `context_window`、`supported_reasoning_levels`（自动将 `ultra` 映射为 DSH 的 `max` 档位）与 `default_verbosity`。
    - **防旧版客户端降级污染**：自动将高版本 `models_cache.json` 备份至 `~/.codex/cache/dsh-codex-subscription/models-cache-backup.json`；当外部旧版进程（如 `dsh-subagent-codex` 内置的 `0.153.4`）将 `models_cache.json` 降级覆写为不含新模型的旧缓存时，自动拦截降级并保留高版本模型目录。
    - **可观测性日志**：自动在 `~/.codex/cache/dsh-codex-subscription/plugin.log`（带 512KB 自动滚动截断，忽略纯 `fetched_at` 心跳刷新）记录目录加载（`catalog.load` / `catalog.snapshot`）、降级拦截（`catalog.downgrade_rejected`）、备份恢复（`catalog.backup_restored`）与凭证续期（`auth.refreshed` / `auth.refresh_failed`）事件。
-3. **支持按配置屏蔽模型（`disabledModels`）与自定义上下文窗口（`contextWindows`）**
+3. **支持按配置屏蔽模型（`disabledModels`）、自定义上下文窗口（`contextWindows`）与原生重试策略（`retryPolicy`）**
    - 在插件配置中通过 `disabledModels` 屏蔽指定模型 `slug`，保持模型选择器干净清爽。
    - 通过 `contextWindows` 按模型 `slug` 自定义向 DSH 会话与自动压缩器暴露的预算窗口（例如设为 `400000`）；未配置时默认使用模型物理上限 `max_context_window`（`872000`）。同时底层 `PiCodexModel.contextWindow` 始终保持物理上限 `872000`，彻底防止 `dsh-llm-pi-ai` 在预算窗口与物理上限之间误判 `CONTEXT_WINDOW_EXCEEDED`。
+   - 通过 `retryPolicy` 暴露 DSH 原生 `RetryPolicySchema`（支持 `normal` / `always` 模式、`maxRetries`、`retryableCodes` 与 `backoff`，并随 `registration.replace` 热更新生效）。未配置时保持宿主默认策略（`mode: "normal"`、`maxRetries: 5`、默认 5 类错误码 `EMPTY_RESPONSE / RATE_LIMIT / SERVER / TIMEOUT / TRANSPORT` 及 `500ms..10000ms` 退避）。**注意：`retryableCodes` 为完整覆盖集合而非追加列表**，显式配置时需同时列出默认错误码与新增错误码。
 4. **原生多轮思考链回放与工具调用**
    - 底层复用 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"`。
 
 ---
 
@@ -36,9 +37,11 @@
 dsh plugin --profile desktop add @lyogvce/dsh-codex-subscription
 ```
 
-### 2. 配置屏蔽模型（`disabledModels`）与上下文窗口（`contextWindows`）
+### 2. 配置屏蔽模型（`disabledModels`）、上下文窗口（`contextWindows`）与重试策略（`retryPolicy`）
 
-默认配置（见 [cordis.patch.yml](cordis.patch.yml)）已屏蔽旧模型与备用模型，仅保留主力模型（如 `gpt-6.1-sol`、`gpt-6-astra`）。未配置 `contextWindows` 时，模型默认使用 `872000` 上下文窗口；若希望主动限制长上下文预算（例如统一设为 400K），可在 Profile 的 `cordis.patch.yml` 中配置：
+默认配置（见 [cordis.patch.yml](cordis.patch.yml)）已屏蔽旧模型与备用模型，仅保留主力模型（如 `gpt-6.1-sol`、`gpt-6-astra`）。未配置 `contextWindows` 时，模型默认使用 `872000` 上下文窗口；未配置 `retryPolicy` 时默认使用宿主 `normal` 重试策略（最多额外重试 5 次，覆盖 `EMPTY_RESPONSE`、`RATE_LIMIT`、`SERVER`、`TIMEOUT`、`TRANSPORT`）。
+
+若希望自定义上下文预算窗口或扩展可重试错误码白名单（**注意：`retryableCodes` 是完整集合而非追加列表，显式填写时需同时保留默认 5 类错误码**），可在 Profile 的 `cordis.patch.yml` 中配置：
 
 ```yaml
 - id: codex-subscription
@@ -53,9 +56,13 @@ dsh plugin --profile desktop add @lyogvce/dsh-codex-subscription
     contextWindows:
       gpt-6.1-sol: 400000
       gpt-6-astra: 400000
+    retryPolicy:
+      mode: normal
+      maxRetries: 5
+      retryableCodes: [ EMPTY_RESPONSE, RATE_LIMIT, SERVER, TIMEOUT, TRANSPORT, QUOTA, ACCOUNT_QUOTA, AUTH, INVALID_REQUEST, PI_AI_ERROR, STREAM_CLOSED, MALFORMED_RESPONSE ]
 ```
 
-可在 DSH 的 **设置 → 插件 → Codex 订阅直连** 页面或 `cordis.patch.yml` 中随时修改 `disabledModels` 与 `contextWindows`，实时生效无需重启。
+可在 DSH 的 **设置 → 插件 → Codex 订阅直连** 页面或 `cordis.patch.yml` 中随时修改 `disabledModels`、`contextWindows` 与 `retryPolicy`，实时热更新生效无需重启。
 
 ---
 

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

@@ -1,9 +1,11 @@
 import { PiCodexModel } from "@earendil-works/pi-ai/providers/openai-codex";
 //#region src/host-types.d.ts
+interface AdapterRegistrationHandle {
+  (): void;
+  replace: (next: string[]) => void;
+}
 interface LlmServiceLike {
-  registerAdapter: (providers: string[], adapter: unknown) => (() => void) & {
-    replace?: (next: string[]) => void;
-  };
+  registerAdapter: (providers: string[], adapter: unknown) => AdapterRegistrationHandle;
   emitAdaptersUpdated?: () => void;
 }
 type PluginLogLevel = 'INFO' | 'WARN' | 'ERROR';
@@ -15,9 +17,10 @@ interface LoggerLike {
 }
 interface PluginContextLike {
   llm: LlmServiceLike;
+  fiber?: unknown;
   logger?: LoggerLike | undefined;
   get?: (name: string) => unknown;
-  on?: (event: string, listener: () => void) => () => void;
+  on?: (event: string, listener: (...args: unknown[]) => unknown) => () => void;
   effect?: (factory: () => void | (() => void), label?: string) => () => void;
   inject?: (services: readonly string[], callback: (ctx: PluginContextLike) => void) => () => void;
 }
@@ -134,12 +137,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

@@ -1,7 +1,7 @@
 import { appendFileSync, chmodSync, mkdirSync, readFileSync, realpathSync, renameSync, statSync, unlinkSync, writeFileSync } from "node:fs";
 import { basename, dirname, join, resolve } from "node:path";
 import z from "@deepseek-ai/schemastery";
-import { LlmError, resolveImageAttachmentAccess, resolveRetryPolicy } from "@deepseek-ai/dsh-llm";
+import { LlmError, RetryPolicySchema, resolveImageAttachmentAccess, resolveRetryPolicy } from "@deepseek-ai/dsh-llm";
 import { PiAiAdapter } from "@deepseek-ai/dsh-llm-pi-ai";
 import { openaiCodexProvider } from "@earendil-works/pi-ai/providers/openai-codex";
 import { randomUUID } from "node:crypto";
@@ -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) });
 	}
@@ -669,7 +705,8 @@ function writePersistedSessionFastModes(codexHome, ctx, sessionFastModes) {
 const Config = z.object({
 	fastMode: z.boolean().default(false).description("新会话默认是否开启 Fast 极速模式（未填时默认 false；单会话可在右下角限额弹窗内独立切换）").volatile(),
 	disabledModels: z.array(z.string()).default([...DEFAULT_DISABLED_MODELS]).description("在模型选择器中屏蔽的模型 slug 列表").volatile(),
-	contextWindows: z.dict(z.number()).default({}).description("按模型 slug 覆盖上下文窗口大小（tokens），未配置时默认使用 872000").volatile()
+	contextWindows: z.dict(z.number()).default({}).description("按模型 slug 覆盖上下文窗口大小（tokens），未配置时默认使用 872000").volatile(),
+	retryPolicy: RetryPolicySchema.description("原生 Agent 步骤重试策略（retryableCodes 为完整覆盖集合，显式配置时需同时包含默认错误码与新增错误码）").volatile()
 });
 function readVolatile(field, fallback) {
 	const value = typeof field?.get === "function" ? field.get() : field;
@@ -683,7 +720,7 @@ var CodexPiAiAdapter = class extends PiAiAdapter {
 	}
 	modelInfo(snapshot, provider, model) {
 		const info = super.modelInfo(snapshot, provider, model);
-		const meta = this.#getModelMetadata(model);
+		const meta = this.#getModelMetadata(snapshot, model);
 		if (meta === void 0) return info;
 		const defaultLevel = meta.defaultReasoningLevel;
 		const hasDefaultLevel = defaultLevel !== void 0 && info.reasoning !== void 0 && info.reasoning.efforts.some((e) => e.id === defaultLevel);
@@ -703,34 +740,148 @@ var CodexPiAiAdapter = class extends PiAiAdapter {
 function apply(ctx, config = {}) {
 	const baseProvider = openaiCodexProvider();
 	const baseModels = baseProvider.getModels();
-	const retryPolicy = resolveRetryPolicy(void 0, "codex-subscription retryPolicy");
 	const getCodexHome = () => resolveCodexHome();
 	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 getRawRetryPolicy = () => readVolatile(config.retryPolicy, void 0);
+	const resolveConfiguredRetryPolicy = () => resolveRetryPolicy(getRawRetryPolicy(), "codex-subscription retryPolicy");
+	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,
@@ -740,13 +891,9 @@ function apply(ctx, config = {}) {
 	let cachedRaw;
 	let lastRejectedRaw;
 	let rawRevision = 0;
-	let snapshotSignature;
-	let currentCatalog = {
-		clientVersion: DEFAULT_CLIENT_VERSION,
-		models: [],
-		metadataById: new Map()
-	};
-	let currentProfilesMap;
+	let activeRetryPolicy = resolveConfiguredRetryPolicy();
+	let currentState;
+	const catalogByProfiles = new WeakMap();
 	const requestTokenAuth = Object.freeze({
 		name: "Codex CLI auth.json token",
 		async resolve({ credential }) {
@@ -758,11 +905,12 @@ function apply(ctx, config = {}) {
 			};
 		}
 	});
-	function wrapOptions(model, options = {}) {
-		const { catalog } = ensureSnapshot();
+	function wrapOptions(catalog, model, options = {}) {
 		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 } : {},
@@ -777,25 +925,28 @@ function apply(ctx, config = {}) {
 		resolveCredentials: () => authManager.resolveCredentials(),
 		buildHeaders: (authHeaders) => buildCodexHeaders(authHeaders, ensureSnapshot().catalog.clientVersion)
 	});
-	const piProvider = Object.freeze({
-		...baseProvider,
-		id: PROVIDER,
-		name: DISPLAY_NAME,
-		auth: Object.freeze({
-			...baseProvider.auth,
-			apiKey: requestTokenAuth
-		}),
-		getModels: () => ensureSnapshot().catalog.models,
-		stream: (model, context, options) => {
-			usageFetcher.invalidate();
-			return baseProvider.stream(model, context, wrapOptions(model, options));
-		},
-		streamSimple: (model, context, options) => {
-			usageFetcher.invalidate();
-			return baseProvider.streamSimple(model, context, wrapOptions(model, options));
-		}
-	});
+	function createSnapshotPiProvider(snapshotCatalog) {
+		return Object.freeze({
+			...baseProvider,
+			id: PROVIDER,
+			name: DISPLAY_NAME,
+			auth: Object.freeze({
+				...baseProvider.auth,
+				apiKey: requestTokenAuth
+			}),
+			getModels: () => snapshotCatalog.models,
+			stream: (model, context, options) => {
+				usageFetcher.invalidate();
+				return baseProvider.stream(model, context, wrapOptions(snapshotCatalog, model, options));
+			},
+			streamSimple: (model, context, options) => {
+				usageFetcher.invalidate();
+				return baseProvider.streamSimple(model, context, wrapOptions(snapshotCatalog, model, options));
+			}
+		});
+	}
 	function ensureSnapshot() {
+		const retryPolicy = activeRetryPolicy;
 		const codexHome = getCodexHome();
 		const cachePath = join(codexHome, "models_cache.json");
 		const disabled = getDisabledModels();
@@ -868,24 +1019,30 @@ function apply(ctx, config = {}) {
 				}
 			}
 		}
-		const nextSnapshotSig = JSON.stringify([
+		const nextCatalogSig = JSON.stringify([
 			rawRevision,
 			disabled,
 			contextWindows
 		]);
-		if (nextSnapshotSig !== snapshotSignature || currentProfilesMap === void 0) {
-			currentCatalog = cachedRaw !== void 0 ? parseModelsCache(cachedRaw, baseModels, disabled, contextWindows) : {
+		let nextCatalog = currentState?.catalog;
+		if (nextCatalog === void 0 || nextCatalogSig !== currentState?.catalogSig) {
+			nextCatalog = cachedRaw !== void 0 ? parseModelsCache(cachedRaw, baseModels, disabled, contextWindows) : {
 				clientVersion: DEFAULT_CLIENT_VERSION,
 				models: [],
 				metadataById: new Map()
 			};
-			appendPluginLog(codexHome, ctx, currentCatalog.models.length === 0 ? "WARN" : "INFO", "catalog.snapshot", {
-				clientVersion: currentCatalog.clientVersion,
-				models: currentCatalog.models.map((m) => m.id),
+			appendPluginLog(codexHome, ctx, nextCatalog.models.length === 0 ? "WARN" : "INFO", "catalog.snapshot", {
+				clientVersion: nextCatalog.clientVersion,
+				models: nextCatalog.models.map((m) => m.id),
 				disabledModels: disabled,
 				contextWindows
 			});
-			currentProfilesMap = new Map([[PROVIDER, Object.freeze({
+		}
+		const retryPolicySig = JSON.stringify(retryPolicy);
+		const nextSnapshotSig = JSON.stringify([nextCatalogSig, retryPolicySig]);
+		if (currentState === void 0 || nextSnapshotSig !== currentState.snapshotSig) {
+			const piProvider = createSnapshotPiProvider(nextCatalog);
+			const profileEntry = Object.freeze({
 				provider: PROVIDER,
 				displayName: DISPLAY_NAME,
 				piProvider,
@@ -898,13 +1055,19 @@ function apply(ctx, config = {}) {
 				streamIdleTimeoutMs: STREAM_IDLE_TIMEOUT_MS,
 				cacheRetention: "short",
 				transport: "sse"
-			})]]);
-			snapshotSignature = nextSnapshotSig;
+			});
+			const profiles = new Map([[PROVIDER, profileEntry]]);
+			catalogByProfiles.set(profiles, nextCatalog);
+			currentState = Object.freeze({
+				catalog: nextCatalog,
+				catalogSig: nextCatalogSig,
+				profiles,
+				snapshotSig: nextSnapshotSig,
+				retryPolicy,
+				retryPolicySig
+			});
 		}
-		return {
-			catalog: currentCatalog,
-			profiles: currentProfilesMap
-		};
+		return currentState;
 	}
 	const adapter = new CodexPiAiAdapter({
 		profiles: () => ensureSnapshot().profiles,
@@ -930,8 +1093,12 @@ function apply(ctx, config = {}) {
 		onReplayDegrade: (provider, model, reason) => {
 			console.error(`[dsh-codex-subscription] replay state for "${provider}/${model}" degraded (${reason})`);
 		}
-	}, (modelId) => ensureSnapshot().catalog.metadataById.get(modelId));
-	ctx.llm.registerAdapter([PROVIDER], adapter);
+	}, (snapshot, modelId) => {
+		const snapProfiles = snapshot?.profiles;
+		return ((snapProfiles !== void 0 ? catalogByProfiles.get(snapProfiles) : void 0) ?? ensureSnapshot().catalog).metadataById.get(modelId);
+	});
+	let committedState = ensureSnapshot();
+	const registration = ctx.llm.registerAdapter([PROVIDER], adapter);
 	const mountUsageRoute = (connCtx) => {
 		const connection = connCtx.get?.("connection");
 		const registerFetch = connection?.fetch?.register;
@@ -960,7 +1127,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 +1139,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,8 +1152,45 @@ 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?.("internal/config", function(rawInput, next) {
+		const raw = typeof next === "function" ? next() : rawInput;
+		if (ctx.fiber !== void 0 && this !== ctx.fiber) return raw;
+		if (raw !== null && typeof raw === "object" && !Array.isArray(raw)) {
+			const candidateRetry = readVolatile(raw.retryPolicy, void 0);
+			resolveRetryPolicy(candidateRetry, "codex-subscription retryPolicy");
+		}
+		return raw;
+	});
 	ctx.on?.("loader/volatile-update", () => {
-		ensureSnapshot();
+		const candidateRetryPolicy = resolveConfiguredRetryPolicy();
+		const prevRetryPolicy = activeRetryPolicy;
+		const prevCommitted = committedState;
+		activeRetryPolicy = candidateRetryPolicy;
+		let nextState;
+		try {
+			nextState = ensureSnapshot();
+			if (nextState.retryPolicySig !== prevCommitted.retryPolicySig) {
+				registration.replace([PROVIDER]);
+				committedState = nextState;
+				return;
+			}
+		} catch (error) {
+			activeRetryPolicy = prevRetryPolicy;
+			currentState = prevCommitted;
+			throw error;
+		}
+		committedState = nextState;
 		ctx.llm.emitAdaptersUpdated?.();
 	});
 }

package/package.json

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