codex-slot 0.1.27 → 0.1.29

package/dist/relay-commands.js

@@ -0,0 +1,166 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.handleRelayAdd = handleRelayAdd;
+exports.handleRelayList = handleRelayList;
+exports.handleRelayRemove = handleRelayRemove;
+exports.handleRelayRename = handleRelayRename;
+exports.handleRelayEnable = handleRelayEnable;
+exports.handleRelayDisable = handleRelayDisable;
+exports.handleUseRelay = handleUseRelay;
+exports.handleUseAuthPool = handleUseAuthPool;
+exports.handleCurrent = handleCurrent;
+const relay_store_1 = require("./relay-store");
+const state_1 = require("./state");
+const text_1 = require("./text");
+function maskApiKey(apiKey) {
+    if (apiKey.length <= 8) {
+        return "****";
+    }
+    return `${apiKey.slice(0, 4)}...${apiKey.slice(-4)}`;
+}
+function formatModelRoute() {
+    const route = (0, state_1.getSelectedModelRoute)();
+    if (route.mode === "relay_slot") {
+        return `relay:${route.relay_slot_id}`;
+    }
+    return "auth_pool";
+}
+/**
+ * 新增 OpenAI-compatible relay slot。
+ *
+ * @param name relay slot 名称。
+ * @param options relay 参数，必须包含 baseUrl 与 apiKey。
+ * @returns 无返回值。
+ * @throws 当 slot 已存在、URL 非法或配置写入失败时抛出异常。
+ */
+function handleRelayAdd(name, options) {
+    const slot = (0, relay_store_1.addRelaySlot)({
+        name,
+        baseUrl: options.baseUrl,
+        apiKey: options.apiKey
+    });
+    console.log((0, text_1.bi)(`已添加中转槽位: ${slot.name}`, `Relay slot added: ${slot.name}`));
+}
+/**
+ * 列出 OpenAI-compatible relay slot。
+ *
+ * @returns 无返回值。
+ * @throws 当配置读取失败时抛出异常。
+ */
+function handleRelayList() {
+    const slots = (0, relay_store_1.listRelaySlots)();
+    const route = (0, state_1.getSelectedModelRoute)();
+    if (slots.length === 0) {
+        console.log((0, text_1.bi)("当前没有中转槽位。", "No relay slots found."));
+        return;
+    }
+    for (const slot of slots) {
+        const selected = route.mode === "relay_slot" && route.relay_slot_id === slot.id ? "*" : " ";
+        const enabled = slot.enabled ? "enabled" : "disabled";
+        console.log(`${selected} ${slot.name}  ${enabled}  ${slot.base_url}  key=${maskApiKey(slot.api_key)}`);
+    }
+}
+/**
+ * 删除 OpenAI-compatible relay slot。
+ *
+ * @param name relay slot 名称。
+ * @returns 无返回值。
+ * @throws 当配置写入失败时抛出异常。
+ */
+function handleRelayRemove(name) {
+    const removed = (0, relay_store_1.removeRelaySlot)(name);
+    if (!removed) {
+        console.log((0, text_1.bi)(`未找到中转槽位: ${name}`, `Relay slot not found: ${name}`));
+        return;
+    }
+    console.log((0, text_1.bi)(`已删除中转槽位: ${name}`, `Relay slot removed: ${name}`));
+}
+/**
+ * 重命名 OpenAI-compatible relay slot。
+ *
+ * @param oldName 原 slot 名称。
+ * @param newName 新 slot 名称。
+ * @returns 无返回值。
+ * @throws 当旧 slot 不存在、新 slot 已存在或写入失败时抛出异常。
+ */
+function handleRelayRename(oldName, newName) {
+    const slot = (0, relay_store_1.renameRelaySlot)(oldName, newName);
+    console.log((0, text_1.bi)(`已重命名中转槽位: ${oldName} -> ${slot.name}`, `Relay slot renamed: ${oldName} -> ${slot.name}`));
+}
+/**
+ * 启用或禁用 OpenAI-compatible relay slot。
+ *
+ * @param name relay slot 名称。
+ * @param enabled 是否启用。
+ * @returns 无返回值。
+ * @throws 当 slot 不存在或写入失败时抛出异常。
+ */
+function handleRelayEnabled(name, enabled) {
+    const slot = (0, relay_store_1.setRelaySlotEnabled)(name, enabled);
+    console.log(enabled
+        ? (0, text_1.bi)(`已启用中转槽位: ${slot.name}`, `Relay slot enabled: ${slot.name}`)
+        : (0, text_1.bi)(`已禁用中转槽位: ${slot.name}`, `Relay slot disabled: ${slot.name}`));
+}
+/**
+ * 启用 OpenAI-compatible relay slot。
+ *
+ * @param name relay slot 名称。
+ * @returns 无返回值。
+ * @throws 当 slot 不存在或写入失败时抛出异常。
+ */
+function handleRelayEnable(name) {
+    handleRelayEnabled(name, true);
+}
+/**
+ * 禁用 OpenAI-compatible relay slot。
+ *
+ * @param name relay slot 名称。
+ * @returns 无返回值。
+ * @throws 当 slot 不存在或写入失败时抛出异常。
+ */
+function handleRelayDisable(name) {
+    handleRelayEnabled(name, false);
+}
+/**
+ * 将模型请求固定到指定 relay slot。
+ *
+ * @param name relay slot 名称。
+ * @returns 无返回值。
+ * @throws 当 relay slot 不存在或状态写入失败时抛出异常。
+ */
+function handleUseRelay(name) {
+    const slot = (0, relay_store_1.findRelaySlot)(name);
+    if (!slot) {
+        throw new Error((0, text_1.bi)(`未找到中转槽位 ${name}`, `Relay slot not found: ${name}`));
+    }
+    if (!slot.enabled) {
+        throw new Error((0, text_1.bi)(`中转槽位 ${name} 已禁用`, `Relay slot is disabled: ${name}`));
+    }
+    (0, state_1.setSelectedModelRoute)({
+        mode: "relay_slot",
+        relay_slot_id: slot.id
+    });
+    console.log((0, text_1.bi)(`模型请求已固定到中转槽位: ${slot.name}`, `Model route fixed to relay slot: ${slot.name}`));
+}
+/**
+ * 恢复模型请求到官方账号自动调度池。
+ *
+ * @returns 无返回值。
+ * @throws 当状态写入失败时抛出异常。
+ */
+function handleUseAuthPool() {
+    (0, state_1.setSelectedModelRoute)({
+        mode: "auth_pool"
+    });
+    console.log((0, text_1.bi)("模型请求已恢复官方账号池。", "Model route restored to auth pool."));
+}
+/**
+ * 输出当前模型出口与 Codex App 登录态选择。
+ *
+ * @returns 无返回值。
+ * @throws 当状态读取失败时抛出异常。
+ */
+function handleCurrent() {
+    console.log(`model_route=${formatModelRoute()}`);
+    console.log(`codex_auth=${(0, state_1.getSelectedCodexAuthAccountId)() ?? "none"}`);
+}

package/dist/relay-proxy-service.js

@@ -0,0 +1,160 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.proxyRelaySlot = void 0;
+exports.buildRelayHeaders = buildRelayHeaders;
+exports.sendRelayRequest = sendRelayRequest;
+exports.createRelayProxyService = createRelayProxyService;
+const undici_1 = require("undici");
+/**
+ * 构造 OpenAI-compatible relay 上游请求头。
+ *
+ * 业务含义：
+ * 1. 客户端发给本地 cslot 的 Authorization 不代表 relay 凭据，不能透传。
+ * 2. relay 模式只使用 relay slot 自己的 api_key 鉴权，并保留其余兼容请求头。
+ *
+ * @param requestHeaders 客户端发到本地服务的原始请求头。
+ * @param apiKey relay slot 的 API key。
+ * @param bodyLength 请求体字节长度；无 body 时不写 content-length。
+ * @returns 可发往 relay 上游的请求头。
+ * @throws 无显式抛出。
+ */
+function buildRelayHeaders(requestHeaders, apiKey, bodyLength) {
+    const headers = {};
+    for (const [headerName, headerValue] of Object.entries(requestHeaders)) {
+        const normalizedName = headerName.toLowerCase();
+        if (headerValue == null ||
+            normalizedName === "authorization" ||
+            normalizedName === "host" ||
+            normalizedName === "connection" ||
+            normalizedName === "content-length") {
+            continue;
+        }
+        headers[normalizedName] = Array.isArray(headerValue)
+            ? headerValue.join(", ")
+            : headerValue;
+    }
+    headers.authorization = `Bearer ${apiKey}`;
+    if (typeof bodyLength === "number") {
+        headers["content-length"] = String(bodyLength);
+    }
+    headers["user-agent"] = "codex-slot/0.1.1";
+    return headers;
+}
+/**
+ * 将本地 `/v1/*` 请求解析成 relay base_url 后的相对路径。
+ *
+ * @param url 本地代理收到的 URL。
+ * @returns relay 上游相对 path 与 query；不属于 `/v1/*` 时返回 `null`。
+ * @throws URL 解析失败时透传异常。
+ */
+function resolveRelayPath(url) {
+    const parsedUrl = new URL(url, "http://127.0.0.1");
+    const openAiPrefix = "/v1";
+    if (!parsedUrl.pathname.startsWith(`${openAiPrefix}/`)) {
+        return null;
+    }
+    return `${parsedUrl.pathname.slice(openAiPrefix.length)}${parsedUrl.search}`;
+}
+/**
+ * 提取上游响应中允许透传给客户端的响应头。
+ *
+ * @param headers 上游响应头对象。
+ * @returns 可透传响应头。
+ * @throws 无显式抛出。
+ */
+function pickResponseHeaders(headers) {
+    const picked = {};
+    const contentType = headers["content-type"];
+    const cacheControl = headers["cache-control"];
+    if (typeof contentType === "string") {
+        picked["content-type"] = contentType;
+    }
+    if (typeof cacheControl === "string") {
+        picked["cache-control"] = cacheControl;
+    }
+    return picked;
+}
+function buildSendResult(statusCode, payload, headers) {
+    return {
+        type: "send",
+        statusCode,
+        payload,
+        headers
+    };
+}
+/**
+ * 向 OpenAI-compatible relay 上游发送一次请求。
+ *
+ * @param options relay 请求参数。
+ * @returns undici 上游响应对象。
+ * @throws 当网络层或 undici 请求失败时透传底层异常。
+ */
+async function sendRelayRequest(options) {
+    const baseUrl = options.baseUrl.replace(/\/+$/, "");
+    const pathWithQuery = options.pathWithQuery.startsWith("/")
+        ? options.pathWithQuery
+        : `/${options.pathWithQuery}`;
+    return await (0, undici_1.request)(`${baseUrl}${pathWithQuery}`, {
+        method: options.method,
+        headers: options.headers,
+        body: options.body && options.body.length > 0 ? options.body : undefined
+    });
+}
+/**
+ * 创建 relay 模型代理服务。
+ *
+ * 业务含义：
+ * 1. relay slot 是手动固定的模型出口，不参与官方账号自动调度。
+ * 2. relay 上游返回 401/429/5xx 时原样返回给客户端，不回退到官方账号。
+ *
+ * @param overrides 可选依赖覆盖项。
+ * @returns relay 代理服务实例。
+ * @throws 无显式抛出。
+ */
+function createRelayProxyService(overrides) {
+    const dependencies = {
+        sendRelayRequest,
+        ...overrides
+    };
+    return {
+        async proxyRelaySlot(options) {
+            const pathWithQuery = resolveRelayPath(options.request.url);
+            if (!pathWithQuery) {
+                return buildSendResult(404, {
+                    error: {
+                        message: "Unsupported relay proxy path",
+                        type: "unsupported_relay_proxy_path"
+                    }
+                });
+            }
+            const bodyLength = options.request.body && options.request.body.length > 0
+                ? options.request.body.length
+                : undefined;
+            let upstream;
+            try {
+                upstream = await dependencies.sendRelayRequest({
+                    baseUrl: options.slot.base_url,
+                    method: options.request.method.toUpperCase(),
+                    pathWithQuery,
+                    headers: buildRelayHeaders(options.request.headers, options.slot.api_key, bodyLength),
+                    body: options.request.body
+                });
+            }
+            catch (error) {
+                return buildSendResult(502, {
+                    error: {
+                        message: error instanceof Error ? error.message : String(error),
+                        type: "relay_request_failed"
+                    }
+                });
+            }
+            return {
+                type: "proxy",
+                statusCode: upstream.statusCode,
+                headers: pickResponseHeaders(upstream.headers),
+                body: upstream.body
+            };
+        }
+    };
+}
+exports.proxyRelaySlot = createRelayProxyService().proxyRelaySlot;

package/dist/relay-store.js

@@ -0,0 +1,138 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.listRelaySlots = listRelaySlots;
+exports.findRelaySlot = findRelaySlot;
+exports.addRelaySlot = addRelaySlot;
+exports.removeRelaySlot = removeRelaySlot;
+exports.renameRelaySlot = renameRelaySlot;
+exports.setRelaySlotEnabled = setRelaySlotEnabled;
+const config_1 = require("./config");
+const state_1 = require("./state");
+const text_1 = require("./text");
+/**
+ * 列出所有 OpenAI-compatible relay 槽位。
+ *
+ * @returns 当前配置中的 relay slot 列表。
+ * @throws 当配置读取失败时抛出异常。
+ */
+function listRelaySlots() {
+    return (0, config_1.loadConfig)().relay_slots;
+}
+/**
+ * 查找指定 relay slot。
+ *
+ * @param slotId relay slot 标识。
+ * @returns 命中时返回 slot；不存在时返回 `null`。
+ * @throws 当配置读取失败时抛出异常。
+ */
+function findRelaySlot(slotId) {
+    return listRelaySlots().find((item) => item.id === slotId) ?? null;
+}
+/**
+ * 新增一个 OpenAI-compatible relay slot。
+ *
+ * @param input relay slot 配置。
+ * @returns 新增后的 relay slot。
+ * @throws 当同名 slot 已存在或配置写入失败时抛出异常。
+ */
+function addRelaySlot(input) {
+    const config = (0, config_1.loadConfig)();
+    if (config.relay_slots.some((item) => item.id === input.name)) {
+        throw new Error((0, text_1.bi)(`中转槽位 ${input.name} 已存在`, `Relay slot already exists: ${input.name}`));
+    }
+    try {
+        new URL(input.baseUrl);
+    }
+    catch {
+        throw new Error((0, text_1.bi)(`中转 base_url 非法: ${input.baseUrl}`, `Invalid relay base_url: ${input.baseUrl}`));
+    }
+    const slot = {
+        id: input.name,
+        name: input.name,
+        base_url: input.baseUrl,
+        api_key: input.apiKey,
+        enabled: true,
+        imported_at: new Date().toISOString()
+    };
+    config.relay_slots.push(slot);
+    (0, config_1.saveConfig)(config);
+    return slot;
+}
+/**
+ * 删除指定 relay slot，并在当前模型出口指向它时恢复官方账号池。
+ *
+ * @param slotId relay slot 标识。
+ * @returns 被删除的 slot；不存在时返回 `null`。
+ * @throws 当配置或状态写入失败时抛出异常。
+ */
+function removeRelaySlot(slotId) {
+    const config = (0, config_1.loadConfig)();
+    const index = config.relay_slots.findIndex((item) => item.id === slotId);
+    if (index < 0) {
+        return null;
+    }
+    const [removed] = config.relay_slots.splice(index, 1);
+    (0, config_1.saveConfig)(config);
+    const route = (0, state_1.getSelectedModelRoute)();
+    if (route.mode === "relay_slot" && route.relay_slot_id === slotId) {
+        (0, state_1.setSelectedModelRoute)({ mode: "auth_pool" });
+    }
+    return removed;
+}
+/**
+ * 重命名 relay slot，并迁移当前模型出口选择。
+ *
+ * @param oldName 原 slot 标识。
+ * @param newName 新 slot 标识。
+ * @returns 重命名后的 relay slot。
+ * @throws 当旧 slot 不存在、新 slot 已存在或写入失败时抛出异常。
+ */
+function renameRelaySlot(oldName, newName) {
+    const config = (0, config_1.loadConfig)();
+    const index = config.relay_slots.findIndex((item) => item.id === oldName);
+    if (index < 0) {
+        throw new Error((0, text_1.bi)(`未找到中转槽位 ${oldName}`, `Relay slot not found: ${oldName}`));
+    }
+    if (config.relay_slots.some((item) => item.id === newName)) {
+        throw new Error((0, text_1.bi)(`中转槽位 ${newName} 已存在`, `Relay slot already exists: ${newName}`));
+    }
+    const renamed = {
+        ...config.relay_slots[index],
+        id: newName,
+        name: newName
+    };
+    config.relay_slots[index] = renamed;
+    (0, config_1.saveConfig)(config);
+    (0, state_1.updateState)((state) => {
+        if (state.selected_model_route?.mode === "relay_slot" &&
+            state.selected_model_route.relay_slot_id === oldName) {
+            state.selected_model_route = {
+                mode: "relay_slot",
+                relay_slot_id: newName
+            };
+        }
+    });
+    return renamed;
+}
+/**
+ * 更新 relay slot 启用状态。
+ *
+ * @param slotId relay slot 标识。
+ * @param enabled 是否启用。
+ * @returns 更新后的 relay slot。
+ * @throws 当 slot 不存在或配置写入失败时抛出异常。
+ */
+function setRelaySlotEnabled(slotId, enabled) {
+    const config = (0, config_1.loadConfig)();
+    const index = config.relay_slots.findIndex((item) => item.id === slotId);
+    if (index < 0) {
+        throw new Error((0, text_1.bi)(`未找到中转槽位 ${slotId}`, `Relay slot not found: ${slotId}`));
+    }
+    const updated = {
+        ...config.relay_slots[index],
+        enabled
+    };
+    config.relay_slots[index] = updated;
+    (0, config_1.saveConfig)(config);
+    return updated;
+}

package/dist/server.js

@@ -7,7 +7,7 @@ exports.startServer = startServer;
 const fastify_1 = __importDefault(require("fastify"));
 const config_1 = require("./config");
 const backend_proxy_service_1 = require("./backend-proxy-service");
-const proxy_retry_service_1 = require("./proxy-retry-service");
+const model_proxy_dispatcher_1 = require("./model-proxy-dispatcher");
 const status_1 = require("./status");
 const scheduler_1 = require("./scheduler");
 const usage_sync_1 = require("./usage-sync");
@@ -91,7 +91,7 @@ async function startServer(port) {
     });
     const codexProxyHandler = async (request, reply) => {
         const requestBody = await readRawRequestBody(request.body);
-        const result = await (0, proxy_retry_service_1.proxyCodexWithRetry)({
+        const result = await (0, model_proxy_dispatcher_1.proxyModelWithRoute)({
             method: request.method,
             url: request.url,
             headers: request.headers,

package/dist/state.js

@@ -17,6 +17,8 @@ exports.clearUsageRefreshError = clearUsageRefreshError;
 exports.getUsageRefreshError = getUsageRefreshError;
 exports.getSelectedCodexAuthAccountId = getSelectedCodexAuthAccountId;
 exports.setSelectedCodexAuthAccountId = setSelectedCodexAuthAccountId;
+exports.getSelectedModelRoute = getSelectedModelRoute;
+exports.setSelectedModelRoute = setSelectedModelRoute;
 exports.getManagedCodexConfigState = getManagedCodexConfigState;
 exports.getManagedCodexAuthState = getManagedCodexAuthState;
 exports.setManagedCodexConfigState = setManagedCodexConfigState;
@@ -44,6 +46,9 @@ function createDefaultState() {
     return {
         state_version: STATE_SCHEMA_VERSION,
         selected_codex_auth_account_id: null,
+        selected_model_route: {
+            mode: "auth_pool"
+        },
         account_blocks: {},
         usage_cache: {},
         usage_refresh_errors: {},
@@ -64,6 +69,7 @@ function normalizeState(parsed) {
     return {
         state_version: STATE_SCHEMA_VERSION,
         selected_codex_auth_account_id: parsed?.selected_codex_auth_account_id ?? defaults.selected_codex_auth_account_id,
+        selected_model_route: normalizeModelRouteSelection(parsed?.selected_model_route ?? defaults.selected_model_route),
         account_blocks: parsed?.account_blocks ?? defaults.account_blocks,
         usage_cache: parsed?.usage_cache ?? defaults.usage_cache,
         usage_refresh_errors: parsed?.usage_refresh_errors ?? defaults.usage_refresh_errors,
@@ -72,6 +78,24 @@ function normalizeState(parsed) {
         managed_codex_config: parsed?.managed_codex_config ?? defaults.managed_codex_config
     };
 }
+/**
+ * 归一化当前模型出口选择，避免历史 state 或手工编辑内容让代理入口进入未知模式。
+ *
+ * @param value state 文件中记录的模型出口选择。
+ * @returns 可安全使用的模型出口选择；非法内容回退到官方账号池。
+ * @throws 无显式抛出。
+ */
+function normalizeModelRouteSelection(value) {
+    if (value?.mode === "relay_slot" && value.relay_slot_id) {
+        return {
+            mode: "relay_slot",
+            relay_slot_id: value.relay_slot_id
+        };
+    }
+    return {
+        mode: "auth_pool"
+    };
+}
 /**
  * 读取 cslot 的本地运行状态；文件不存在时返回默认空状态。
  *
@@ -261,6 +285,32 @@ function setSelectedCodexAuthAccountId(accountId) {
         state.selected_codex_auth_account_id = accountId;
     });
 }
+/**
+ * 读取当前模型请求出口选择。
+ *
+ * 业务含义：
+ * 1. `auth_pool` 表示 `/v1/*` 继续走官方账号自动调度。
+ * 2. `relay_slot` 表示 `/v1/*` 固定走指定 OpenAI-compatible 中转槽位。
+ * 3. 该选择不影响 Codex App 主登录态与 `/backend-api/*` 插件链路。
+ *
+ * @returns 当前模型出口选择。
+ * @throws 当 state 文件读取或解析失败时透传底层异常。
+ */
+function getSelectedModelRoute() {
+    return normalizeModelRouteSelection(loadState().selected_model_route);
+}
+/**
+ * 保存当前模型请求出口选择。
+ *
+ * @param selection 模型出口选择；`null` 表示恢复官方账号池。
+ * @returns 无返回值。
+ * @throws 当 state 文件写入失败时透传底层异常。
+ */
+function setSelectedModelRoute(selection) {
+    updateState((state) => {
+        state.selected_model_route = normalizeModelRouteSelection(selection);
+    });
+}
 /**
  * 读取当前记录的 Codex `config.toml` 接管快照。
  *