@mmnto/cli 0.17.0 → 0.19.0

package/dist/orchestrators/shell-orchestrator.js

@@ -0,0 +1,131 @@
+import { spawn } from 'node:child_process';
+import * as crypto from 'node:crypto';
+import * as fs from 'node:fs';
+import * as path from 'node:path';
+import { z } from 'zod';
+import { log } from '../ui.js';
+import { isQuotaError } from './orchestrator.js';
+// ─── Constants ───────────────────────────────────────
+const LLM_TIMEOUT_MS = 180_000;
+const LLM_MAX_OUTPUT = 50 * 1024 * 1024; // 50 MB — safety cap on streamed output
+const TEMP_ID_BYTES = 4;
+/** execFileSync on Windows can't resolve executables without `shell: true`. */
+const IS_WIN = process.platform === 'win32';
+// ─── Gemini CLI JSON parsing ─────────────────────────
+const GeminiModelStatsSchema = z.object({
+    tokens: z.object({ input: z.number(), candidates: z.number() }).optional(),
+    api: z.object({ totalLatencyMs: z.number() }).optional(),
+});
+const GeminiOutputSchema = z.object({
+    response: z.string(),
+    stats: z.object({ models: z.record(GeminiModelStatsSchema) }),
+});
+/**
+ * Try to parse Gemini CLI JSON output. Returns extracted data or null if
+ * the output is not valid Gemini JSON (e.g. raw text from a non-Gemini orchestrator).
+ */
+export function tryParseGeminiJson(raw) {
+    let data;
+    try {
+        data = JSON.parse(raw);
+    }
+    catch {
+        return null;
+    }
+    const result = GeminiOutputSchema.safeParse(data);
+    if (!result.success)
+        return null;
+    const allModelStats = Object.values(result.data.stats.models);
+    if (allModelStats.length === 0)
+        return null;
+    const inputTokens = allModelStats.reduce((sum, s) => sum + (s.tokens?.input ?? 0), 0);
+    const outputTokens = allModelStats.reduce((sum, s) => sum + (s.tokens?.candidates ?? 0), 0);
+    const latencyMs = allModelStats.reduce((sum, s) => sum + (s.api?.totalLatencyMs ?? 0), 0) || null;
+    return {
+        content: result.data.response,
+        inputTokens,
+        outputTokens,
+        latencyMs,
+    };
+}
+export async function invokeShellOrchestrator(opts) {
+    const { prompt, command, model, cwd, tag, totemDir } = opts;
+    const tmpName = `totem-${tag.toLowerCase()}-${crypto.randomBytes(TEMP_ID_BYTES).toString('hex')}.md`;
+    const tempDir = path.join(cwd, totemDir, 'temp');
+    fs.mkdirSync(tempDir, { recursive: true });
+    const tempPath = path.join(tempDir, tmpName);
+    fs.writeFileSync(tempPath, prompt, { encoding: 'utf-8', mode: 0o600 });
+    const quotedPath = IS_WIN
+        ? `"${tempPath.replace(/"/g, '""')}"`
+        : `'${tempPath.replace(/'/g, "'\\''")}'`;
+    const resolvedCmd = command.replace(/\{file\}/g, quotedPath).replace(/\{model\}/g, model);
+    log.info(tag, 'Invoking orchestrator (this may take 15-60 seconds)...');
+    const startMs = Date.now();
+    try {
+        const raw = await new Promise((resolve, reject) => {
+            const child = spawn(resolvedCmd, {
+                cwd,
+                shell: true,
+                stdio: ['ignore', 'pipe', 'pipe'],
+            });
+            let stdout = '';
+            let stderr = '';
+            let timedOut = false;
+            child.stdout.on('data', (chunk) => {
+                if (stdout.length < LLM_MAX_OUTPUT)
+                    stdout += chunk.toString();
+            });
+            child.stderr.on('data', (chunk) => {
+                if (stderr.length < LLM_MAX_OUTPUT)
+                    stderr += chunk.toString();
+            });
+            const timer = setTimeout(() => {
+                timedOut = true;
+                child.kill();
+            }, LLM_TIMEOUT_MS);
+            child.on('error', (err) => {
+                clearTimeout(timer);
+                reject(new Error(`[Totem Error] Shell orchestrator command failed: ${err.message}\n${stderr}`));
+            });
+            child.on('close', (code) => {
+                clearTimeout(timer);
+                if (timedOut) {
+                    reject(Object.assign(new Error(`[Totem Error] Orchestrator timed out after ${LLM_TIMEOUT_MS / 1000}s.\n${stderr}`), { code: 'ETIMEDOUT' }));
+                    return;
+                }
+                if (code !== 0) {
+                    const fullError = new Error(`Process exited with code ${code}\n${stderr}`);
+                    if (isQuotaError(fullError)) {
+                        fullError.name = 'QuotaError';
+                        reject(fullError);
+                    }
+                    else {
+                        reject(new Error(`[Totem Error] Shell orchestrator command failed: ${fullError.message}`));
+                    }
+                    return;
+                }
+                resolve(stdout);
+            });
+        });
+        const wallMs = Date.now() - startMs;
+        const gemini = tryParseGeminiJson(raw);
+        if (gemini) {
+            return {
+                content: gemini.content,
+                inputTokens: gemini.inputTokens,
+                outputTokens: gemini.outputTokens,
+                durationMs: gemini.latencyMs ?? wallMs,
+            };
+        }
+        return { content: raw.trim(), inputTokens: null, outputTokens: null, durationMs: wallMs };
+    }
+    finally {
+        try {
+            fs.unlinkSync(tempPath);
+        }
+        catch {
+            // Temp cleanup is best-effort
+        }
+    }
+}
+//# sourceMappingURL=shell-orchestrator.js.map

package/dist/orchestrators/shell-orchestrator.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"shell-orchestrator.js","sourceRoot":"","sources":["../../src/orchestrators/shell-orchestrator.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,KAAK,EAAE,MAAM,oBAAoB,CAAC;AAC3C,OAAO,KAAK,MAAM,MAAM,aAAa,CAAC;AACtC,OAAO,KAAK,EAAE,MAAM,SAAS,CAAC;AAC9B,OAAO,KAAK,IAAI,MAAM,WAAW,CAAC;AAElC,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAExB,OAAO,EAAE,GAAG,EAAE,MAAM,UAAU,CAAC;AAE/B,OAAO,EAAE,YAAY,EAAE,MAAM,mBAAmB,CAAC;AAEjD,wDAAwD;AAExD,MAAM,cAAc,GAAG,OAAO,CAAC;AAC/B,MAAM,cAAc,GAAG,EAAE,GAAG,IAAI,GAAG,IAAI,CAAC,CAAC,wCAAwC;AACjF,MAAM,aAAa,GAAG,CAAC,CAAC;AAExB,+EAA+E;AAC/E,MAAM,MAAM,GAAG,OAAO,CAAC,QAAQ,KAAK,OAAO,CAAC;AAE5C,wDAAwD;AAExD,MAAM,sBAAsB,GAAG,CAAC,CAAC,MAAM,CAAC;IACtC,MAAM,EAAE,CAAC,CAAC,MAAM,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE,EAAE,UAAU,EAAE,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,QAAQ,EAAE;IAC1E,GAAG,EAAE,CAAC,CAAC,MAAM,CAAC,EAAE,cAAc,EAAE,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,QAAQ,EAAE;CACzD,CAAC,CAAC;AAEH,MAAM,kBAAkB,GAAG,CAAC,CAAC,MAAM,CAAC;IAClC,QAAQ,EAAE,CAAC,CAAC,MAAM,EAAE;IACpB,KAAK,EAAE,CAAC,CAAC,MAAM,CAAC,EAAE,MAAM,EAAE,CAAC,CAAC,MAAM,CAAC,sBAAsB,CAAC,EAAE,CAAC;CAC9D,CAAC,CAAC;AAEH;;;GAGG;AACH,MAAM,UAAU,kBAAkB,CAChC,GAAW;IAEX,IAAI,IAAa,CAAC;IAClB,IAAI,CAAC;QACH,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;IACzB,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,IAAI,CAAC;IACd,CAAC;IAED,MAAM,MAAM,GAAG,kBAAkB,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;IAClD,IAAI,CAAC,MAAM,CAAC,OAAO;QAAE,OAAO,IAAI,CAAC;IAEjC,MAAM,aAAa,GAAG,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;IAC9D,IAAI,aAAa,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,IAAI,CAAC;IAE5C,MAAM,WAAW,GAAG,aAAa,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,GAAG,GAAG,CAAC,CAAC,CAAC,MAAM,EAAE,KAAK,IAAI,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;IACtF,MAAM,YAAY,GAAG,aAAa,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,GAAG,GAAG,CAAC,CAAC,CAAC,MAAM,EAAE,UAAU,IAAI,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;IAC5F,MAAM,SAAS,GAAG,aAAa,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,GAAG,GAAG,CAAC,CAAC,CAAC,GAAG,EAAE,cAAc,IAAI,CAAC,CAAC,EAAE,CAAC,CAAC,IAAI,IAAI,CAAC;IAElG,OAAO;QACL,OAAO,EAAE,MAAM,CAAC,IAAI,CAAC,QAAQ;QAC7B,WAAW;QACX,YAAY;QACZ,SAAS;KACV,CAAC;AACJ,CAAC;AAQD,MAAM,CAAC,KAAK,UAAU,uBAAuB,CAC3C,IAA8B;IAE9B,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,KAAK,EAAE,GAAG,EAAE,GAAG,EAAE,QAAQ,EAAE,GAAG,IAAI,CAAC;IAC5D,MAAM,OAAO,GAAG,SAAS,GAAG,CAAC,WAAW,EAAE,IAAI,MAAM,CAAC,WAAW,CAAC,aAAa,CAAC,CAAC,QAAQ,CAAC,KAAK,CAAC,KAAK,CAAC;IACrG,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,QAAQ,EAAE,MAAM,CAAC,CAAC;IACjD,EAAE,CAAC,SAAS,CAAC,OAAO,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;IAC3C,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;IAE7C,EAAE,CAAC,aAAa,CAAC,QAAQ,EAAE,MAAM,EAAE,EAAE,QAAQ,EAAE,OAAO,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC,CAAC;IAEvE,MAAM,UAAU,GAAG,MAAM;QACvB,CAAC,CAAC,IAAI,QAAQ,CAAC,OAAO,CAAC,IAAI,EAAE,IAAI,CAAC,GAAG;QACrC,CAAC,CAAC,IAAI,QAAQ,CAAC,OAAO,CAAC,IAAI,EAAE,OAAO,CAAC,GAAG,CAAC;IAC3C,MAAM,WAAW,GAAG,OAAO,CAAC,OAAO,CAAC,WAAW,EAAE,UAAU,CAAC,CAAC,OAAO,CAAC,YAAY,EAAE,KAAK,CAAC,CAAC;IAE1F,GAAG,CAAC,IAAI,CAAC,GAAG,EAAE,wDAAwD,CAAC,CAAC;IAExE,MAAM,OAAO,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IAE3B,IAAI,CAAC;QACH,MAAM,GAAG,GAAG,MAAM,IAAI,OAAO,CAAS,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YACxD,MAAM,KAAK,GAAG,KAAK,CAAC,WAAW,EAAE;gBAC/B,GAAG;gBACH,KAAK,EAAE,IAAI;gBACX,KAAK,EAAE,CAAC,QAAQ,EAAE,MAAM,EAAE,MAAM,CAAC;aAClC,CAAC,CAAC;YAEH,IAAI,MAAM,GAAG,EAAE,CAAC;YAChB,IAAI,MAAM,GAAG,EAAE,CAAC;YAChB,IAAI,QAAQ,GAAG,KAAK,CAAC;YAErB,KAAK,CAAC,MAAM,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,KAAa,EAAE,EAAE;gBACxC,IAAI,MAAM,CAAC,MAAM,GAAG,cAAc;oBAAE,MAAM,IAAI,KAAK,CAAC,QAAQ,EAAE,CAAC;YACjE,CAAC,CAAC,CAAC;YACH,KAAK,CAAC,MAAM,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,KAAa,EAAE,EAAE;gBACxC,IAAI,MAAM,CAAC,MAAM,GAAG,cAAc;oBAAE,MAAM,IAAI,KAAK,CAAC,QAAQ,EAAE,CAAC;YACjE,CAAC,CAAC,CAAC;YAEH,MAAM,KAAK,GAAG,UAAU,CAAC,GAAG,EAAE;gBAC5B,QAAQ,GAAG,IAAI,CAAC;gBAChB,KAAK,CAAC,IAAI,EAAE,CAAC;YACf,CAAC,EAAE,cAAc,CAAC,CAAC;YAEnB,KAAK,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,GAAG,EAAE,EAAE;gBACxB,YAAY,CAAC,KAAK,CAAC,CAAC;gBACpB,MAAM,CACJ,IAAI,KAAK,CAAC,oDAAoD,GAAG,CAAC,OAAO,KAAK,MAAM,EAAE,CAAC,CACxF,CAAC;YACJ,CAAC,CAAC,CAAC;YAEH,KAAK,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,IAAI,EAAE,EAAE;gBACzB,YAAY,CAAC,KAAK,CAAC,CAAC;gBACpB,IAAI,QAAQ,EAAE,CAAC;oBACb,MAAM,CACJ,MAAM,CAAC,MAAM,CACX,IAAI,KAAK,CACP,8CAA8C,cAAc,GAAG,IAAI,OAAO,MAAM,EAAE,CACnF,EACD,EAAE,IAAI,EAAE,WAAW,EAAE,CACtB,CACF,CAAC;oBACF,OAAO;gBACT,CAAC;gBAED,IAAI,IAAI,KAAK,CAAC,EAAE,CAAC;oBACf,MAAM,SAAS,GAAG,IAAI,KAAK,CAAC,4BAA4B,IAAI,KAAK,MAAM,EAAE,CAAC,CAAC;oBAC3E,IAAI,YAAY,CAAC,SAAS,CAAC,EAAE,CAAC;wBAC5B,SAAS,CAAC,IAAI,GAAG,YAAY,CAAC;wBAC9B,MAAM,CAAC,SAAS,CAAC,CAAC;oBACpB,CAAC;yBAAM,CAAC;wBACN,MAAM,CACJ,IAAI,KAAK,CAAC,oDAAoD,SAAS,CAAC,OAAO,EAAE,CAAC,CACnF,CAAC;oBACJ,CAAC;oBACD,OAAO;gBACT,CAAC;gBAED,OAAO,CAAC,MAAM,CAAC,CAAC;YAClB,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,MAAM,MAAM,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,OAAO,CAAC;QAEpC,MAAM,MAAM,GAAG,kBAAkB,CAAC,GAAG,CAAC,CAAC;QACvC,IAAI,MAAM,EAAE,CAAC;YACX,OAAO;gBACL,OAAO,EAAE,MAAM,CAAC,OAAO;gBACvB,WAAW,EAAE,MAAM,CAAC,WAAW;gBAC/B,YAAY,EAAE,MAAM,CAAC,YAAY;gBACjC,UAAU,EAAE,MAAM,CAAC,SAAS,IAAI,MAAM;aACvC,CAAC;QACJ,CAAC;QAED,OAAO,EAAE,OAAO,EAAE,GAAG,CAAC,IAAI,EAAE,EAAE,WAAW,EAAE,IAAI,EAAE,YAAY,EAAE,IAAI,EAAE,UAAU,EAAE,MAAM,EAAE,CAAC;IAC5F,CAAC;YAAS,CAAC;QACT,IAAI,CAAC;YACH,EAAE,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC;QAC1B,CAAC;QAAC,MAAM,CAAC;YACP,8BAA8B;QAChC,CAAC;IACH,CAAC;AACH,CAAC"}

package/dist/orchestrators/shell-orchestrator.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=shell-orchestrator.test.d.ts.map

package/dist/orchestrators/shell-orchestrator.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"shell-orchestrator.test.d.ts","sourceRoot":"","sources":["../../src/orchestrators/shell-orchestrator.test.ts"],"names":[],"mappings":""}

package/dist/orchestrators/shell-orchestrator.test.js

@@ -0,0 +1,267 @@
+import { EventEmitter } from 'node:events';
+import * as fs from 'node:fs';
+import * as os from 'node:os';
+import * as path from 'node:path';
+import { afterEach, beforeEach, describe, expect, it, vi } from 'vitest';
+import { invokeShellOrchestrator, tryParseGeminiJson } from './shell-orchestrator.js';
+function createMockChild() {
+    const child = new EventEmitter();
+    child.stdout = new EventEmitter();
+    child.stderr = new EventEmitter();
+    child.kill = vi.fn();
+    return child;
+}
+let mockChild;
+vi.mock('node:child_process', () => ({
+    spawn: vi.fn(() => mockChild),
+}));
+const { spawn } = await import('node:child_process');
+const mockedSpawn = vi.mocked(spawn);
+// ─── invokeShellOrchestrator ─────────────────────────
+describe('invokeShellOrchestrator', () => {
+    let tmpDir;
+    const totemDir = '.totem';
+    beforeEach(() => {
+        tmpDir = fs.mkdtempSync(path.join(os.tmpdir(), 'totem-orch-'));
+        vi.spyOn(console, 'error').mockImplementation(() => { });
+        mockChild = createMockChild();
+        mockedSpawn.mockClear();
+    });
+    afterEach(() => {
+        fs.rmSync(tmpDir, { recursive: true, force: true });
+        vi.restoreAllMocks();
+    });
+    /** Emit stdout data and close with success */
+    function emitSuccess(data) {
+        process.nextTick(() => {
+            mockChild.stdout.emit('data', Buffer.from(data));
+            mockChild.emit('close', 0);
+        });
+    }
+    /** Emit close with non-zero exit code and optional stderr */
+    function emitFailure(code, stderr = '') {
+        process.nextTick(() => {
+            if (stderr)
+                mockChild.stderr.emit('data', Buffer.from(stderr));
+            mockChild.emit('close', code);
+        });
+    }
+    it('returns raw content when output is not Gemini JSON', async () => {
+        emitSuccess('  The answer is 42.  ');
+        const result = await invokeShellOrchestrator({
+            prompt: 'prompt text',
+            command: 'echo {file}',
+            model: 'test-model',
+            cwd: tmpDir,
+            tag: 'Test',
+            totemDir,
+        });
+        expect(result.content).toBe('The answer is 42.');
+        expect(result.inputTokens).toBeNull();
+        expect(result.outputTokens).toBeNull();
+        expect(result.durationMs).toBeGreaterThanOrEqual(0);
+    });
+    it('parses Gemini JSON output and returns structured result', async () => {
+        const geminiOutput = JSON.stringify({
+            response: 'Gemini says hello.',
+            stats: {
+                models: {
+                    'gemini-2.5-pro': {
+                        tokens: { input: 100, candidates: 50 },
+                        api: { totalLatencyMs: 2000 },
+                    },
+                },
+            },
+        });
+        emitSuccess(geminiOutput);
+        const result = await invokeShellOrchestrator({
+            prompt: 'prompt',
+            command: 'gemini -e none < {file}',
+            model: 'gemini-2.5-pro',
+            cwd: tmpDir,
+            tag: 'Test',
+            totemDir,
+        });
+        expect(result.content).toBe('Gemini says hello.');
+        expect(result.inputTokens).toBe(100);
+        expect(result.outputTokens).toBe(50);
+        expect(result.durationMs).toBe(2000);
+    });
+    it('substitutes {file} and {model} in command', async () => {
+        emitSuccess('ok');
+        await invokeShellOrchestrator({
+            prompt: 'prompt',
+            command: 'llm --model {model} < {file}',
+            model: 'my-model',
+            cwd: tmpDir,
+            tag: 'Test',
+            totemDir,
+        });
+        const cmd = mockedSpawn.mock.calls[0][0];
+        expect(cmd).toContain('my-model');
+        expect(cmd).not.toContain('{model}');
+        expect(cmd).not.toContain('{file}');
+    });
+    it('writes prompt to temp file and cleans up after', async () => {
+        emitSuccess('result');
+        await invokeShellOrchestrator({
+            prompt: 'my prompt content',
+            command: 'cat {file}',
+            model: 'model',
+            cwd: tmpDir,
+            tag: 'Test',
+            totemDir,
+        });
+        // Temp file should be cleaned up
+        const tempDir = path.join(tmpDir, totemDir, 'temp');
+        if (fs.existsSync(tempDir)) {
+            const files = fs.readdirSync(tempDir).filter((f) => f.startsWith('totem-test-'));
+            expect(files).toHaveLength(0);
+        }
+    });
+    it('throws QuotaError for quota-related failures', async () => {
+        emitFailure(1, '429 Too Many Requests quota exceeded');
+        try {
+            await invokeShellOrchestrator({
+                prompt: 'prompt',
+                command: 'cmd',
+                model: 'model',
+                cwd: tmpDir,
+                tag: 'Test',
+                totemDir,
+            });
+            expect.fail('Should have thrown');
+        }
+        catch (err) {
+            expect(err.name).toBe('QuotaError');
+        }
+    });
+    it('throws generic error for non-zero exit code', async () => {
+        emitFailure(1, 'something went wrong');
+        await expect(invokeShellOrchestrator({
+            prompt: 'prompt',
+            command: 'cmd',
+            model: 'model',
+            cwd: tmpDir,
+            tag: 'Test',
+            totemDir,
+        })).rejects.toThrow('[Totem Error] Shell orchestrator command failed');
+    });
+    it('throws error on spawn error event', async () => {
+        process.nextTick(() => {
+            mockChild.emit('error', new Error('command not found'));
+        });
+        await expect(invokeShellOrchestrator({
+            prompt: 'prompt',
+            command: 'cmd',
+            model: 'model',
+            cwd: tmpDir,
+            tag: 'Test',
+            totemDir,
+        })).rejects.toThrow('command not found');
+    });
+    it('throws descriptive error for timeout', async () => {
+        vi.useFakeTimers();
+        // Simulate kill → close (rejection now happens in the close handler)
+        mockChild.kill = vi.fn(() => {
+            process.nextTick(() => mockChild.emit('close', null));
+        });
+        // Capture the rejection before advancing timers
+        const promise = invokeShellOrchestrator({
+            prompt: 'prompt',
+            command: 'cmd',
+            model: 'model',
+            cwd: tmpDir,
+            tag: 'Test',
+            totemDir,
+        }).catch((err) => err);
+        await vi.advanceTimersByTimeAsync(180_001);
+        const err = await promise;
+        expect(err).toBeInstanceOf(Error);
+        expect(err.message).toContain('timed out after 180s');
+        expect(mockChild.kill).toHaveBeenCalled();
+        vi.useRealTimers();
+    });
+});
+// ─── tryParseGeminiJson ──────────────────────────────
+describe('tryParseGeminiJson', () => {
+    it('returns null for non-JSON input', () => {
+        expect(tryParseGeminiJson('plain text output')).toBeNull();
+    });
+    it('returns null for JSON that does not match Gemini schema', () => {
+        expect(tryParseGeminiJson('{"foo": "bar"}')).toBeNull();
+    });
+    it('returns null when stats.models is empty', () => {
+        const input = JSON.stringify({
+            response: 'hello',
+            stats: { models: {} },
+        });
+        expect(tryParseGeminiJson(input)).toBeNull();
+    });
+    it('parses valid Gemini output with token stats', () => {
+        const input = JSON.stringify({
+            response: 'The answer is 42.',
+            stats: {
+                models: {
+                    'gemini-2.5-pro': {
+                        tokens: { input: 500, candidates: 200 },
+                        api: { totalLatencyMs: 3000 },
+                    },
+                },
+            },
+        });
+        const result = tryParseGeminiJson(input);
+        expect(result).toEqual({
+            content: 'The answer is 42.',
+            inputTokens: 500,
+            outputTokens: 200,
+            latencyMs: 3000,
+        });
+    });
+    it('aggregates stats across multiple models', () => {
+        const input = JSON.stringify({
+            response: 'multi-model',
+            stats: {
+                models: {
+                    'model-a': { tokens: { input: 100, candidates: 50 }, api: { totalLatencyMs: 1000 } },
+                    'model-b': { tokens: { input: 200, candidates: 75 }, api: { totalLatencyMs: 2000 } },
+                },
+            },
+        });
+        const result = tryParseGeminiJson(input);
+        expect(result).toEqual({
+            content: 'multi-model',
+            inputTokens: 300,
+            outputTokens: 125,
+            latencyMs: 3000,
+        });
+    });
+    it('returns latencyMs null when api stats are missing', () => {
+        const input = JSON.stringify({
+            response: 'no api stats',
+            stats: {
+                models: {
+                    'gemini-flash': { tokens: { input: 10, candidates: 5 } },
+                },
+            },
+        });
+        const result = tryParseGeminiJson(input);
+        expect(result).not.toBeNull();
+        expect(result.latencyMs).toBeNull();
+    });
+    it('defaults missing token counts to zero', () => {
+        const input = JSON.stringify({
+            response: 'no tokens key',
+            stats: {
+                models: {
+                    'gemini-flash': { api: { totalLatencyMs: 500 } },
+                },
+            },
+        });
+        const result = tryParseGeminiJson(input);
+        expect(result).not.toBeNull();
+        expect(result.inputTokens).toBe(0);
+        expect(result.outputTokens).toBe(0);
+    });
+});
+//# sourceMappingURL=shell-orchestrator.test.js.map

package/dist/orchestrators/shell-orchestrator.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"shell-orchestrator.test.js","sourceRoot":"","sources":["../../src/orchestrators/shell-orchestrator.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AAC3C,OAAO,KAAK,EAAE,MAAM,SAAS,CAAC;AAC9B,OAAO,KAAK,EAAE,MAAM,SAAS,CAAC;AAC9B,OAAO,KAAK,IAAI,MAAM,WAAW,CAAC;AAElC,OAAO,EAAE,SAAS,EAAE,UAAU,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAEzE,OAAO,EAAE,uBAAuB,EAAE,kBAAkB,EAAE,MAAM,yBAAyB,CAAC;AAUtF,SAAS,eAAe;IACtB,MAAM,KAAK,GAAG,IAAI,YAAY,EAAe,CAAC;IAC9C,KAAK,CAAC,MAAM,GAAG,IAAI,YAAY,EAAE,CAAC;IAClC,KAAK,CAAC,MAAM,GAAG,IAAI,YAAY,EAAE,CAAC;IAClC,KAAK,CAAC,IAAI,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;IACrB,OAAO,KAAK,CAAC;AACf,CAAC;AAED,IAAI,SAAoB,CAAC;AAEzB,EAAE,CAAC,IAAI,CAAC,oBAAoB,EAAE,GAAG,EAAE,CAAC,CAAC;IACnC,KAAK,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC;CAC9B,CAAC,CAAC,CAAC;AAEJ,MAAM,EAAE,KAAK,EAAE,GAAG,MAAM,MAAM,CAAC,oBAAoB,CAAC,CAAC;AACrD,MAAM,WAAW,GAAG,EAAE,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;AAErC,wDAAwD;AAExD,QAAQ,CAAC,yBAAyB,EAAE,GAAG,EAAE;IACvC,IAAI,MAAc,CAAC;IACnB,MAAM,QAAQ,GAAG,QAAQ,CAAC;IAE1B,UAAU,CAAC,GAAG,EAAE;QACd,MAAM,GAAG,EAAE,CAAC,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,EAAE,aAAa,CAAC,CAAC,CAAC;QAC/D,EAAE,CAAC,KAAK,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC,kBAAkB,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;QACxD,SAAS,GAAG,eAAe,EAAE,CAAC;QAC9B,WAAW,CAAC,SAAS,EAAE,CAAC;IAC1B,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,EAAE,CAAC,MAAM,CAAC,MAAM,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;QACpD,EAAE,CAAC,eAAe,EAAE,CAAC;IACvB,CAAC,CAAC,CAAC;IAEH,8CAA8C;IAC9C,SAAS,WAAW,CAAC,IAAY;QAC/B,OAAO,CAAC,QAAQ,CAAC,GAAG,EAAE;YACpB,SAAS,CAAC,MAAM,CAAC,IAAI,CAAC,MAAM,EAAE,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;YACjD,SAAS,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC;QAC7B,CAAC,CAAC,CAAC;IACL,CAAC;IAED,6DAA6D;IAC7D,SAAS,WAAW,CAAC,IAAY,EAAE,MAAM,GAAG,EAAE;QAC5C,OAAO,CAAC,QAAQ,CAAC,GAAG,EAAE;YACpB,IAAI,MAAM;gBAAE,SAAS,CAAC,MAAM,CAAC,IAAI,CAAC,MAAM,EAAE,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC;YAC/D,SAAS,CAAC,IAAI,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;QAChC,CAAC,CAAC,CAAC;IACL,CAAC;IAED,EAAE,CAAC,oDAAoD,EAAE,KAAK,IAAI,EAAE;QAClE,WAAW,CAAC,uBAAuB,CAAC,CAAC;QACrC,MAAM,MAAM,GAAG,MAAM,uBAAuB,CAAC;YAC3C,MAAM,EAAE,aAAa;YACrB,OAAO,EAAE,aAAa;YACtB,KAAK,EAAE,YAAY;YACnB,GAAG,EAAE,MAAM;YACX,GAAG,EAAE,MAAM;YACX,QAAQ;SACT,CAAC,CAAC;QACH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;QACjD,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC,QAAQ,EAAE,CAAC;QACtC,MAAM,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,QAAQ,EAAE,CAAC;QACvC,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,sBAAsB,CAAC,CAAC,CAAC,CAAC;IACtD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yDAAyD,EAAE,KAAK,IAAI,EAAE;QACvE,MAAM,YAAY,GAAG,IAAI,CAAC,SAAS,CAAC;YAClC,QAAQ,EAAE,oBAAoB;YAC9B,KAAK,EAAE;gBACL,MAAM,EAAE;oBACN,gBAAgB,EAAE;wBAChB,MAAM,EAAE,EAAE,KAAK,EAAE,GAAG,EAAE,UAAU,EAAE,EAAE,EAAE;wBACtC,GAAG,EAAE,EAAE,cAAc,EAAE,IAAI,EAAE;qBAC9B;iBACF;aACF;SACF,CAAC,CAAC;QACH,WAAW,CAAC,YAAY,CAAC,CAAC;QAC1B,MAAM,MAAM,GAAG,MAAM,uBAAuB,CAAC;YAC3C,MAAM,EAAE,QAAQ;YAChB,OAAO,EAAE,yBAAyB;YAClC,KAAK,EAAE,gBAAgB;YACvB,GAAG,EAAE,MAAM;YACX,GAAG,EAAE,MAAM;YACX,QAAQ;SACT,CAAC,CAAC;QACH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC;QAClD,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACrC,MAAM,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACrC,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACvC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2CAA2C,EAAE,KAAK,IAAI,EAAE;QACzD,WAAW,CAAC,IAAI,CAAC,CAAC;QAClB,MAAM,uBAAuB,CAAC;YAC5B,MAAM,EAAE,QAAQ;YAChB,OAAO,EAAE,8BAA8B;YACvC,KAAK,EAAE,UAAU;YACjB,GAAG,EAAE,MAAM;YACX,GAAG,EAAE,MAAM;YACX,QAAQ;SACT,CAAC,CAAC;QACH,MAAM,GAAG,GAAG,WAAW,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAE,CAAC,CAAC,CAAW,CAAC;QACpD,MAAM,CAAC,GAAG,CAAC,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC;QAClC,MAAM,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC;QACrC,MAAM,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;IACtC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gDAAgD,EAAE,KAAK,IAAI,EAAE;QAC9D,WAAW,CAAC,QAAQ,CAAC,CAAC;QACtB,MAAM,uBAAuB,CAAC;YAC5B,MAAM,EAAE,mBAAmB;YAC3B,OAAO,EAAE,YAAY;YACrB,KAAK,EAAE,OAAO;YACd,GAAG,EAAE,MAAM;YACX,GAAG,EAAE,MAAM;YACX,QAAQ;SACT,CAAC,CAAC;QACH,iCAAiC;QACjC,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,QAAQ,EAAE,MAAM,CAAC,CAAC;QACpD,IAAI,EAAE,CAAC,UAAU,CAAC,OAAO,CAAC,EAAE,CAAC;YAC3B,MAAM,KAAK,GAAG,EAAE,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,UAAU,CAAC,aAAa,CAAC,CAAC,CAAC;YACjF,MAAM,CAAC,KAAK,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAChC,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8CAA8C,EAAE,KAAK,IAAI,EAAE;QAC5D,WAAW,CAAC,CAAC,EAAE,sCAAsC,CAAC,CAAC;QACvD,IAAI,CAAC;YACH,MAAM,uBAAuB,CAAC;gBAC5B,MAAM,EAAE,QAAQ;gBAChB,OAAO,EAAE,KAAK;gBACd,KAAK,EAAE,OAAO;gBACd,GAAG,EAAE,MAAM;gBACX,GAAG,EAAE,MAAM;gBACX,QAAQ;aACT,CAAC,CAAC;YACH,MAAM,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC;QACpC,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,MAAM,CAAE,GAAa,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;QACjD,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6CAA6C,EAAE,KAAK,IAAI,EAAE;QAC3D,WAAW,CAAC,CAAC,EAAE,sBAAsB,CAAC,CAAC;QACvC,MAAM,MAAM,CACV,uBAAuB,CAAC;YACtB,MAAM,EAAE,QAAQ;YAChB,OAAO,EAAE,KAAK;YACd,KAAK,EAAE,OAAO;YACd,GAAG,EAAE,MAAM;YACX,GAAG,EAAE,MAAM;YACX,QAAQ;SACT,CAAC,CACH,CAAC,OAAO,CAAC,OAAO,CAAC,iDAAiD,CAAC,CAAC;IACvE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,mCAAmC,EAAE,KAAK,IAAI,EAAE;QACjD,OAAO,CAAC,QAAQ,CAAC,GAAG,EAAE;YACpB,SAAS,CAAC,IAAI,CAAC,OAAO,EAAE,IAAI,KAAK,CAAC,mBAAmB,CAAC,CAAC,CAAC;QAC1D,CAAC,CAAC,CAAC;QACH,MAAM,MAAM,CACV,uBAAuB,CAAC;YACtB,MAAM,EAAE,QAAQ;YAChB,OAAO,EAAE,KAAK;YACd,KAAK,EAAE,OAAO;YACd,GAAG,EAAE,MAAM;YACX,GAAG,EAAE,MAAM;YACX,QAAQ;SACT,CAAC,CACH,CAAC,OAAO,CAAC,OAAO,CAAC,mBAAmB,CAAC,CAAC;IACzC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sCAAsC,EAAE,KAAK,IAAI,EAAE;QACpD,EAAE,CAAC,aAAa,EAAE,CAAC;QAEnB,qEAAqE;QACrE,SAAS,CAAC,IAAI,GAAG,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE;YAC1B,OAAO,CAAC,QAAQ,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,IAAI,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC,CAAC;QACxD,CAAC,CAAC,CAAC;QAEH,gDAAgD;QAChD,MAAM,OAAO,GAAG,uBAAuB,CAAC;YACtC,MAAM,EAAE,QAAQ;YAChB,OAAO,EAAE,KAAK;YACd,KAAK,EAAE,OAAO;YACd,GAAG,EAAE,MAAM;YACX,GAAG,EAAE,MAAM;YACX,QAAQ;SACT,CAAC,CAAC,KAAK,CAAC,CAAC,GAAU,EAAE,EAAE,CAAC,GAAG,CAAC,CAAC;QAE9B,MAAM,EAAE,CAAC,wBAAwB,CAAC,OAAO,CAAC,CAAC;QAE3C,MAAM,GAAG,GAAG,MAAM,OAAO,CAAC;QAC1B,MAAM,CAAC,GAAG,CAAC,CAAC,cAAc,CAAC,KAAK,CAAC,CAAC;QAClC,MAAM,CAAE,GAAa,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,sBAAsB,CAAC,CAAC;QACjE,MAAM,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,gBAAgB,EAAE,CAAC;QAE1C,EAAE,CAAC,aAAa,EAAE,CAAC;IACrB,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,wDAAwD;AAExD,QAAQ,CAAC,oBAAoB,EAAE,GAAG,EAAE;IAClC,EAAE,CAAC,iCAAiC,EAAE,GAAG,EAAE;QACzC,MAAM,CAAC,kBAAkB,CAAC,mBAAmB,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAC;IAC7D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yDAAyD,EAAE,GAAG,EAAE;QACjE,MAAM,CAAC,kBAAkB,CAAC,gBAAgB,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAC;IAC1D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yCAAyC,EAAE,GAAG,EAAE;QACjD,MAAM,KAAK,GAAG,IAAI,CAAC,SAAS,CAAC;YAC3B,QAAQ,EAAE,OAAO;YACjB,KAAK,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE;SACtB,CAAC,CAAC;QACH,MAAM,CAAC,kBAAkB,CAAC,KAAK,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAC;IAC/C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6CAA6C,EAAE,GAAG,EAAE;QACrD,MAAM,KAAK,GAAG,IAAI,CAAC,SAAS,CAAC;YAC3B,QAAQ,EAAE,mBAAmB;YAC7B,KAAK,EAAE;gBACL,MAAM,EAAE;oBACN,gBAAgB,EAAE;wBAChB,MAAM,EAAE,EAAE,KAAK,EAAE,GAAG,EAAE,UAAU,EAAE,GAAG,EAAE;wBACvC,GAAG,EAAE,EAAE,cAAc,EAAE,IAAI,EAAE;qBAC9B;iBACF;aACF;SACF,CAAC,CAAC;QACH,MAAM,MAAM,GAAG,kBAAkB,CAAC,KAAK,CAAC,CAAC;QACzC,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;YACrB,OAAO,EAAE,mBAAmB;YAC5B,WAAW,EAAE,GAAG;YAChB,YAAY,EAAE,GAAG;YACjB,SAAS,EAAE,IAAI;SAChB,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yCAAyC,EAAE,GAAG,EAAE;QACjD,MAAM,KAAK,GAAG,IAAI,CAAC,SAAS,CAAC;YAC3B,QAAQ,EAAE,aAAa;YACvB,KAAK,EAAE;gBACL,MAAM,EAAE;oBACN,SAAS,EAAE,EAAE,MAAM,EAAE,EAAE,KAAK,EAAE,GAAG,EAAE,UAAU,EAAE,EAAE,EAAE,EAAE,GAAG,EAAE,EAAE,cAAc,EAAE,IAAI,EAAE,EAAE;oBACpF,SAAS,EAAE,EAAE,MAAM,EAAE,EAAE,KAAK,EAAE,GAAG,EAAE,UAAU,EAAE,EAAE,EAAE,EAAE,GAAG,EAAE,EAAE,cAAc,EAAE,IAAI,EAAE,EAAE;iBACrF;aACF;SACF,CAAC,CAAC;QACH,MAAM,MAAM,GAAG,kBAAkB,CAAC,KAAK,CAAC,CAAC;QACzC,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;YACrB,OAAO,EAAE,aAAa;YACtB,WAAW,EAAE,GAAG;YAChB,YAAY,EAAE,GAAG;YACjB,SAAS,EAAE,IAAI;SAChB,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,mDAAmD,EAAE,GAAG,EAAE;QAC3D,MAAM,KAAK,GAAG,IAAI,CAAC,SAAS,CAAC;YAC3B,QAAQ,EAAE,cAAc;YACxB,KAAK,EAAE;gBACL,MAAM,EAAE;oBACN,cAAc,EAAE,EAAE,MAAM,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,UAAU,EAAE,CAAC,EAAE,EAAE;iBACzD;aACF;SACF,CAAC,CAAC;QACH,MAAM,MAAM,GAAG,kBAAkB,CAAC,KAAK,CAAC,CAAC;QACzC,MAAM,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,QAAQ,EAAE,CAAC;QAC9B,MAAM,CAAC,MAAO,CAAC,SAAS,CAAC,CAAC,QAAQ,EAAE,CAAC;IACvC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uCAAuC,EAAE,GAAG,EAAE;QAC/C,MAAM,KAAK,GAAG,IAAI,CAAC,SAAS,CAAC;YAC3B,QAAQ,EAAE,eAAe;YACzB,KAAK,EAAE;gBACL,MAAM,EAAE;oBACN,cAAc,EAAE,EAAE,GAAG,EAAE,EAAE,cAAc,EAAE,GAAG,EAAE,EAAE;iBACjD;aACF;SACF,CAAC,CAAC;QACH,MAAM,MAAM,GAAG,kBAAkB,CAAC,KAAK,CAAC,CAAC;QACzC,MAAM,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,QAAQ,EAAE,CAAC;QAC9B,MAAM,CAAC,MAAO,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACpC,MAAM,CAAC,MAAO,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IACvC,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/utils.d.ts

@@ -16,23 +16,6 @@ export declare function loadConfig(configPath: string): Promise<TotemConfig>;
  */
 export declare function resolveConfigPath(cwd: string): string;
 export { requireEmbedding } from '@mmnto/totem';
-export interface OrchestratorResult {
-    content: string;
-    inputTokens: number | null;
-    outputTokens: number | null;
-    durationMs: number;
-}
-/**
- * Try to parse Gemini CLI JSON output. Returns extracted data or null if
- * the output is not valid Gemini JSON (e.g. raw text from a non-Gemini orchestrator).
- */
-export declare function tryParseGeminiJson(raw: string): {
-    content: string;
-    inputTokens: number;
-    outputTokens: number;
-    latencyMs: number | null;
-} | null;
-export declare function invokeShellOrchestrator(prompt: string, command: string, model: string, cwd: string, tag: string, totemDir: string): OrchestratorResult;
 /**
  * Reap orphaned temp files older than `maxAgeMs` from `.totem/temp/`.
  * Fire-and-forget — never blocks the CLI critical path.
@@ -55,7 +38,7 @@ export interface OrchestratorRunOptions {
 }
 /**
  * Validate orchestrator config, then either output raw context (--raw) or
- * invoke the shell orchestrator and return the LLM content.
+ * invoke the configured orchestrator provider and return the LLM content.
  *
  * Returns `undefined` in --raw mode (prompt already written to output).
  * Returns the LLM response content string otherwise.
@@ -68,5 +51,5 @@ export declare function runOrchestrator(opts: {
     config: TotemConfig;
     cwd: string;
     totalResults?: number;
-}): string | undefined;
+}): Promise<string | undefined>;
 //# sourceMappingURL=utils.d.ts.map

package/dist/utils.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"utils.d.ts","sourceRoot":"","sources":["../src/utils.ts"],"names":[],"mappings":"AAOA,OAAO,KAAK,EAAE,YAAY,EAAE,WAAW,EAAE,MAAM,cAAc,CAAC;AAa9D,+EAA+E;AAC/E,eAAO,MAAM,MAAM,SAA+B,CAAC;AAEnD,yCAAyC;AACzC,eAAO,MAAM,aAAa,QAAS,CAAC;AAEpC;;GAEG;AACH,wBAAgB,OAAO,CAAC,GAAG,EAAE,MAAM,GAAG,IAAI,CAiBzC;AAED;;GAEG;AACH,wBAAsB,UAAU,CAAC,UAAU,EAAE,MAAM,GAAG,OAAO,CAAC,WAAW,CAAC,CAMzE;AAED;;GAEG;AACH,wBAAgB,iBAAiB,CAAC,GAAG,EAAE,MAAM,GAAG,MAAM,CAMrD;AAGD,OAAO,EAAE,gBAAgB,EAAE,MAAM,cAAc,CAAC;AAchD,MAAM,WAAW,kBAAkB;IACjC,OAAO,EAAE,MAAM,CAAC;IAChB,WAAW,EAAE,MAAM,GAAG,IAAI,CAAC;IAC3B,YAAY,EAAE,MAAM,GAAG,IAAI,CAAC;IAC5B,UAAU,EAAE,MAAM,CAAC;CACpB;AAwBD;;;GAGG;AACH,wBAAgB,kBAAkB,CAChC,GAAG,EAAE,MAAM,GACV;IAAE,OAAO,EAAE,MAAM,CAAC;IAAC,WAAW,EAAE,MAAM,CAAC;IAAC,YAAY,EAAE,MAAM,CAAC;IAAC,SAAS,EAAE,MAAM,GAAG,IAAI,CAAA;CAAE,GAAG,IAAI,CAwBjG;AAID,wBAAgB,uBAAuB,CACrC,MAAM,EAAE,MAAM,EACd,OAAO,EAAE,MAAM,EACf,KAAK,EAAE,MAAM,EACb,GAAG,EAAE,MAAM,EACX,GAAG,EAAE,MAAM,EACX,QAAQ,EAAE,MAAM,GACf,kBAAkB,CA6EpB;AAOD;;;GAGG;AACH,wBAAsB,qBAAqB,CACzC,GAAG,EAAE,MAAM,EACX,QAAQ,EAAE,MAAM,EAChB,QAAQ,GAAE,MAAwB,GACjC,OAAO,CAAC,MAAM,CAAC,CA8BjB;AAMD;;;GAGG;AACH,wBAAgB,eAAe,CAC7B,WAAW,EAAE,MAAM,EACnB,aAAa,EAAE,MAAM,EACrB,GAAG,EAAE,MAAM,EACX,QAAQ,EAAE,MAAM,GACf,MAAM,CAaR;AAID,wBAAgB,WAAW,CAAC,OAAO,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,MAAM,GAAG,IAAI,CAUnE;AAKD,OAAO,EAAE,QAAQ,EAAE,MAAM,cAAc,CAAC;AAKxC,OAAO,EAAE,OAAO,EAAE,MAAM,cAAc,CAAC;AAOvC,wBAAgB,aAAa,CAC3B,OAAO,EAAE,YAAY,EAAE,EACvB,OAAO,EAAE,MAAM,EACf,SAAS,CAAC,EAAE,OAAO,GAClB,MAAM,CAqBR;AAID,MAAM,WAAW,sBAAsB;IACrC,GAAG,CAAC,EAAE,OAAO,CAAC;IACd,GAAG,CAAC,EAAE,MAAM,CAAC;IACb,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,KAAK,CAAC,EAAE,OAAO,CAAC;CACjB;AAYD;;;;;;;GAOG;AACH,wBAAgB,eAAe,CAAC,IAAI,EAAE;IACpC,MAAM,EAAE,MAAM,CAAC;IACf,GAAG,EAAE,MAAM,CAAC;IACZ,OAAO,EAAE,sBAAsB,CAAC;IAChC,MAAM,EAAE,WAAW,CAAC;IACpB,GAAG,EAAE,MAAM,CAAC;IACZ,YAAY,CAAC,EAAE,MAAM,CAAC;CACvB,GAAG,MAAM,GAAG,SAAS,CA6KrB"}
+{"version":3,"file":"utils.d.ts","sourceRoot":"","sources":["../src/utils.ts"],"names":[],"mappings":"AAIA,OAAO,KAAK,EAAE,YAAY,EAAE,WAAW,EAAE,MAAM,cAAc,CAAC;AAY9D,+EAA+E;AAC/E,eAAO,MAAM,MAAM,SAA+B,CAAC;AAEnD,yCAAyC;AACzC,eAAO,MAAM,aAAa,QAAS,CAAC;AAEpC;;GAEG;AACH,wBAAgB,OAAO,CAAC,GAAG,EAAE,MAAM,GAAG,IAAI,CAiBzC;AAED;;GAEG;AACH,wBAAsB,UAAU,CAAC,UAAU,EAAE,MAAM,GAAG,OAAO,CAAC,WAAW,CAAC,CAMzE;AAED;;GAEG;AACH,wBAAgB,iBAAiB,CAAC,GAAG,EAAE,MAAM,GAAG,MAAM,CAMrD;AAGD,OAAO,EAAE,gBAAgB,EAAE,MAAM,cAAc,CAAC;AA+BhD;;;GAGG;AACH,wBAAsB,qBAAqB,CACzC,GAAG,EAAE,MAAM,EACX,QAAQ,EAAE,MAAM,EAChB,QAAQ,GAAE,MAAwB,GACjC,OAAO,CAAC,MAAM,CAAC,CA8BjB;AAMD;;;GAGG;AACH,wBAAgB,eAAe,CAC7B,WAAW,EAAE,MAAM,EACnB,aAAa,EAAE,MAAM,EACrB,GAAG,EAAE,MAAM,EACX,QAAQ,EAAE,MAAM,GACf,MAAM,CAaR;AAID,wBAAgB,WAAW,CAAC,OAAO,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,MAAM,GAAG,IAAI,CAUnE;AAKD,OAAO,EAAE,QAAQ,EAAE,MAAM,cAAc,CAAC;AAKxC,OAAO,EAAE,OAAO,EAAE,MAAM,cAAc,CAAC;AAOvC,wBAAgB,aAAa,CAC3B,OAAO,EAAE,YAAY,EAAE,EACvB,OAAO,EAAE,MAAM,EACf,SAAS,CAAC,EAAE,OAAO,GAClB,MAAM,CAqBR;AAID,MAAM,WAAW,sBAAsB;IACrC,GAAG,CAAC,EAAE,OAAO,CAAC;IACd,GAAG,CAAC,EAAE,MAAM,CAAC;IACb,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,KAAK,CAAC,EAAE,OAAO,CAAC;CACjB;AAYD;;;;;;;GAOG;AACH,wBAAsB,eAAe,CAAC,IAAI,EAAE;IAC1C,MAAM,EAAE,MAAM,CAAC;IACf,GAAG,EAAE,MAAM,CAAC;IACZ,OAAO,EAAE,sBAAsB,CAAC;IAChC,MAAM,EAAE,WAAW,CAAC;IACpB,GAAG,EAAE,MAAM,CAAC;IACZ,YAAY,CAAC,EAAE,MAAM,CAAC;CACvB,GAAG,OAAO,CAAC,MAAM,GAAG,SAAS,CAAC,CA2J9B"}

package/dist/utils.js

@@ -1,14 +1,10 @@
-import { execSync } from 'node:child_process';
 import * as crypto from 'node:crypto';
 import * as fs from 'node:fs';
 import * as path from 'node:path';
-import { z } from 'zod';
 import { TotemConfigSchema } from '@mmnto/totem';
+import { createOrchestrator } from './orchestrators/orchestrator.js';
 import { bold, log } from './ui.js';
 // ─── Shared constants ────────────────────────────────────
-const LLM_TIMEOUT_MS = 180_000;
-const LLM_MAX_BUFFER = 10 * 1024 * 1024; // 10MB — Gemini JSON responses can exceed Node's 1MB default
-const TEMP_ID_BYTES = 4;
 const MODEL_NAME_RE = /^[\w./:_-]+$/;
 const TELEMETRY_FILE = 'telemetry.jsonl';
 /** execFileSync on Windows can't resolve executables without `shell: true`. */
@@ -70,108 +66,6 @@ function appendTelemetry(entry, cwd, totemDir) {
         log.warn('Totem', `Failed to write telemetry: ${msg}`);
     }
 }
-const GeminiModelStatsSchema = z.object({
-    tokens: z.object({ input: z.number(), candidates: z.number() }).optional(),
-    api: z.object({ totalLatencyMs: z.number() }).optional(),
-});
-const GeminiOutputSchema = z.object({
-    response: z.string(),
-    stats: z.object({ models: z.record(GeminiModelStatsSchema) }),
-});
-/**
- * Try to parse Gemini CLI JSON output. Returns extracted data or null if
- * the output is not valid Gemini JSON (e.g. raw text from a non-Gemini orchestrator).
- */
-export function tryParseGeminiJson(raw) {
-    let data;
-    try {
-        data = JSON.parse(raw);
-    }
-    catch {
-        return null;
-    }
-    const result = GeminiOutputSchema.safeParse(data);
-    if (!result.success)
-        return null;
-    const allModelStats = Object.values(result.data.stats.models);
-    if (allModelStats.length === 0)
-        return null;
-    const inputTokens = allModelStats.reduce((sum, s) => sum + (s.tokens?.input ?? 0), 0);
-    const outputTokens = allModelStats.reduce((sum, s) => sum + (s.tokens?.candidates ?? 0), 0);
-    const latencyMs = allModelStats.reduce((sum, s) => sum + (s.api?.totalLatencyMs ?? 0), 0) || null;
-    return {
-        content: result.data.response,
-        inputTokens,
-        outputTokens,
-        latencyMs,
-    };
-}
-// ─── Shell orchestrator ─────────────────────────────────
-export function invokeShellOrchestrator(prompt, command, model, cwd, tag, totemDir) {
-    const tmpName = `totem-${tag.toLowerCase()}-${crypto.randomBytes(TEMP_ID_BYTES).toString('hex')}.md`;
-    const tempDir = path.join(cwd, totemDir, 'temp');
-    fs.mkdirSync(tempDir, { recursive: true });
-    const tempPath = path.join(tempDir, tmpName);
-    try {
-        fs.writeFileSync(tempPath, prompt, { encoding: 'utf-8', mode: 0o600 });
-        const quotedPath = IS_WIN
-            ? `"${tempPath.replace(/"/g, '""')}"`
-            : `'${tempPath.replace(/'/g, "'\\''")}'`;
-        const resolvedCmd = command.replace(/\{file\}/g, quotedPath).replace(/\{model\}/g, model);
-        log.info(tag, 'Invoking orchestrator (this may take 15-60 seconds)...');
-        const startMs = Date.now();
-        const raw = execSync(resolvedCmd, {
-            cwd,
-            encoding: 'utf-8',
-            timeout: LLM_TIMEOUT_MS,
-            maxBuffer: LLM_MAX_BUFFER,
-            stdio: ['pipe', 'pipe', 'pipe'],
-        });
-        const wallMs = Date.now() - startMs;
-        const gemini = tryParseGeminiJson(raw);
-        if (gemini) {
-            return {
-                content: gemini.content,
-                inputTokens: gemini.inputTokens,
-                outputTokens: gemini.outputTokens,
-                durationMs: gemini.latencyMs ?? wallMs,
-            };
-        }
-        return { content: raw.trim(), inputTokens: null, outputTokens: null, durationMs: wallMs };
-    }
-    catch (err) {
-        const execErr = err;
-        const stderr = execErr.stderr ? execErr.stderr.toString() : '';
-        const msg = err instanceof Error ? err.message : String(err);
-        const fullError = `${msg}\n${stderr}`;
-        // Detect buffer overflow — Node kills the process when stdout exceeds maxBuffer
-        if (execErr.code === 'ENOBUFS') {
-            throw new Error(`[Totem Error] Orchestrator response exceeded the ${LLM_MAX_BUFFER / 1024 / 1024}MB output buffer.\n` +
-                `This usually means the LLM produced an unexpectedly large response.\n${stderr}`);
-        }
-        // Detect timeout
-        if (execErr.code === 'ETIMEDOUT') {
-            throw new Error(`[Totem Error] Orchestrator timed out after ${LLM_TIMEOUT_MS / 1000}s.\n${stderr}`);
-        }
-        const lowerMsg = fullError.toLowerCase();
-        if (lowerMsg.includes('quota') ||
-            lowerMsg.includes('429') ||
-            lowerMsg.includes('too many requests')) {
-            const quotaErr = new Error(fullError);
-            quotaErr.name = 'QuotaError';
-            throw quotaErr;
-        }
-        throw new Error(`[Totem Error] Shell orchestrator command failed: ${fullError}`);
-    }
-    finally {
-        try {
-            fs.unlinkSync(tempPath);
-        }
-        catch {
-            // Temp cleanup is best-effort
-        }
-    }
-}
 // ─── Orphaned temp file cleanup ──────────────────────────
 const TEMP_FILE_RE = /^totem-.*\.md$/;
 const TEMP_MAX_AGE_MS = 24 * 60 * 60 * 1000; // 24 hours
@@ -282,13 +176,13 @@ const DEFAULT_TTLS = {
 };
 /**
  * Validate orchestrator config, then either output raw context (--raw) or
- * invoke the shell orchestrator and return the LLM content.
+ * invoke the configured orchestrator provider and return the LLM content.
  *
  * Returns `undefined` in --raw mode (prompt already written to output).
  * Returns the LLM response content string otherwise.
  * Callers are responsible for writing output via `writeOutput()`.
  */
-export function runOrchestrator(opts) {
+export async function runOrchestrator(opts) {
     const { prompt, tag, options, config, cwd } = opts;
     // --raw mode: output context only
     if (options.raw) {
@@ -302,9 +196,7 @@ export function runOrchestrator(opts) {
         throw new Error(`[Totem Error] No orchestrator configured. Add an 'orchestrator' block to totem.config.ts.\n` +
             `Example:\n  orchestrator: {\n    provider: 'shell',\n    command: 'gemini --model {model} -e none < {file}',\n    defaultModel: 'gemini-2.5-pro',\n  }`);
     }
-    if (config.orchestrator.provider !== 'shell') {
-        throw new Error(`[Totem Error] Unsupported orchestrator provider: '${config.orchestrator.provider}'. Only 'shell' is supported.`);
-    }
+    const invoke = createOrchestrator(config.orchestrator);
     const tagKey = tag.toLowerCase();
     let model = options.model ?? config.orchestrator.overrides?.[tagKey] ?? config.orchestrator.defaultModel;
     if (!model) {
@@ -342,7 +234,7 @@ export function runOrchestrator(opts) {
     }
     let result;
     try {
-        result = invokeShellOrchestrator(prompt, config.orchestrator.command, model, cwd, tag, config.totemDir);
+        result = await invoke({ prompt, model, cwd, tag, totemDir: config.totemDir });
     }
     catch (err) {
         if (err instanceof Error && err.name === 'QuotaError') {
@@ -350,7 +242,7 @@ export function runOrchestrator(opts) {
             if (fallback && model !== fallback) {
                 log.warn(tag, `Quota exhausted for ${model}. Retrying with fallback model: ${bold(fallback)}...`);
                 try {
-                    result = invokeShellOrchestrator(prompt, config.orchestrator.command, fallback, cwd, tag, config.totemDir);
+                    result = await invoke({ prompt, model: fallback, cwd, tag, totemDir: config.totemDir });
                     // Note: we update `model` to the fallback so telemetry logs the correct one
                     model = fallback;
                 }