@armadra/agent 0.6.0 → 0.6.2

package/dist/auth/chatgpt/backend-client.js

@@ -2,13 +2,14 @@
  * ChatGPT 后端的只读查询（[W6-O]）：模型列表（`ama models discover chatgpt`）与 codex 的 `wham/usage`。
  *
  * - SIWC：`GET {base}/models`，只留 `visibility === "list"`（没有该字段的也留）；
- * - codex：`GET {base}/models?client_version=<ama 版本>`，`models[].slug`（同样按 visibility 筛）；
+ * - codex：`GET {base}/models?client_version=<Codex CLI 版本>`（`codexClientVersion()`；后端按每个模型的
+ *   `minimal_client_version` 过滤，发 ama 自己的版本号会一个都拿不到），`models[].slug`（同样按 visibility 筛），
+ *   另取 `context_window`、`input_modalities`、`supported_reasoning_levels` 作元数据（models.dev 补不到时用）；
  *   `GET https://chatgpt.com/backend-api/wham/usage`（由 codex base 推出），私有格式，解析失败返回 undefined。
- * 只取 slug 与显示名，上下文窗口不猜。请求带 Bearer；codex 另带 `ChatGPT-Account-ID` 与 `originator`。
+ * SIWC 只取 slug 与显示名。请求带 Bearer；codex 另带 `ChatGPT-Account-ID` 与 `originator`。
  */
 import { parseUsagePayload } from "../../ai/apis/chatgpt-rate-limits.js";
-import { AMA_VERSION } from "../../version.js";
-import { DEFAULT_ORIGINATOR } from "./presets.js";
+import { DEFAULT_CODEX_CLIENT_VERSION, DEFAULT_ORIGINATOR } from "./presets.js";
 export const BACKEND_TIMEOUT_MS = 15_000;
 function headers(auth) {
     const out = {
@@ -26,10 +27,37 @@ function listed(item) {
     const visibility = item["visibility"];
     return visibility === undefined || visibility === "list";
 }
+/** codex 模型条目里的元数据；字段缺失或形状不对就不填。 */
+function codexMetadata(item) {
+    const out = {};
+    const window = item["context_window"];
+    if (typeof window === "number" && Number.isInteger(window) && window > 0)
+        out.contextWindow = window;
+    const modalities = item["input_modalities"];
+    if (Array.isArray(modalities)) {
+        const input = ["text", "image"].filter((x) => modalities.includes(x));
+        if (input.includes("text"))
+            out.input = input;
+    }
+    const levels = item["supported_reasoning_levels"];
+    if (Array.isArray(levels)) {
+        const efforts = levels
+            .map((level) => typeof level === "string"
+            ? level
+            : typeof level === "object" && level !== null
+                ? level["effort"]
+                : undefined)
+            .filter((x) => typeof x === "string" && x !== "");
+        if (efforts.length > 0)
+            out.reasoningLevels = [...new Set(efforts)];
+    }
+    return out;
+}
 export async function listChatGptModels(fetchFn, baseUrl, auth) {
     const base = baseUrl.replace(/\/+$/, "");
+    const version = auth.clientVersion ?? DEFAULT_CODEX_CLIENT_VERSION;
     const url = auth.flavor === "codex"
-        ? `${base}/models?client_version=${encodeURIComponent(AMA_VERSION)}`
+        ? `${base}/models?client_version=${encodeURIComponent(version)}`
         : `${base}/models`;
     const response = await fetchFn(url, {
         headers: headers(auth),
@@ -49,7 +77,11 @@ export async function listChatGptModels(fetchFn, baseUrl, auth) {
             continue;
         seen.add(id);
         const name = item["display_name"];
-        out.push(typeof name === "string" && name !== "" ? { id, name } : { id });
+        out.push({
+            id,
+            ...(typeof name === "string" && name !== "" ? { name } : {}),
+            ...(auth.flavor === "codex" ? codexMetadata(item) : {}),
+        });
     }
     return out;
 }

package/dist/auth/chatgpt/cli.d.ts

@@ -4,7 +4,10 @@
  * - login：flavor 取 `--flavor` > 用户级 `auth.chatgpt.flavor` > siwc；`--paste` / `--device`（只 codex）/ 缺省
  *   浏览器；codex 首次在 TTY 下一次性确认「非官方、仅个人使用」（非 TTY 需 `--yes`），`acknowledgedAt` 入条目；
  *   成功后在刷新锁下写 auth.json（0600）。
- * - logout：SIWC 先撤销 refresh token（失败照样删本地，提示到 ChatGPT 设置里断开）；codex 只删本地。
+ *   成功后先删旧的发现缓存，再查一次模型列表写缓存（`<dataDir>/models/discovered/chatgpt.json`；0 个也写空表并带
+ *   flavor，免得残留另一种登录方式的缓存；查询失败静默）。codex 返回 0 个时提示 codexClientVersion 可能过旧。
+ * - logout：SIWC 先撤销 refresh token（失败照样删本地，提示到 ChatGPT 设置里断开）；codex 只删本地；
+ *   同时删发现缓存。
  * - status：flavor、计划、掩码邮箱、access token 剩余、needsLogin；codex 另查 `wham/usage` 显示配额。
  *
  * 输出里从不出现 token、code、账户 id；错误按码渲染（i18n），不回显响应原文。

package/dist/auth/chatgpt/cli.js

@@ -4,7 +4,10 @@
  * - login：flavor 取 `--flavor` > 用户级 `auth.chatgpt.flavor` > siwc；`--paste` / `--device`（只 codex）/ 缺省
  *   浏览器；codex 首次在 TTY 下一次性确认「非官方、仅个人使用」（非 TTY 需 `--yes`），`acknowledgedAt` 入条目；
  *   成功后在刷新锁下写 auth.json（0600）。
- * - logout：SIWC 先撤销 refresh token（失败照样删本地，提示到 ChatGPT 设置里断开）；codex 只删本地。
+ *   成功后先删旧的发现缓存，再查一次模型列表写缓存（`<dataDir>/models/discovered/chatgpt.json`；0 个也写空表并带
+ *   flavor，免得残留另一种登录方式的缓存；查询失败静默）。codex 返回 0 个时提示 codexClientVersion 可能过旧。
+ * - logout：SIWC 先撤销 refresh token（失败照样删本地，提示到 ChatGPT 设置里断开）；codex 只删本地；
+ *   同时删发现缓存。
  * - status：flavor、计划、掩码邮箱、access token 剩余、needsLogin；codex 另查 `wham/usage` 显示配额。
  *
  * 输出里从不出现 token、code、账户 id；错误按码渲染（i18n），不回显响应原文。
@@ -19,10 +22,11 @@ import { UsageError } from "../../cli/args.js";
 import { ExitCode } from "../../cli/exit-codes.js";
 import { openBrowser } from "../oauth/browser.js";
 import { readOAuthEntry, withRefreshLock, writeOAuthEntry } from "../oauth/token-store.js";
-import { fetchCodexUsage } from "./backend-client.js";
+import { clearDiscoveredCache, writeDiscoveredCache } from "../../ai/providers/discovered-cache.js";
+import { fetchCodexUsage, listChatGptModels } from "./backend-client.js";
 import { maskEmail } from "./claims.js";
 import { loginChatGpt, revokeChatGpt } from "./login.js";
-import { CHATGPT_PROVIDER_ID, chatgptBaseUrl } from "./presets.js";
+import { CHATGPT_PROVIDER_ID, chatgptBaseUrl, codexClientVersion } from "./presets.js";
 import { quotaParts } from "./quota-text.js";
 function userAuthConfig(io) {
     try {
@@ -55,7 +59,7 @@ export function loginErrorText(error, flavor) {
         case "oauth_timeout":
             return m.timeout;
         case "oauth_denied":
-            return m.denied(detail.error ?? "error");
+            return m.denied(detail.error ?? "error", flavor === "siwc");
         case "oauth_state_mismatch":
             return m.stateMismatch;
         case "oauth_no_code":
@@ -183,10 +187,39 @@ export async function runLogin(args, io, deps = {}) {
     if (entry.planType === "free")
         io.stdout(`${m.freePlanHint(entry.planType)}\n`);
     if (flavor === "siwc")
-        io.stdout(`${m.siwcLimitHint}\n`);
-    io.stdout(`${m.modelHint}\n`);
+        io.stdout(`${m.siwcLimitHint}\n${m.siwcEligibilityHint}\n`);
+    const count = await cacheModels(io, entry, config, deps);
+    const done = count === 0 && flavor === "codex"
+        ? m.codexNoModels(codexClientVersion(config, io.env))
+        : count === undefined || count === 0
+            ? m.modelHint
+            : m.modelsCached(count);
+    io.stdout(`${done}\n`);
     return ExitCode.Ok;
 }
+/**
+ * 登录后查一次账户可用的模型（只读的模型列表接口，不消耗额度）并写发现缓存，`/model` 由此列出；
+ * 任何失败静默（返回 undefined，调用方改提示 `ama models discover chatgpt`）。
+ */
+async function cacheModels(io, entry, config, deps) {
+    const dataDir = resolveDataDir({ env: io.env });
+    try {
+        // 先删旧缓存：换了登录方式时另一 flavor 的缓存不能留下
+        clearDiscoveredCache(dataDir, CHATGPT_PROVIDER_ID);
+        const models = await listChatGptModels(deps.fetch ?? fetch, chatgptBaseUrl(entry.flavor, io.env), {
+            flavor: entry.flavor,
+            accessToken: entry.accessToken,
+            accountId: entry.accountId,
+            originator: config?.originator,
+            clientVersion: codexClientVersion(config, io.env),
+        });
+        writeDiscoveredCache(dataDir, CHATGPT_PROVIDER_ID, { models, flavor: entry.flavor });
+        return models.length;
+    }
+    catch {
+        return undefined;
+    }
+}
 export async function runLogout(args, io, deps = {}) {
     const m = msg().auth.logout;
     const provider = providerOf(args, "logout");
@@ -205,6 +238,7 @@ export async function runLogout(args, io, deps = {}) {
         io.stderr(`ama: ${loginErrorText(error, entry.flavor)}\n`);
         return ExitCode.RuntimeError;
     }
+    clearDiscoveredCache(resolveDataDir({ env: io.env }), provider);
     const text = revoked === undefined
         ? m.local(provider, args.path)
         : revoked

package/dist/auth/chatgpt/follow.d.ts

@@ -0,0 +1,16 @@
+/**
+ * ChatGPT 订阅：请求渠道跟随当前登录方式。
+ *
+ * `siwc` 与 `codex` 两条渠道协议相同（openai-responses），只差端点、请求头与请求体白名单。会话里的模型是启动
+ * （或选模型）时按当时的登录物化的；用户中途 `ama auth logout` 再换另一种方式登录后，同一会话的下一次请求
+ * 拿到的 token 属于新的 flavor。这里在**请求时**按 token 的 flavor（活 token 登记）把模型改到对应渠道：
+ *
+ * - 模型引用没有显式 `@渠道`（`channelPinned` 不为 true）→ 改到登录 flavor 的渠道（端点、渠道头、compat）；
+ * - 显式写了另一渠道 → 不动，协议层报 `chatgpt_flavor_mismatch`（chatgpt-backend.ts）；
+ * - 不是 ChatGPT 后端、token 不是 OAuth、flavor 已一致 → 原样返回。
+ *
+ * 会话里记录的模型（`model_change`）不算显式：恢复时由 startup-steps 去掉渠道再解析。
+ */
+import type { Model, ProviderData } from "../../ai/types.js";
+/** 按 `apiKey` 对应登录的 flavor 改写模型的渠道；无需改动时返回原对象。 */
+export declare function followChatGptLogin(model: Model, provider: ProviderData | undefined, apiKey: string | undefined): Model;

package/dist/auth/chatgpt/follow.js

@@ -0,0 +1,51 @@
+/**
+ * ChatGPT 订阅：请求渠道跟随当前登录方式。
+ *
+ * `siwc` 与 `codex` 两条渠道协议相同（openai-responses），只差端点、请求头与请求体白名单。会话里的模型是启动
+ * （或选模型）时按当时的登录物化的；用户中途 `ama auth logout` 再换另一种方式登录后，同一会话的下一次请求
+ * 拿到的 token 属于新的 flavor。这里在**请求时**按 token 的 flavor（活 token 登记）把模型改到对应渠道：
+ *
+ * - 模型引用没有显式 `@渠道`（`channelPinned` 不为 true）→ 改到登录 flavor 的渠道（端点、渠道头、compat）；
+ * - 显式写了另一渠道 → 不动，协议层报 `chatgpt_flavor_mismatch`（chatgpt-backend.ts）；
+ * - 不是 ChatGPT 后端、token 不是 OAuth、flavor 已一致 → 原样返回。
+ *
+ * 会话里记录的模型（`model_change`）不算显式：恢复时由 startup-steps 去掉渠道再解析。
+ */
+import { liveToken } from "../oauth/live.js";
+function backendOf(compat) {
+    if (typeof compat !== "object" || compat === null)
+        return undefined;
+    const value = compat.chatgptBackend;
+    return value === "siwc" || value === "codex" ? value : undefined;
+}
+/** 按 `apiKey` 对应登录的 flavor 改写模型的渠道；无需改动时返回原对象。 */
+export function followChatGptLogin(model, provider, apiKey) {
+    const current = backendOf(model.compat);
+    if (current === undefined || model.channelPinned === true)
+        return model;
+    const flavor = liveToken(apiKey)?.flavor;
+    if (flavor === undefined || flavor === current)
+        return model;
+    const channels = provider?.channels ?? [];
+    const target = channels.find((c) => backendOf(c.compat) === flavor) ?? channels.find((c) => c.name === flavor);
+    if (target === undefined)
+        return model;
+    const previous = channels.find((c) => c.name === model.channel);
+    // 去掉旧渠道带来的头（codex 的 originator 等），换上新渠道的
+    const headers = { ...model.headers };
+    for (const name of Object.keys(previous?.headers ?? {}))
+        delete headers[name];
+    Object.assign(headers, target.headers);
+    const next = {
+        ...model,
+        api: target.api,
+        baseUrl: target.baseUrl,
+        channel: target.name,
+        compat: { ...model.compat, ...target.compat, chatgptBackend: flavor },
+    };
+    if (Object.keys(headers).length > 0)
+        next.headers = headers;
+    else
+        delete next.headers;
+    return next;
+}

package/dist/auth/chatgpt/presets.d.ts

@@ -9,6 +9,10 @@
  *
  * 覆盖：`AMA_CHATGPT_CLIENT_ID` / `AMA_CHATGPT_ISSUER`（测试用）/ `AMA_CHATGPT_BASE_URL` > 配置
  * `auth.chatgpt.*` > 条目里签发的 client id > 缺省。
+ *
+ * codex 后端的模型列表按 `client_version` 过滤（每个模型带 `minimal_client_version`，版本过低一个都不给）：
+ * 发的是 Codex CLI 的版本号而不是 ama 的，`AMA_CHATGPT_CODEX_CLIENT_VERSION` > `auth.chatgpt.codexClientVersion`
+ * > `DEFAULT_CODEX_CLIENT_VERSION`。推理请求不带版本（Codex 自己也只在 User-Agent 里带）。
  */
 import type { ChatGptAuthConfig, ChatGptFlavor, OAuthAuthEntry } from "../../config/types-w6.js";
 export declare const CHATGPT_PROVIDER_ID = "chatgpt";
@@ -19,6 +23,8 @@ export declare const SIWC_RESOURCE = "https://api.openai.com/v1";
 export declare const SIWC_REQUIRED_SCOPE = "chatgpt.tokens.use.direct";
 export declare const AGENT_NAME_HINT = "ama";
 export declare const DEFAULT_ORIGINATOR = "codex_cli_rs";
+/** codex 后端 `GET /models?client_version=` 用的 Codex CLI 版本号（模型列表按它过滤）。 */
+export declare const DEFAULT_CODEX_CLIENT_VERSION = "0.160.0";
 export declare const CHATGPT_BASE_URLS: Readonly<Record<ChatGptFlavor, string>>;
 export interface AuthorizeInput {
     redirectUri: string;
@@ -61,6 +67,8 @@ export interface PresetOverrides {
     port?: number | undefined;
 }
 export declare function chatgptBaseUrl(flavor: ChatGptFlavor, env?: Readonly<Record<string, string | undefined>>): string;
+/** codex 模型列表的 `client_version`：环境变量 > 用户级配置 > 缺省。 */
+export declare function codexClientVersion(config?: ChatGptAuthConfig, env?: Readonly<Record<string, string | undefined>>): string;
 export declare function resolveChatGptPreset(flavor: ChatGptFlavor, overrides?: PresetOverrides): ChatGptPreset;
 /** 授权 URL（参数顺序固定，便于测试）。 */
 export declare function authorizeUrl(preset: ChatGptPreset, input: AuthorizeInput): string;

package/dist/auth/chatgpt/presets.js

@@ -9,6 +9,10 @@
  *
  * 覆盖：`AMA_CHATGPT_CLIENT_ID` / `AMA_CHATGPT_ISSUER`（测试用）/ `AMA_CHATGPT_BASE_URL` > 配置
  * `auth.chatgpt.*` > 条目里签发的 client id > 缺省。
+ *
+ * codex 后端的模型列表按 `client_version` 过滤（每个模型带 `minimal_client_version`，版本过低一个都不给）：
+ * 发的是 Codex CLI 的版本号而不是 ama 的，`AMA_CHATGPT_CODEX_CLIENT_VERSION` > `auth.chatgpt.codexClientVersion`
+ * > `DEFAULT_CODEX_CLIENT_VERSION`。推理请求不带版本（Codex 自己也只在 User-Agent 里带）。
  */
 export const CHATGPT_PROVIDER_ID = "chatgpt";
 export const DEFAULT_ISSUER = "https://auth.openai.com";
@@ -18,6 +22,8 @@ export const SIWC_RESOURCE = "https://api.openai.com/v1";
 export const SIWC_REQUIRED_SCOPE = "chatgpt.tokens.use.direct";
 export const AGENT_NAME_HINT = "ama";
 export const DEFAULT_ORIGINATOR = "codex_cli_rs";
+/** codex 后端 `GET /models?client_version=` 用的 Codex CLI 版本号（模型列表按它过滤）。 */
+export const DEFAULT_CODEX_CLIENT_VERSION = "0.160.0";
 export const CHATGPT_BASE_URLS = {
     siwc: "https://api.openai.com/v1",
     codex: "https://chatgpt.com/backend-api/codex",
@@ -29,6 +35,12 @@ function nonEmpty(value) {
 export function chatgptBaseUrl(flavor, env = process.env) {
     return nonEmpty(env["AMA_CHATGPT_BASE_URL"]) ?? CHATGPT_BASE_URLS[flavor];
 }
+/** codex 模型列表的 `client_version`：环境变量 > 用户级配置 > 缺省。 */
+export function codexClientVersion(config, env = process.env) {
+    return (nonEmpty(env["AMA_CHATGPT_CODEX_CLIENT_VERSION"]) ??
+        nonEmpty(config?.codexClientVersion) ??
+        DEFAULT_CODEX_CLIENT_VERSION);
+}
 export function resolveChatGptPreset(flavor, overrides = {}) {
     const env = overrides.env ?? process.env;
     const config = overrides.config;

package/dist/auth/testing/fake-oauth.d.ts

@@ -4,7 +4,8 @@
  * OAuth：authorize（记参数、302 回 redirect_uri；SIWC 注册时带签发的 client_id 与 scope）、token（校验 PKCE；
  * 授权码 / 刷新；刷新轮换，旧 refresh token 再用 → `refresh_token_reused`）、deviceauth、discovery + JWKS
  * （现场生成 RSA 密钥对）、revoke。后端：`/v1/responses`、`/codex/responses`、`/v1/models`、`/codex/models`、
- * `/wham/usage`，按脚本回放。
+ * `/wham/usage`，按脚本回放；给了 `codexModels` 时 `/codex/models` 改按 `client_version` 过滤（不带 → 400，
+ * 低于条目的 `minimal_client_version` 不给），与真实 codex 后端一致。
  */
 export interface RecordedRequest {
     method: string;
@@ -37,6 +38,8 @@ export interface FakeOAuthOptions {
     /** 设备码轮询先返回几次 403。 */
     devicePending?: number;
     marker?: string;
+    /** codex 模型目录（`/codex/models` 按 `client_version` 过滤后返回 `{ models }`）。 */
+    codexModels?: Record<string, unknown>[];
 }
 export declare class FakeOAuthServer {
     readonly options: FakeOAuthOptions;

package/dist/auth/testing/fake-oauth.js

@@ -4,10 +4,22 @@
  * OAuth：authorize（记参数、302 回 redirect_uri；SIWC 注册时带签发的 client_id 与 scope）、token（校验 PKCE；
  * 授权码 / 刷新；刷新轮换，旧 refresh token 再用 → `refresh_token_reused`）、deviceauth、discovery + JWKS
  * （现场生成 RSA 密钥对）、revoke。后端：`/v1/responses`、`/codex/responses`、`/v1/models`、`/codex/models`、
- * `/wham/usage`，按脚本回放。
+ * `/wham/usage`，按脚本回放；给了 `codexModels` 时 `/codex/models` 改按 `client_version` 过滤（不带 → 400，
+ * 低于条目的 `minimal_client_version` 不给），与真实 codex 后端一致。
  */
 import { createHash, generateKeyPairSync, randomBytes, sign } from "node:crypto";
 import { createServer } from "node:http";
+/** `a.b.c` 比较；非数字段按 0。 */
+function compareVersion(a, b) {
+    const pa = a.split(".").map((x) => Number.parseInt(x, 10) || 0);
+    const pb = b.split(".").map((x) => Number.parseInt(x, 10) || 0);
+    for (let i = 0; i < Math.max(pa.length, pb.length); i++) {
+        const d = (pa[i] ?? 0) - (pb[i] ?? 0);
+        if (d !== 0)
+            return d;
+    }
+    return 0;
+}
 function b64url(value) {
     return Buffer.from(JSON.stringify(value)).toString("base64url");
 }
@@ -212,6 +224,17 @@ export class FakeOAuthServer {
                     code_challenge: "x",
                 });
             }
+            case "/codex/models": {
+                const catalog = this.options.codexModels;
+                if (catalog === undefined)
+                    return this.backend(res);
+                const version = url.searchParams.get("client_version");
+                if (version === null)
+                    return json(400, { error: { message: "client_version required" } });
+                return json(200, {
+                    models: catalog.filter((m) => compareVersion(version, String(m["minimal_client_version"] ?? "0")) >= 0),
+                });
+            }
             case "/revoke":
                 this.revoked.push(params()["token"] ?? "");
                 res.writeHead(200).end();