mcp-codex-worker 0.1.0

package/dist/src/services/codex-runtime.js

@@ -0,0 +1,371 @@
+import process from 'node:process';
+import { copyFile, mkdtemp, mkdir, readFile, writeFile } from 'node:fs/promises';
+import { basename, join } from 'node:path';
+import { tmpdir } from 'node:os';
+import { CODEX_APP_SERVER_ARGS_ENV, CODEX_APP_SERVER_COMMAND_ENV, REQUEST_TIMEOUT_MS, } from '../config/defaults.js';
+import { AppServerClient } from './app-server-client.js';
+import { appendFleetDeveloperInstructions } from './fleet-mode.js';
+import { buildModelCatalog, describeAllowedModels, resolveModel, } from './model-catalog.js';
+import { ProfileManager, } from './profile-manager.js';
+function parseArgs(raw) {
+    if (!raw || !raw.trim()) {
+        return ['app-server', '--listen', 'stdio://'];
+    }
+    const trimmed = raw.trim();
+    if (trimmed.startsWith('[')) {
+        try {
+            const parsed = JSON.parse(trimmed);
+            if (Array.isArray(parsed) && parsed.every((value) => typeof value === 'string')) {
+                return parsed;
+            }
+        }
+        catch {
+            // fall through to token split
+        }
+    }
+    return trimmed.split(/\s+/g).filter(Boolean);
+}
+function isRateLimitSignal(error) {
+    const message = error instanceof Error ? error.message : String(error);
+    const normalized = message.toLowerCase();
+    return (normalized.includes('rate limit') ||
+        normalized.includes('usage limit') ||
+        normalized.includes('usagelimitexceeded') ||
+        normalized.includes('429'));
+}
+function isAuthSignal(error) {
+    const message = error instanceof Error ? error.message : String(error);
+    const normalized = message.toLowerCase();
+    return (normalized.includes('unauthorized') ||
+        normalized.includes('requires auth') ||
+        normalized.includes('authentication') ||
+        normalized.includes('login'));
+}
+function isConfigCompatSignal(error) {
+    const message = error instanceof Error ? error.message : String(error);
+    const normalized = message.toLowerCase();
+    return (normalized.includes('error deriving config') ||
+        normalized.includes('agentroletoml') ||
+        (normalized.includes('invalid type: integer') && normalized.includes('agents')));
+}
+function sanitizeConfigForCompatibility(rawConfig) {
+    const lines = rawConfig.split(/\r?\n/g);
+    const sanitized = [];
+    let skippingAgentsSection = false;
+    for (const line of lines) {
+        const trimmed = line.trim();
+        const isSection = trimmed.startsWith('[') && trimmed.endsWith(']');
+        if (isSection) {
+            if (trimmed === '[agents]' || trimmed.startsWith('[agents.')) {
+                skippingAgentsSection = true;
+                continue;
+            }
+            skippingAgentsSection = false;
+        }
+        if (skippingAgentsSection) {
+            continue;
+        }
+        sanitized.push(line);
+    }
+    return sanitized.join('\n');
+}
+export class CodexRuntime {
+    profileManager;
+    command;
+    args;
+    env;
+    clients = new Map();
+    effectiveCodexHomes = new Map();
+    modelCatalog;
+    knownThreadIds = new Set();
+    loadedThreadIds = new Set();
+    constructor(commandOptions = {}) {
+        this.profileManager = ProfileManager.fromEnvironment();
+        this.command = commandOptions.command ?? process.env[CODEX_APP_SERVER_COMMAND_ENV] ?? 'codex';
+        this.args = commandOptions.args ?? parseArgs(process.env[CODEX_APP_SERVER_ARGS_ENV]);
+        this.env = commandOptions.env ?? process.env;
+    }
+    async shutdown() {
+        await Promise.all([...this.clients.values()].map((client) => client.stop()));
+        this.clients.clear();
+        this.loadedThreadIds.clear();
+    }
+    getPendingServerRequests(includeResolved = false) {
+        return this.getCurrentClient().listServerRequests(includeResolved);
+    }
+    getPendingServerRequest(id) {
+        return this.getCurrentClient().getServerRequest(id);
+    }
+    async respondToServerRequest(id, payload) {
+        await this.getCurrentClient().respondToServerRequest(id, payload);
+    }
+    async waitForOperation(operationId, timeoutMs, pollIntervalMs) {
+        return this.getCurrentClient().waitForOperation(operationId, {
+            timeoutMs,
+            pollIntervalMs,
+        });
+    }
+    getOperation(operationId) {
+        return this.getCurrentClient().getOperation(operationId);
+    }
+    async request(method, params, options = {}) {
+        const response = await this.withFailover(async (client, profile) => {
+            try {
+                return await client.request(method, params, { timeoutMs: options.timeoutMs ?? REQUEST_TIMEOUT_MS });
+            }
+            catch (error) {
+                if (!isConfigCompatSignal(error)) {
+                    throw error;
+                }
+                const compatClient = await this.activateCompatibilityClient(profile);
+                return await compatClient.request(method, params, { timeoutMs: options.timeoutMs ?? REQUEST_TIMEOUT_MS });
+            }
+        });
+        this.trackLoadedThread(method, response, params);
+        return response;
+    }
+    async requestWithBridge(method, params, options = {}) {
+        const response = await this.withFailover(async (client, profile) => {
+            try {
+                return await client.requestWithBridge(method, params, {
+                    timeoutMs: options.timeoutMs ?? REQUEST_TIMEOUT_MS,
+                    bridgeTimeoutMs: options.bridgeTimeoutMs,
+                    threadId: options.threadId,
+                });
+            }
+            catch (error) {
+                if (!isConfigCompatSignal(error)) {
+                    throw error;
+                }
+                const compatClient = await this.activateCompatibilityClient(profile);
+                return await compatClient.requestWithBridge(method, params, {
+                    timeoutMs: options.timeoutMs ?? REQUEST_TIMEOUT_MS,
+                    bridgeTimeoutMs: options.bridgeTimeoutMs,
+                    threadId: options.threadId,
+                });
+            }
+        });
+        this.trackLoadedThread(method, response.result ?? response, params);
+        return response;
+    }
+    async listModels(includeHidden = false) {
+        const response = await this.request('model/list', { includeHidden });
+        const models = (response.data ?? []).map((row) => ({
+            id: String(row.id),
+            model: String(row.model),
+            displayName: String(row.displayName),
+            hidden: Boolean(row.hidden),
+            upgrade: typeof row.upgrade === 'string' ? row.upgrade : null,
+            isDefault: Boolean(row.isDefault),
+        }));
+        this.modelCatalog = buildModelCatalog(models);
+        return includeHidden ? models : models.filter((model) => !model.hidden);
+    }
+    async resolveRequestedModel(requestedModel) {
+        if (!this.modelCatalog) {
+            await this.listModels(true);
+        }
+        const catalog = this.modelCatalog;
+        if (!catalog) {
+            throw new Error('Failed to build model catalog from model/list.');
+        }
+        try {
+            const resolution = resolveModel(catalog, requestedModel);
+            return {
+                resolvedModel: resolution.resolved,
+                remappedFrom: resolution.remappedFrom,
+                allowedModels: catalog.visibleModels.map((model) => model.id).sort(),
+            };
+        }
+        catch (error) {
+            if (error instanceof Error && error.message.startsWith('Unsupported model')) {
+                throw new Error(`Unsupported model "${requestedModel}". Allowed models: ${describeAllowedModels(catalog)}`);
+            }
+            throw error;
+        }
+    }
+    async listVisibleModelIds() {
+        if (!this.modelCatalog) {
+            await this.listModels(false);
+        }
+        return (this.modelCatalog?.visibleModels ?? []).map((model) => model.id).sort();
+    }
+    buildThreadStartParams(input) {
+        return this.resolveRequestedModel(input.model).then((resolved) => {
+            const params = {
+                model: resolved.resolvedModel,
+                modelProvider: 'openai',
+                cwd: input.cwd ?? process.cwd(),
+                approvalPolicy: 'on-request',
+                sandbox: 'workspace-write',
+                developerInstructions: appendFleetDeveloperInstructions(input.developerInstructions),
+                experimentalRawEvents: false,
+                persistExtendedHistory: false,
+            };
+            return {
+                params,
+                remappedFrom: resolved.remappedFrom,
+            };
+        });
+    }
+    buildThreadResumeParams(input) {
+        return this.resolveRequestedModel(input.model).then((resolved) => {
+            const params = {
+                threadId: input.threadId,
+                model: resolved.resolvedModel,
+                modelProvider: 'openai',
+                cwd: input.cwd ?? process.cwd(),
+                approvalPolicy: 'on-request',
+                sandbox: 'workspace-write',
+                developerInstructions: appendFleetDeveloperInstructions(input.developerInstructions),
+                persistExtendedHistory: false,
+            };
+            return {
+                params,
+                remappedFrom: resolved.remappedFrom,
+            };
+        });
+    }
+    buildTurnStartParams(input) {
+        return this.resolveRequestedModel(input.model).then((resolved) => ({
+            params: {
+                threadId: input.threadId,
+                model: resolved.resolvedModel,
+                input: [{
+                        type: 'text',
+                        text: input.userInput,
+                        text_elements: [],
+                    }],
+            },
+            remappedFrom: resolved.remappedFrom,
+        }));
+    }
+    buildTurnSteerParams(input) {
+        return {
+            threadId: input.threadId,
+            expectedTurnId: input.expectedTurnId,
+            input: [{
+                    type: 'text',
+                    text: input.userInput,
+                    text_elements: [],
+                }],
+        };
+    }
+    async getThreadEvents(threadId) {
+        return this.getCurrentClient().getThreadEvents(threadId);
+    }
+    rememberThreadId(threadId) {
+        this.knownThreadIds.add(threadId);
+    }
+    listKnownThreadIds() {
+        return [...this.knownThreadIds].sort();
+    }
+    async ensureThreadLoaded(threadId, model) {
+        if (!this.knownThreadIds.has(threadId) || this.loadedThreadIds.has(threadId)) {
+            return;
+        }
+        const built = await this.buildThreadResumeParams({ threadId, model });
+        await this.request('thread/resume', built.params);
+    }
+    async withFailover(operation) {
+        let attempt = 0;
+        let lastError;
+        while (attempt < this.profileManager.getProfiles().length) {
+            const profile = this.profileManager.getCurrentProfile();
+            const client = await this.getClient(profile);
+            try {
+                return await operation(client, profile);
+            }
+            catch (error) {
+                lastError = error;
+                if (!isRateLimitSignal(error) && !isAuthSignal(error)) {
+                    throw error;
+                }
+                await client.stop().catch(() => { });
+                this.clients.delete(profile.codexHome);
+                this.loadedThreadIds.clear();
+                const reason = isRateLimitSignal(error) ? 'rate_limit' : 'auth';
+                const rotated = this.profileManager.markFailure(reason);
+                if (!rotated.rotated) {
+                    throw error;
+                }
+                attempt += 1;
+            }
+        }
+        throw lastError instanceof Error
+            ? lastError
+            : new Error(String(lastError));
+    }
+    getCurrentClient() {
+        const profile = this.profileManager.getCurrentProfile();
+        const existing = this.clients.get(profile.codexHome);
+        if (!existing) {
+            throw new Error(`No active Codex app-server client for profile ${profile.codexHome}`);
+        }
+        return existing;
+    }
+    async getClient(profile) {
+        const existing = this.clients.get(profile.codexHome);
+        if (existing) {
+            return existing;
+        }
+        const codexHome = this.effectiveCodexHomes.get(profile.codexHome) ?? profile.codexHome;
+        const client = new AppServerClient({
+            command: this.command,
+            args: this.args,
+            env: this.env,
+            codexHome,
+            appVersion: '0.1.0',
+        });
+        await client.start();
+        this.clients.set(profile.codexHome, client);
+        return client;
+    }
+    async activateCompatibilityClient(profile) {
+        const existing = this.clients.get(profile.codexHome);
+        if (existing) {
+            await existing.stop().catch(() => { });
+            this.clients.delete(profile.codexHome);
+        }
+        if (!this.effectiveCodexHomes.has(profile.codexHome)) {
+            const compatHome = await this.createCompatibilityHome(profile.codexHome);
+            this.effectiveCodexHomes.set(profile.codexHome, compatHome);
+        }
+        this.loadedThreadIds.clear();
+        return await this.getClient(profile);
+    }
+    async createCompatibilityHome(sourceCodexHome) {
+        const compatHome = await mkdtemp(join(tmpdir(), 'mcp-codex-compat-'));
+        await mkdir(compatHome, { recursive: true });
+        const configPath = join(sourceCodexHome, 'config.toml');
+        const rawConfig = await readFile(configPath, 'utf8');
+        await writeFile(join(compatHome, 'config.toml'), sanitizeConfigForCompatibility(rawConfig), 'utf8');
+        for (const fileName of ['auth.json', 'version.json']) {
+            try {
+                await copyFile(join(sourceCodexHome, fileName), join(compatHome, basename(fileName)));
+            }
+            catch {
+                // optional compatibility files
+            }
+        }
+        return compatHome;
+    }
+    trackLoadedThread(method, response, params) {
+        if (method === 'thread/start' || method === 'thread/resume') {
+            const object = response;
+            const threadId = object.thread?.id;
+            if (typeof threadId === 'string') {
+                this.rememberThreadId(threadId);
+                this.loadedThreadIds.add(threadId);
+            }
+            return;
+        }
+        if (method === 'turn/start') {
+            const object = params;
+            if (typeof object.threadId === 'string') {
+                this.loadedThreadIds.add(object.threadId);
+            }
+        }
+    }
+}
+//# sourceMappingURL=codex-runtime.js.map

package/dist/src/services/codex-runtime.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"codex-runtime.js","sourceRoot":"","sources":["../../../src/services/codex-runtime.ts"],"names":[],"mappings":"AAAA,OAAO,OAAO,MAAM,cAAc,CAAC;AACnC,OAAO,EAAE,QAAQ,EAAE,OAAO,EAAE,KAAK,EAAE,QAAQ,EAAE,SAAS,EAAE,MAAM,kBAAkB,CAAC;AACjF,OAAO,EAAE,QAAQ,EAAE,IAAI,EAAE,MAAM,WAAW,CAAC;AAC3C,OAAO,EAAE,MAAM,EAAE,MAAM,SAAS,CAAC;AAEjC,OAAO,EACL,yBAAyB,EACzB,4BAA4B,EAC5B,kBAAkB,GACnB,MAAM,uBAAuB,CAAC;AAE/B,OAAO,EAAE,eAAe,EAA+B,MAAM,wBAAwB,CAAC;AACtF,OAAO,EAAE,gCAAgC,EAAE,MAAM,iBAAiB,CAAC;AACnE,OAAO,EACL,iBAAiB,EACjB,qBAAqB,EACrB,YAAY,GAGb,MAAM,oBAAoB,CAAC;AAC5B,OAAO,EACL,cAAc,GAEf,MAAM,sBAAsB,CAAC;AAiB9B,SAAS,SAAS,CAAC,GAAuB;IACxC,IAAI,CAAC,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI,EAAE,EAAE,CAAC;QACxB,OAAO,CAAC,YAAY,EAAE,UAAU,EAAE,UAAU,CAAC,CAAC;IAChD,CAAC;IAED,MAAM,OAAO,GAAG,GAAG,CAAC,IAAI,EAAE,CAAC;IAC3B,IAAI,OAAO,CAAC,UAAU,CAAC,GAAG,CAAC,EAAE,CAAC;QAC5B,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAY,CAAC;YAC9C,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,MAAM,CAAC,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,OAAO,KAAK,KAAK,QAAQ,CAAC,EAAE,CAAC;gBAChF,OAAO,MAAM,CAAC;YAChB,CAAC;QACH,CAAC;QAAC,MAAM,CAAC;YACP,8BAA8B;QAChC,CAAC;IACH,CAAC;IAED,OAAO,OAAO,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;AAC/C,CAAC;AAED,SAAS,iBAAiB,CAAC,KAAc;IACvC,MAAM,OAAO,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;IACvE,MAAM,UAAU,GAAG,OAAO,CAAC,WAAW,EAAE,CAAC;IACzC,OAAO,CACL,UAAU,CAAC,QAAQ,CAAC,YAAY,CAAC;QACjC,UAAU,CAAC,QAAQ,CAAC,aAAa,CAAC;QAClC,UAAU,CAAC,QAAQ,CAAC,oBAAoB,CAAC;QACzC,UAAU,CAAC,QAAQ,CAAC,KAAK,CAAC,CAC3B,CAAC;AACJ,CAAC;AAED,SAAS,YAAY,CAAC,KAAc;IAClC,MAAM,OAAO,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;IACvE,MAAM,UAAU,GAAG,OAAO,CAAC,WAAW,EAAE,CAAC;IACzC,OAAO,CACL,UAAU,CAAC,QAAQ,CAAC,cAAc,CAAC;QACnC,UAAU,CAAC,QAAQ,CAAC,eAAe,CAAC;QACpC,UAAU,CAAC,QAAQ,CAAC,gBAAgB,CAAC;QACrC,UAAU,CAAC,QAAQ,CAAC,OAAO,CAAC,CAC7B,CAAC;AACJ,CAAC;AAED,SAAS,oBAAoB,CAAC,KAAc;IAC1C,MAAM,OAAO,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;IACvE,MAAM,UAAU,GAAG,OAAO,CAAC,WAAW,EAAE,CAAC;IACzC,OAAO,CACL,UAAU,CAAC,QAAQ,CAAC,uBAAuB,CAAC;QAC5C,UAAU,CAAC,QAAQ,CAAC,eAAe,CAAC;QACpC,CAAC,UAAU,CAAC,QAAQ,CAAC,uBAAuB,CAAC,IAAI,UAAU,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,CAChF,CAAC;AACJ,CAAC;AAED,SAAS,8BAA8B,CAAC,SAAiB;IACvD,MAAM,KAAK,GAAG,SAAS,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;IACxC,MAAM,SAAS,GAAa,EAAE,CAAC;IAC/B,IAAI,qBAAqB,GAAG,KAAK,CAAC;IAElC,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;QACzB,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,EAAE,CAAC;QAC5B,MAAM,SAAS,GAAG,OAAO,CAAC,UAAU,CAAC,GAAG,CAAC,IAAI,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC;QACnE,IAAI,SAAS,EAAE,CAAC;YACd,IAAI,OAAO,KAAK,UAAU,IAAI,OAAO,CAAC,UAAU,CAAC,UAAU,CAAC,EAAE,CAAC;gBAC7D,qBAAqB,GAAG,IAAI,CAAC;gBAC7B,SAAS;YACX,CAAC;YACD,qBAAqB,GAAG,KAAK,CAAC;QAChC,CAAC;QAED,IAAI,qBAAqB,EAAE,CAAC;YAC1B,SAAS;QACX,CAAC;QAED,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACvB,CAAC;IAED,OAAO,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AAC9B,CAAC;AAQD,MAAM,OAAO,YAAY;IACN,cAAc,CAAiB;IAC/B,OAAO,CAAS;IAChB,IAAI,CAAW;IACf,GAAG,CAAoB;IACvB,OAAO,GAAG,IAAI,GAAG,EAA2B,CAAC;IAC7C,mBAAmB,GAAG,IAAI,GAAG,EAAkB,CAAC;IACzD,YAAY,CAA4B;IAC/B,cAAc,GAAG,IAAI,GAAG,EAAU,CAAC;IACnC,eAAe,GAAG,IAAI,GAAG,EAAU,CAAC;IAErD,YAAY,iBAAwC,EAAE;QACpD,IAAI,CAAC,cAAc,GAAG,cAAc,CAAC,eAAe,EAAE,CAAC;QACvD,IAAI,CAAC,OAAO,GAAG,cAAc,CAAC,OAAO,IAAI,OAAO,CAAC,GAAG,CAAC,4BAA4B,CAAC,IAAI,OAAO,CAAC;QAC9F,IAAI,CAAC,IAAI,GAAG,cAAc,CAAC,IAAI,IAAI,SAAS,CAAC,OAAO,CAAC,GAAG,CAAC,yBAAyB,CAAC,CAAC,CAAC;QACrF,IAAI,CAAC,GAAG,GAAG,cAAc,CAAC,GAAG,IAAI,OAAO,CAAC,GAAG,CAAC;IAC/C,CAAC;IAED,KAAK,CAAC,QAAQ;QACZ,MAAM,OAAO,CAAC,GAAG,CACf,CAAC,GAAG,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC,CAC1D,CAAC;QACF,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;QACrB,IAAI,CAAC,eAAe,CAAC,KAAK,EAAE,CAAC;IAC/B,CAAC;IAED,wBAAwB,CAAC,eAAe,GAAG,KAAK;QAC9C,OAAO,IAAI,CAAC,gBAAgB,EAAE,CAAC,kBAAkB,CAAC,eAAe,CAAC,CAAC;IACrE,CAAC;IAED,uBAAuB,CAAC,EAAmB;QACzC,OAAO,IAAI,CAAC,gBAAgB,EAAE,CAAC,gBAAgB,CAAC,EAAE,CAAC,CAAC;IACtD,CAAC;IAED,KAAK,CAAC,sBAAsB,CAAC,EAAmB,EAAE,OAAgB;QAChE,MAAM,IAAI,CAAC,gBAAgB,EAAE,CAAC,sBAAsB,CAAC,EAAE,EAAE,OAAO,CAAC,CAAC;IACpE,CAAC;IAED,KAAK,CAAC,gBAAgB,CAAC,WAAmB,EAAE,SAAiB,EAAE,cAAsB;QACnF,OAAO,IAAI,CAAC,gBAAgB,EAAE,CAAC,gBAAgB,CAAC,WAAW,EAAE;YAC3D,SAAS;YACT,cAAc;SACf,CAAC,CAAC;IACL,CAAC;IAED,YAAY,CAAC,WAAmB;QAC9B,OAAO,IAAI,CAAC,gBAAgB,EAAE,CAAC,YAAY,CAAC,WAAW,CAAC,CAAC;IAC3D,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,MAAc,EAAE,MAAe,EAAE,UAAiC,EAAE;QAChF,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,KAAK,EAAE,MAAM,EAAE,OAAO,EAAE,EAAE;YACjE,IAAI,CAAC;gBACH,OAAO,MAAM,MAAM,CAAC,OAAO,CACzB,MAAM,EACN,MAAM,EACN,EAAE,SAAS,EAAE,OAAO,CAAC,SAAS,IAAI,kBAAkB,EAAE,CACvD,CAAC;YACJ,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,IAAI,CAAC,oBAAoB,CAAC,KAAK,CAAC,EAAE,CAAC;oBACjC,MAAM,KAAK,CAAC;gBACd,CAAC;gBAED,MAAM,YAAY,GAAG,MAAM,IAAI,CAAC,2BAA2B,CAAC,OAAO,CAAC,CAAC;gBACrE,OAAO,MAAM,YAAY,CAAC,OAAO,CAC/B,MAAM,EACN,MAAM,EACN,EAAE,SAAS,EAAE,OAAO,CAAC,SAAS,IAAI,kBAAkB,EAAE,CACvD,CAAC;YACJ,CAAC;QACH,CAAC,CAAC,CAAC;QACH,IAAI,CAAC,iBAAiB,CAAC,MAAM,EAAE,QAAQ,EAAE,MAAM,CAAC,CAAC;QACjD,OAAO,QAAQ,CAAC;IAClB,CAAC;IAED,KAAK,CAAC,iBAAiB,CACrB,MAAc,EACd,MAAe,EACf,UAAgC,EAAE;QAElC,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,KAAK,EAAE,MAAM,EAAE,OAAO,EAAE,EAAE;YACjE,IAAI,CAAC;gBACH,OAAO,MAAM,MAAM,CAAC,iBAAiB,CACnC,MAAM,EACN,MAAM,EACN;oBACE,SAAS,EAAE,OAAO,CAAC,SAAS,IAAI,kBAAkB;oBAClD,eAAe,EAAE,OAAO,CAAC,eAAe;oBACxC,QAAQ,EAAE,OAAO,CAAC,QAAQ;iBAC3B,CACF,CAAC;YACJ,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,IAAI,CAAC,oBAAoB,CAAC,KAAK,CAAC,EAAE,CAAC;oBACjC,MAAM,KAAK,CAAC;gBACd,CAAC;gBAED,MAAM,YAAY,GAAG,MAAM,IAAI,CAAC,2BAA2B,CAAC,OAAO,CAAC,CAAC;gBACrE,OAAO,MAAM,YAAY,CAAC,iBAAiB,CACzC,MAAM,EACN,MAAM,EACN;oBACE,SAAS,EAAE,OAAO,CAAC,SAAS,IAAI,kBAAkB;oBAClD,eAAe,EAAE,OAAO,CAAC,eAAe;oBACxC,QAAQ,EAAE,OAAO,CAAC,QAAQ;iBAC3B,CACF,CAAC;YACJ,CAAC;QACH,CAAC,CAAC,CAAC;QACH,IAAI,CAAC,iBAAiB,CAAC,MAAM,EAAE,QAAQ,CAAC,MAAM,IAAI,QAAQ,EAAE,MAAM,CAAC,CAAC;QACpE,OAAO,QAAQ,CAAC;IAClB,CAAC;IAED,KAAK,CAAC,UAAU,CAAC,aAAa,GAAG,KAAK;QACpC,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,YAAY,EAAE,EAAE,aAAa,EAAE,CAElE,CAAC;QACF,MAAM,MAAM,GAAG,CAAC,QAAQ,CAAC,IAAI,IAAI,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;YACjD,EAAE,EAAE,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC;YAClB,KAAK,EAAE,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC;YACxB,WAAW,EAAE,MAAM,CAAC,GAAG,CAAC,WAAW,CAAC;YACpC,MAAM,EAAE,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC;YAC3B,OAAO,EAAE,OAAO,GAAG,CAAC,OAAO,KAAK,QAAQ,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI;YAC7D,SAAS,EAAE,OAAO,CAAC,GAAG,CAAC,SAAS,CAAC;SAClC,CAAC,CAAC,CAAC;QACJ,IAAI,CAAC,YAAY,GAAG,iBAAiB,CAAC,MAAM,CAAC,CAAC;QAC9C,OAAO,aAAa,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;IAC1E,CAAC;IAED,KAAK,CAAC,qBAAqB,CAAC,cAAmC;QAC7D,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE,CAAC;YACvB,MAAM,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC;QAC9B,CAAC;QACD,MAAM,OAAO,GAAG,IAAI,CAAC,YAAY,CAAC;QAClC,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,MAAM,IAAI,KAAK,CAAC,gDAAgD,CAAC,CAAC;QACpE,CAAC;QAED,IAAI,CAAC;YACH,MAAM,UAAU,GAAG,YAAY,CAAC,OAAO,EAAE,cAAc,CAAC,CAAC;YACzD,OAAO;gBACL,aAAa,EAAE,UAAU,CAAC,QAAQ;gBAClC,YAAY,EAAE,UAAU,CAAC,YAAY;gBACrC,aAAa,EAAE,OAAO,CAAC,aAAa,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC,IAAI,EAAE;aACrE,CAAC;QACJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,KAAK,YAAY,KAAK,IAAI,KAAK,CAAC,OAAO,CAAC,UAAU,CAAC,mBAAmB,CAAC,EAAE,CAAC;gBAC5E,MAAM,IAAI,KAAK,CACb,sBAAsB,cAAc,sBAAsB,qBAAqB,CAAC,OAAO,CAAC,EAAE,CAC3F,CAAC;YACJ,CAAC;YACD,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;IAED,KAAK,CAAC,mBAAmB;QACvB,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE,CAAC;YACvB,MAAM,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;QAC/B,CAAC;QACD,OAAO,CAAC,IAAI,CAAC,YAAY,EAAE,aAAa,IAAI,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;IAClF,CAAC;IAED,sBAAsB,CAAC,KAItB;QACC,OAAO,IAAI,CAAC,qBAAqB,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,CAAC,QAAQ,EAAE,EAAE;YAC/D,MAAM,MAAM,GAA4B;gBACtC,KAAK,EAAE,QAAQ,CAAC,aAAa;gBAC7B,aAAa,EAAE,QAAQ;gBACvB,GAAG,EAAE,KAAK,CAAC,GAAG,IAAI,OAAO,CAAC,GAAG,EAAE;gBAC/B,cAAc,EAAE,YAAY;gBAC5B,OAAO,EAAE,iBAAiB;gBAC1B,qBAAqB,EAAE,gCAAgC,CAAC,KAAK,CAAC,qBAAqB,CAAC;gBACpF,qBAAqB,EAAE,KAAK;gBAC5B,sBAAsB,EAAE,KAAK;aAC9B,CAAC;YACF,OAAO;gBACL,MAAM;gBACN,YAAY,EAAE,QAAQ,CAAC,YAAY;aACpC,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC;IAED,uBAAuB,CAAC,KAKvB;QACC,OAAO,IAAI,CAAC,qBAAqB,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,CAAC,QAAQ,EAAE,EAAE;YAC/D,MAAM,MAAM,GAA4B;gBACtC,QAAQ,EAAE,KAAK,CAAC,QAAQ;gBACxB,KAAK,EAAE,QAAQ,CAAC,aAAa;gBAC7B,aAAa,EAAE,QAAQ;gBACvB,GAAG,EAAE,KAAK,CAAC,GAAG,IAAI,OAAO,CAAC,GAAG,EAAE;gBAC/B,cAAc,EAAE,YAAY;gBAC5B,OAAO,EAAE,iBAAiB;gBAC1B,qBAAqB,EAAE,gCAAgC,CAAC,KAAK,CAAC,qBAAqB,CAAC;gBACpF,sBAAsB,EAAE,KAAK;aAC9B,CAAC;YACF,OAAO;gBACL,MAAM;gBACN,YAAY,EAAE,QAAQ,CAAC,YAAY;aACpC,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC;IAED,oBAAoB,CAAC,KAIpB;QACC,OAAO,IAAI,CAAC,qBAAqB,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC;YACjE,MAAM,EAAE;gBACN,QAAQ,EAAE,KAAK,CAAC,QAAQ;gBACxB,KAAK,EAAE,QAAQ,CAAC,aAAa;gBAC7B,KAAK,EAAE,CAAC;wBACN,IAAI,EAAE,MAAM;wBACZ,IAAI,EAAE,KAAK,CAAC,SAAS;wBACrB,aAAa,EAAE,EAAE;qBAClB,CAAC;aACH;YACD,YAAY,EAAE,QAAQ,CAAC,YAAY;SACpC,CAAC,CAAC,CAAC;IACN,CAAC;IAED,oBAAoB,CAAC,KAIpB;QACC,OAAO;YACL,QAAQ,EAAE,KAAK,CAAC,QAAQ;YACxB,cAAc,EAAE,KAAK,CAAC,cAAc;YACpC,KAAK,EAAE,CAAC;oBACN,IAAI,EAAE,MAAM;oBACZ,IAAI,EAAE,KAAK,CAAC,SAAS;oBACrB,aAAa,EAAE,EAAE;iBAClB,CAAC;SACH,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,eAAe,CAAC,QAAgB;QACpC,OAAO,IAAI,CAAC,gBAAgB,EAAE,CAAC,eAAe,CAAC,QAAQ,CAAC,CAAC;IAC3D,CAAC;IAED,gBAAgB,CAAC,QAAgB;QAC/B,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;IACpC,CAAC;IAED,kBAAkB;QAChB,OAAO,CAAC,GAAG,IAAI,CAAC,cAAc,CAAC,CAAC,IAAI,EAAE,CAAC;IACzC,CAAC;IAED,KAAK,CAAC,kBAAkB,CAAC,QAAgB,EAAE,KAA0B;QACnE,IAAI,CAAC,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,QAAQ,CAAC,IAAI,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,CAAC;YAC7E,OAAO;QACT,CAAC;QAED,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,uBAAuB,CAAC,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC,CAAC;QACtE,MAAM,IAAI,CAAC,OAAO,CAAC,eAAe,EAAE,KAAK,CAAC,MAAM,CAAC,CAAC;IACpD,CAAC;IAEO,KAAK,CAAC,YAAY,CAAI,SAAyE;QACrG,IAAI,OAAO,GAAG,CAAC,CAAC;QAChB,IAAI,SAAkB,CAAC;QACvB,OAAO,OAAO,GAAG,IAAI,CAAC,cAAc,CAAC,WAAW,EAAE,CAAC,MAAM,EAAE,CAAC;YAC1D,MAAM,OAAO,GAAG,IAAI,CAAC,cAAc,CAAC,iBAAiB,EAAE,CAAC;YACxD,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;YAC7C,IAAI,CAAC;gBACH,OAAO,MAAM,SAAS,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;YAC1C,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,SAAS,GAAG,KAAK,CAAC;gBAClB,IAAI,CAAC,iBAAiB,CAAC,KAAK,CAAC,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,EAAE,CAAC;oBACtD,MAAM,KAAK,CAAC;gBACd,CAAC;gBAED,MAAM,MAAM,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;gBACpC,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;gBACvC,IAAI,CAAC,eAAe,CAAC,KAAK,EAAE,CAAC;gBAC7B,MAAM,MAAM,GAAG,iBAAiB,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,MAAM,CAAC;gBAChE,MAAM,OAAO,GAAG,IAAI,CAAC,cAAc,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC;gBACxD,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC;oBACrB,MAAM,KAAK,CAAC;gBACd,CAAC;gBACD,OAAO,IAAI,CAAC,CAAC;YACf,CAAC;QACH,CAAC;QAED,MAAM,SAAS,YAAY,KAAK;YAC9B,CAAC,CAAC,SAAS;YACX,CAAC,CAAC,IAAI,KAAK,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC;IACnC,CAAC;IAEO,gBAAgB;QACtB,MAAM,OAAO,GAAG,IAAI,CAAC,cAAc,CAAC,iBAAiB,EAAE,CAAC;QACxD,MAAM,QAAQ,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;QACrD,IAAI,CAAC,QAAQ,EAAE,CAAC;YACd,MAAM,IAAI,KAAK,CAAC,iDAAiD,OAAO,CAAC,SAAS,EAAE,CAAC,CAAC;QACxF,CAAC;QACD,OAAO,QAAQ,CAAC;IAClB,CAAC;IAEO,KAAK,CAAC,SAAS,CAAC,OAAqB;QAC3C,MAAM,QAAQ,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;QACrD,IAAI,QAAQ,EAAE,CAAC;YACb,OAAO,QAAQ,CAAC;QAClB,CAAC;QAED,MAAM,SAAS,GAAG,IAAI,CAAC,mBAAmB,CAAC,GAAG,CAAC,OAAO,CAAC,SAAS,CAAC,IAAI,OAAO,CAAC,SAAS,CAAC;QACvF,MAAM,MAAM,GAAG,IAAI,eAAe,CAAC;YACjC,OAAO,EAAE,IAAI,CAAC,OAAO;YACrB,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,GAAG,EAAE,IAAI,CAAC,GAAG;YACb,SAAS;YACT,UAAU,EAAE,OAAO;SACpB,CAAC,CAAC;QACH,MAAM,MAAM,CAAC,KAAK,EAAE,CAAC;QACrB,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,SAAS,EAAE,MAAM,CAAC,CAAC;QAC5C,OAAO,MAAM,CAAC;IAChB,CAAC;IAEO,KAAK,CAAC,2BAA2B,CAAC,OAAqB;QAC7D,MAAM,QAAQ,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;QACrD,IAAI,QAAQ,EAAE,CAAC;YACb,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;YACtC,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;QACzC,CAAC;QAED,IAAI,CAAC,IAAI,CAAC,mBAAmB,CAAC,GAAG,CAAC,OAAO,CAAC,SAAS,CAAC,EAAE,CAAC;YACrD,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,uBAAuB,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;YACzE,IAAI,CAAC,mBAAmB,CAAC,GAAG,CAAC,OAAO,CAAC,SAAS,EAAE,UAAU,CAAC,CAAC;QAC9D,CAAC;QAED,IAAI,CAAC,eAAe,CAAC,KAAK,EAAE,CAAC;QAC7B,OAAO,MAAM,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;IACvC,CAAC;IAEO,KAAK,CAAC,uBAAuB,CAAC,eAAuB;QAC3D,MAAM,UAAU,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC,MAAM,EAAE,EAAE,mBAAmB,CAAC,CAAC,CAAC;QACtE,MAAM,KAAK,CAAC,UAAU,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QAE7C,MAAM,UAAU,GAAG,IAAI,CAAC,eAAe,EAAE,aAAa,CAAC,CAAC;QACxD,MAAM,SAAS,GAAG,MAAM,QAAQ,CAAC,UAAU,EAAE,MAAM,CAAC,CAAC;QACrD,MAAM,SAAS,CAAC,IAAI,CAAC,UAAU,EAAE,aAAa,CAAC,EAAE,8BAA8B,CAAC,SAAS,CAAC,EAAE,MAAM,CAAC,CAAC;QAEpG,KAAK,MAAM,QAAQ,IAAI,CAAC,WAAW,EAAE,cAAc,CAAC,EAAE,CAAC;YACrD,IAAI,CAAC;gBACH,MAAM,QAAQ,CAAC,IAAI,CAAC,eAAe,EAAE,QAAQ,CAAC,EAAE,IAAI,CAAC,UAAU,EAAE,QAAQ,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC;YACxF,CAAC;YAAC,MAAM,CAAC;gBACP,+BAA+B;YACjC,CAAC;QACH,CAAC;QAED,OAAO,UAAU,CAAC;IACpB,CAAC;IAEO,iBAAiB,CAAC,MAAc,EAAE,QAAiB,EAAE,MAAe;QAC1E,IAAI,MAAM,KAAK,cAAc,IAAI,MAAM,KAAK,eAAe,EAAE,CAAC;YAC5D,MAAM,MAAM,GAAG,QAAwC,CAAC;YACxD,MAAM,QAAQ,GAAG,MAAM,CAAC,MAAM,EAAE,EAAE,CAAC;YACnC,IAAI,OAAO,QAAQ,KAAK,QAAQ,EAAE,CAAC;gBACjC,IAAI,CAAC,gBAAgB,CAAC,QAAQ,CAAC,CAAC;gBAChC,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;YACrC,CAAC;YACD,OAAO;QACT,CAAC;QAED,IAAI,MAAM,KAAK,YAAY,EAAE,CAAC;YAC5B,MAAM,MAAM,GAAG,MAA+B,CAAC;YAC/C,IAAI,OAAO,MAAM,CAAC,QAAQ,KAAK,QAAQ,EAAE,CAAC;gBACxC,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;YAC5C,CAAC;QACH,CAAC;IACH,CAAC;CACF"}

package/dist/src/services/fleet-mode.d.ts

@@ -0,0 +1,3 @@
+export declare const FLEET_DEVELOPER_INSTRUCTIONS_SUFFIX: string;
+export declare function isFleetModeEnabled(value?: string | undefined): boolean;
+export declare function appendFleetDeveloperInstructions(developerInstructions: string | null | undefined): string | undefined;

package/dist/src/services/fleet-mode.js

@@ -0,0 +1,21 @@
+import { CODEX_ENABLE_FLEET_ENV } from '../config/defaults.js';
+const TRUTHY_VALUES = new Set(['1', 'true', 'yes', 'on']);
+export const FLEET_DEVELOPER_INSTRUCTIONS_SUFFIX = [
+    '',
+    '[codex-worker-fleet]',
+    'This thread is running under mcp-codex-worker fleet mode.',
+].join('\n');
+export function isFleetModeEnabled(value = process.env[CODEX_ENABLE_FLEET_ENV]) {
+    return value !== undefined && TRUTHY_VALUES.has(value.trim().toLowerCase());
+}
+export function appendFleetDeveloperInstructions(developerInstructions) {
+    if (!isFleetModeEnabled()) {
+        return developerInstructions ?? undefined;
+    }
+    const base = developerInstructions?.trim();
+    if (!base) {
+        return FLEET_DEVELOPER_INSTRUCTIONS_SUFFIX.trim();
+    }
+    return `${base}\n${FLEET_DEVELOPER_INSTRUCTIONS_SUFFIX}`;
+}
+//# sourceMappingURL=fleet-mode.js.map

package/dist/src/services/fleet-mode.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"fleet-mode.js","sourceRoot":"","sources":["../../../src/services/fleet-mode.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,sBAAsB,EAAE,MAAM,uBAAuB,CAAC;AAE/D,MAAM,aAAa,GAAG,IAAI,GAAG,CAAC,CAAC,GAAG,EAAE,MAAM,EAAE,KAAK,EAAE,IAAI,CAAC,CAAC,CAAC;AAE1D,MAAM,CAAC,MAAM,mCAAmC,GAAG;IACjD,EAAE;IACF,sBAAsB;IACtB,2DAA2D;CAC5D,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AAEb,MAAM,UAAU,kBAAkB,CAAC,KAAK,GAAG,OAAO,CAAC,GAAG,CAAC,sBAAsB,CAAC;IAC5E,OAAO,KAAK,KAAK,SAAS,IAAI,aAAa,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC,CAAC;AAC9E,CAAC;AAED,MAAM,UAAU,gCAAgC,CAC9C,qBAAgD;IAEhD,IAAI,CAAC,kBAAkB,EAAE,EAAE,CAAC;QAC1B,OAAO,qBAAqB,IAAI,SAAS,CAAC;IAC5C,CAAC;IAED,MAAM,IAAI,GAAG,qBAAqB,EAAE,IAAI,EAAE,CAAC;IAC3C,IAAI,CAAC,IAAI,EAAE,CAAC;QACV,OAAO,mCAAmC,CAAC,IAAI,EAAE,CAAC;IACpD,CAAC;IAED,OAAO,GAAG,IAAI,KAAK,mCAAmC,EAAE,CAAC;AAC3D,CAAC"}

package/dist/src/services/model-catalog.d.ts

@@ -0,0 +1,22 @@
+export interface ModelInfo {
+    id: string;
+    model: string;
+    displayName: string;
+    hidden: boolean;
+    upgrade: string | null;
+    isDefault: boolean;
+}
+export interface ModelCatalog {
+    visibleModels: ModelInfo[];
+    byId: Map<string, ModelInfo>;
+    upgradeAliases: Map<string, string>;
+    defaultModelId?: string | undefined;
+}
+export interface ModelResolution {
+    requested: string;
+    resolved: string;
+    remappedFrom?: string | undefined;
+}
+export declare function buildModelCatalog(models: ModelInfo[]): ModelCatalog;
+export declare function describeAllowedModels(catalog: ModelCatalog): string;
+export declare function resolveModel(catalog: ModelCatalog, requestedModel?: string | undefined): ModelResolution;

package/dist/src/services/model-catalog.js

@@ -0,0 +1,51 @@
+export function buildModelCatalog(models) {
+    const byId = new Map(models.map((model) => [model.id, model]));
+    const visibleModels = models.filter((model) => !model.hidden);
+    const upgradeAliases = new Map();
+    for (const model of models) {
+        if (!model.upgrade) {
+            continue;
+        }
+        if (byId.has(model.upgrade)) {
+            upgradeAliases.set(model.id, model.upgrade);
+        }
+    }
+    const defaultModelId = visibleModels.find((model) => model.isDefault)?.id ?? visibleModels[0]?.id;
+    return {
+        visibleModels,
+        byId,
+        upgradeAliases,
+        defaultModelId,
+    };
+}
+export function describeAllowedModels(catalog) {
+    return catalog.visibleModels.map((model) => model.id).sort().join(', ');
+}
+export function resolveModel(catalog, requestedModel) {
+    const requested = requestedModel ?? catalog.defaultModelId;
+    if (!requested) {
+        throw new Error('No models available from model/list.');
+    }
+    const direct = catalog.byId.get(requested);
+    if (direct && !direct.hidden) {
+        const upgraded = catalog.upgradeAliases.get(requested);
+        if (upgraded) {
+            return {
+                requested,
+                resolved: upgraded,
+                remappedFrom: requested,
+            };
+        }
+        return { requested, resolved: requested };
+    }
+    const upgraded = catalog.upgradeAliases.get(requested);
+    if (upgraded) {
+        return {
+            requested,
+            resolved: upgraded,
+            remappedFrom: requested,
+        };
+    }
+    throw new Error(`Unsupported model "${requested}". Allowed models: ${describeAllowedModels(catalog)}`);
+}
+//# sourceMappingURL=model-catalog.js.map

package/dist/src/services/model-catalog.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"model-catalog.js","sourceRoot":"","sources":["../../../src/services/model-catalog.ts"],"names":[],"mappings":"AAsBA,MAAM,UAAU,iBAAiB,CAAC,MAAmB;IACnD,MAAM,IAAI,GAAG,IAAI,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC,KAAK,CAAC,EAAE,EAAE,KAAK,CAAC,CAAC,CAAC,CAAC;IAC/D,MAAM,aAAa,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;IAC9D,MAAM,cAAc,GAAG,IAAI,GAAG,EAAkB,CAAC;IAEjD,KAAK,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;QAC3B,IAAI,CAAC,KAAK,CAAC,OAAO,EAAE,CAAC;YACnB,SAAS;QACX,CAAC;QACD,IAAI,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,OAAO,CAAC,EAAE,CAAC;YAC5B,cAAc,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC;QAC9C,CAAC;IACH,CAAC;IAED,MAAM,cAAc,GAAG,aAAa,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,SAAS,CAAC,EAAE,EAAE,IAAI,aAAa,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC;IAElG,OAAO;QACL,aAAa;QACb,IAAI;QACJ,cAAc;QACd,cAAc;KACf,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,qBAAqB,CAAC,OAAqB;IACzD,OAAO,OAAO,CAAC,aAAa,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AAC1E,CAAC;AAED,MAAM,UAAU,YAAY,CAC1B,OAAqB,EACrB,cAAmC;IAEnC,MAAM,SAAS,GAAG,cAAc,IAAI,OAAO,CAAC,cAAc,CAAC;IAC3D,IAAI,CAAC,SAAS,EAAE,CAAC;QACf,MAAM,IAAI,KAAK,CAAC,sCAAsC,CAAC,CAAC;IAC1D,CAAC;IAED,MAAM,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;IAC3C,IAAI,MAAM,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC;QAC7B,MAAM,QAAQ,GAAG,OAAO,CAAC,cAAc,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QACvD,IAAI,QAAQ,EAAE,CAAC;YACb,OAAO;gBACL,SAAS;gBACT,QAAQ,EAAE,QAAQ;gBAClB,YAAY,EAAE,SAAS;aACxB,CAAC;QACJ,CAAC;QACD,OAAO,EAAE,SAAS,EAAE,QAAQ,EAAE,SAAS,EAAE,CAAC;IAC5C,CAAC;IAED,MAAM,QAAQ,GAAG,OAAO,CAAC,cAAc,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;IACvD,IAAI,QAAQ,EAAE,CAAC;QACb,OAAO;YACL,SAAS;YACT,QAAQ,EAAE,QAAQ;YAClB,YAAY,EAAE,SAAS;SACxB,CAAC;IACJ,CAAC;IAED,MAAM,IAAI,KAAK,CACb,sBAAsB,SAAS,sBAAsB,qBAAqB,CAAC,OAAO,CAAC,EAAE,CACtF,CAAC;AACJ,CAAC"}

package/dist/src/services/profile-manager.d.ts

@@ -0,0 +1,29 @@
+export interface CodexProfile {
+    id: string;
+    codexHome: string;
+    cooldownUntil?: number | undefined;
+    failureCount: number;
+    lastFailureReason?: string | undefined;
+}
+interface ProfileManagerOptions {
+    codexHomes?: string[] | undefined;
+    cooldownMs?: number | undefined;
+    now?: (() => number) | undefined;
+}
+export declare class ProfileManager {
+    private readonly cooldownMs;
+    private readonly now;
+    private readonly profiles;
+    private currentIndex;
+    constructor(options?: ProfileManagerOptions);
+    static fromEnvironment(now?: () => number): ProfileManager;
+    getCurrentProfile(): CodexProfile;
+    getProfiles(): CodexProfile[];
+    markFailure(reason: string): {
+        rotated: boolean;
+        profile?: CodexProfile;
+    };
+    private resetExpiredCooldowns;
+    private findFirstAvailableIndex;
+}
+export {};

package/dist/src/services/profile-manager.js

@@ -0,0 +1,82 @@
+import { homedir } from 'node:os';
+import { join } from 'node:path';
+import { CODEX_HOME_DIRS_ENV, PROFILE_COOLDOWN_MS } from '../config/defaults.js';
+function defaultCodexHome() {
+    return join(homedir(), '.codex');
+}
+function dedupeHomes(codexHomes) {
+    const seen = new Set();
+    const result = [];
+    for (const home of codexHomes) {
+        const normalized = home.trim();
+        if (!normalized || seen.has(normalized)) {
+            continue;
+        }
+        seen.add(normalized);
+        result.push(normalized);
+    }
+    return result.length > 0 ? result : [defaultCodexHome()];
+}
+export class ProfileManager {
+    cooldownMs;
+    now;
+    profiles;
+    currentIndex = 0;
+    constructor(options = {}) {
+        this.cooldownMs = options.cooldownMs ?? PROFILE_COOLDOWN_MS;
+        this.now = options.now ?? Date.now;
+        this.profiles = dedupeHomes(options.codexHomes ?? [defaultCodexHome()]).map((codexHome, index) => ({
+            id: `profile-${index + 1}`,
+            codexHome,
+            failureCount: 0,
+        }));
+        this.currentIndex = this.findFirstAvailableIndex();
+    }
+    static fromEnvironment(now) {
+        const raw = process.env[CODEX_HOME_DIRS_ENV];
+        const homes = raw
+            ? raw.split(':').map((entry) => entry.trim()).filter(Boolean)
+            : [defaultCodexHome()];
+        return new ProfileManager({ codexHomes: homes, now });
+    }
+    getCurrentProfile() {
+        this.resetExpiredCooldowns();
+        return { ...this.profiles[this.currentIndex] };
+    }
+    getProfiles() {
+        this.resetExpiredCooldowns();
+        return this.profiles.map((profile) => ({ ...profile }));
+    }
+    markFailure(reason) {
+        const current = this.profiles[this.currentIndex];
+        if (!current) {
+            return { rotated: false };
+        }
+        current.failureCount += 1;
+        current.lastFailureReason = reason;
+        current.cooldownUntil = this.now() + this.cooldownMs;
+        const nextIndex = this.findFirstAvailableIndex();
+        if (nextIndex === -1) {
+            return { rotated: false };
+        }
+        this.currentIndex = nextIndex;
+        return { rotated: true, profile: this.getCurrentProfile() };
+    }
+    resetExpiredCooldowns() {
+        const now = this.now();
+        for (const profile of this.profiles) {
+            if (profile.cooldownUntil !== undefined && profile.cooldownUntil <= now) {
+                profile.cooldownUntil = undefined;
+            }
+        }
+        const nextIndex = this.findFirstAvailableIndex();
+        if (nextIndex !== -1) {
+            this.currentIndex = nextIndex;
+        }
+    }
+    findFirstAvailableIndex() {
+        const now = this.now();
+        return this.profiles.findIndex((profile) => profile.cooldownUntil === undefined || profile.cooldownUntil <= now);
+    }
+}
+//# sourceMappingURL=profile-manager.js.map

package/dist/src/services/profile-manager.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"profile-manager.js","sourceRoot":"","sources":["../../../src/services/profile-manager.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,MAAM,SAAS,CAAC;AAClC,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAC;AAEjC,OAAO,EAAE,mBAAmB,EAAE,mBAAmB,EAAE,MAAM,uBAAuB,CAAC;AAgBjF,SAAS,gBAAgB;IACvB,OAAO,IAAI,CAAC,OAAO,EAAE,EAAE,QAAQ,CAAC,CAAC;AACnC,CAAC;AAED,SAAS,WAAW,CAAC,UAAoB;IACvC,MAAM,IAAI,GAAG,IAAI,GAAG,EAAU,CAAC;IAC/B,MAAM,MAAM,GAAa,EAAE,CAAC;IAE5B,KAAK,MAAM,IAAI,IAAI,UAAU,EAAE,CAAC;QAC9B,MAAM,UAAU,GAAG,IAAI,CAAC,IAAI,EAAE,CAAC;QAC/B,IAAI,CAAC,UAAU,IAAI,IAAI,CAAC,GAAG,CAAC,UAAU,CAAC,EAAE,CAAC;YACxC,SAAS;QACX,CAAC;QACD,IAAI,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;QACrB,MAAM,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;IAC1B,CAAC;IAED,OAAO,MAAM,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,gBAAgB,EAAE,CAAC,CAAC;AAC3D,CAAC;AAED,MAAM,OAAO,cAAc;IACR,UAAU,CAAS;IACnB,GAAG,CAAe;IAClB,QAAQ,CAAiB;IAClC,YAAY,GAAG,CAAC,CAAC;IAEzB,YAAY,UAAiC,EAAE;QAC7C,IAAI,CAAC,UAAU,GAAG,OAAO,CAAC,UAAU,IAAI,mBAAmB,CAAC;QAC5D,IAAI,CAAC,GAAG,GAAG,OAAO,CAAC,GAAG,IAAI,IAAI,CAAC,GAAG,CAAC;QAEnC,IAAI,CAAC,QAAQ,GAAG,WAAW,CAAC,OAAO,CAAC,UAAU,IAAI,CAAC,gBAAgB,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,SAAS,EAAE,KAAK,EAAE,EAAE,CAAC,CAAC;YACjG,EAAE,EAAE,WAAW,KAAK,GAAG,CAAC,EAAE;YAC1B,SAAS;YACT,YAAY,EAAE,CAAC;SAChB,CAAC,CAAC,CAAC;QACJ,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC,uBAAuB,EAAE,CAAC;IACrD,CAAC;IAED,MAAM,CAAC,eAAe,CAAC,GAAkB;QACvC,MAAM,GAAG,GAAG,OAAO,CAAC,GAAG,CAAC,mBAAmB,CAAC,CAAC;QAC7C,MAAM,KAAK,GAAG,GAAG;YACf,CAAC,CAAC,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC;YAC7D,CAAC,CAAC,CAAC,gBAAgB,EAAE,CAAC,CAAC;QACzB,OAAO,IAAI,cAAc,CAAC,EAAE,UAAU,EAAE,KAAK,EAAE,GAAG,EAAE,CAAC,CAAC;IACxD,CAAC;IAED,iBAAiB;QACf,IAAI,CAAC,qBAAqB,EAAE,CAAC;QAC7B,OAAO,EAAE,GAAG,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,YAAY,CAAE,EAAE,CAAC;IAClD,CAAC;IAED,WAAW;QACT,IAAI,CAAC,qBAAqB,EAAE,CAAC;QAC7B,OAAO,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,EAAE,GAAG,OAAO,EAAE,CAAC,CAAC,CAAC;IAC1D,CAAC;IAED,WAAW,CAAC,MAAc;QACxB,MAAM,OAAO,GAAG,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;QACjD,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC;QAC5B,CAAC;QAED,OAAO,CAAC,YAAY,IAAI,CAAC,CAAC;QAC1B,OAAO,CAAC,iBAAiB,GAAG,MAAM,CAAC;QACnC,OAAO,CAAC,aAAa,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,UAAU,CAAC;QAErD,MAAM,SAAS,GAAG,IAAI,CAAC,uBAAuB,EAAE,CAAC;QACjD,IAAI,SAAS,KAAK,CAAC,CAAC,EAAE,CAAC;YACrB,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC;QAC5B,CAAC;QAED,IAAI,CAAC,YAAY,GAAG,SAAS,CAAC;QAC9B,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,IAAI,CAAC,iBAAiB,EAAE,EAAE,CAAC;IAC9D,CAAC;IAEO,qBAAqB;QAC3B,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACvB,KAAK,MAAM,OAAO,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;YACpC,IAAI,OAAO,CAAC,aAAa,KAAK,SAAS,IAAI,OAAO,CAAC,aAAa,IAAI,GAAG,EAAE,CAAC;gBACxE,OAAO,CAAC,aAAa,GAAG,SAAS,CAAC;YACpC,CAAC;QACH,CAAC;QACD,MAAM,SAAS,GAAG,IAAI,CAAC,uBAAuB,EAAE,CAAC;QACjD,IAAI,SAAS,KAAK,CAAC,CAAC,EAAE,CAAC;YACrB,IAAI,CAAC,YAAY,GAAG,SAAS,CAAC;QAChC,CAAC;IACH,CAAC;IAEO,uBAAuB;QAC7B,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACvB,OAAO,IAAI,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,aAAa,KAAK,SAAS,IAAI,OAAO,CAAC,aAAa,IAAI,GAAG,CAAC,CAAC;IACnH,CAAC;CACF"}