@mmnto/cli 0.18.0 → 0.20.0

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (70) hide show
  1. package/LICENSE +190 -21
  2. package/dist/commands/compile.d.ts.map +1 -1
  3. package/dist/commands/compile.js +15 -0
  4. package/dist/commands/compile.js.map +1 -1
  5. package/dist/commands/docs.d.ts +6 -1
  6. package/dist/commands/docs.d.ts.map +1 -1
  7. package/dist/commands/docs.js +58 -5
  8. package/dist/commands/docs.js.map +1 -1
  9. package/dist/commands/docs.test.js +141 -89
  10. package/dist/commands/docs.test.js.map +1 -1
  11. package/dist/commands/extract.d.ts +1 -0
  12. package/dist/commands/extract.d.ts.map +1 -1
  13. package/dist/commands/extract.js +15 -5
  14. package/dist/commands/extract.js.map +1 -1
  15. package/dist/commands/extract.test.js +54 -1
  16. package/dist/commands/extract.test.js.map +1 -1
  17. package/dist/commands/shield.d.ts.map +1 -1
  18. package/dist/commands/shield.js +3 -1
  19. package/dist/commands/shield.js.map +1 -1
  20. package/dist/commands/wrap.js +1 -1
  21. package/dist/commands/wrap.js.map +1 -1
  22. package/dist/commands/wrap.test.js +2 -2
  23. package/dist/commands/wrap.test.js.map +1 -1
  24. package/dist/git.d.ts.map +1 -1
  25. package/dist/git.js +35 -24
  26. package/dist/git.js.map +1 -1
  27. package/dist/index.js +3 -3
  28. package/dist/index.js.map +1 -1
  29. package/dist/orchestrators/anthropic-orchestrator.d.ts +9 -0
  30. package/dist/orchestrators/anthropic-orchestrator.d.ts.map +1 -0
  31. package/dist/orchestrators/anthropic-orchestrator.js +62 -0
  32. package/dist/orchestrators/anthropic-orchestrator.js.map +1 -0
  33. package/dist/orchestrators/anthropic-orchestrator.test.d.ts +2 -0
  34. package/dist/orchestrators/anthropic-orchestrator.test.d.ts.map +1 -0
  35. package/dist/orchestrators/anthropic-orchestrator.test.js +109 -0
  36. package/dist/orchestrators/anthropic-orchestrator.test.js.map +1 -0
  37. package/dist/orchestrators/gemini-orchestrator.d.ts +9 -0
  38. package/dist/orchestrators/gemini-orchestrator.d.ts.map +1 -0
  39. package/dist/orchestrators/gemini-orchestrator.js +58 -0
  40. package/dist/orchestrators/gemini-orchestrator.js.map +1 -0
  41. package/dist/orchestrators/gemini-orchestrator.test.d.ts +2 -0
  42. package/dist/orchestrators/gemini-orchestrator.test.d.ts.map +1 -0
  43. package/dist/orchestrators/gemini-orchestrator.test.js +120 -0
  44. package/dist/orchestrators/gemini-orchestrator.test.js.map +1 -0
  45. package/dist/orchestrators/orchestrator.d.ts +56 -0
  46. package/dist/orchestrators/orchestrator.d.ts.map +1 -0
  47. package/dist/orchestrators/orchestrator.js +96 -0
  48. package/dist/orchestrators/orchestrator.js.map +1 -0
  49. package/dist/orchestrators/orchestrator.test.d.ts.map +1 -0
  50. package/dist/orchestrators/orchestrator.test.js +222 -0
  51. package/dist/orchestrators/orchestrator.test.js.map +1 -0
  52. package/dist/orchestrators/shell-orchestrator.d.ts +16 -0
  53. package/dist/orchestrators/shell-orchestrator.d.ts.map +1 -0
  54. package/dist/orchestrators/shell-orchestrator.js +131 -0
  55. package/dist/orchestrators/shell-orchestrator.js.map +1 -0
  56. package/dist/orchestrators/shell-orchestrator.test.d.ts +2 -0
  57. package/dist/orchestrators/shell-orchestrator.test.d.ts.map +1 -0
  58. package/dist/orchestrators/shell-orchestrator.test.js +267 -0
  59. package/dist/orchestrators/shell-orchestrator.test.js.map +1 -0
  60. package/dist/utils.d.ts +1 -18
  61. package/dist/utils.d.ts.map +1 -1
  62. package/dist/utils.js +31 -146
  63. package/dist/utils.js.map +1 -1
  64. package/dist/utils.test.js +183 -82
  65. package/dist/utils.test.js.map +1 -1
  66. package/package.json +17 -3
  67. package/dist/orchestrator.test.d.ts.map +0 -1
  68. package/dist/orchestrator.test.js +0 -130
  69. package/dist/orchestrator.test.js.map +0 -1
  70. /package/dist/{orchestrator.test.d.ts → orchestrators/orchestrator.test.d.ts} +0 -0
@@ -0,0 +1,131 @@
1
+ import { spawn } from 'node:child_process';
2
+ import * as crypto from 'node:crypto';
3
+ import * as fs from 'node:fs';
4
+ import * as path from 'node:path';
5
+ import { z } from 'zod';
6
+ import { log } from '../ui.js';
7
+ import { isQuotaError } from './orchestrator.js';
8
+ // ─── Constants ───────────────────────────────────────
9
+ const LLM_TIMEOUT_MS = 180_000;
10
+ const LLM_MAX_OUTPUT = 50 * 1024 * 1024; // 50 MB — safety cap on streamed output
11
+ const TEMP_ID_BYTES = 4;
12
+ /** execFileSync on Windows can't resolve executables without `shell: true`. */
13
+ const IS_WIN = process.platform === 'win32';
14
+ // ─── Gemini CLI JSON parsing ─────────────────────────
15
+ const GeminiModelStatsSchema = z.object({
16
+ tokens: z.object({ input: z.number(), candidates: z.number() }).optional(),
17
+ api: z.object({ totalLatencyMs: z.number() }).optional(),
18
+ });
19
+ const GeminiOutputSchema = z.object({
20
+ response: z.string(),
21
+ stats: z.object({ models: z.record(GeminiModelStatsSchema) }),
22
+ });
23
+ /**
24
+ * Try to parse Gemini CLI JSON output. Returns extracted data or null if
25
+ * the output is not valid Gemini JSON (e.g. raw text from a non-Gemini orchestrator).
26
+ */
27
+ export function tryParseGeminiJson(raw) {
28
+ let data;
29
+ try {
30
+ data = JSON.parse(raw);
31
+ }
32
+ catch {
33
+ return null;
34
+ }
35
+ const result = GeminiOutputSchema.safeParse(data);
36
+ if (!result.success)
37
+ return null;
38
+ const allModelStats = Object.values(result.data.stats.models);
39
+ if (allModelStats.length === 0)
40
+ return null;
41
+ const inputTokens = allModelStats.reduce((sum, s) => sum + (s.tokens?.input ?? 0), 0);
42
+ const outputTokens = allModelStats.reduce((sum, s) => sum + (s.tokens?.candidates ?? 0), 0);
43
+ const latencyMs = allModelStats.reduce((sum, s) => sum + (s.api?.totalLatencyMs ?? 0), 0) || null;
44
+ return {
45
+ content: result.data.response,
46
+ inputTokens,
47
+ outputTokens,
48
+ latencyMs,
49
+ };
50
+ }
51
+ export async function invokeShellOrchestrator(opts) {
52
+ const { prompt, command, model, cwd, tag, totemDir } = opts;
53
+ const tmpName = `totem-${tag.toLowerCase()}-${crypto.randomBytes(TEMP_ID_BYTES).toString('hex')}.md`;
54
+ const tempDir = path.join(cwd, totemDir, 'temp');
55
+ fs.mkdirSync(tempDir, { recursive: true });
56
+ const tempPath = path.join(tempDir, tmpName);
57
+ fs.writeFileSync(tempPath, prompt, { encoding: 'utf-8', mode: 0o600 });
58
+ const quotedPath = IS_WIN
59
+ ? `"${tempPath.replace(/"/g, '""')}"`
60
+ : `'${tempPath.replace(/'/g, "'\\''")}'`;
61
+ const resolvedCmd = command.replace(/\{file\}/g, quotedPath).replace(/\{model\}/g, model);
62
+ log.info(tag, 'Invoking orchestrator (this may take 15-60 seconds)...');
63
+ const startMs = Date.now();
64
+ try {
65
+ const raw = await new Promise((resolve, reject) => {
66
+ const child = spawn(resolvedCmd, {
67
+ cwd,
68
+ shell: true,
69
+ stdio: ['ignore', 'pipe', 'pipe'],
70
+ });
71
+ let stdout = '';
72
+ let stderr = '';
73
+ let timedOut = false;
74
+ child.stdout.on('data', (chunk) => {
75
+ if (stdout.length < LLM_MAX_OUTPUT)
76
+ stdout += chunk.toString();
77
+ });
78
+ child.stderr.on('data', (chunk) => {
79
+ if (stderr.length < LLM_MAX_OUTPUT)
80
+ stderr += chunk.toString();
81
+ });
82
+ const timer = setTimeout(() => {
83
+ timedOut = true;
84
+ child.kill();
85
+ }, LLM_TIMEOUT_MS);
86
+ child.on('error', (err) => {
87
+ clearTimeout(timer);
88
+ reject(new Error(`[Totem Error] Shell orchestrator command failed: ${err.message}\n${stderr}`));
89
+ });
90
+ child.on('close', (code) => {
91
+ clearTimeout(timer);
92
+ if (timedOut) {
93
+ reject(Object.assign(new Error(`[Totem Error] Orchestrator timed out after ${LLM_TIMEOUT_MS / 1000}s.\n${stderr}`), { code: 'ETIMEDOUT' }));
94
+ return;
95
+ }
96
+ if (code !== 0) {
97
+ const fullError = new Error(`Process exited with code ${code}\n${stderr}`);
98
+ if (isQuotaError(fullError)) {
99
+ fullError.name = 'QuotaError';
100
+ reject(fullError);
101
+ }
102
+ else {
103
+ reject(new Error(`[Totem Error] Shell orchestrator command failed: ${fullError.message}`));
104
+ }
105
+ return;
106
+ }
107
+ resolve(stdout);
108
+ });
109
+ });
110
+ const wallMs = Date.now() - startMs;
111
+ const gemini = tryParseGeminiJson(raw);
112
+ if (gemini) {
113
+ return {
114
+ content: gemini.content,
115
+ inputTokens: gemini.inputTokens,
116
+ outputTokens: gemini.outputTokens,
117
+ durationMs: gemini.latencyMs ?? wallMs,
118
+ };
119
+ }
120
+ return { content: raw.trim(), inputTokens: null, outputTokens: null, durationMs: wallMs };
121
+ }
122
+ finally {
123
+ try {
124
+ fs.unlinkSync(tempPath);
125
+ }
126
+ catch {
127
+ // Temp cleanup is best-effort
128
+ }
129
+ }
130
+ }
131
+ //# sourceMappingURL=shell-orchestrator.js.map
@@ -0,0 +1 @@
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"}
@@ -0,0 +1,2 @@
1
+ export {};
2
+ //# sourceMappingURL=shell-orchestrator.test.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"shell-orchestrator.test.d.ts","sourceRoot":"","sources":["../../src/orchestrators/shell-orchestrator.test.ts"],"names":[],"mappings":""}
@@ -0,0 +1,267 @@
1
+ import { EventEmitter } from 'node:events';
2
+ import * as fs from 'node:fs';
3
+ import * as os from 'node:os';
4
+ import * as path from 'node:path';
5
+ import { afterEach, beforeEach, describe, expect, it, vi } from 'vitest';
6
+ import { invokeShellOrchestrator, tryParseGeminiJson } from './shell-orchestrator.js';
7
+ function createMockChild() {
8
+ const child = new EventEmitter();
9
+ child.stdout = new EventEmitter();
10
+ child.stderr = new EventEmitter();
11
+ child.kill = vi.fn();
12
+ return child;
13
+ }
14
+ let mockChild;
15
+ vi.mock('node:child_process', () => ({
16
+ spawn: vi.fn(() => mockChild),
17
+ }));
18
+ const { spawn } = await import('node:child_process');
19
+ const mockedSpawn = vi.mocked(spawn);
20
+ // ─── invokeShellOrchestrator ─────────────────────────
21
+ describe('invokeShellOrchestrator', () => {
22
+ let tmpDir;
23
+ const totemDir = '.totem';
24
+ beforeEach(() => {
25
+ tmpDir = fs.mkdtempSync(path.join(os.tmpdir(), 'totem-orch-'));
26
+ vi.spyOn(console, 'error').mockImplementation(() => { });
27
+ mockChild = createMockChild();
28
+ mockedSpawn.mockClear();
29
+ });
30
+ afterEach(() => {
31
+ fs.rmSync(tmpDir, { recursive: true, force: true });
32
+ vi.restoreAllMocks();
33
+ });
34
+ /** Emit stdout data and close with success */
35
+ function emitSuccess(data) {
36
+ process.nextTick(() => {
37
+ mockChild.stdout.emit('data', Buffer.from(data));
38
+ mockChild.emit('close', 0);
39
+ });
40
+ }
41
+ /** Emit close with non-zero exit code and optional stderr */
42
+ function emitFailure(code, stderr = '') {
43
+ process.nextTick(() => {
44
+ if (stderr)
45
+ mockChild.stderr.emit('data', Buffer.from(stderr));
46
+ mockChild.emit('close', code);
47
+ });
48
+ }
49
+ it('returns raw content when output is not Gemini JSON', async () => {
50
+ emitSuccess(' The answer is 42. ');
51
+ const result = await invokeShellOrchestrator({
52
+ prompt: 'prompt text',
53
+ command: 'echo {file}',
54
+ model: 'test-model',
55
+ cwd: tmpDir,
56
+ tag: 'Test',
57
+ totemDir,
58
+ });
59
+ expect(result.content).toBe('The answer is 42.');
60
+ expect(result.inputTokens).toBeNull();
61
+ expect(result.outputTokens).toBeNull();
62
+ expect(result.durationMs).toBeGreaterThanOrEqual(0);
63
+ });
64
+ it('parses Gemini JSON output and returns structured result', async () => {
65
+ const geminiOutput = JSON.stringify({
66
+ response: 'Gemini says hello.',
67
+ stats: {
68
+ models: {
69
+ 'gemini-2.5-pro': {
70
+ tokens: { input: 100, candidates: 50 },
71
+ api: { totalLatencyMs: 2000 },
72
+ },
73
+ },
74
+ },
75
+ });
76
+ emitSuccess(geminiOutput);
77
+ const result = await invokeShellOrchestrator({
78
+ prompt: 'prompt',
79
+ command: 'gemini -e none < {file}',
80
+ model: 'gemini-2.5-pro',
81
+ cwd: tmpDir,
82
+ tag: 'Test',
83
+ totemDir,
84
+ });
85
+ expect(result.content).toBe('Gemini says hello.');
86
+ expect(result.inputTokens).toBe(100);
87
+ expect(result.outputTokens).toBe(50);
88
+ expect(result.durationMs).toBe(2000);
89
+ });
90
+ it('substitutes {file} and {model} in command', async () => {
91
+ emitSuccess('ok');
92
+ await invokeShellOrchestrator({
93
+ prompt: 'prompt',
94
+ command: 'llm --model {model} < {file}',
95
+ model: 'my-model',
96
+ cwd: tmpDir,
97
+ tag: 'Test',
98
+ totemDir,
99
+ });
100
+ const cmd = mockedSpawn.mock.calls[0][0];
101
+ expect(cmd).toContain('my-model');
102
+ expect(cmd).not.toContain('{model}');
103
+ expect(cmd).not.toContain('{file}');
104
+ });
105
+ it('writes prompt to temp file and cleans up after', async () => {
106
+ emitSuccess('result');
107
+ await invokeShellOrchestrator({
108
+ prompt: 'my prompt content',
109
+ command: 'cat {file}',
110
+ model: 'model',
111
+ cwd: tmpDir,
112
+ tag: 'Test',
113
+ totemDir,
114
+ });
115
+ // Temp file should be cleaned up
116
+ const tempDir = path.join(tmpDir, totemDir, 'temp');
117
+ if (fs.existsSync(tempDir)) {
118
+ const files = fs.readdirSync(tempDir).filter((f) => f.startsWith('totem-test-'));
119
+ expect(files).toHaveLength(0);
120
+ }
121
+ });
122
+ it('throws QuotaError for quota-related failures', async () => {
123
+ emitFailure(1, '429 Too Many Requests quota exceeded');
124
+ try {
125
+ await invokeShellOrchestrator({
126
+ prompt: 'prompt',
127
+ command: 'cmd',
128
+ model: 'model',
129
+ cwd: tmpDir,
130
+ tag: 'Test',
131
+ totemDir,
132
+ });
133
+ expect.fail('Should have thrown');
134
+ }
135
+ catch (err) {
136
+ expect(err.name).toBe('QuotaError');
137
+ }
138
+ });
139
+ it('throws generic error for non-zero exit code', async () => {
140
+ emitFailure(1, 'something went wrong');
141
+ await expect(invokeShellOrchestrator({
142
+ prompt: 'prompt',
143
+ command: 'cmd',
144
+ model: 'model',
145
+ cwd: tmpDir,
146
+ tag: 'Test',
147
+ totemDir,
148
+ })).rejects.toThrow('[Totem Error] Shell orchestrator command failed');
149
+ });
150
+ it('throws error on spawn error event', async () => {
151
+ process.nextTick(() => {
152
+ mockChild.emit('error', new Error('command not found'));
153
+ });
154
+ await expect(invokeShellOrchestrator({
155
+ prompt: 'prompt',
156
+ command: 'cmd',
157
+ model: 'model',
158
+ cwd: tmpDir,
159
+ tag: 'Test',
160
+ totemDir,
161
+ })).rejects.toThrow('command not found');
162
+ });
163
+ it('throws descriptive error for timeout', async () => {
164
+ vi.useFakeTimers();
165
+ // Simulate kill → close (rejection now happens in the close handler)
166
+ mockChild.kill = vi.fn(() => {
167
+ process.nextTick(() => mockChild.emit('close', null));
168
+ });
169
+ // Capture the rejection before advancing timers
170
+ const promise = invokeShellOrchestrator({
171
+ prompt: 'prompt',
172
+ command: 'cmd',
173
+ model: 'model',
174
+ cwd: tmpDir,
175
+ tag: 'Test',
176
+ totemDir,
177
+ }).catch((err) => err);
178
+ await vi.advanceTimersByTimeAsync(180_001);
179
+ const err = await promise;
180
+ expect(err).toBeInstanceOf(Error);
181
+ expect(err.message).toContain('timed out after 180s');
182
+ expect(mockChild.kill).toHaveBeenCalled();
183
+ vi.useRealTimers();
184
+ });
185
+ });
186
+ // ─── tryParseGeminiJson ──────────────────────────────
187
+ describe('tryParseGeminiJson', () => {
188
+ it('returns null for non-JSON input', () => {
189
+ expect(tryParseGeminiJson('plain text output')).toBeNull();
190
+ });
191
+ it('returns null for JSON that does not match Gemini schema', () => {
192
+ expect(tryParseGeminiJson('{"foo": "bar"}')).toBeNull();
193
+ });
194
+ it('returns null when stats.models is empty', () => {
195
+ const input = JSON.stringify({
196
+ response: 'hello',
197
+ stats: { models: {} },
198
+ });
199
+ expect(tryParseGeminiJson(input)).toBeNull();
200
+ });
201
+ it('parses valid Gemini output with token stats', () => {
202
+ const input = JSON.stringify({
203
+ response: 'The answer is 42.',
204
+ stats: {
205
+ models: {
206
+ 'gemini-2.5-pro': {
207
+ tokens: { input: 500, candidates: 200 },
208
+ api: { totalLatencyMs: 3000 },
209
+ },
210
+ },
211
+ },
212
+ });
213
+ const result = tryParseGeminiJson(input);
214
+ expect(result).toEqual({
215
+ content: 'The answer is 42.',
216
+ inputTokens: 500,
217
+ outputTokens: 200,
218
+ latencyMs: 3000,
219
+ });
220
+ });
221
+ it('aggregates stats across multiple models', () => {
222
+ const input = JSON.stringify({
223
+ response: 'multi-model',
224
+ stats: {
225
+ models: {
226
+ 'model-a': { tokens: { input: 100, candidates: 50 }, api: { totalLatencyMs: 1000 } },
227
+ 'model-b': { tokens: { input: 200, candidates: 75 }, api: { totalLatencyMs: 2000 } },
228
+ },
229
+ },
230
+ });
231
+ const result = tryParseGeminiJson(input);
232
+ expect(result).toEqual({
233
+ content: 'multi-model',
234
+ inputTokens: 300,
235
+ outputTokens: 125,
236
+ latencyMs: 3000,
237
+ });
238
+ });
239
+ it('returns latencyMs null when api stats are missing', () => {
240
+ const input = JSON.stringify({
241
+ response: 'no api stats',
242
+ stats: {
243
+ models: {
244
+ 'gemini-flash': { tokens: { input: 10, candidates: 5 } },
245
+ },
246
+ },
247
+ });
248
+ const result = tryParseGeminiJson(input);
249
+ expect(result).not.toBeNull();
250
+ expect(result.latencyMs).toBeNull();
251
+ });
252
+ it('defaults missing token counts to zero', () => {
253
+ const input = JSON.stringify({
254
+ response: 'no tokens key',
255
+ stats: {
256
+ models: {
257
+ 'gemini-flash': { api: { totalLatencyMs: 500 } },
258
+ },
259
+ },
260
+ });
261
+ const result = tryParseGeminiJson(input);
262
+ expect(result).not.toBeNull();
263
+ expect(result.inputTokens).toBe(0);
264
+ expect(result.outputTokens).toBe(0);
265
+ });
266
+ });
267
+ //# sourceMappingURL=shell-orchestrator.test.js.map
@@ -0,0 +1 @@
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 CHANGED
@@ -16,23 +16,6 @@ export declare function loadConfig(configPath: string): Promise<TotemConfig>;
16
16
  */
17
17
  export declare function resolveConfigPath(cwd: string): string;
18
18
  export { requireEmbedding } from '@mmnto/totem';
19
- export interface OrchestratorResult {
20
- content: string;
21
- inputTokens: number | null;
22
- outputTokens: number | null;
23
- durationMs: number;
24
- }
25
- /**
26
- * Try to parse Gemini CLI JSON output. Returns extracted data or null if
27
- * the output is not valid Gemini JSON (e.g. raw text from a non-Gemini orchestrator).
28
- */
29
- export declare function tryParseGeminiJson(raw: string): {
30
- content: string;
31
- inputTokens: number;
32
- outputTokens: number;
33
- latencyMs: number | null;
34
- } | null;
35
- export declare function invokeShellOrchestrator(prompt: string, command: string, model: string, cwd: string, tag: string, totemDir: string): Promise<OrchestratorResult>;
36
19
  /**
37
20
  * Reap orphaned temp files older than `maxAgeMs` from `.totem/temp/`.
38
21
  * Fire-and-forget — never blocks the CLI critical path.
@@ -55,7 +38,7 @@ export interface OrchestratorRunOptions {
55
38
  }
56
39
  /**
57
40
  * Validate orchestrator config, then either output raw context (--raw) or
58
- * invoke the shell orchestrator and return the LLM content.
41
+ * invoke the configured orchestrator provider and return the LLM content.
59
42
  *
60
43
  * Returns `undefined` in --raw mode (prompt already written to output).
61
44
  * Returns the LLM response content string otherwise.
@@ -1 +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,wBAAsB,uBAAuB,CAC3C,MAAM,EAAE,MAAM,EACd,OAAO,EAAE,MAAM,EACf,KAAK,EAAE,MAAM,EACb,GAAG,EAAE,MAAM,EACX,GAAG,EAAE,MAAM,EACX,QAAQ,EAAE,MAAM,GACf,OAAO,CAAC,kBAAkB,CAAC,CAuG7B;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,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,CA6K9B"}
1
+ {"version":3,"file":"utils.d.ts","sourceRoot":"","sources":["../src/utils.ts"],"names":[],"mappings":"AAIA,OAAO,KAAK,EAAE,YAAY,EAAE,WAAW,EAAE,MAAM,cAAc,CAAC;AAW9D,+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,CAsK9B"}