@tnnevol/dsh-codex-auth 0.1.7-rc.2 → 0.1.7-rc.2.2

package/lib/index.js

@@ -9,7 +9,7 @@ import { PiAiAdapter } from "@deepseek-ai/dsh-llm-pi-ai";
 /** Read-only Codex account quota information for the local settings card. */
 const CODEX_USAGE_URL = "https://chatgpt.com/backend-api/wham/usage";
 const CODEX_USAGE_TIMEOUT_MS = 1e4;
-function record$1(value) {
+function record$2(value) {
 	return typeof value === "object" && value !== null && !Array.isArray(value) ? value : void 0;
 }
 function number$1(value) {
@@ -29,7 +29,7 @@ function percentage(value) {
 	return Math.max(0, Math.min(100, value));
 }
 function window(value) {
-	const source = record$1(value);
+	const source = record$2(value);
 	if (source === void 0) return void 0;
 	const result = {};
 	const usedPercent = number$1(source["used_percent"]);
@@ -45,7 +45,7 @@ function window(value) {
 	return Object.keys(result).length > 0 ? result : void 0;
 }
 function credits(value) {
-	const source = record$1(value);
+	const source = record$2(value);
 	if (source === void 0) return void 0;
 	const result = {};
 	const hasCredits = boolean(source["has_credits"]);
@@ -58,9 +58,9 @@ function credits(value) {
 }
 /** Normalize the evolving private WHAM response into a small UI-safe shape. */
 function normalizeCodexUsagePayload(value) {
-	const source = record$1(value);
+	const source = record$2(value);
 	if (source === void 0) throw new Error("Codex usage response was not an object");
-	const rateLimit = record$1(source["rate_limit"]);
+	const rateLimit = record$2(source["rate_limit"]);
 	const result = {};
 	const planType = string$1(source["plan_type"]);
 	const allowed = boolean(rateLimit?.["allowed"]);
@@ -171,14 +171,14 @@ const LLM_PI_AI_THINKING_LEVELS = /* @__PURE__ */ new Set([
 ]);
 /** Rows the account endpoint returns for machinery, not human selection. */
 const INTERNAL_CODEX_MODEL_IDS = /* @__PURE__ */ new Set(["codex-auto-review", "gpt-reserve"]);
-function record(value) {
+function record$1(value) {
 	return typeof value === "object" && value !== null && !Array.isArray(value) ? value : void 0;
 }
 function accessToken(auth) {
 	return typeof auth?.apiKey === "string" && auth.apiKey.length > 0 ? auth.apiKey : void 0;
 }
 function accountId(credential) {
-	const value = record(credential)?.accountId;
+	const value = record$1(credential)?.accountId;
 	return typeof value === "string" && value.length > 0 ? value : void 0;
 }
 function string(value) {
@@ -310,7 +310,7 @@ async function fetchCodexModelCatalog(store, fetchImpl = fetch) {
 			signal: controller.signal
 		});
 		if (!response.ok) throw new Error(`Codex model catalog request failed with status ${String(response.status)}`);
-		const parsed = record(await response.json());
+		const parsed = record$1(await response.json());
 		if (parsed === void 0 || !Array.isArray(parsed.models)) throw new Error("Codex model catalog response was not an object with a models array");
 		return { models: parsed.models };
 	} finally {
@@ -820,6 +820,205 @@ var CodexCredentialMirror = class {
 	}
 };
 //#endregion
+//#region src/contracts/model-capabilities.ts
+/** 深思等级的共同前缀：`minimal..max`，各模型的真实线路值。 */
+const FULL_LEVELS = {
+	minimal: "low",
+	low: "low",
+	medium: "medium",
+	high: "high",
+	xhigh: "xhigh",
+	max: "max"
+};
+const CONTEXT_WINDOW = 272e3;
+const MAX_TOKENS = 128e3;
+/** 七个模型当前都是图文输入；单独抽出来避免每行重复书写造成漂移。 */
+const TEXT_AND_IMAGE = ["text", "image"];
+/**
+* Codex Auth 声明并保证的模型集合，顺序即补丁中的书写顺序。
+*
+* `gpt-6-astra` 的目录条目把 `off` 映射为 `null`，即该模型确实没有 `off`；
+* `gpt-5.5` 的目录不提供 `max`，因此它的等级到 `xhigh` 为止。
+*/
+const CODEX_MODEL_CAPABILITIES = [
+	{
+		id: "gpt-6-astra",
+		name: "GPT-6-Astra",
+		contextWindow: CONTEXT_WINDOW,
+		maxTokens: MAX_TOKENS,
+		input: TEXT_AND_IMAGE,
+		reasoningEfforts: { ...FULL_LEVELS }
+	},
+	{
+		id: "gpt-6-sol",
+		name: "GPT-6-Sol",
+		contextWindow: CONTEXT_WINDOW,
+		maxTokens: MAX_TOKENS,
+		input: TEXT_AND_IMAGE,
+		reasoningEfforts: {
+			off: "none",
+			...FULL_LEVELS
+		}
+	},
+	{
+		id: "gpt-6-luna",
+		name: "GPT-6-Luna",
+		contextWindow: CONTEXT_WINDOW,
+		maxTokens: MAX_TOKENS,
+		input: TEXT_AND_IMAGE,
+		reasoningEfforts: {
+			off: "none",
+			...FULL_LEVELS
+		}
+	},
+	{
+		id: "gpt-5.6-sol",
+		name: "GPT-5.6 Sol",
+		contextWindow: CONTEXT_WINDOW,
+		maxTokens: MAX_TOKENS,
+		input: TEXT_AND_IMAGE,
+		reasoningEfforts: {
+			off: null,
+			...FULL_LEVELS
+		}
+	},
+	{
+		id: "gpt-5.6-terra",
+		name: "GPT-5.6 Terra",
+		contextWindow: CONTEXT_WINDOW,
+		maxTokens: MAX_TOKENS,
+		input: TEXT_AND_IMAGE,
+		reasoningEfforts: {
+			off: null,
+			...FULL_LEVELS
+		}
+	},
+	{
+		id: "gpt-5.6-luna",
+		name: "GPT-5.6 Luna",
+		contextWindow: CONTEXT_WINDOW,
+		maxTokens: MAX_TOKENS,
+		input: TEXT_AND_IMAGE,
+		reasoningEfforts: {
+			off: null,
+			...FULL_LEVELS
+		}
+	},
+	{
+		id: "gpt-5.5",
+		name: "GPT-5.5",
+		contextWindow: CONTEXT_WINDOW,
+		maxTokens: MAX_TOKENS,
+		input: TEXT_AND_IMAGE,
+		reasoningEfforts: {
+			off: null,
+			minimal: "low",
+			low: "low",
+			medium: "medium",
+			high: "high",
+			xhigh: "xhigh"
+		}
+	}
+];
+/** 按 id 取能力；未知 id 返回 `undefined`（调用方决定是否省略字段而不是猜）。 */
+function codexModelCapability(id) {
+	return CODEX_MODEL_CAPABILITIES.find((model) => model.id === id);
+}
+//#endregion
+//#region src/host/model-capability-heal.ts
+/**
+* Repair a Codex model list that lost its capability fields.
+*
+* The official Models page writes a provider's whole `models` array into the
+* profile's patch document. That array *replaces* the shipped baseline
+* wholesale (DSH merges objects field by field but replaces arrays), so any
+* row that only carries `id`/`name` hides the route's real context window,
+* output cap and thinking levels from every selector — and the model picker
+* shows grey `256K`/`32K` placeholders while the composer offers no reasoning
+* levels. Rows like that were produced by the official candidate picker while
+* its Codex candidates carried no capabilities; the bridge that feeds it is
+* fixed, but the rows it already wrote are still in user profiles.
+*
+* This repair is deliberately **additive**:
+*
+* - it only fills a field that is *absent* — a value the user set, even a
+*   custom one, is never overwritten, so a deliberate `input: [text]` survives;
+* - it never adds, removes, renames or reorders models, so a user's own
+*   selection, order and display names are preserved;
+* - it only touches models this plugin's contract describes, so an id from a
+*   newer account catalog is left exactly as the user has it;
+* - it returns `undefined` when nothing needs filling, which makes the caller
+*   a no-op and the whole operation idempotent — the write it triggers cannot
+*   echo back into another write.
+*
+* @module dsh-codex-auth/model-capability-heal
+*/
+/** The settings namespace owning the Codex route. */
+const CODEX_SETTINGS_NAMESPACE = "llm-pi-ai";
+/** Capability fields the repair may fill; all other fields are user-owned. */
+const FILLABLE = [
+	"contextWindow",
+	"maxTokens",
+	"input",
+	"reasoningEfforts"
+];
+function record(value) {
+	return typeof value === "object" && value !== null && !Array.isArray(value) ? value : void 0;
+}
+/**
+* Plan the capability repair for one provider's `models` array.
+*
+* @param models - the row list as stored in the settings document.
+* @returns the repaired rows plus the ids that were filled, or `undefined`
+*   when every described model already states its capabilities.
+*/
+function planCodexCapabilityRepair(models) {
+	if (!Array.isArray(models)) return void 0;
+	const filled = [];
+	let changed = false;
+	const repaired = models.map((row) => {
+		const model = record(row);
+		const id = model?.id;
+		if (model === void 0 || typeof id !== "string") return row;
+		const capability = codexModelCapability(id);
+		if (capability === void 0) return row;
+		if (FILLABLE.filter((field) => model[field] === void 0).length === 0) return row;
+		const add = {};
+		if (model.contextWindow === void 0) add.contextWindow = capability.contextWindow;
+		if (model.maxTokens === void 0) add.maxTokens = capability.maxTokens;
+		if (model.input === void 0) add.input = [...capability.input];
+		if (model.reasoningEfforts === void 0) add.reasoningEfforts = { ...capability.reasoningEfforts };
+		changed = true;
+		filled.push(id);
+		return {
+			...model,
+			...add
+		};
+	});
+	return changed ? {
+		models: repaired,
+		filled
+	} : void 0;
+}
+/**
+* Fill in missing Codex capabilities in the stored settings document.
+*
+* A missing namespace entry, a route that states its capabilities already, or
+* any error is reported as "nothing to do": this runs at startup beside the
+* credential mirror, so it must never fail activation or block boot. The
+* failure mode it guards against is cosmetic-plus-silent (degraded selectors),
+* never data loss.
+*
+* @param settings - the Host settings service.
+* @returns the ids that were filled, empty when no write was needed.
+*/
+async function healCodexModelCapabilities(settings) {
+	const repair = planCodexCapabilityRepair(record(record(record(settings.describe().find((row) => row.ns === CODEX_SETTINGS_NAMESPACE)?.value)?.providers)?.[CODEX_PROVIDER])?.models);
+	if (repair === void 0) return [];
+	await settings.update(CODEX_SETTINGS_NAMESPACE, { providers: { [CODEX_PROVIDER]: { models: repair.models } } });
+	return repair.filled;
+}
+//#endregion
 //#region src/host/adapter.ts
 /** OpenAI Codex adapter assembled from dsh's public pi-ai extension seam. */
 /** Keep the Codex stream open while the provider is still producing output. */
@@ -892,6 +1091,39 @@ const inject = [
 	"agentDefaultModel",
 	"llm"
 ];
+/**
+* One-shot capability repair, run once the settings entries are addressable.
+*
+* Activation order between this plugin and `llm-pi-ai` is not guaranteed, so
+* the descriptor may not exist yet on the first attempt. The retry listens for
+* the settings service's own update notification and disconnects immediately
+* after the first attempt that can see the namespace: this has to stay a
+* one-shot, because rewriting a document while the user is editing it would
+* make their save lose its revision check.
+*/
+function scheduleCodexCapabilityHeal(ctx) {
+	let settled = false;
+	const attempt = () => {
+		const settings = ctx.settings;
+		if (!settings.describe().some((row) => row.ns === "llm-pi-ai")) return false;
+		settled = true;
+		healCodexModelCapabilities(settings).then((filled) => {
+			if (filled.length > 0) ctx.logger.info(`dsh-codex-auth: filled missing Codex model capabilities for ${filled.join(", ")}`);
+		}).catch((error) => {
+			ctx.logger.warn("dsh-codex-auth: unable to fill Codex model capabilities", error);
+		});
+		return true;
+	};
+	ctx.effect(() => {
+		if (attempt()) return () => {};
+		const off = ctx.on("settings/document-updated", () => {
+			if (!settled) attempt();
+		});
+		return () => {
+			off();
+		};
+	}, "dsh-codex-auth: Codex model capability repair");
+}
 function apply(ctx) {
 	const store = new CodexCredentialStore();
 	const mirror = new CodexCredentialMirror(ctx.credentials, store);
@@ -910,6 +1142,7 @@ function apply(ctx) {
 	registerCodexAuthRoutes(ctx, store, mirror);
 	registerCodexModelRefreshRoute(ctx, store, { update: (ns, patch) => ctx.settings.update(ns, patch) });
 	registerCodexGlobalModelRoute(ctx, ctx.agentDefaultModel, ctx.llm);
+	scheduleCodexCapabilityHeal(ctx);
 }
 //#endregion
 export { CODEX_API_KEY_ENV, CODEX_API_KEY_REF, CODEX_AUTH_LOGIN_PATH, CODEX_AUTH_LOGOUT_PATH, CODEX_AUTH_STATUS_PATH, CODEX_GLOBAL_MODEL_PATH, CODEX_STREAM_IDLE_TIMEOUT_MS, CodexCredentialMirror, CodexCredentialStore, CodexUsageService, CodexWebAuth, apply, codexAuthStatus, createCodexAdapter, inject, loginCodex, logoutCodex, name, normalizeCodexUsagePayload, registerCodexAuthRoutes, trustedRequest };