@mastra/code-sdk 1.15.0-alpha.8 → 1.15.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.
- package/dist/acp/agent.js +2 -1
- package/dist/acp/agent.js.map +1 -1
- package/dist/acp/event-mapper.js +0 -2
- package/dist/acp/event-mapper.js.map +1 -1
- package/dist/acp.js +1 -2
- package/dist/agent-connections/registry.js.map +1 -1
- package/dist/agent-connections/signal-provider.js.map +1 -1
- package/dist/agent-connections/state-processor.js.map +1 -1
- package/dist/agent-connections/tools.js +26 -24
- package/dist/agent-connections/tools.js.map +1 -1
- package/dist/agents/background-completion.d.ts +15 -2
- package/dist/agents/background-completion.d.ts.map +1 -1
- package/dist/agents/background-completion.js +19 -2
- package/dist/agents/background-completion.js.map +1 -1
- package/dist/agents/instructions.js +3 -2
- package/dist/agents/instructions.js.map +1 -1
- package/dist/agents/mastracode-gateway.js +35 -27
- package/dist/agents/mastracode-gateway.js.map +1 -1
- package/dist/agents/memory.d.ts.map +1 -1
- package/dist/agents/memory.js +17 -3
- package/dist/agents/memory.js.map +1 -1
- package/dist/agents/prompts/agent-instructions.js +1 -1
- package/dist/agents/prompts/agent-instructions.js.map +1 -1
- package/dist/agents/prompts/index.js +3 -2
- package/dist/agents/prompts/index.js.map +1 -1
- package/dist/agents/prompts/plan.js +2 -1
- package/dist/agents/prompts/plan.js.map +1 -1
- package/dist/agents/sandbox-filesystem.d.ts +28 -0
- package/dist/agents/sandbox-filesystem.d.ts.map +1 -1
- package/dist/agents/sandbox-filesystem.js +81 -2
- package/dist/agents/sandbox-filesystem.js.map +1 -1
- package/dist/agents/tools.d.ts.map +1 -1
- package/dist/agents/tools.js +18 -12
- package/dist/agents/tools.js.map +1 -1
- package/dist/agents/workspace.js +2 -1
- package/dist/agents/workspace.js.map +1 -1
- package/dist/auth/account-rotation-processor.js +2 -1
- package/dist/auth/account-rotation-processor.js.map +1 -1
- package/dist/auth/providers/anthropic.js +2 -2
- package/dist/auth/providers/anthropic.js.map +1 -1
- package/dist/auth/providers/github-copilot.js +1 -1
- package/dist/auth/providers/github-copilot.js.map +1 -1
- package/dist/auth/providers/openai-codex.js +3 -2
- package/dist/auth/providers/openai-codex.js.map +1 -1
- package/dist/auth/providers/xai.js +1 -1
- package/dist/auth/providers/xai.js.map +1 -1
- package/dist/clipboard/index.js +5 -4
- package/dist/clipboard/index.js.map +1 -1
- package/dist/evals/context-builder.js +1 -3
- package/dist/evals/context-builder.js.map +1 -1
- package/dist/evals/scorers/outcome.js +2 -1
- package/dist/evals/scorers/outcome.js.map +1 -1
- package/dist/factory/remote-controller.js.map +1 -1
- package/dist/factory/remote-session.d.ts +1 -1
- package/dist/factory/remote-session.d.ts.map +1 -1
- package/dist/factory/remote-session.js +15 -13
- package/dist/factory/remote-session.js.map +1 -1
- package/dist/headless/cli.js +2 -1
- package/dist/headless/cli.js.map +1 -1
- package/dist/headless/run-mc.js +9 -9
- package/dist/headless/run-mc.js.map +1 -1
- package/dist/hooks/executor.js +3 -1
- package/dist/hooks/executor.js.map +1 -1
- package/dist/hooks/manager.js +40 -26
- package/dist/hooks/manager.js.map +1 -1
- package/dist/index.d.ts.map +1 -1
- package/dist/index.js +38 -21
- package/dist/index.js.map +1 -1
- package/dist/knowledge-inspector.js +2 -2
- package/dist/knowledge-inspector.js.map +1 -1
- package/dist/lsp/server.js +4 -2
- package/dist/lsp/server.js.map +1 -1
- package/dist/mcp/manager.d.ts.map +1 -1
- package/dist/mcp/manager.js +14 -2
- package/dist/mcp/manager.js.map +1 -1
- package/dist/onboarding/settings.js +5 -2
- package/dist/onboarding/settings.js.map +1 -1
- package/dist/platform/index.js +5 -3
- package/dist/platform/index.js.map +1 -1
- package/dist/plugins/loader.js +3 -1
- package/dist/plugins/loader.js.map +1 -1
- package/dist/plugins/manager.js +2 -1
- package/dist/plugins/manager.js.map +1 -1
- package/dist/plugins/package-link.js.map +1 -1
- package/dist/plugins/scaffold.js +4 -2
- package/dist/plugins/scaffold.js.map +1 -1
- package/dist/plugins/signal-lane.js.map +1 -1
- package/dist/process-memory-diagnostics.js +2 -1
- package/dist/process-memory-diagnostics.js.map +1 -1
- package/dist/providers/amazon-bedrock-gateway.js +2 -1
- package/dist/providers/amazon-bedrock-gateway.js.map +1 -1
- package/dist/providers/amazon-bedrock.js +2 -2
- package/dist/providers/amazon-bedrock.js.map +1 -1
- package/dist/providers/claude-max.js +13 -9
- package/dist/providers/claude-max.js.map +1 -1
- package/dist/providers/github-copilot-gateway.js.map +1 -1
- package/dist/providers/github-copilot.js +13 -12
- package/dist/providers/github-copilot.js.map +1 -1
- package/dist/providers/openai-codex.js +14 -11
- package/dist/providers/openai-codex.js.map +1 -1
- package/dist/providers/provider-thinking.js.map +1 -1
- package/dist/providers/thinking-request.js +1 -1
- package/dist/providers/thinking-request.js.map +1 -1
- package/dist/schedules/run-script.js +1 -1
- package/dist/schedules/run-script.js.map +1 -1
- package/dist/schedules/scheduler.js.map +1 -1
- package/dist/tools/request-sandbox-access.js +7 -5
- package/dist/tools/request-sandbox-access.js.map +1 -1
- package/dist/tools/utils.js +3 -1
- package/dist/tools/utils.js.map +1 -1
- package/dist/tools/web-search.d.ts +13 -11
- package/dist/tools/web-search.d.ts.map +1 -1
- package/dist/tools/web-search.js +21 -15
- package/dist/tools/web-search.js.map +1 -1
- package/dist/utils/debug-log.js +2 -2
- package/dist/utils/debug-log.js.map +1 -1
- package/dist/utils/gateway-sync.js +1 -1
- package/dist/utils/gateway-sync.js.map +1 -1
- package/dist/utils/local-observability.d.ts +1 -1
- package/dist/utils/local-observability.d.ts.map +1 -1
- package/dist/utils/local-observability.js +1 -1
- package/dist/utils/local-observability.js.map +1 -1
- package/dist/utils/prune-cli.js +10 -8
- package/dist/utils/prune-cli.js.map +1 -1
- package/dist/utils/signals-pubsub.js.map +1 -1
- package/dist/utils/slash-command-processor.js +1 -1
- package/dist/utils/slash-command-processor.js.map +1 -1
- package/dist/utils/stdin-pipe.js +1 -1
- package/dist/utils/stdin-pipe.js.map +1 -1
- package/dist/utils/storage-factory.d.ts.map +1 -1
- package/dist/utils/storage-factory.js +1 -1
- package/dist/utils/storage-factory.js.map +1 -1
- package/dist/utils/storage-maintenance.js +1 -1
- package/dist/utils/storage-maintenance.js.map +1 -1
- package/dist/utils/update-check.js +2 -1
- package/dist/utils/update-check.js.map +1 -1
- package/dist/utils/wake-session.d.ts +1 -1
- package/dist/utils/wake-session.d.ts.map +1 -1
- package/dist/utils/wake-session.js +8 -1
- package/dist/utils/wake-session.js.map +1 -1
- package/dist/workflows/register-primitives.js +1 -1
- package/dist/workflows/register-primitives.js.map +1 -1
- package/package.json +19 -19
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"outcome.js","names":[],"sources":["../../../src/evals/scorers/outcome.ts"],"sourcesContent":["/**\n * MastraCode Outcome Scorer (Code-based, Agent type, always-on)\n *\n * Grades the objective outcome of a coding session by checking hard facts:\n * - Did the build/typecheck pass?\n * - Did tests pass?\n * - What was the tool error rate?\n * - Did the agent get stuck in a loop?\n * - Was there a regression (build passed earlier, failed later)?\n * - Did the agent use ask_user unnecessarily? (autonomy)\n *\n * This scorer is always-on (no sampling) because it's cheap — it only\n * reads exit codes and error states from tool results already in the messages.\n *\n * Scoring philosophy (from Anthropic's eval guidance):\n * \"Grade outcomes, not paths\" — we check what happened, not how.\n * Partial credit: a session that passes build but fails tests scores\n * higher than one that fails build entirely.\n *\n * Dimensions that don't apply to a session (e.g. build for a docs-only task)\n * are excluded from the weighted average, so sessions aren't penalized for\n * irrelevant criteria.\n */\n\nimport type { MastraDBMessage } from '@mastra/core/agent';\nimport { createScorer, filterRun } from '@mastra/core/evals';\n\nimport { isBuildCommand, isTestCommand, getExitCode } from './classify-command.js';\nimport type { ExtractedToolCall } from './extract-tools.js';\nimport { extractToolCalls } from './extract-tools.js';\n\n// ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\n// CONFIGURATION — Adjust weights and thresholds here\n// ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\n\n/**\n * Dimension weights. Used to compute the weighted average of applicable dimensions.\n * Dimensions that don't apply (e.g. no build ran) are excluded from the average.\n */\nconst WEIGHTS = {\n build: 0.3,\n tests: 0.25,\n toolErrors: 0.2,\n loops: 0.1,\n regression: 0.1,\n autonomy: 0.05,\n} as const;\n\n/**\n * Scoring thresholds and multipliers.\n */\nconst THRESHOLDS = {\n /** Multiplier for tool error rate penalty. Score = max(0, 1 - rate × this). */\n toolErrorPenaltyMultiplier: 1.0,\n\n /** How many consecutive identical calls (or same-error repeats) before it's a \"loop\". */\n loopMinRepetitions: 3,\n\n /** Score penalty per detected loop. */\n loopPenaltyPerOccurrence: 0.3,\n\n /** Score deducted per ask_user call. */\n autonomyPenaltyPerAsk: 0.25,\n\n /** Score when a command ran but exit code is ambiguous. */\n ambiguousExitScore: 0.75,\n\n /** Minimum tool calls required to produce a meaningful score. */\n minToolCalls: 1,\n\n /** Tool names whose errors are expected/benign (not penalized). */\n benignErrorTools: ['search_content', 'find_files', 'lsp_inspect'] as string[],\n} as const;\n\n// ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\n\ntype DimensionResult = { score: number; detail: string; applicable: boolean };\n\n/**\n * Check if the primary verb of an execute_command is a build/typecheck command.\n * Splits compound commands (&&, ;) and checks the first segment.\n */\n/** Extract command string from tool call args, then delegate to shared classifier. */\nfunction isBuildArgs(args: Record<string, unknown>): boolean {\n return isBuildCommand(String(args.command ?? ''));\n}\n\nfunction isTestArgs(args: Record<string, unknown>): boolean {\n return isTestCommand(String(args.command ?? ''));\n}\n\n/**\n * Score build/typecheck outcome (final state).\n */\nfunction scoreBuild(execResults: ExtractedToolCall[]): DimensionResult {\n const builds = execResults.filter(r => r.toolName === 'execute_command' && isBuildArgs(r.args));\n if (builds.length === 0) return { score: 0, detail: 'No build/typecheck ran', applicable: false };\n\n const last = builds[builds.length - 1]!;\n if (last.isError) return { score: 0, detail: 'Build/typecheck tool errored', applicable: true };\n\n const exitCode = getExitCode(last.result);\n if (exitCode === 0) return { score: 1, detail: 'Build/typecheck passed', applicable: true };\n if (exitCode !== null && exitCode !== 0)\n return { score: 0, detail: `Build failed (exit ${exitCode})`, applicable: true };\n\n // Fallback: check result text for TS errors\n const text = typeof last.result === 'string' ? last.result : JSON.stringify(last.result ?? '');\n if (/error TS\\d+|Cannot find module|is not assignable/i.test(text)) {\n return { score: 0, detail: 'Build failed (TypeScript errors in output)', applicable: true };\n }\n\n return { score: THRESHOLDS.ambiguousExitScore, detail: 'Build ran, outcome unclear', applicable: true };\n}\n\n/**\n * Score test outcome (final state).\n */\nfunction scoreTests(execResults: ExtractedToolCall[]): DimensionResult {\n const tests = execResults.filter(r => r.toolName === 'execute_command' && isTestArgs(r.args));\n if (tests.length === 0) return { score: 0, detail: 'No tests ran', applicable: false };\n\n const last = tests[tests.length - 1]!;\n if (last.isError) return { score: 0, detail: 'Test command errored', applicable: true };\n\n const exitCode = getExitCode(last.result);\n if (exitCode === 0) return { score: 1, detail: 'Tests passed', applicable: true };\n if (exitCode !== null && exitCode !== 0)\n return { score: 0, detail: `Tests failed (exit ${exitCode})`, applicable: true };\n\n const text = typeof last.result === 'string' ? last.result : JSON.stringify(last.result ?? '');\n if (/\\d+ (?:tests? )?passed|✓|PASS/i.test(text) && !/\\bfail(?:ed|ure)?\\b|\\berror\\b/i.test(text)) {\n return { score: 1, detail: 'Tests passed (inferred)', applicable: true };\n }\n if (/\\bFAIL\\b|failed|✗|✘/i.test(text)) {\n return { score: 0, detail: 'Tests failed (inferred)', applicable: true };\n }\n\n return { score: THRESHOLDS.ambiguousExitScore, detail: 'Tests ran, outcome unclear', applicable: true };\n}\n\n/**\n * Score tool error rate, excluding benign errors from expected-to-fail tools.\n */\nfunction scoreToolErrors(results: ExtractedToolCall[]): DimensionResult {\n if (results.length === 0) return { score: 1, detail: 'No tool calls', applicable: false };\n\n // Only count errors from non-benign tools\n const nonBenign = results.filter(r => !THRESHOLDS.benignErrorTools.includes(r.toolName));\n if (nonBenign.length === 0) return { score: 1, detail: 'All tool calls are benign-error tools', applicable: true };\n\n const errors = nonBenign.filter(r => r.isError);\n const rate = errors.length / nonBenign.length;\n\n if (rate === 0) return { score: 1, detail: 'No tool errors', applicable: true };\n\n const score = Math.max(0, 1 - rate * THRESHOLDS.toolErrorPenaltyMultiplier);\n return {\n score,\n detail: `${errors.length}/${nonBenign.length} tools errored (${(rate * 100).toFixed(0)}%)`,\n applicable: true,\n };\n}\n\n/**\n * Detect stuck loops: same tool + same args appearing 3+ times consecutively,\n * or same tool + same args erroring 3+ times total.\n */\nfunction scoreStuckLoops(results: ExtractedToolCall[]): DimensionResult {\n if (results.length < 3) return { score: 1, detail: 'Too few calls for loop detection', applicable: true };\n\n const fingerprint = (r: ExtractedToolCall) => `${r.toolName}:${JSON.stringify(r.args)}`;\n\n // Check consecutive identical calls\n let maxConsecutive = 1;\n let currentRun = 1;\n for (let i = 1; i < results.length; i++) {\n if (fingerprint(results[i]!) === fingerprint(results[i - 1]!)) {\n currentRun++;\n maxConsecutive = Math.max(maxConsecutive, currentRun);\n } else {\n currentRun = 1;\n }\n }\n\n // Check repeated error fingerprints\n const errorCounts = new Map<string, number>();\n for (const r of results) {\n if (!r.isError) continue;\n const fp = fingerprint(r);\n errorCounts.set(fp, (errorCounts.get(fp) ?? 0) + 1);\n }\n const maxErrorRepeat = Math.max(0, ...[...errorCounts.values()]);\n\n const minReps = THRESHOLDS.loopMinRepetitions;\n const loopSeverity = Math.max(\n maxConsecutive >= minReps ? maxConsecutive - (minReps - 1) : 0,\n maxErrorRepeat >= minReps ? maxErrorRepeat - (minReps - 1) : 0,\n );\n\n if (loopSeverity === 0) return { score: 1, detail: 'No stuck loops', applicable: true };\n\n const score = Math.max(0, 1 - loopSeverity * THRESHOLDS.loopPenaltyPerOccurrence);\n const details: string[] = [];\n if (maxConsecutive >= minReps) details.push(`${maxConsecutive} consecutive identical calls`);\n if (maxErrorRepeat >= minReps) details.push(`same error repeated ${maxErrorRepeat}x`);\n return { score, detail: details.join('; '), applicable: true };\n}\n\n/**\n * Detect regressions: build/test passed at some point, then failed later.\n * Gives partial credit if the agent recovered (final state passes).\n */\nfunction scoreRegression(execResults: ExtractedToolCall[]): DimensionResult {\n const regressions: { label: string; recovered: boolean }[] = [];\n\n for (const [label, filter] of [\n ['Build', isBuildArgs],\n ['Tests', isTestArgs],\n ] as const) {\n const cmds = execResults.filter(r => r.toolName === 'execute_command' && filter(r.args));\n if (cmds.length < 2) continue;\n\n let sawPass = false;\n let sawRegression = false;\n for (const cmd of cmds) {\n const exit = getExitCode(cmd.result);\n if (exit === 0) {\n if (sawRegression) {\n // Recovered from regression\n regressions.push({ label, recovered: true });\n sawRegression = false;\n }\n sawPass = true;\n }\n if (sawPass && exit !== null && exit !== 0) {\n sawRegression = true;\n }\n }\n\n if (sawRegression) {\n // Regression persisted at end of session\n regressions.push({ label, recovered: false });\n }\n }\n\n if (regressions.length === 0) return { score: 1, detail: 'No regressions', applicable: true };\n\n // Persistent regressions are severe (0), recovered regressions get partial credit (0.5)\n const persistentCount = regressions.filter(r => !r.recovered).length;\n const recoveredCount = regressions.filter(r => r.recovered).length;\n const score = Math.max(0, 1 - persistentCount * 0.5 - recoveredCount * 0.1);\n const details = regressions.map(\n r => `${r.label} ${r.recovered ? 'regressed then recovered' : 'regressed (persisted)'}`,\n );\n return { score, detail: details.join('; '), applicable: true };\n}\n\n/**\n * Score autonomy: penalize ask_user calls.\n * 0 calls = 1.0, each call reduces by 0.25 (up to a floor of 0).\n */\nfunction scoreAutonomy(results: ExtractedToolCall[]): DimensionResult {\n const askUserCalls = results.filter(r => r.toolName === 'ask_user');\n const count = askUserCalls.length;\n\n if (count === 0) return { score: 1, detail: 'No ask_user calls (fully autonomous)', applicable: true };\n\n const score = Math.max(0, 1 - count * THRESHOLDS.autonomyPenaltyPerAsk);\n return { score, detail: `${count} ask_user call${count > 1 ? 's' : ''}`, applicable: true };\n}\n\n/**\n * Compute weighted average of only applicable dimensions.\n * Returns null if no dimensions are applicable.\n */\nfunction weightedAverage(dimensions: Record<string, DimensionResult>, weights: Record<string, number>): number | null {\n let totalWeight = 0;\n let weightedSum = 0;\n\n for (const [key, result] of Object.entries(dimensions)) {\n if (!result.applicable) continue;\n const weight = weights[key] ?? 0;\n totalWeight += weight;\n weightedSum += result.score * weight;\n }\n\n if (totalWeight === 0) return null;\n return weightedSum / totalWeight;\n}\n\nexport function createOutcomeScorer() {\n return createScorer({\n id: 'mastracode-outcome',\n name: 'MastraCode Outcome',\n description:\n 'Grades coding session outcomes: build/test pass, tool errors, stuck loops, regressions, autonomy. Always-on.',\n type: 'agent',\n prepareRun: filterRun({ partTypes: ['tool-invocation', 'text'] }),\n })\n .preprocess(async ({ run }) => {\n const messages = (run.output ?? []) as MastraDBMessage[];\n const allResults = extractToolCalls(messages);\n const totalCalls = allResults.length;\n\n // H1: Empty sessions or sessions with no tool calls → score 0\n if (totalCalls < THRESHOLDS.minToolCalls) {\n const empty: DimensionResult = { score: 0, detail: 'No tool calls', applicable: false };\n return {\n empty: true as const,\n build: empty,\n tests: empty,\n toolErrors: empty,\n loops: empty,\n regression: empty,\n autonomy: empty,\n totalCalls,\n };\n }\n\n const execResults = allResults.filter(r => r.toolName === 'execute_command');\n\n return {\n empty: false as const,\n build: scoreBuild(execResults),\n tests: scoreTests(execResults),\n toolErrors: scoreToolErrors(allResults),\n loops: scoreStuckLoops(allResults),\n regression: scoreRegression(execResults),\n autonomy: scoreAutonomy(allResults),\n totalCalls,\n };\n })\n .generateScore(({ results }) => {\n const p = results.preprocessStepResult;\n if (p.empty) return 0;\n\n const score = weightedAverage(\n {\n build: p.build,\n tests: p.tests,\n toolErrors: p.toolErrors,\n loops: p.loops,\n regression: p.regression,\n autonomy: p.autonomy,\n },\n WEIGHTS,\n );\n\n return score === null ? 0 : Math.round(score * 100) / 100;\n })\n .generateReason(({ results, score }) => {\n const p = results.preprocessStepResult;\n if (p.empty) {\n return `Score: 0 (${p.totalCalls} tool calls — below minimum threshold for scoring)`;\n }\n\n const pct = (w: number) => `${(w * 100).toFixed(0)}%`;\n const dimLine = (name: string, weight: number, r: DimensionResult) =>\n r.applicable\n ? `${name} (${pct(weight)}): ${r.detail} [${r.score}]`\n : `${name}: ${r.detail} [N/A — excluded from average]`;\n\n const parts: string[] = [`Score: ${score} (${p.totalCalls} tool calls total)`];\n parts.push(dimLine('Build', WEIGHTS.build, p.build));\n parts.push(dimLine('Tests', WEIGHTS.tests, p.tests));\n parts.push(dimLine('Tool errors', WEIGHTS.toolErrors, p.toolErrors));\n parts.push(dimLine('Loops', WEIGHTS.loops, p.loops));\n parts.push(dimLine('Regression', WEIGHTS.regression, p.regression));\n parts.push(dimLine('Autonomy', WEIGHTS.autonomy, p.autonomy));\n\n return parts.join('\\n');\n });\n}\n"],"mappings":";;;;;;;;AAuCA,MAAM,UAAU;CACd,OAAO;CACP,OAAO;CACP,YAAY;CACZ,OAAO;CACP,YAAY;CACZ,UAAU;AACZ;;;;AAKA,MAAM,aAAa;;CAEjB,4BAA4B;;CAG5B,oBAAoB;;CAGpB,0BAA0B;;CAG1B,uBAAuB;;CAGvB,oBAAoB;;CAGpB,cAAc;;CAGd,kBAAkB;EAAC;EAAkB;EAAc;CAAa;AAClE;;;;;;AAWA,SAAS,YAAY,MAAwC;CAC3D,OAAO,eAAe,OAAO,KAAK,WAAW,EAAE,CAAC;AAClD;AAEA,SAAS,WAAW,MAAwC;CAC1D,OAAO,cAAc,OAAO,KAAK,WAAW,EAAE,CAAC;AACjD;;;;AAKA,SAAS,WAAW,aAAmD;CACrE,MAAM,SAAS,YAAY,QAAO,MAAK,EAAE,aAAa,qBAAqB,YAAY,EAAE,IAAI,CAAC;CAC9F,IAAI,OAAO,WAAW,GAAG,OAAO;EAAE,OAAO;EAAG,QAAQ;EAA0B,YAAY;CAAM;CAEhG,MAAM,OAAO,OAAO,OAAO,SAAS;CACpC,IAAI,KAAK,SAAS,OAAO;EAAE,OAAO;EAAG,QAAQ;EAAgC,YAAY;CAAK;CAE9F,MAAM,WAAW,YAAY,KAAK,MAAM;CACxC,IAAI,aAAa,GAAG,OAAO;EAAE,OAAO;EAAG,QAAQ;EAA0B,YAAY;CAAK;CAC1F,IAAI,aAAa,QAAQ,aAAa,GACpC,OAAO;EAAE,OAAO;EAAG,QAAQ,sBAAsB,SAAS;EAAI,YAAY;CAAK;CAGjF,MAAM,OAAO,OAAO,KAAK,WAAW,WAAW,KAAK,SAAS,KAAK,UAAU,KAAK,UAAU,EAAE;CAC7F,IAAI,oDAAoD,KAAK,IAAI,GAC/D,OAAO;EAAE,OAAO;EAAG,QAAQ;EAA8C,YAAY;CAAK;CAG5F,OAAO;EAAE,OAAO,WAAW;EAAoB,QAAQ;EAA8B,YAAY;CAAK;AACxG;;;;AAKA,SAAS,WAAW,aAAmD;CACrE,MAAM,QAAQ,YAAY,QAAO,MAAK,EAAE,aAAa,qBAAqB,WAAW,EAAE,IAAI,CAAC;CAC5F,IAAI,MAAM,WAAW,GAAG,OAAO;EAAE,OAAO;EAAG,QAAQ;EAAgB,YAAY;CAAM;CAErF,MAAM,OAAO,MAAM,MAAM,SAAS;CAClC,IAAI,KAAK,SAAS,OAAO;EAAE,OAAO;EAAG,QAAQ;EAAwB,YAAY;CAAK;CAEtF,MAAM,WAAW,YAAY,KAAK,MAAM;CACxC,IAAI,aAAa,GAAG,OAAO;EAAE,OAAO;EAAG,QAAQ;EAAgB,YAAY;CAAK;CAChF,IAAI,aAAa,QAAQ,aAAa,GACpC,OAAO;EAAE,OAAO;EAAG,QAAQ,sBAAsB,SAAS;EAAI,YAAY;CAAK;CAEjF,MAAM,OAAO,OAAO,KAAK,WAAW,WAAW,KAAK,SAAS,KAAK,UAAU,KAAK,UAAU,EAAE;CAC7F,IAAI,iCAAiC,KAAK,IAAI,KAAK,CAAC,iCAAiC,KAAK,IAAI,GAC5F,OAAO;EAAE,OAAO;EAAG,QAAQ;EAA2B,YAAY;CAAK;CAEzE,IAAI,uBAAuB,KAAK,IAAI,GAClC,OAAO;EAAE,OAAO;EAAG,QAAQ;EAA2B,YAAY;CAAK;CAGzE,OAAO;EAAE,OAAO,WAAW;EAAoB,QAAQ;EAA8B,YAAY;CAAK;AACxG;;;;AAKA,SAAS,gBAAgB,SAA+C;CACtE,IAAI,QAAQ,WAAW,GAAG,OAAO;EAAE,OAAO;EAAG,QAAQ;EAAiB,YAAY;CAAM;CAGxF,MAAM,YAAY,QAAQ,QAAO,MAAK,CAAC,WAAW,iBAAiB,SAAS,EAAE,QAAQ,CAAC;CACvF,IAAI,UAAU,WAAW,GAAG,OAAO;EAAE,OAAO;EAAG,QAAQ;EAAyC,YAAY;CAAK;CAEjH,MAAM,SAAS,UAAU,QAAO,MAAK,EAAE,OAAO;CAC9C,MAAM,OAAO,OAAO,SAAS,UAAU;CAEvC,IAAI,SAAS,GAAG,OAAO;EAAE,OAAO;EAAG,QAAQ;EAAkB,YAAY;CAAK;CAG9E,OAAO;EACL,OAFY,KAAK,IAAI,GAAG,IAAI,OAAO,WAAW,0BAE1C;EACJ,QAAQ,GAAG,OAAO,OAAO,GAAG,UAAU,OAAO,mBAAmB,OAAO,IAAA,CAAK,QAAQ,CAAC,EAAE;EACvF,YAAY;CACd;AACF;;;;;AAMA,SAAS,gBAAgB,SAA+C;CACtE,IAAI,QAAQ,SAAS,GAAG,OAAO;EAAE,OAAO;EAAG,QAAQ;EAAoC,YAAY;CAAK;CAExG,MAAM,eAAe,MAAyB,GAAG,EAAE,SAAS,GAAG,KAAK,UAAU,EAAE,IAAI;CAGpF,IAAI,iBAAiB;CACrB,IAAI,aAAa;CACjB,KAAK,IAAI,IAAI,GAAG,IAAI,QAAQ,QAAQ,KAClC,IAAI,YAAY,QAAQ,EAAG,MAAM,YAAY,QAAQ,IAAI,EAAG,GAAG;EAC7D;EACA,iBAAiB,KAAK,IAAI,gBAAgB,UAAU;CACtD,OACE,aAAa;CAKjB,MAAM,8BAAc,IAAI,IAAoB;CAC5C,KAAK,MAAM,KAAK,SAAS;EACvB,IAAI,CAAC,EAAE,SAAS;EAChB,MAAM,KAAK,YAAY,CAAC;EACxB,YAAY,IAAI,KAAK,YAAY,IAAI,EAAE,KAAK,KAAK,CAAC;CACpD;CACA,MAAM,iBAAiB,KAAK,IAAI,GAAG,GAAG,CAAC,GAAG,YAAY,OAAO,CAAC,CAAC;CAE/D,MAAM,UAAU,WAAW;CAC3B,MAAM,eAAe,KAAK,IACxB,kBAAkB,UAAU,kBAAkB,UAAU,KAAK,GAC7D,kBAAkB,UAAU,kBAAkB,UAAU,KAAK,CAC/D;CAEA,IAAI,iBAAiB,GAAG,OAAO;EAAE,OAAO;EAAG,QAAQ;EAAkB,YAAY;CAAK;CAEtF,MAAM,QAAQ,KAAK,IAAI,GAAG,IAAI,eAAe,WAAW,wBAAwB;CAChF,MAAM,UAAoB,CAAC;CAC3B,IAAI,kBAAkB,SAAS,QAAQ,KAAK,GAAG,eAAe,6BAA6B;CAC3F,IAAI,kBAAkB,SAAS,QAAQ,KAAK,uBAAuB,eAAe,EAAE;CACpF,OAAO;EAAE;EAAO,QAAQ,QAAQ,KAAK,IAAI;EAAG,YAAY;CAAK;AAC/D;;;;;AAMA,SAAS,gBAAgB,aAAmD;CAC1E,MAAM,cAAuD,CAAC;CAE9D,KAAK,MAAM,CAAC,OAAO,WAAW,CAC5B,CAAC,SAAS,WAAW,GACrB,CAAC,SAAS,UAAU,CACtB,GAAY;EACV,MAAM,OAAO,YAAY,QAAO,MAAK,EAAE,aAAa,qBAAqB,OAAO,EAAE,IAAI,CAAC;EACvF,IAAI,KAAK,SAAS,GAAG;EAErB,IAAI,UAAU;EACd,IAAI,gBAAgB;EACpB,KAAK,MAAM,OAAO,MAAM;GACtB,MAAM,OAAO,YAAY,IAAI,MAAM;GACnC,IAAI,SAAS,GAAG;IACd,IAAI,eAAe;KAEjB,YAAY,KAAK;MAAE;MAAO,WAAW;KAAK,CAAC;KAC3C,gBAAgB;IAClB;IACA,UAAU;GACZ;GACA,IAAI,WAAW,SAAS,QAAQ,SAAS,GACvC,gBAAgB;EAEpB;EAEA,IAAI,eAEF,YAAY,KAAK;GAAE;GAAO,WAAW;EAAM,CAAC;CAEhD;CAEA,IAAI,YAAY,WAAW,GAAG,OAAO;EAAE,OAAO;EAAG,QAAQ;EAAkB,YAAY;CAAK;CAG5F,MAAM,kBAAkB,YAAY,QAAO,MAAK,CAAC,EAAE,SAAS,CAAC,CAAC;CAC9D,MAAM,iBAAiB,YAAY,QAAO,MAAK,EAAE,SAAS,CAAC,CAAC;CAK5D,OAAO;EAAE,OAJK,KAAK,IAAI,GAAG,IAAI,kBAAkB,KAAM,iBAAiB,EAI1D;EAAG,QAHA,YAAY,KAC1B,MAAK,GAAG,EAAE,MAAM,GAAG,EAAE,YAAY,6BAA6B,yBAElC,CAAC,CAAC,KAAK,IAAI;EAAG,YAAY;CAAK;AAC/D;;;;;AAMA,SAAS,cAAc,SAA+C;CAEpE,MAAM,QADe,QAAQ,QAAO,MAAK,EAAE,aAAa,UAC/B,CAAC,CAAC;CAE3B,IAAI,UAAU,GAAG,OAAO;EAAE,OAAO;EAAG,QAAQ;EAAwC,YAAY;CAAK;CAGrG,OAAO;EAAE,OADK,KAAK,IAAI,GAAG,IAAI,QAAQ,WAAW,qBACpC;EAAG,QAAQ,GAAG,MAAM,gBAAgB,QAAQ,IAAI,MAAM;EAAM,YAAY;CAAK;AAC5F;;;;;AAMA,SAAS,gBAAgB,YAA6C,SAAgD;CACpH,IAAI,cAAc;CAClB,IAAI,cAAc;CAElB,KAAK,MAAM,CAAC,KAAK,WAAW,OAAO,QAAQ,UAAU,GAAG;EACtD,IAAI,CAAC,OAAO,YAAY;EACxB,MAAM,SAAS,QAAQ,QAAQ;EAC/B,eAAe;EACf,eAAe,OAAO,QAAQ;CAChC;CAEA,IAAI,gBAAgB,GAAG,OAAO;CAC9B,OAAO,cAAc;AACvB;AAEA,SAAgB,sBAAsB;CACpC,OAAO,aAAa;EAClB,IAAI;EACJ,MAAM;EACN,aACE;EACF,MAAM;EACN,YAAY,UAAU,EAAE,WAAW,CAAC,mBAAmB,MAAM,EAAE,CAAC;CAClE,CAAC,CAAC,CACC,WAAW,OAAO,EAAE,UAAU;EAE7B,MAAM,aAAa,iBADD,IAAI,UAAU,CAAC,CACW;EAC5C,MAAM,aAAa,WAAW;EAG9B,IAAI,aAAa,WAAW,cAAc;GACxC,MAAM,QAAyB;IAAE,OAAO;IAAG,QAAQ;IAAiB,YAAY;GAAM;GACtF,OAAO;IACL,OAAO;IACP,OAAO;IACP,OAAO;IACP,YAAY;IACZ,OAAO;IACP,YAAY;IACZ,UAAU;IACV;GACF;EACF;EAEA,MAAM,cAAc,WAAW,QAAO,MAAK,EAAE,aAAa,iBAAiB;EAE3E,OAAO;GACL,OAAO;GACP,OAAO,WAAW,WAAW;GAC7B,OAAO,WAAW,WAAW;GAC7B,YAAY,gBAAgB,UAAU;GACtC,OAAO,gBAAgB,UAAU;GACjC,YAAY,gBAAgB,WAAW;GACvC,UAAU,cAAc,UAAU;GAClC;EACF;CACF,CAAC,CAAC,CACD,eAAe,EAAE,cAAc;EAC9B,MAAM,IAAI,QAAQ;EAClB,IAAI,EAAE,OAAO,OAAO;EAEpB,MAAM,QAAQ,gBACZ;GACE,OAAO,EAAE;GACT,OAAO,EAAE;GACT,YAAY,EAAE;GACd,OAAO,EAAE;GACT,YAAY,EAAE;GACd,UAAU,EAAE;EACd,GACA,OACF;EAEA,OAAO,UAAU,OAAO,IAAI,KAAK,MAAM,QAAQ,GAAG,IAAI;CACxD,CAAC,CAAC,CACD,gBAAgB,EAAE,SAAS,YAAY;EACtC,MAAM,IAAI,QAAQ;EAClB,IAAI,EAAE,OACJ,OAAO,aAAa,EAAE,WAAW;EAGnC,MAAM,OAAO,MAAc,IAAI,IAAI,IAAA,CAAK,QAAQ,CAAC,EAAE;EACnD,MAAM,WAAW,MAAc,QAAgB,MAC7C,EAAE,aACE,GAAG,KAAK,IAAI,IAAI,MAAM,EAAE,KAAK,EAAE,OAAO,IAAI,EAAE,MAAM,KAClD,GAAG,KAAK,IAAI,EAAE,OAAO;EAE3B,MAAM,QAAkB,CAAC,UAAU,MAAM,IAAI,EAAE,WAAW,mBAAmB;EAC7E,MAAM,KAAK,QAAQ,SAAS,QAAQ,OAAO,EAAE,KAAK,CAAC;EACnD,MAAM,KAAK,QAAQ,SAAS,QAAQ,OAAO,EAAE,KAAK,CAAC;EACnD,MAAM,KAAK,QAAQ,eAAe,QAAQ,YAAY,EAAE,UAAU,CAAC;EACnE,MAAM,KAAK,QAAQ,SAAS,QAAQ,OAAO,EAAE,KAAK,CAAC;EACnD,MAAM,KAAK,QAAQ,cAAc,QAAQ,YAAY,EAAE,UAAU,CAAC;EAClE,MAAM,KAAK,QAAQ,YAAY,QAAQ,UAAU,EAAE,QAAQ,CAAC;EAE5D,OAAO,MAAM,KAAK,IAAI;CACxB,CAAC;AACL"}
|
|
1
|
+
{"version":3,"file":"outcome.js","names":[],"sources":["../../../src/evals/scorers/outcome.ts"],"sourcesContent":["/**\n * MastraCode Outcome Scorer (Code-based, Agent type, always-on)\n *\n * Grades the objective outcome of a coding session by checking hard facts:\n * - Did the build/typecheck pass?\n * - Did tests pass?\n * - What was the tool error rate?\n * - Did the agent get stuck in a loop?\n * - Was there a regression (build passed earlier, failed later)?\n * - Did the agent use ask_user unnecessarily? (autonomy)\n *\n * This scorer is always-on (no sampling) because it's cheap — it only\n * reads exit codes and error states from tool results already in the messages.\n *\n * Scoring philosophy (from Anthropic's eval guidance):\n * \"Grade outcomes, not paths\" — we check what happened, not how.\n * Partial credit: a session that passes build but fails tests scores\n * higher than one that fails build entirely.\n *\n * Dimensions that don't apply to a session (e.g. build for a docs-only task)\n * are excluded from the weighted average, so sessions aren't penalized for\n * irrelevant criteria.\n */\n\nimport type { MastraDBMessage } from '@mastra/core/agent';\nimport { createScorer, filterRun } from '@mastra/core/evals';\n\nimport { isBuildCommand, isTestCommand, getExitCode } from './classify-command.js';\nimport type { ExtractedToolCall } from './extract-tools.js';\nimport { extractToolCalls } from './extract-tools.js';\n\n// ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\n// CONFIGURATION — Adjust weights and thresholds here\n// ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\n\n/**\n * Dimension weights. Used to compute the weighted average of applicable dimensions.\n * Dimensions that don't apply (e.g. no build ran) are excluded from the average.\n */\nconst WEIGHTS = {\n build: 0.3,\n tests: 0.25,\n toolErrors: 0.2,\n loops: 0.1,\n regression: 0.1,\n autonomy: 0.05,\n} as const;\n\n/**\n * Scoring thresholds and multipliers.\n */\nconst THRESHOLDS = {\n /** Multiplier for tool error rate penalty. Score = max(0, 1 - rate × this). */\n toolErrorPenaltyMultiplier: 1.0,\n\n /** How many consecutive identical calls (or same-error repeats) before it's a \"loop\". */\n loopMinRepetitions: 3,\n\n /** Score penalty per detected loop. */\n loopPenaltyPerOccurrence: 0.3,\n\n /** Score deducted per ask_user call. */\n autonomyPenaltyPerAsk: 0.25,\n\n /** Score when a command ran but exit code is ambiguous. */\n ambiguousExitScore: 0.75,\n\n /** Minimum tool calls required to produce a meaningful score. */\n minToolCalls: 1,\n\n /** Tool names whose errors are expected/benign (not penalized). */\n benignErrorTools: ['search_content', 'find_files', 'lsp_inspect'] as string[],\n} as const;\n\n// ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\n\ntype DimensionResult = { score: number; detail: string; applicable: boolean };\n\n/**\n * Check if the primary verb of an execute_command is a build/typecheck command.\n * Splits compound commands (&&, ;) and checks the first segment.\n */\n/** Extract command string from tool call args, then delegate to shared classifier. */\nfunction isBuildArgs(args: Record<string, unknown>): boolean {\n return isBuildCommand(String(args.command ?? ''));\n}\n\nfunction isTestArgs(args: Record<string, unknown>): boolean {\n return isTestCommand(String(args.command ?? ''));\n}\n\n/**\n * Score build/typecheck outcome (final state).\n */\nfunction scoreBuild(execResults: ExtractedToolCall[]): DimensionResult {\n const builds = execResults.filter(r => r.toolName === 'execute_command' && isBuildArgs(r.args));\n if (builds.length === 0) return { score: 0, detail: 'No build/typecheck ran', applicable: false };\n\n const last = builds[builds.length - 1]!;\n if (last.isError) return { score: 0, detail: 'Build/typecheck tool errored', applicable: true };\n\n const exitCode = getExitCode(last.result);\n if (exitCode === 0) return { score: 1, detail: 'Build/typecheck passed', applicable: true };\n if (exitCode !== null && exitCode !== 0)\n return { score: 0, detail: `Build failed (exit ${exitCode})`, applicable: true };\n\n // Fallback: check result text for TS errors\n const text = typeof last.result === 'string' ? last.result : JSON.stringify(last.result ?? '');\n if (/error TS\\d+|Cannot find module|is not assignable/i.test(text)) {\n return { score: 0, detail: 'Build failed (TypeScript errors in output)', applicable: true };\n }\n\n return { score: THRESHOLDS.ambiguousExitScore, detail: 'Build ran, outcome unclear', applicable: true };\n}\n\n/**\n * Score test outcome (final state).\n */\nfunction scoreTests(execResults: ExtractedToolCall[]): DimensionResult {\n const tests = execResults.filter(r => r.toolName === 'execute_command' && isTestArgs(r.args));\n if (tests.length === 0) return { score: 0, detail: 'No tests ran', applicable: false };\n\n const last = tests[tests.length - 1]!;\n if (last.isError) return { score: 0, detail: 'Test command errored', applicable: true };\n\n const exitCode = getExitCode(last.result);\n if (exitCode === 0) return { score: 1, detail: 'Tests passed', applicable: true };\n if (exitCode !== null && exitCode !== 0)\n return { score: 0, detail: `Tests failed (exit ${exitCode})`, applicable: true };\n\n const text = typeof last.result === 'string' ? last.result : JSON.stringify(last.result ?? '');\n if (/\\d+ (?:tests? )?passed|✓|PASS/i.test(text) && !/\\bfail(?:ed|ure)?\\b|\\berror\\b/i.test(text)) {\n return { score: 1, detail: 'Tests passed (inferred)', applicable: true };\n }\n if (/\\bFAIL\\b|failed|✗|✘/i.test(text)) {\n return { score: 0, detail: 'Tests failed (inferred)', applicable: true };\n }\n\n return { score: THRESHOLDS.ambiguousExitScore, detail: 'Tests ran, outcome unclear', applicable: true };\n}\n\n/**\n * Score tool error rate, excluding benign errors from expected-to-fail tools.\n */\nfunction scoreToolErrors(results: ExtractedToolCall[]): DimensionResult {\n if (results.length === 0) return { score: 1, detail: 'No tool calls', applicable: false };\n\n // Only count errors from non-benign tools\n const nonBenign = results.filter(r => !THRESHOLDS.benignErrorTools.includes(r.toolName));\n if (nonBenign.length === 0) return { score: 1, detail: 'All tool calls are benign-error tools', applicable: true };\n\n const errors = nonBenign.filter(r => r.isError);\n const rate = errors.length / nonBenign.length;\n\n if (rate === 0) return { score: 1, detail: 'No tool errors', applicable: true };\n\n const score = Math.max(0, 1 - rate * THRESHOLDS.toolErrorPenaltyMultiplier);\n return {\n score,\n detail: `${errors.length}/${nonBenign.length} tools errored (${(rate * 100).toFixed(0)}%)`,\n applicable: true,\n };\n}\n\n/**\n * Detect stuck loops: same tool + same args appearing 3+ times consecutively,\n * or same tool + same args erroring 3+ times total.\n */\nfunction scoreStuckLoops(results: ExtractedToolCall[]): DimensionResult {\n if (results.length < 3) return { score: 1, detail: 'Too few calls for loop detection', applicable: true };\n\n const fingerprint = (r: ExtractedToolCall) => `${r.toolName}:${JSON.stringify(r.args)}`;\n\n // Check consecutive identical calls\n let maxConsecutive = 1;\n let currentRun = 1;\n for (let i = 1; i < results.length; i++) {\n if (fingerprint(results[i]!) === fingerprint(results[i - 1]!)) {\n currentRun++;\n maxConsecutive = Math.max(maxConsecutive, currentRun);\n } else {\n currentRun = 1;\n }\n }\n\n // Check repeated error fingerprints\n const errorCounts = new Map<string, number>();\n for (const r of results) {\n if (!r.isError) continue;\n const fp = fingerprint(r);\n errorCounts.set(fp, (errorCounts.get(fp) ?? 0) + 1);\n }\n const maxErrorRepeat = Math.max(0, ...[...errorCounts.values()]);\n\n const minReps = THRESHOLDS.loopMinRepetitions;\n const loopSeverity = Math.max(\n maxConsecutive >= minReps ? maxConsecutive - (minReps - 1) : 0,\n maxErrorRepeat >= minReps ? maxErrorRepeat - (minReps - 1) : 0,\n );\n\n if (loopSeverity === 0) return { score: 1, detail: 'No stuck loops', applicable: true };\n\n const score = Math.max(0, 1 - loopSeverity * THRESHOLDS.loopPenaltyPerOccurrence);\n const details: string[] = [];\n if (maxConsecutive >= minReps) details.push(`${maxConsecutive} consecutive identical calls`);\n if (maxErrorRepeat >= minReps) details.push(`same error repeated ${maxErrorRepeat}x`);\n return { score, detail: details.join('; '), applicable: true };\n}\n\n/**\n * Detect regressions: build/test passed at some point, then failed later.\n * Gives partial credit if the agent recovered (final state passes).\n */\nfunction scoreRegression(execResults: ExtractedToolCall[]): DimensionResult {\n const regressions: { label: string; recovered: boolean }[] = [];\n\n for (const [label, filter] of [\n ['Build', isBuildArgs],\n ['Tests', isTestArgs],\n ] as const) {\n const cmds = execResults.filter(r => r.toolName === 'execute_command' && filter(r.args));\n if (cmds.length < 2) continue;\n\n let sawPass = false;\n let sawRegression = false;\n for (const cmd of cmds) {\n const exit = getExitCode(cmd.result);\n if (exit === 0) {\n if (sawRegression) {\n // Recovered from regression\n regressions.push({ label, recovered: true });\n sawRegression = false;\n }\n sawPass = true;\n }\n if (sawPass && exit !== null && exit !== 0) {\n sawRegression = true;\n }\n }\n\n if (sawRegression) {\n // Regression persisted at end of session\n regressions.push({ label, recovered: false });\n }\n }\n\n if (regressions.length === 0) return { score: 1, detail: 'No regressions', applicable: true };\n\n // Persistent regressions are severe (0), recovered regressions get partial credit (0.5)\n const persistentCount = regressions.filter(r => !r.recovered).length;\n const recoveredCount = regressions.filter(r => r.recovered).length;\n const score = Math.max(0, 1 - persistentCount * 0.5 - recoveredCount * 0.1);\n const details = regressions.map(\n r => `${r.label} ${r.recovered ? 'regressed then recovered' : 'regressed (persisted)'}`,\n );\n return { score, detail: details.join('; '), applicable: true };\n}\n\n/**\n * Score autonomy: penalize ask_user calls.\n * 0 calls = 1.0, each call reduces by 0.25 (up to a floor of 0).\n */\nfunction scoreAutonomy(results: ExtractedToolCall[]): DimensionResult {\n const askUserCalls = results.filter(r => r.toolName === 'ask_user');\n const count = askUserCalls.length;\n\n if (count === 0) return { score: 1, detail: 'No ask_user calls (fully autonomous)', applicable: true };\n\n const score = Math.max(0, 1 - count * THRESHOLDS.autonomyPenaltyPerAsk);\n return { score, detail: `${count} ask_user call${count > 1 ? 's' : ''}`, applicable: true };\n}\n\n/**\n * Compute weighted average of only applicable dimensions.\n * Returns null if no dimensions are applicable.\n */\nfunction weightedAverage(dimensions: Record<string, DimensionResult>, weights: Record<string, number>): number | null {\n let totalWeight = 0;\n let weightedSum = 0;\n\n for (const [key, result] of Object.entries(dimensions)) {\n if (!result.applicable) continue;\n const weight = weights[key] ?? 0;\n totalWeight += weight;\n weightedSum += result.score * weight;\n }\n\n if (totalWeight === 0) return null;\n return weightedSum / totalWeight;\n}\n\nexport function createOutcomeScorer() {\n return createScorer({\n id: 'mastracode-outcome',\n name: 'MastraCode Outcome',\n description:\n 'Grades coding session outcomes: build/test pass, tool errors, stuck loops, regressions, autonomy. Always-on.',\n type: 'agent',\n prepareRun: filterRun({ partTypes: ['tool-invocation', 'text'] }),\n })\n .preprocess(async ({ run }) => {\n const messages = (run.output ?? []) as MastraDBMessage[];\n const allResults = extractToolCalls(messages);\n const totalCalls = allResults.length;\n\n // H1: Empty sessions or sessions with no tool calls → score 0\n if (totalCalls < THRESHOLDS.minToolCalls) {\n const empty: DimensionResult = { score: 0, detail: 'No tool calls', applicable: false };\n return {\n empty: true as const,\n build: empty,\n tests: empty,\n toolErrors: empty,\n loops: empty,\n regression: empty,\n autonomy: empty,\n totalCalls,\n };\n }\n\n const execResults = allResults.filter(r => r.toolName === 'execute_command');\n\n return {\n empty: false as const,\n build: scoreBuild(execResults),\n tests: scoreTests(execResults),\n toolErrors: scoreToolErrors(allResults),\n loops: scoreStuckLoops(allResults),\n regression: scoreRegression(execResults),\n autonomy: scoreAutonomy(allResults),\n totalCalls,\n };\n })\n .generateScore(({ results }) => {\n const p = results.preprocessStepResult;\n if (p.empty) return 0;\n\n const score = weightedAverage(\n {\n build: p.build,\n tests: p.tests,\n toolErrors: p.toolErrors,\n loops: p.loops,\n regression: p.regression,\n autonomy: p.autonomy,\n },\n WEIGHTS,\n );\n\n return score === null ? 0 : Math.round(score * 100) / 100;\n })\n .generateReason(({ results, score }) => {\n const p = results.preprocessStepResult;\n if (p.empty) {\n return `Score: 0 (${p.totalCalls} tool calls — below minimum threshold for scoring)`;\n }\n\n const pct = (w: number) => `${(w * 100).toFixed(0)}%`;\n const dimLine = (name: string, weight: number, r: DimensionResult) =>\n r.applicable\n ? `${name} (${pct(weight)}): ${r.detail} [${r.score}]`\n : `${name}: ${r.detail} [N/A — excluded from average]`;\n\n const parts: string[] = [`Score: ${score} (${p.totalCalls} tool calls total)`];\n parts.push(dimLine('Build', WEIGHTS.build, p.build));\n parts.push(dimLine('Tests', WEIGHTS.tests, p.tests));\n parts.push(dimLine('Tool errors', WEIGHTS.toolErrors, p.toolErrors));\n parts.push(dimLine('Loops', WEIGHTS.loops, p.loops));\n parts.push(dimLine('Regression', WEIGHTS.regression, p.regression));\n parts.push(dimLine('Autonomy', WEIGHTS.autonomy, p.autonomy));\n\n return parts.join('\\n');\n });\n}\n"],"mappings":";;;;;;;;AAuCA,MAAM,UAAU;CACd,OAAO;CACP,OAAO;CACP,YAAY;CACZ,OAAO;CACP,YAAY;CACZ,UAAU;AACZ;;;;AAKA,MAAM,aAAa;;CAEjB,4BAA4B;;CAG5B,oBAAoB;;CAGpB,0BAA0B;;CAG1B,uBAAuB;;CAGvB,oBAAoB;;CAGpB,cAAc;;CAGd,kBAAkB;EAAC;EAAkB;EAAc;CAAa;AAClE;;;;;;AAWA,SAAS,YAAY,MAAwC;CAC3D,OAAO,eAAe,OAAO,KAAK,WAAW,EAAE,CAAC;AAClD;AAEA,SAAS,WAAW,MAAwC;CAC1D,OAAO,cAAc,OAAO,KAAK,WAAW,EAAE,CAAC;AACjD;;;;AAKA,SAAS,WAAW,aAAmD;CACrE,MAAM,SAAS,YAAY,QAAO,MAAK,EAAE,aAAa,qBAAqB,YAAY,EAAE,IAAI,CAAC;CAC9F,IAAI,OAAO,WAAW,GAAG,OAAO;EAAE,OAAO;EAAG,QAAQ;EAA0B,YAAY;CAAM;CAEhG,MAAM,OAAO,OAAO,OAAO,SAAS;CACpC,IAAI,KAAK,SAAS,OAAO;EAAE,OAAO;EAAG,QAAQ;EAAgC,YAAY;CAAK;CAE9F,MAAM,WAAW,YAAY,KAAK,MAAM;CACxC,IAAI,aAAa,GAAG,OAAO;EAAE,OAAO;EAAG,QAAQ;EAA0B,YAAY;CAAK;CAC1F,IAAI,aAAa,QAAQ,aAAa,GACpC,OAAO;EAAE,OAAO;EAAG,QAAQ,sBAAsB,SAAS;EAAI,YAAY;CAAK;CAGjF,MAAM,OAAO,OAAO,KAAK,WAAW,WAAW,KAAK,SAAS,KAAK,UAAU,KAAK,UAAU,EAAE;CAC7F,IAAI,oDAAoD,KAAK,IAAI,GAC/D,OAAO;EAAE,OAAO;EAAG,QAAQ;EAA8C,YAAY;CAAK;CAG5F,OAAO;EAAE,OAAO,WAAW;EAAoB,QAAQ;EAA8B,YAAY;CAAK;AACxG;;;;AAKA,SAAS,WAAW,aAAmD;CACrE,MAAM,QAAQ,YAAY,QAAO,MAAK,EAAE,aAAa,qBAAqB,WAAW,EAAE,IAAI,CAAC;CAC5F,IAAI,MAAM,WAAW,GAAG,OAAO;EAAE,OAAO;EAAG,QAAQ;EAAgB,YAAY;CAAM;CAErF,MAAM,OAAO,MAAM,MAAM,SAAS;CAClC,IAAI,KAAK,SAAS,OAAO;EAAE,OAAO;EAAG,QAAQ;EAAwB,YAAY;CAAK;CAEtF,MAAM,WAAW,YAAY,KAAK,MAAM;CACxC,IAAI,aAAa,GAAG,OAAO;EAAE,OAAO;EAAG,QAAQ;EAAgB,YAAY;CAAK;CAChF,IAAI,aAAa,QAAQ,aAAa,GACpC,OAAO;EAAE,OAAO;EAAG,QAAQ,sBAAsB,SAAS;EAAI,YAAY;CAAK;CAEjF,MAAM,OAAO,OAAO,KAAK,WAAW,WAAW,KAAK,SAAS,KAAK,UAAU,KAAK,UAAU,EAAE;CAC7F,IAAI,iCAAiC,KAAK,IAAI,KAAK,CAAC,iCAAiC,KAAK,IAAI,GAC5F,OAAO;EAAE,OAAO;EAAG,QAAQ;EAA2B,YAAY;CAAK;CAEzE,IAAI,uBAAuB,KAAK,IAAI,GAClC,OAAO;EAAE,OAAO;EAAG,QAAQ;EAA2B,YAAY;CAAK;CAGzE,OAAO;EAAE,OAAO,WAAW;EAAoB,QAAQ;EAA8B,YAAY;CAAK;AACxG;;;;AAKA,SAAS,gBAAgB,SAA+C;CACtE,IAAI,QAAQ,WAAW,GAAG,OAAO;EAAE,OAAO;EAAG,QAAQ;EAAiB,YAAY;CAAM;CAGxF,MAAM,YAAY,QAAQ,QAAO,MAAK,CAAC,WAAW,iBAAiB,SAAS,EAAE,QAAQ,CAAC;CACvF,IAAI,UAAU,WAAW,GAAG,OAAO;EAAE,OAAO;EAAG,QAAQ;EAAyC,YAAY;CAAK;CAEjH,MAAM,SAAS,UAAU,QAAO,MAAK,EAAE,OAAO;CAC9C,MAAM,OAAO,OAAO,SAAS,UAAU;CAEvC,IAAI,SAAS,GAAG,OAAO;EAAE,OAAO;EAAG,QAAQ;EAAkB,YAAY;CAAK;CAG9E,OAAO;EACL,OAFY,KAAK,IAAI,GAAG,IAAI,OAAO,WAAW,0BAE1C;EACJ,QAAQ,GAAG,OAAO,OAAO,GAAG,UAAU,OAAO,mBAAmB,OAAO,IAAA,CAAK,QAAQ,CAAC,EAAE;EACvF,YAAY;CACd;AACF;;;;;AAMA,SAAS,gBAAgB,SAA+C;CACtE,IAAI,QAAQ,SAAS,GAAG,OAAO;EAAE,OAAO;EAAG,QAAQ;EAAoC,YAAY;CAAK;CAExG,MAAM,eAAe,MAAyB,GAAG,EAAE,SAAS,GAAG,KAAK,UAAU,EAAE,IAAI;CAGpF,IAAI,iBAAiB;CACrB,IAAI,aAAa;CACjB,KAAK,IAAI,IAAI,GAAG,IAAI,QAAQ,QAAQ,KAClC,IAAI,YAAY,QAAQ,EAAG,MAAM,YAAY,QAAQ,IAAI,EAAG,GAAG;EAC7D;EACA,iBAAiB,KAAK,IAAI,gBAAgB,UAAU;CACtD,OACE,aAAa;CAKjB,MAAM,8BAAc,IAAI,IAAoB;CAC5C,KAAK,MAAM,KAAK,SAAS;EACvB,IAAI,CAAC,EAAE,SAAS;EAChB,MAAM,KAAK,YAAY,CAAC;EACxB,YAAY,IAAI,KAAK,YAAY,IAAI,EAAE,KAAK,KAAK,CAAC;CACpD;CACA,MAAM,iBAAiB,KAAK,IAAI,GAAG,GAAG,CAAC,GAAG,YAAY,OAAO,CAAC,CAAC;CAE/D,MAAM,UAAU,WAAW;CAC3B,MAAM,eAAe,KAAK,IACxB,kBAAkB,UAAU,kBAAkB,UAAU,KAAK,GAC7D,kBAAkB,UAAU,kBAAkB,UAAU,KAAK,CAC/D;CAEA,IAAI,iBAAiB,GAAG,OAAO;EAAE,OAAO;EAAG,QAAQ;EAAkB,YAAY;CAAK;CAEtF,MAAM,QAAQ,KAAK,IAAI,GAAG,IAAI,eAAe,WAAW,wBAAwB;CAChF,MAAM,UAAoB,CAAC;CAC3B,IAAI,kBAAkB,SAAS,QAAQ,KAAK,GAAG,eAAe,6BAA6B;CAC3F,IAAI,kBAAkB,SAAS,QAAQ,KAAK,uBAAuB,eAAe,EAAE;CACpF,OAAO;EAAE;EAAO,QAAQ,QAAQ,KAAK,IAAI;EAAG,YAAY;CAAK;AAC/D;;;;;AAMA,SAAS,gBAAgB,aAAmD;CAC1E,MAAM,cAAuD,CAAC;CAE9D,KAAK,MAAM,CAAC,OAAO,WAAW,CAC5B,CAAC,SAAS,WAAW,GACrB,CAAC,SAAS,UAAU,CACtB,GAAY;EACV,MAAM,OAAO,YAAY,QAAO,MAAK,EAAE,aAAa,qBAAqB,OAAO,EAAE,IAAI,CAAC;EACvF,IAAI,KAAK,SAAS,GAAG;EAErB,IAAI,UAAU;EACd,IAAI,gBAAgB;EACpB,KAAK,MAAM,OAAO,MAAM;GACtB,MAAM,OAAO,YAAY,IAAI,MAAM;GACnC,IAAI,SAAS,GAAG;IACd,IAAI,eAAe;KAEjB,YAAY,KAAK;MAAE;MAAO,WAAW;KAAK,CAAC;KAC3C,gBAAgB;IAClB;IACA,UAAU;GACZ;GACA,IAAI,WAAW,SAAS,QAAQ,SAAS,GACvC,gBAAgB;EAEpB;EAEA,IAAI,eAEF,YAAY,KAAK;GAAE;GAAO,WAAW;EAAM,CAAC;CAEhD;CAEA,IAAI,YAAY,WAAW,GAAG,OAAO;EAAE,OAAO;EAAG,QAAQ;EAAkB,YAAY;CAAK;CAG5F,MAAM,kBAAkB,YAAY,QAAO,MAAK,CAAC,EAAE,SAAS,CAAC,CAAC;CAC9D,MAAM,iBAAiB,YAAY,QAAO,MAAK,EAAE,SAAS,CAAC,CAAC;CAK5D,OAAO;EAAE,OAJK,KAAK,IAAI,GAAG,IAAI,kBAAkB,KAAM,iBAAiB,EAI1D;EAAG,QAHA,YAAY,KAC1B,MAAK,GAAG,EAAE,MAAM,GAAG,EAAE,YAAY,6BAA6B,yBAElC,CAAC,CAAC,KAAK,IAAI;EAAG,YAAY;CAAK;AAC/D;;;;;AAMA,SAAS,cAAc,SAA+C;CAEpE,MAAM,QADe,QAAQ,QAAO,MAAK,EAAE,aAAa,UAC/B,CAAC,CAAC;CAE3B,IAAI,UAAU,GAAG,OAAO;EAAE,OAAO;EAAG,QAAQ;EAAwC,YAAY;CAAK;CAGrG,OAAO;EAAE,OADK,KAAK,IAAI,GAAG,IAAI,QAAQ,WAAW,qBACpC;EAAG,QAAQ,GAAG,MAAM,gBAAgB,QAAQ,IAAI,MAAM;EAAM,YAAY;CAAK;AAC5F;;;;;AAMA,SAAS,gBAAgB,YAA6C,SAAgD;CACpH,IAAI,cAAc;CAClB,IAAI,cAAc;CAElB,KAAK,MAAM,CAAC,KAAK,WAAW,OAAO,QAAQ,UAAU,GAAG;EACtD,IAAI,CAAC,OAAO,YAAY;EACxB,MAAM,SAAS,QAAQ,QAAQ;EAC/B,eAAe;EACf,eAAe,OAAO,QAAQ;CAChC;CAEA,IAAI,gBAAgB,GAAG,OAAO;CAC9B,OAAO,cAAc;AACvB;AAEA,SAAgB,sBAAsB;CACpC,OAAO,aAAa;EAClB,IAAI;EACJ,MAAM;EACN,aACE;EACF,MAAM;EACN,YAAY,UAAU,EAAE,WAAW,CAAC,mBAAmB,MAAM,EAAE,CAAC;CAClE,CAAC,CAAC,CACC,WAAW,OAAO,EAAE,UAAU;EAC7B,MAAM,WAAY,IAAI,UAAU,CAAC;EACjC,MAAM,aAAa,iBAAiB,QAAQ;EAC5C,MAAM,aAAa,WAAW;EAG9B,IAAI,aAAa,WAAW,cAAc;GACxC,MAAM,QAAyB;IAAE,OAAO;IAAG,QAAQ;IAAiB,YAAY;GAAM;GACtF,OAAO;IACL,OAAO;IACP,OAAO;IACP,OAAO;IACP,YAAY;IACZ,OAAO;IACP,YAAY;IACZ,UAAU;IACV;GACF;EACF;EAEA,MAAM,cAAc,WAAW,QAAO,MAAK,EAAE,aAAa,iBAAiB;EAE3E,OAAO;GACL,OAAO;GACP,OAAO,WAAW,WAAW;GAC7B,OAAO,WAAW,WAAW;GAC7B,YAAY,gBAAgB,UAAU;GACtC,OAAO,gBAAgB,UAAU;GACjC,YAAY,gBAAgB,WAAW;GACvC,UAAU,cAAc,UAAU;GAClC;EACF;CACF,CAAC,CAAC,CACD,eAAe,EAAE,cAAc;EAC9B,MAAM,IAAI,QAAQ;EAClB,IAAI,EAAE,OAAO,OAAO;EAEpB,MAAM,QAAQ,gBACZ;GACE,OAAO,EAAE;GACT,OAAO,EAAE;GACT,YAAY,EAAE;GACd,OAAO,EAAE;GACT,YAAY,EAAE;GACd,UAAU,EAAE;EACd,GACA,OACF;EAEA,OAAO,UAAU,OAAO,IAAI,KAAK,MAAM,QAAQ,GAAG,IAAI;CACxD,CAAC,CAAC,CACD,gBAAgB,EAAE,SAAS,YAAY;EACtC,MAAM,IAAI,QAAQ;EAClB,IAAI,EAAE,OACJ,OAAO,aAAa,EAAE,WAAW;EAGnC,MAAM,OAAO,MAAc,IAAI,IAAI,IAAA,CAAK,QAAQ,CAAC,EAAE;EACnD,MAAM,WAAW,MAAc,QAAgB,MAC7C,EAAE,aACE,GAAG,KAAK,IAAI,IAAI,MAAM,EAAE,KAAK,EAAE,OAAO,IAAI,EAAE,MAAM,KAClD,GAAG,KAAK,IAAI,EAAE,OAAO;EAE3B,MAAM,QAAkB,CAAC,UAAU,MAAM,IAAI,EAAE,WAAW,mBAAmB;EAC7E,MAAM,KAAK,QAAQ,SAAS,QAAQ,OAAO,EAAE,KAAK,CAAC;EACnD,MAAM,KAAK,QAAQ,SAAS,QAAQ,OAAO,EAAE,KAAK,CAAC;EACnD,MAAM,KAAK,QAAQ,eAAe,QAAQ,YAAY,EAAE,UAAU,CAAC;EACnE,MAAM,KAAK,QAAQ,SAAS,QAAQ,OAAO,EAAE,KAAK,CAAC;EACnD,MAAM,KAAK,QAAQ,cAAc,QAAQ,YAAY,EAAE,UAAU,CAAC;EAClE,MAAM,KAAK,QAAQ,YAAY,QAAQ,UAAU,EAAE,QAAQ,CAAC;EAE5D,OAAO,MAAM,KAAK,IAAI;CACxB,CAAC;AACL"}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"remote-controller.js","names":[
|
|
1
|
+
{"version":3,"file":"remote-controller.js","names":[],"sources":["../../src/factory/remote-controller.ts"],"sourcesContent":["import type { AvailableModel, ModelAuthStatus } from '@mastra/core/agent-controller';\n\nimport type { FactoryAddress } from './client.js';\nimport { createFactoryClient, FACTORY_CONTROLLER_ID } from './client.js';\n\nexport interface RemoteMode {\n id: string;\n name?: string;\n description?: string;\n default?: boolean;\n defaultModelId?: string;\n metadata?: Record<string, unknown>;\n}\n\ntype ControllerHandle = ReturnType<ReturnType<typeof createFactoryClient>['getAgentController']>;\n\n/**\n * The Factory-side counterpart to the local agent controller. Modes, models, and\n * provider auth come from the Factory, never from local settings or auth storage.\n */\nexport class RemoteController {\n readonly id = FACTORY_CONTROLLER_ID;\n readonly #remote: ControllerHandle;\n #modes: RemoteMode[] = [];\n #models: Promise<AvailableModel[]> | undefined;\n\n constructor(address: FactoryAddress, remote?: ControllerHandle) {\n this.#remote = remote ?? createFactoryClient(address).getAgentController(FACTORY_CONTROLLER_ID);\n }\n\n /** Load the mode list; `listModes()` is synchronous so it must be primed. */\n async connect(): Promise<void> {\n this.#modes = (await this.#remote.listModes()) as RemoteMode[];\n }\n\n listModes(): RemoteMode[] {\n return this.#modes;\n }\n\n listAvailableModels(): Promise<AvailableModel[]> {\n this.#models ??= this.#remote.listModels().then(models => models as AvailableModel[]);\n this.#models.catch(() => (this.#models = undefined));\n return this.#models;\n }\n\n invalidateAvailableModelsCache(): void {\n this.#models = undefined;\n }\n\n /** Auth is the Factory's: report what it reports, or assume ok and let the run surface errors. */\n async getCurrentModelAuthStatus(session: { model: { get(): string } }): Promise<ModelAuthStatus> {\n const modelId = session.model.get();\n const models = await this.listAvailableModels().catch(() => []);\n const model = models.find(m => m.id === modelId);\n return model ? { hasAuth: model.hasApiKey, apiKeyEnvVar: model.apiKeyEnvVar } : { hasAuth: true };\n }\n}\n"],"mappings":";;;;;;AAoBA,IAAa,mBAAb,MAA8B;CAC5B,KAAc;CACd;CACA,SAAuB,CAAC;CACxB;CAEA,YAAY,SAAyB,QAA2B;EAC9D,KAAK,UAAU,UAAU,oBAAoB,OAAO,CAAC,CAAC,mBAAA,MAAwC;CAChG;;CAGA,MAAM,UAAyB;EAC7B,KAAK,SAAU,MAAM,KAAK,QAAQ,UAAU;CAC9C;CAEA,YAA0B;EACxB,OAAO,KAAK;CACd;CAEA,sBAAiD;EAC/C,KAAK,YAAY,KAAK,QAAQ,WAAW,CAAC,CAAC,MAAK,WAAU,MAA0B;EACpF,KAAK,QAAQ,YAAa,KAAK,UAAU,KAAA,CAAU;EACnD,OAAO,KAAK;CACd;CAEA,iCAAuC;EACrC,KAAK,UAAU,KAAA;CACjB;;CAGA,MAAM,0BAA0B,SAAiE;EAC/F,MAAM,UAAU,QAAQ,MAAM,IAAI;EAElC,MAAM,SAAQ,MADO,KAAK,oBAAoB,CAAC,CAAC,YAAY,CAAC,CAAC,EAAA,CACzC,MAAK,MAAK,EAAE,OAAO,OAAO;EAC/C,OAAO,QAAQ;GAAE,SAAS,MAAM;GAAW,cAAc,MAAM;EAAa,IAAI,EAAE,SAAS,KAAK;CAClG;AACF"}
|
|
@@ -19,7 +19,7 @@ type Files = Array<{
|
|
|
19
19
|
*/
|
|
20
20
|
export declare class RemoteSession {
|
|
21
21
|
#private;
|
|
22
|
-
constructor(address: FactoryAddress,
|
|
22
|
+
constructor(address: FactoryAddress, sessionId: string, remote?: RemoteHandle);
|
|
23
23
|
/** Create/resume the remote session, hydrate the mirror, and open the event stream. */
|
|
24
24
|
connect(): Promise<void>;
|
|
25
25
|
/** Modes come from the controller; used to resolve the current mode's name and color. */
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"remote-session.d.ts","sourceRoot":"","sources":["../../src/factory/remote-session.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAuC,uBAAuB,EAAE,MAAM,mBAAmB,CAAC;AACtG,OAAO,KAAK,EACV,2BAA2B,EAE3B,4BAA4B,EAC5B,kCAAkC,EAClC,qBAAqB,EACrB,eAAe,EACf,OAAO,EACR,MAAM,+BAA+B,CAAC;AAGvC,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,aAAa,CAAC;AAClD,OAAO,EAAE,mBAAmB,EAAyB,MAAM,aAAa,CAAC;AAEzE,KAAK,YAAY,GAAG,UAAU,CAAC,UAAU,CAAC,UAAU,CAAC,OAAO,mBAAmB,CAAC,CAAC,oBAAoB,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC;AACpH,KAAK,UAAU,GAAG,uBAAuB,CAAC;AAC1C,KAAK,KAAK,GAAG,KAAK,CAAC;IAAE,IAAI,EAAE,MAAM,CAAC;IAAC,SAAS,EAAE,MAAM,CAAC;IAAC,QAAQ,CAAC,EAAE,MAAM,CAAA;CAAE,CAAC,CAAC;AAgD3E;;;;;;;GAOG;AACH,qBAAa,aAAa;;
|
|
1
|
+
{"version":3,"file":"remote-session.d.ts","sourceRoot":"","sources":["../../src/factory/remote-session.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAuC,uBAAuB,EAAE,MAAM,mBAAmB,CAAC;AACtG,OAAO,KAAK,EACV,2BAA2B,EAE3B,4BAA4B,EAC5B,kCAAkC,EAClC,qBAAqB,EACrB,eAAe,EACf,OAAO,EACR,MAAM,+BAA+B,CAAC;AAGvC,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,aAAa,CAAC;AAClD,OAAO,EAAE,mBAAmB,EAAyB,MAAM,aAAa,CAAC;AAEzE,KAAK,YAAY,GAAG,UAAU,CAAC,UAAU,CAAC,UAAU,CAAC,OAAO,mBAAmB,CAAC,CAAC,oBAAoB,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC;AACpH,KAAK,UAAU,GAAG,uBAAuB,CAAC;AAC1C,KAAK,KAAK,GAAG,KAAK,CAAC;IAAE,IAAI,EAAE,MAAM,CAAC;IAAC,SAAS,EAAE,MAAM,CAAC;IAAC,QAAQ,CAAC,EAAE,MAAM,CAAA;CAAE,CAAC,CAAC;AAgD3E;;;;;;;GAOG;AACH,qBAAa,aAAa;;gBAkBZ,OAAO,EAAE,cAAc,EAAE,SAAS,EAAE,MAAM,EAAE,MAAM,CAAC,EAAE,YAAY;IAM7E,uFAAuF;IACjF,OAAO,IAAI,OAAO,CAAC,IAAI,CAAC;IAa9B,yFAAyF;IACzF,QAAQ,CAAC,KAAK,EAAE,UAAU,EAAE,GAAG,IAAI;IAInC,UAAU,IAAI,IAAI;IA0FlB,SAAS,CAAC,QAAQ,EAAE,4BAA4B,GAAG,MAAM,IAAI;IAK7D,gBAAgB,IAAI,MAAM,IAAI;IAI9B,QAAQ,CAAC,QAAQ;;;;;;MAMf;IAEF,QAAQ,CAAC,MAAM;;;;;;gCAemB;YAAE,QAAQ,EAAE,MAAM,CAAA;SAAE;6BAExB;YAAE,KAAK,CAAC,EAAE,MAAM,CAAA;SAAE;+BAMf;YAAE,QAAQ,EAAE,MAAM,CAAA;SAAE;4CAIR;YAAE,cAAc,CAAC,EAAE,MAAM,CAAC;YAAC,KAAK,CAAC,EAAE,MAAM,CAAA;SAAE;4BAM1D;YAAE,KAAK,EAAE,MAAM,CAAA;SAAE;+BAGd;YAAE,QAAQ,EAAE,MAAM,CAAA;SAAE;4CAIb;YAAE,QAAQ,EAAE,MAAM,CAAC;YAAC,KAAK,CAAC,EAAE,MAAM,CAAA;SAAE;yCAElC;YAAE,KAAK,CAAC,EAAE,MAAM,CAAA;SAAE;;;;;;;;;;;;qDAgBL;YAAE,cAAc,EAAE,MAAM,CAAA;SAAE;;MAI5E;IAEL,QAAQ,CAAC,IAAI;;;;;;;0BAIO;YAAE,MAAM,EAAE,MAAM,CAAA;SAAE;6BAGT;YAAE,MAAM,EAAE,MAAM,CAAA;SAAE;MAI7C;IAEF,QAAQ,CAAC,KAAK;;;;2BAIO;YAAE,OAAO,EAAE,MAAM,CAAA;SAAE;0BAGd,MAAM,YAAW;YAAE,aAAa,CAAC,EAAE,GAAG,CAAA;SAAE;;MAKhE;IAEF,QAAQ,CAAC,KAAK;;uBAES,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC;yBAIrB,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,eAAe,MAAM,OAAO;iBAK3D,OAAO,WAAW,kCAAkC,CAAC,GAAG,EAAE,OAAO,CAAC,KAAG,OAAO,CAAC,OAAO,CAAC;MAKpG;IAEF,QAAQ,CAAC,YAAY;;;;;;;;;MAWnB;IAEF,QAAQ,CAAC,GAAG;;;;MAIV;IAEF,QAAQ,CAAC,MAAM;;;MAAkE;IAEjF,QAAQ,CAAC,WAAW;;8BAEI;YAAE,UAAU,EAAE,MAAM,CAAA;SAAE;;;;;;;;;;8BACtB;YAAE,UAAU,EAAE,MAAM,CAAA;SAAE;;MAE5C;IAEF,QAAQ,CAAC,WAAW;;+CAE2B;YAAE,QAAQ,EAAE,GAAG,CAAC;YAAC,MAAM,EAAE,GAAG,CAAA;SAAE;2CAIlC;YAAE,QAAQ,EAAE,MAAM,CAAC;YAAC,MAAM,EAAE,GAAG,CAAA;SAAE;MAI1E;IAEF,eAAe,IAAI,MAAM,GAAG,IAAI;IAIhC,OAAO,IAAI,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC;IAIjC,aAAa;IAIb,SAAS;;;;IAIH,WAAW,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE,EAAE;QAAE,OAAO,EAAE,MAAM,CAAC;QAAC,KAAK,CAAC,EAAE,KAAK,CAAA;KAAE;IAIlE,YAAY,CAAC,EAAE,OAAO,EAAE,EAAE;QAAE,OAAO,EAAE,MAAM,CAAA;KAAE;IAInD,UAAU,CAAC,KAAK,EAAE,UAAU,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,CAAC;;;;;;;;IAqBhD,KAAK,CAAC,EAAE,OAAO,EAAE,EAAE;QAAE,OAAO,EAAE,MAAM,CAAA;KAAE;IAItC,QAAQ,CAAC,EAAE,OAAO,EAAE,EAAE;QAAE,OAAO,EAAE,MAAM,CAAA;KAAE;IAI/C,qBAAqB,CAAC,EACpB,QAAQ,EACR,UAAU,GACX,EAAE;QACD,QAAQ,EAAE,SAAS,GAAG,SAAS,GAAG,uBAAuB,CAAC;QAC1D,UAAU,EAAE,MAAM,CAAC;KACpB;;;IAcK,uBAAuB,CAAC,EAAE,UAAU,EAAE,UAAU,EAAE,EAAE;QAAE,UAAU,EAAE,OAAO,CAAC;QAAC,UAAU,CAAC,EAAE,MAAM,CAAA;KAAE;IAStG,iBAAiB,IAAI,OAAO;IAI5B,mBAAmB,IAAI,IAAI;IAE3B,KAAK,IAAI,IAAI;IAMb,QAAQ,IAAI,IAAI;CAGjB"}
|
|
@@ -50,6 +50,8 @@ function toThread(thread, resourceId) {
|
|
|
50
50
|
*/
|
|
51
51
|
var RemoteSession = class {
|
|
52
52
|
#remote;
|
|
53
|
+
#sessionId;
|
|
54
|
+
/** Whose threads and memory the session works on; the Factory reports it when the session is created. */
|
|
53
55
|
#resourceId;
|
|
54
56
|
#listeners = /* @__PURE__ */ new Set();
|
|
55
57
|
#unsubscribe;
|
|
@@ -66,13 +68,15 @@ var RemoteSession = class {
|
|
|
66
68
|
};
|
|
67
69
|
#pendingUserIds = [];
|
|
68
70
|
#userIdMap = /* @__PURE__ */ new Map();
|
|
69
|
-
constructor(address,
|
|
70
|
-
this.#
|
|
71
|
-
this.#
|
|
71
|
+
constructor(address, sessionId, remote) {
|
|
72
|
+
this.#sessionId = sessionId;
|
|
73
|
+
this.#resourceId = sessionId;
|
|
74
|
+
this.#remote = remote ?? createFactoryClient(address).getAgentController("code").session(sessionId);
|
|
72
75
|
}
|
|
73
76
|
/** Create/resume the remote session, hydrate the mirror, and open the event stream. */
|
|
74
77
|
async connect() {
|
|
75
|
-
await this.#remote.create();
|
|
78
|
+
const created = await this.#remote.create();
|
|
79
|
+
if (created?.resourceId) this.#resourceId = created.resourceId;
|
|
76
80
|
await Promise.all([this.#hydrate(), this.#refreshPermissions()]);
|
|
77
81
|
const subscription = await this.#remote.subscribe({
|
|
78
82
|
reconnect: true,
|
|
@@ -166,9 +170,7 @@ var RemoteSession = class {
|
|
|
166
170
|
case "agent_start":
|
|
167
171
|
this.#running = true;
|
|
168
172
|
break;
|
|
169
|
-
case "agent_end":
|
|
170
|
-
this.#running = false;
|
|
171
|
-
break;
|
|
173
|
+
case "agent_end": this.#running = false;
|
|
172
174
|
}
|
|
173
175
|
for (const listener of this.#listeners) Promise.resolve(listener(event)).catch(() => {});
|
|
174
176
|
}
|
|
@@ -182,13 +184,13 @@ var RemoteSession = class {
|
|
|
182
184
|
identity = {
|
|
183
185
|
getResourceId: () => this.#resourceId,
|
|
184
186
|
getDefaultResourceId: () => this.#resourceId,
|
|
185
|
-
getId: () => this.#
|
|
186
|
-
getOwnerId: () => this.#
|
|
187
|
+
getId: () => this.#sessionId,
|
|
188
|
+
getOwnerId: () => this.#sessionId,
|
|
187
189
|
setResourceId: () => {}
|
|
188
190
|
};
|
|
189
191
|
thread = (() => {
|
|
190
192
|
const remote = () => this.#remote;
|
|
191
|
-
const resourceId = this.#resourceId;
|
|
193
|
+
const resourceId = () => this.#resourceId;
|
|
192
194
|
return {
|
|
193
195
|
getId: () => this.#threadId,
|
|
194
196
|
isSet: () => this.#threadId !== null,
|
|
@@ -200,10 +202,10 @@ var RemoteSession = class {
|
|
|
200
202
|
if (!this.#threadId) await this.thread.create();
|
|
201
203
|
return this.#threadId;
|
|
202
204
|
},
|
|
203
|
-
list: async () => (await remote().listThreads()).map((t) => toThread(t, resourceId)),
|
|
205
|
+
list: async () => (await remote().listThreads()).map((t) => toThread(t, resourceId())),
|
|
204
206
|
getById: async ({ threadId }) => (await this.thread.list()).find((t) => t.id === threadId) ?? null,
|
|
205
207
|
create: async ({ title } = {}) => {
|
|
206
|
-
const thread = toThread(await remote().createThread(title), resourceId);
|
|
208
|
+
const thread = toThread(await remote().createThread(title), resourceId());
|
|
207
209
|
this.#threadId = thread.id;
|
|
208
210
|
return thread;
|
|
209
211
|
},
|
|
@@ -215,7 +217,7 @@ var RemoteSession = class {
|
|
|
215
217
|
const thread = toThread(await remote().cloneThread({
|
|
216
218
|
sourceThreadId,
|
|
217
219
|
title
|
|
218
|
-
}), resourceId);
|
|
220
|
+
}), resourceId());
|
|
219
221
|
this.#threadId = thread.id;
|
|
220
222
|
return thread;
|
|
221
223
|
},
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"remote-session.js","names":["#remote","#resourceId","#listeners","#userIdMap","#hydrate","#refreshPermissions","#receive","#unsubscribe","#modes","#threadId","#modeId","#modelId","#running","#state","#displayState","#permissionRules","#pendingUserIds","#adoptLocalUserId","#emitError"],"sources":["../../src/factory/remote-session.ts"],"sourcesContent":["import { randomUUID } from 'node:crypto';\n\nimport type { AgentControllerEvent as ClientEvent, AgentControllerModeInfo } from '@mastra/client-js';\nimport type {\n AgentControllerDisplayState,\n AgentControllerEvent,\n AgentControllerEventListener,\n AgentControllerRequestStateUpdater,\n AgentControllerThread,\n PermissionRules,\n Session,\n} from '@mastra/core/agent-controller';\nimport { defaultDisplayState } from '@mastra/core/agent-controller';\n\nimport type { FactoryAddress } from './client.js';\nimport { createFactoryClient, FACTORY_CONTROLLER_ID } from './client.js';\n\ntype RemoteHandle = ReturnType<ReturnType<ReturnType<typeof createFactoryClient>['getAgentController']>['session']>;\ntype RemoteMode = AgentControllerModeInfo;\ntype Files = Array<{ data: string; mediaType: string; filename?: string }>;\n\nconst MAP_FIELDS = [\n 'activeTools',\n 'toolInputBuffers',\n 'pendingSuspensions',\n 'pendingApprovals',\n 'activeSubagents',\n 'modifiedFiles',\n] as const;\n\n/** The wire flattens the display state's Maps to objects; restore them. */\nfunction hydrateDisplayState(wire: Record<string, unknown>): AgentControllerDisplayState {\n const state: Record<string, unknown> = { ...defaultDisplayState(), ...wire };\n for (const field of MAP_FIELDS) {\n const value = wire[field];\n state[field] = value instanceof Map ? value : new Map(Object.entries((value as object | undefined) ?? {}));\n }\n return state as unknown as AgentControllerDisplayState;\n}\n\n/** The wire flattens errors to `{ name, message }`; restore them. */\nfunction hydrateEvent(event: ClientEvent): AgentControllerEvent {\n const out: Record<string, unknown> = { ...event };\n if (out.displayState && typeof out.displayState === 'object') {\n out.displayState = hydrateDisplayState(out.displayState as Record<string, unknown>);\n }\n const error = out.error as { name?: string; message?: string } | undefined;\n if (error && typeof error === 'object' && !(error instanceof Error)) {\n out.error = Object.assign(new Error(error.message ?? 'Remote error'), { name: error.name ?? 'Error' });\n }\n return out as unknown as AgentControllerEvent;\n}\n\nfunction toThread(\n thread: { id: string; title?: string; resourceId?: string; createdAt?: unknown; updatedAt?: unknown },\n resourceId: string,\n): AgentControllerThread {\n const date = (value: unknown) => (value ? new Date(value as string) : new Date());\n return {\n id: thread.id,\n resourceId: thread.resourceId ?? resourceId,\n title: thread.title ?? '',\n createdAt: date(thread.createdAt),\n updatedAt: date(thread.updatedAt),\n } as AgentControllerThread;\n}\n\n/**\n * Stands in for an in-process `Session` while the agent, tools, and storage\n * run inside a Factory. Reads are served from state mirrored off the session's\n * event stream; every action is forwarded to the Factory over HTTP.\n *\n * Only the surface the TUI drives is implemented. Anything else resolves to a\n * no-op so local-only features (thread locks, grants) degrade instead of crash.\n */\nexport class RemoteSession {\n readonly #remote: RemoteHandle;\n readonly #resourceId: string;\n readonly #listeners = new Set<AgentControllerEventListener>();\n #unsubscribe: (() => void) | undefined;\n #threadId: string | null = null;\n #modeId = '';\n #modelId = '';\n #running = false;\n #state: Record<string, unknown> = {};\n #displayState: AgentControllerDisplayState = defaultDisplayState();\n #modes: RemoteMode[] = [];\n #permissionRules: PermissionRules = { categories: {}, tools: {} };\n #pendingUserIds: string[] = [];\n readonly #userIdMap = new Map<string, string>();\n\n constructor(address: FactoryAddress, resourceId: string, remote?: RemoteHandle) {\n this.#resourceId = resourceId;\n this.#remote = remote ?? createFactoryClient(address).getAgentController(FACTORY_CONTROLLER_ID).session(resourceId);\n }\n\n /** Create/resume the remote session, hydrate the mirror, and open the event stream. */\n async connect(): Promise<void> {\n await this.#remote.create();\n await Promise.all([this.#hydrate(), this.#refreshPermissions()]);\n const subscription = await this.#remote.subscribe({\n reconnect: true,\n onEvent: event => this.#receive(hydrateEvent(event)),\n // The server does not replay missed events; resync after a drop.\n onReconnect: () => void this.#hydrate().catch(() => {}),\n });\n this.#unsubscribe = subscription.unsubscribe;\n }\n\n /** Modes come from the controller; used to resolve the current mode's name and color. */\n setModes(modes: RemoteMode[]): void {\n this.#modes = modes;\n }\n\n disconnect(): void {\n this.#unsubscribe?.();\n this.#unsubscribe = undefined;\n }\n\n async #hydrate(): Promise<void> {\n const snapshot = await this.#remote.state();\n this.#threadId = snapshot.threadId ?? null;\n this.#modeId = snapshot.modeId;\n this.#modelId = snapshot.modelId;\n this.#running = Boolean(snapshot.running);\n // Settings (yolo, thinkingLevel, notifications, smartEditing) are session\n // state; the TUI reads them through state.get().\n if (snapshot.settings) this.#state = { ...this.#state, ...snapshot.settings };\n this.#displayState = {\n ...this.#displayState,\n ...(snapshot.tasks ? { tasks: snapshot.tasks } : {}),\n ...(snapshot.tokenUsage ? { tokenUsage: snapshot.tokenUsage } : {}),\n } as AgentControllerDisplayState;\n }\n\n async #refreshPermissions(): Promise<void> {\n const rules = await this.#remote.getPermissions();\n this.#permissionRules = {\n categories: (rules.categories ?? {}) as PermissionRules['categories'],\n tools: rules.tools ?? {},\n };\n }\n\n /** Surface a failed remote command in the transcript instead of dropping it. */\n #emitError(message: string): void {\n this.#receive({ type: 'error', error: new Error(message) } as AgentControllerEvent);\n }\n\n /**\n * The TUI renders a user message as soon as it is sent, keyed by the id\n * `sendSignal` returned. The Factory echoes it back under its own id, so map\n * that echo onto the local id to keep it from rendering twice.\n */\n #adoptLocalUserId(e: Record<string, any>): void {\n const message = e.message as\n | { id?: string; role?: string; content?: { metadata?: { signal?: { type?: string } } } }\n | undefined;\n const isUser =\n message?.role === 'user' || (message?.role === 'signal' && message.content?.metadata?.signal?.type === 'user');\n if (!message?.id || !isUser) return;\n let localId = this.#userIdMap.get(message.id);\n if (!localId && e.type === 'message_start') {\n localId = this.#pendingUserIds.shift();\n if (localId) this.#userIdMap.set(message.id, localId);\n }\n if (localId) e.message = { ...message, id: localId };\n }\n\n #receive(event: AgentControllerEvent): void {\n const e = event as AgentControllerEvent & Record<string, any>;\n if (e.type === 'message_start' || e.type === 'message_update' || e.type === 'message_end') {\n this.#adoptLocalUserId(e);\n }\n switch (e.type) {\n case 'mode_changed':\n this.#modeId = e.modeId;\n break;\n case 'model_changed':\n this.#modelId = e.modelId;\n break;\n case 'thread_changed':\n this.#threadId = e.threadId;\n break;\n case 'thread_created':\n this.#threadId = e.thread.id;\n break;\n case 'state_changed':\n this.#state = { ...this.#state, ...e.state };\n break;\n case 'display_state_changed':\n this.#displayState = e.displayState;\n break;\n case 'agent_start':\n this.#running = true;\n break;\n case 'agent_end':\n this.#running = false;\n break;\n }\n for (const listener of this.#listeners) {\n void Promise.resolve(listener(event)).catch(() => {});\n }\n }\n\n subscribe(listener: AgentControllerEventListener): () => void {\n this.#listeners.add(listener);\n return () => this.#listeners.delete(listener);\n }\n\n onBeforeAgentEnd(): () => void {\n return () => {};\n }\n\n readonly identity = {\n getResourceId: () => this.#resourceId,\n getDefaultResourceId: () => this.#resourceId,\n getId: () => this.#resourceId,\n getOwnerId: () => this.#resourceId,\n setResourceId: () => {},\n };\n\n readonly thread = (() => {\n const remote = () => this.#remote;\n const resourceId = this.#resourceId;\n return {\n getId: () => this.#threadId,\n isSet: () => this.#threadId !== null,\n requireId: () => {\n if (!this.#threadId) throw new Error('No active thread');\n return this.#threadId;\n },\n ensureId: async () => {\n if (!this.#threadId) await this.thread.create();\n return this.#threadId!;\n },\n list: async () => (await remote().listThreads()).map(t => toThread(t, resourceId)),\n getById: async ({ threadId }: { threadId: string }) =>\n (await this.thread.list()).find(t => t.id === threadId) ?? null,\n create: async ({ title }: { title?: string } = {}) => {\n const created = await remote().createThread(title);\n const thread = toThread(created as any, resourceId);\n this.#threadId = thread.id;\n return thread;\n },\n switch: async ({ threadId }: { threadId: string }) => {\n await remote().switchThread(threadId);\n this.#threadId = threadId;\n },\n clone: async ({ sourceThreadId, title }: { sourceThreadId?: string; title?: string } = {}) => {\n const created = await remote().cloneThread({ sourceThreadId, title });\n const thread = toThread(created as any, resourceId);\n this.#threadId = thread.id;\n return thread;\n },\n rename: async ({ title }: { title: string }) => {\n if (this.#threadId) await remote().renameThread(this.#threadId, title);\n },\n delete: async ({ threadId }: { threadId: string }) => {\n await remote().deleteThread(threadId);\n if (this.#threadId === threadId) this.#threadId = null;\n },\n listMessages: ({ threadId, limit }: { threadId: string; limit?: number }) =>\n remote().listMessages(threadId, limit),\n listActiveMessages: async ({ limit }: { limit?: number } = {}) =>\n this.#threadId ? remote().listMessages(this.#threadId, limit) : [],\n firstUserMessage: async () => null,\n firstUserMessages: async () => new Map(),\n // Thread settings live in the Factory; local overrides are not persisted.\n getSetting: async () => undefined,\n getSettingOn: async () => undefined,\n setSetting: async () => {},\n setSettingOn: async () => {},\n deleteSetting: async () => {},\n cleanupSubscription: () => {},\n detachFromCurrent: () => {},\n clear: () => {\n this.#threadId = null;\n },\n clearAndReleaseLock: async () => {},\n cloneToCurrentResource: async ({ sourceThreadId }: { sourceThreadId: string }) =>\n this.thread.clone({ sourceThreadId }),\n loadMetadata: async () => {},\n };\n })();\n\n readonly mode = {\n get: () => this.#modeId,\n hasSelection: () => Boolean(this.#modeId),\n resolve: () => this.#modes.find(mode => mode.id === this.#modeId) ?? { id: this.#modeId, name: this.#modeId },\n set: ({ modeId }: { modeId: string }) => {\n this.#modeId = modeId;\n },\n switch: async ({ modeId }: { modeId: string }) => {\n await this.#remote.switchMode(modeId);\n this.#modeId = modeId;\n },\n };\n\n readonly model = {\n get: () => this.#modelId,\n hasSelection: () => Boolean(this.#modelId),\n displayName: () => this.#modelId,\n set: ({ modelId }: { modelId: string }) => {\n this.#modelId = modelId;\n },\n switch: async (modelId: string, options: { thinkingLevel?: any } = {}) => {\n await this.#remote.switchModel(modelId, options);\n this.#modelId = modelId;\n },\n syncFromPersisted: async () => {},\n };\n\n readonly state = {\n get: () => this.#state,\n set: async (updates: Record<string, unknown>) => {\n this.#state = { ...this.#state, ...updates };\n await this.#remote.setState(updates);\n },\n setIf: async (updates: Record<string, unknown>, shouldApply: () => boolean) => {\n if (!shouldApply()) return false;\n await this.state.set(updates);\n return true;\n },\n update: async <TResult>(updater: AgentControllerRequestStateUpdater<any, TResult>): Promise<TResult> => {\n const { updates, result } = await updater(this.#state);\n if (updates) await this.state.set(updates);\n return result;\n },\n };\n\n readonly displayState = {\n get: () => this.#displayState,\n clearModifiedFiles: () => {\n this.#displayState = { ...this.#displayState, modifiedFiles: new Map() };\n },\n restoreTasks: () => {},\n clearPendingSuspensions: () => {},\n deletePendingSuspension: () => {},\n clearPendingApprovals: () => {},\n resetThread: () => {},\n apply: () => {},\n };\n\n readonly run = {\n isRunning: () => this.#running,\n getRunId: () => null,\n getTraceId: () => null,\n };\n\n readonly stream = { isActive: () => this.#running, isOpen: () => this.#running };\n\n readonly suspensions = {\n hasPending: () => this.#displayState.pendingSuspensions.size > 0,\n get: ({ toolCallId }: { toolCallId: string }) => this.#displayState.pendingSuspensions.get(toolCallId),\n has: ({ toolCallId }: { toolCallId: string }) => this.#displayState.pendingSuspensions.has(toolCallId),\n delete: () => {},\n };\n\n readonly permissions = {\n getRules: () => this.#permissionRules,\n setForCategory: async ({ category, policy }: { category: any; policy: any }) => {\n await this.#remote.setPermissionForCategory(category, policy);\n await this.#refreshPermissions();\n },\n setForTool: async ({ toolName, policy }: { toolName: string; policy: any }) => {\n await this.#remote.setPermissionForTool(toolName, policy);\n await this.#refreshPermissions();\n },\n };\n\n getCurrentRunId(): string | null {\n return null;\n }\n\n getTags(): Record<string, string> {\n return {};\n }\n\n getTokenUsage() {\n return this.#displayState.tokenUsage;\n }\n\n getGrants() {\n return { categories: [], tools: [] };\n }\n\n async sendMessage({ content, files }: { content: string; files?: Files }) {\n await this.#remote.sendMessage({ content, files });\n }\n\n async queueMessage({ content }: { content: string }) {\n await this.#remote.followUp(content);\n }\n\n sendSignal(input: Parameters<Session['sendSignal']>[0]) {\n const content: unknown = 'content' in input ? input.content : undefined;\n const text =\n typeof content === 'string'\n ? content\n : Array.isArray(content)\n ? content.map(part => (typeof part === 'string' ? part : ((part as { text?: string })?.text ?? ''))).join('')\n : JSON.stringify(content);\n const files = 'files' in input ? (input.files as Files | undefined) : undefined;\n const message = files?.length ? { content: text, files } : text;\n const id = randomUUID();\n this.#pendingUserIds.push(id);\n const sent = this.#running ? this.#remote.followUp(message) : this.#remote.sendMessage(message);\n sent.catch(() => {\n this.#pendingUserIds = this.#pendingUserIds.filter(pending => pending !== id);\n });\n const accepted = sent.then(() => ({ accepted: true as const }));\n accepted.catch(() => {});\n return { id, type: 'user' as const, accepted, persisted: sent };\n }\n\n async steer({ content }: { content: string }) {\n await this.#remote.steer(content);\n }\n\n async followUp({ content }: { content: string }) {\n await this.#remote.followUp(content);\n }\n\n respondToToolApproval({\n decision,\n toolCallId,\n }: {\n decision: 'approve' | 'decline' | 'always_allow_category';\n toolCallId: string;\n }) {\n void (async () => {\n const ack = await this.#remote.approveTool(toolCallId, decision !== 'decline', {\n alwaysAllowCategory: decision === 'always_allow_category',\n });\n if (ack && 'accepted' in ack && ack.accepted === false) {\n this.#emitError(`Tool approval was not applied: ${(ack as { reason?: string }).reason ?? 'rejected'}`);\n }\n })().catch(error =>\n this.#emitError(`Tool approval failed: ${error instanceof Error ? error.message : String(error)}`),\n );\n return { accepted: true as const };\n }\n\n async respondToToolSuspension({ resumeData, toolCallId }: { resumeData: unknown; toolCallId?: string }) {\n const id = toolCallId ?? this.#displayState.pendingSuspensions.keys().next().value;\n if (!id) throw new Error('No pending tool suspension to respond to');\n await this.#remote.respondToToolSuspension(\n id,\n resumeData as Parameters<RemoteHandle['respondToToolSuspension']>[1],\n );\n }\n\n claimToolResponse(): boolean {\n return true;\n }\n\n releaseToolResponse(): void {}\n\n abort(): void {\n void this.#remote\n .abort()\n .catch(error => this.#emitError(`Abort failed: ${error instanceof Error ? error.message : String(error)}`));\n }\n\n abortRun(): void {\n this.abort();\n }\n}\n"],"mappings":";;;;AAqBA,MAAM,aAAa;CACjB;CACA;CACA;CACA;CACA;CACA;AACF;;AAGA,SAAS,oBAAoB,MAA4D;CACvF,MAAM,QAAiC;EAAE,GAAG,oBAAoB;EAAG,GAAG;CAAK;CAC3E,KAAK,MAAM,SAAS,YAAY;EAC9B,MAAM,QAAQ,KAAK;EACnB,MAAM,SAAS,iBAAiB,MAAM,QAAQ,IAAI,IAAI,OAAO,QAAS,SAAgC,CAAC,CAAC,CAAC;CAC3G;CACA,OAAO;AACT;;AAGA,SAAS,aAAa,OAA0C;CAC9D,MAAM,MAA+B,EAAE,GAAG,MAAM;CAChD,IAAI,IAAI,gBAAgB,OAAO,IAAI,iBAAiB,UAClD,IAAI,eAAe,oBAAoB,IAAI,YAAuC;CAEpF,MAAM,QAAQ,IAAI;CAClB,IAAI,SAAS,OAAO,UAAU,YAAY,EAAE,iBAAiB,QAC3D,IAAI,QAAQ,OAAO,OAAO,IAAI,MAAM,MAAM,WAAW,cAAc,GAAG,EAAE,MAAM,MAAM,QAAQ,QAAQ,CAAC;CAEvG,OAAO;AACT;AAEA,SAAS,SACP,QACA,YACuB;CACvB,MAAM,QAAQ,UAAoB,QAAQ,IAAI,KAAK,KAAe,oBAAI,IAAI,KAAK;CAC/E,OAAO;EACL,IAAI,OAAO;EACX,YAAY,OAAO,cAAc;EACjC,OAAO,OAAO,SAAS;EACvB,WAAW,KAAK,OAAO,SAAS;EAChC,WAAW,KAAK,OAAO,SAAS;CAClC;AACF;;;;;;;;;AAUA,IAAa,gBAAb,MAA2B;CACzB;CACA;CACA,6BAAsB,IAAI,IAAkC;CAC5D;CACA,YAA2B;CAC3B,UAAU;CACV,WAAW;CACX,WAAW;CACX,SAAkC,CAAC;CACnC,gBAA6C,oBAAoB;CACjE,SAAuB,CAAC;CACxB,mBAAoC;EAAE,YAAY,CAAC;EAAG,OAAO,CAAC;CAAE;CAChE,kBAA4B,CAAC;CAC7B,6BAAsB,IAAI,IAAoB;CAE9C,YAAY,SAAyB,YAAoB,QAAuB;EAC9E,KAAKC,cAAc;EACnB,KAAKD,UAAU,UAAU,oBAAoB,OAAO,CAAC,CAAC,mBAAA,MAAwC,CAAC,CAAC,QAAQ,UAAU;CACpH;;CAGA,MAAM,UAAyB;EAC7B,MAAM,KAAKA,QAAQ,OAAO;EAC1B,MAAM,QAAQ,IAAI,CAAC,KAAKI,SAAS,GAAG,KAAKC,oBAAoB,CAAC,CAAC;EAC/D,MAAM,eAAe,MAAM,KAAKL,QAAQ,UAAU;GAChD,WAAW;GACX,UAAS,UAAS,KAAKM,SAAS,aAAa,KAAK,CAAC;GAEnD,mBAAmB,KAAK,KAAKF,SAAS,CAAC,CAAC,YAAY,CAAC,CAAC;EACxD,CAAC;EACD,KAAKG,eAAe,aAAa;CACnC;;CAGA,SAAS,OAA2B;EAClC,KAAKC,SAAS;CAChB;CAEA,aAAmB;EACjB,KAAKD,eAAe;EACpB,KAAKA,eAAe,KAAA;CACtB;CAEA,MAAMH,WAA0B;EAC9B,MAAM,WAAW,MAAM,KAAKJ,QAAQ,MAAM;EAC1C,KAAKS,YAAY,SAAS,YAAY;EACtC,KAAKC,UAAU,SAAS;EACxB,KAAKC,WAAW,SAAS;EACzB,KAAKC,WAAW,QAAQ,SAAS,OAAO;EAGxC,IAAI,SAAS,UAAU,KAAKC,SAAS;GAAE,GAAG,KAAKA;GAAQ,GAAG,SAAS;EAAS;EAC5E,KAAKC,gBAAgB;GACnB,GAAG,KAAKA;GACR,GAAI,SAAS,QAAQ,EAAE,OAAO,SAAS,MAAM,IAAI,CAAC;GAClD,GAAI,SAAS,aAAa,EAAE,YAAY,SAAS,WAAW,IAAI,CAAC;EACnE;CACF;CAEA,MAAMT,sBAAqC;EACzC,MAAM,QAAQ,MAAM,KAAKL,QAAQ,eAAe;EAChD,KAAKe,mBAAmB;GACtB,YAAa,MAAM,cAAc,CAAC;GAClC,OAAO,MAAM,SAAS,CAAC;EACzB;CACF;;CAGA,WAAW,SAAuB;EAChC,KAAKT,SAAS;GAAE,MAAM;GAAS,OAAO,IAAI,MAAM,OAAO;EAAE,CAAyB;CACpF;;;;;;CAOA,kBAAkB,GAA8B;EAC9C,MAAM,UAAU,EAAE;EAGlB,MAAM,SACJ,SAAS,SAAS,UAAW,SAAS,SAAS,YAAY,QAAQ,SAAS,UAAU,QAAQ,SAAS;EACzG,IAAI,CAAC,SAAS,MAAM,CAAC,QAAQ;EAC7B,IAAI,UAAU,KAAKH,WAAW,IAAI,QAAQ,EAAE;EAC5C,IAAI,CAAC,WAAW,EAAE,SAAS,iBAAiB;GAC1C,UAAU,KAAKa,gBAAgB,MAAM;GACrC,IAAI,SAAS,KAAKb,WAAW,IAAI,QAAQ,IAAI,OAAO;EACtD;EACA,IAAI,SAAS,EAAE,UAAU;GAAE,GAAG;GAAS,IAAI;EAAQ;CACrD;CAEA,SAAS,OAAmC;EAC1C,MAAM,IAAI;EACV,IAAI,EAAE,SAAS,mBAAmB,EAAE,SAAS,oBAAoB,EAAE,SAAS,eAC1E,KAAKc,kBAAkB,CAAC;EAE1B,QAAQ,EAAE,MAAV;GACE,KAAK;IACH,KAAKP,UAAU,EAAE;IACjB;GACF,KAAK;IACH,KAAKC,WAAW,EAAE;IAClB;GACF,KAAK;IACH,KAAKF,YAAY,EAAE;IACnB;GACF,KAAK;IACH,KAAKA,YAAY,EAAE,OAAO;IAC1B;GACF,KAAK;IACH,KAAKI,SAAS;KAAE,GAAG,KAAKA;KAAQ,GAAG,EAAE;IAAM;IAC3C;GACF,KAAK;IACH,KAAKC,gBAAgB,EAAE;IACvB;GACF,KAAK;IACH,KAAKF,WAAW;IAChB;GACF,KAAK;IACH,KAAKA,WAAW;IAChB;EACJ;EACA,KAAK,MAAM,YAAY,KAAKV,YAC1B,QAAa,QAAQ,SAAS,KAAK,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC;CAExD;CAEA,UAAU,UAAoD;EAC5D,KAAKA,WAAW,IAAI,QAAQ;EAC5B,aAAa,KAAKA,WAAW,OAAO,QAAQ;CAC9C;CAEA,mBAA+B;EAC7B,aAAa,CAAC;CAChB;CAEA,WAAoB;EAClB,qBAAqB,KAAKD;EAC1B,4BAA4B,KAAKA;EACjC,aAAa,KAAKA;EAClB,kBAAkB,KAAKA;EACvB,qBAAqB,CAAC;CACxB;CAEA,gBAAyB;EACvB,MAAM,eAAe,KAAKD;EAC1B,MAAM,aAAa,KAAKC;EACxB,OAAO;GACL,aAAa,KAAKQ;GAClB,aAAa,KAAKA,cAAc;GAChC,iBAAiB;IACf,IAAI,CAAC,KAAKA,WAAW,MAAM,IAAI,MAAM,kBAAkB;IACvD,OAAO,KAAKA;GACd;GACA,UAAU,YAAY;IACpB,IAAI,CAAC,KAAKA,WAAW,MAAM,KAAK,OAAO,OAAO;IAC9C,OAAO,KAAKA;GACd;GACA,MAAM,aAAa,MAAM,OAAO,CAAC,CAAC,YAAY,EAAA,CAAG,KAAI,MAAK,SAAS,GAAG,UAAU,CAAC;GACjF,SAAS,OAAO,EAAE,gBACf,MAAM,KAAK,OAAO,KAAK,EAAA,CAAG,MAAK,MAAK,EAAE,OAAO,QAAQ,KAAK;GAC7D,QAAQ,OAAO,EAAE,UAA8B,CAAC,MAAM;IAEpD,MAAM,SAAS,SAAS,MADF,OAAO,CAAC,CAAC,aAAa,KAAK,GACT,UAAU;IAClD,KAAKA,YAAY,OAAO;IACxB,OAAO;GACT;GACA,QAAQ,OAAO,EAAE,eAAqC;IACpD,MAAM,OAAO,CAAC,CAAC,aAAa,QAAQ;IACpC,KAAKA,YAAY;GACnB;GACA,OAAO,OAAO,EAAE,gBAAgB,UAAuD,CAAC,MAAM;IAE5F,MAAM,SAAS,SAAS,MADF,OAAO,CAAC,CAAC,YAAY;KAAE;KAAgB;IAAM,CAAC,GAC5B,UAAU;IAClD,KAAKA,YAAY,OAAO;IACxB,OAAO;GACT;GACA,QAAQ,OAAO,EAAE,YAA+B;IAC9C,IAAI,KAAKA,WAAW,MAAM,OAAO,CAAC,CAAC,aAAa,KAAKA,WAAW,KAAK;GACvE;GACA,QAAQ,OAAO,EAAE,eAAqC;IACpD,MAAM,OAAO,CAAC,CAAC,aAAa,QAAQ;IACpC,IAAI,KAAKA,cAAc,UAAU,KAAKA,YAAY;GACpD;GACA,eAAe,EAAE,UAAU,YACzB,OAAO,CAAC,CAAC,aAAa,UAAU,KAAK;GACvC,oBAAoB,OAAO,EAAE,UAA8B,CAAC,MAC1D,KAAKA,YAAY,OAAO,CAAC,CAAC,aAAa,KAAKA,WAAW,KAAK,IAAI,CAAC;GACnE,kBAAkB,YAAY;GAC9B,mBAAmB,4BAAY,IAAI,IAAI;GAEvC,YAAY,YAAY,KAAA;GACxB,cAAc,YAAY,KAAA;GAC1B,YAAY,YAAY,CAAC;GACzB,cAAc,YAAY,CAAC;GAC3B,eAAe,YAAY,CAAC;GAC5B,2BAA2B,CAAC;GAC5B,yBAAyB,CAAC;GAC1B,aAAa;IACX,KAAKA,YAAY;GACnB;GACA,qBAAqB,YAAY,CAAC;GAClC,wBAAwB,OAAO,EAAE,qBAC/B,KAAK,OAAO,MAAM,EAAE,eAAe,CAAC;GACtC,cAAc,YAAY,CAAC;EAC7B;CACF,EAAA,CAAG;CAEH,OAAgB;EACd,WAAW,KAAKC;EAChB,oBAAoB,QAAQ,KAAKA,OAAO;EACxC,eAAe,KAAKF,OAAO,MAAK,SAAQ,KAAK,OAAO,KAAKE,OAAO,KAAK;GAAE,IAAI,KAAKA;GAAS,MAAM,KAAKA;EAAQ;EAC5G,MAAM,EAAE,aAAiC;GACvC,KAAKA,UAAU;EACjB;EACA,QAAQ,OAAO,EAAE,aAAiC;GAChD,MAAM,KAAKV,QAAQ,WAAW,MAAM;GACpC,KAAKU,UAAU;EACjB;CACF;CAEA,QAAiB;EACf,WAAW,KAAKC;EAChB,oBAAoB,QAAQ,KAAKA,QAAQ;EACzC,mBAAmB,KAAKA;EACxB,MAAM,EAAE,cAAmC;GACzC,KAAKA,WAAW;EAClB;EACA,QAAQ,OAAO,SAAiB,UAAmC,CAAC,MAAM;GACxE,MAAM,KAAKX,QAAQ,YAAY,SAAS,OAAO;GAC/C,KAAKW,WAAW;EAClB;EACA,mBAAmB,YAAY,CAAC;CAClC;CAEA,QAAiB;EACf,WAAW,KAAKE;EAChB,KAAK,OAAO,YAAqC;GAC/C,KAAKA,SAAS;IAAE,GAAG,KAAKA;IAAQ,GAAG;GAAQ;GAC3C,MAAM,KAAKb,QAAQ,SAAS,OAAO;EACrC;EACA,OAAO,OAAO,SAAkC,gBAA+B;GAC7E,IAAI,CAAC,YAAY,GAAG,OAAO;GAC3B,MAAM,KAAK,MAAM,IAAI,OAAO;GAC5B,OAAO;EACT;EACA,QAAQ,OAAgB,YAAgF;GACtG,MAAM,EAAE,SAAS,WAAW,MAAM,QAAQ,KAAKa,MAAM;GACrD,IAAI,SAAS,MAAM,KAAK,MAAM,IAAI,OAAO;GACzC,OAAO;EACT;CACF;CAEA,eAAwB;EACtB,WAAW,KAAKC;EAChB,0BAA0B;GACxB,KAAKA,gBAAgB;IAAE,GAAG,KAAKA;IAAe,+BAAe,IAAI,IAAI;GAAE;EACzE;EACA,oBAAoB,CAAC;EACrB,+BAA+B,CAAC;EAChC,+BAA+B,CAAC;EAChC,6BAA6B,CAAC;EAC9B,mBAAmB,CAAC;EACpB,aAAa,CAAC;CAChB;CAEA,MAAe;EACb,iBAAiB,KAAKF;EACtB,gBAAgB;EAChB,kBAAkB;CACpB;CAEA,SAAkB;EAAE,gBAAgB,KAAKA;EAAU,cAAc,KAAKA;CAAS;CAE/E,cAAuB;EACrB,kBAAkB,KAAKE,cAAc,mBAAmB,OAAO;EAC/D,MAAM,EAAE,iBAAyC,KAAKA,cAAc,mBAAmB,IAAI,UAAU;EACrG,MAAM,EAAE,iBAAyC,KAAKA,cAAc,mBAAmB,IAAI,UAAU;EACrG,cAAc,CAAC;CACjB;CAEA,cAAuB;EACrB,gBAAgB,KAAKC;EACrB,gBAAgB,OAAO,EAAE,UAAU,aAA6C;GAC9E,MAAM,KAAKf,QAAQ,yBAAyB,UAAU,MAAM;GAC5D,MAAM,KAAKK,oBAAoB;EACjC;EACA,YAAY,OAAO,EAAE,UAAU,aAAgD;GAC7E,MAAM,KAAKL,QAAQ,qBAAqB,UAAU,MAAM;GACxD,MAAM,KAAKK,oBAAoB;EACjC;CACF;CAEA,kBAAiC;EAC/B,OAAO;CACT;CAEA,UAAkC;EAChC,OAAO,CAAC;CACV;CAEA,gBAAgB;EACd,OAAO,KAAKS,cAAc;CAC5B;CAEA,YAAY;EACV,OAAO;GAAE,YAAY,CAAC;GAAG,OAAO,CAAC;EAAE;CACrC;CAEA,MAAM,YAAY,EAAE,SAAS,SAA6C;EACxE,MAAM,KAAKd,QAAQ,YAAY;GAAE;GAAS;EAAM,CAAC;CACnD;CAEA,MAAM,aAAa,EAAE,WAAgC;EACnD,MAAM,KAAKA,QAAQ,SAAS,OAAO;CACrC;CAEA,WAAW,OAA6C;EACtD,MAAM,UAAmB,aAAa,QAAQ,MAAM,UAAU,KAAA;EAC9D,MAAM,OACJ,OAAO,YAAY,WACf,UACA,MAAM,QAAQ,OAAO,IACnB,QAAQ,KAAI,SAAS,OAAO,SAAS,WAAW,OAAS,MAA4B,QAAQ,EAAI,CAAC,CAAC,KAAK,EAAE,IAC1G,KAAK,UAAU,OAAO;EAC9B,MAAM,QAAQ,WAAW,QAAS,MAAM,QAA8B,KAAA;EACtE,MAAM,UAAU,OAAO,SAAS;GAAE,SAAS;GAAM;EAAM,IAAI;EAC3D,MAAM,KAAK,WAAW;EACtB,KAAKgB,gBAAgB,KAAK,EAAE;EAC5B,MAAM,OAAO,KAAKJ,WAAW,KAAKZ,QAAQ,SAAS,OAAO,IAAI,KAAKA,QAAQ,YAAY,OAAO;EAC9F,KAAK,YAAY;GACf,KAAKgB,kBAAkB,KAAKA,gBAAgB,QAAO,YAAW,YAAY,EAAE;EAC9E,CAAC;EACD,MAAM,WAAW,KAAK,YAAY,EAAE,UAAU,KAAc,EAAE;EAC9D,SAAS,YAAY,CAAC,CAAC;EACvB,OAAO;GAAE;GAAI,MAAM;GAAiB;GAAU,WAAW;EAAK;CAChE;CAEA,MAAM,MAAM,EAAE,WAAgC;EAC5C,MAAM,KAAKhB,QAAQ,MAAM,OAAO;CAClC;CAEA,MAAM,SAAS,EAAE,WAAgC;EAC/C,MAAM,KAAKA,QAAQ,SAAS,OAAO;CACrC;CAEA,sBAAsB,EACpB,UACA,cAIC;EACD,CAAM,YAAY;GAChB,MAAM,MAAM,MAAM,KAAKA,QAAQ,YAAY,YAAY,aAAa,WAAW,EAC7E,qBAAqB,aAAa,wBACpC,CAAC;GACD,IAAI,OAAO,cAAc,OAAO,IAAI,aAAa,OAC/C,KAAKkB,WAAW,kCAAmC,IAA4B,UAAU,YAAY;EAEzG,EAAA,CAAG,CAAC,CAAC,OAAM,UACT,KAAKA,WAAW,yBAAyB,iBAAiB,QAAQ,MAAM,UAAU,OAAO,KAAK,GAAG,CACnG;EACA,OAAO,EAAE,UAAU,KAAc;CACnC;CAEA,MAAM,wBAAwB,EAAE,YAAY,cAA4D;EACtG,MAAM,KAAK,cAAc,KAAKJ,cAAc,mBAAmB,KAAK,CAAC,CAAC,KAAK,CAAC,CAAC;EAC7E,IAAI,CAAC,IAAI,MAAM,IAAI,MAAM,0CAA0C;EACnE,MAAM,KAAKd,QAAQ,wBACjB,IACA,UACF;CACF;CAEA,oBAA6B;EAC3B,OAAO;CACT;CAEA,sBAA4B,CAAC;CAE7B,QAAc;EACZ,KAAUA,QACP,MAAM,CAAC,CACP,OAAM,UAAS,KAAKkB,WAAW,iBAAiB,iBAAiB,QAAQ,MAAM,UAAU,OAAO,KAAK,GAAG,CAAC;CAC9G;CAEA,WAAiB;EACf,KAAK,MAAM;CACb;AACF"}
|
|
1
|
+
{"version":3,"file":"remote-session.js","names":[],"sources":["../../src/factory/remote-session.ts"],"sourcesContent":["import { randomUUID } from 'node:crypto';\n\nimport type { AgentControllerEvent as ClientEvent, AgentControllerModeInfo } from '@mastra/client-js';\nimport type {\n AgentControllerDisplayState,\n AgentControllerEvent,\n AgentControllerEventListener,\n AgentControllerRequestStateUpdater,\n AgentControllerThread,\n PermissionRules,\n Session,\n} from '@mastra/core/agent-controller';\nimport { defaultDisplayState } from '@mastra/core/agent-controller';\n\nimport type { FactoryAddress } from './client.js';\nimport { createFactoryClient, FACTORY_CONTROLLER_ID } from './client.js';\n\ntype RemoteHandle = ReturnType<ReturnType<ReturnType<typeof createFactoryClient>['getAgentController']>['session']>;\ntype RemoteMode = AgentControllerModeInfo;\ntype Files = Array<{ data: string; mediaType: string; filename?: string }>;\n\nconst MAP_FIELDS = [\n 'activeTools',\n 'toolInputBuffers',\n 'pendingSuspensions',\n 'pendingApprovals',\n 'activeSubagents',\n 'modifiedFiles',\n] as const;\n\n/** The wire flattens the display state's Maps to objects; restore them. */\nfunction hydrateDisplayState(wire: Record<string, unknown>): AgentControllerDisplayState {\n const state: Record<string, unknown> = { ...defaultDisplayState(), ...wire };\n for (const field of MAP_FIELDS) {\n const value = wire[field];\n state[field] = value instanceof Map ? value : new Map(Object.entries((value as object | undefined) ?? {}));\n }\n return state as unknown as AgentControllerDisplayState;\n}\n\n/** The wire flattens errors to `{ name, message }`; restore them. */\nfunction hydrateEvent(event: ClientEvent): AgentControllerEvent {\n const out: Record<string, unknown> = { ...event };\n if (out.displayState && typeof out.displayState === 'object') {\n out.displayState = hydrateDisplayState(out.displayState as Record<string, unknown>);\n }\n const error = out.error as { name?: string; message?: string } | undefined;\n if (error && typeof error === 'object' && !(error instanceof Error)) {\n out.error = Object.assign(new Error(error.message ?? 'Remote error'), { name: error.name ?? 'Error' });\n }\n return out as unknown as AgentControllerEvent;\n}\n\nfunction toThread(\n thread: { id: string; title?: string; resourceId?: string; createdAt?: unknown; updatedAt?: unknown },\n resourceId: string,\n): AgentControllerThread {\n const date = (value: unknown) => (value ? new Date(value as string) : new Date());\n return {\n id: thread.id,\n resourceId: thread.resourceId ?? resourceId,\n title: thread.title ?? '',\n createdAt: date(thread.createdAt),\n updatedAt: date(thread.updatedAt),\n } as AgentControllerThread;\n}\n\n/**\n * Stands in for an in-process `Session` while the agent, tools, and storage\n * run inside a Factory. Reads are served from state mirrored off the session's\n * event stream; every action is forwarded to the Factory over HTTP.\n *\n * Only the surface the TUI drives is implemented. Anything else resolves to a\n * no-op so local-only features (thread locks, grants) degrade instead of crash.\n */\nexport class RemoteSession {\n readonly #remote: RemoteHandle;\n readonly #sessionId: string;\n /** Whose threads and memory the session works on; the Factory reports it when the session is created. */\n #resourceId: string;\n readonly #listeners = new Set<AgentControllerEventListener>();\n #unsubscribe: (() => void) | undefined;\n #threadId: string | null = null;\n #modeId = '';\n #modelId = '';\n #running = false;\n #state: Record<string, unknown> = {};\n #displayState: AgentControllerDisplayState = defaultDisplayState();\n #modes: RemoteMode[] = [];\n #permissionRules: PermissionRules = { categories: {}, tools: {} };\n #pendingUserIds: string[] = [];\n readonly #userIdMap = new Map<string, string>();\n\n constructor(address: FactoryAddress, sessionId: string, remote?: RemoteHandle) {\n this.#sessionId = sessionId;\n this.#resourceId = sessionId;\n this.#remote = remote ?? createFactoryClient(address).getAgentController(FACTORY_CONTROLLER_ID).session(sessionId);\n }\n\n /** Create/resume the remote session, hydrate the mirror, and open the event stream. */\n async connect(): Promise<void> {\n const created = await this.#remote.create();\n if (created?.resourceId) this.#resourceId = created.resourceId;\n await Promise.all([this.#hydrate(), this.#refreshPermissions()]);\n const subscription = await this.#remote.subscribe({\n reconnect: true,\n onEvent: event => this.#receive(hydrateEvent(event)),\n // The server does not replay missed events; resync after a drop.\n onReconnect: () => void this.#hydrate().catch(() => {}),\n });\n this.#unsubscribe = subscription.unsubscribe;\n }\n\n /** Modes come from the controller; used to resolve the current mode's name and color. */\n setModes(modes: RemoteMode[]): void {\n this.#modes = modes;\n }\n\n disconnect(): void {\n this.#unsubscribe?.();\n this.#unsubscribe = undefined;\n }\n\n async #hydrate(): Promise<void> {\n const snapshot = await this.#remote.state();\n this.#threadId = snapshot.threadId ?? null;\n this.#modeId = snapshot.modeId;\n this.#modelId = snapshot.modelId;\n this.#running = Boolean(snapshot.running);\n // Settings (yolo, thinkingLevel, notifications, smartEditing) are session\n // state; the TUI reads them through state.get().\n if (snapshot.settings) this.#state = { ...this.#state, ...snapshot.settings };\n this.#displayState = {\n ...this.#displayState,\n ...(snapshot.tasks ? { tasks: snapshot.tasks } : {}),\n ...(snapshot.tokenUsage ? { tokenUsage: snapshot.tokenUsage } : {}),\n } as AgentControllerDisplayState;\n }\n\n async #refreshPermissions(): Promise<void> {\n const rules = await this.#remote.getPermissions();\n this.#permissionRules = {\n categories: (rules.categories ?? {}) as PermissionRules['categories'],\n tools: rules.tools ?? {},\n };\n }\n\n /** Surface a failed remote command in the transcript instead of dropping it. */\n #emitError(message: string): void {\n this.#receive({ type: 'error', error: new Error(message) } as AgentControllerEvent);\n }\n\n /**\n * The TUI renders a user message as soon as it is sent, keyed by the id\n * `sendSignal` returned. The Factory echoes it back under its own id, so map\n * that echo onto the local id to keep it from rendering twice.\n */\n #adoptLocalUserId(e: Record<string, any>): void {\n const message = e.message as\n | { id?: string; role?: string; content?: { metadata?: { signal?: { type?: string } } } }\n | undefined;\n const isUser =\n message?.role === 'user' || (message?.role === 'signal' && message.content?.metadata?.signal?.type === 'user');\n if (!message?.id || !isUser) return;\n let localId = this.#userIdMap.get(message.id);\n if (!localId && e.type === 'message_start') {\n localId = this.#pendingUserIds.shift();\n if (localId) this.#userIdMap.set(message.id, localId);\n }\n if (localId) e.message = { ...message, id: localId };\n }\n\n #receive(event: AgentControllerEvent): void {\n const e = event as AgentControllerEvent & Record<string, any>;\n if (e.type === 'message_start' || e.type === 'message_update' || e.type === 'message_end') {\n this.#adoptLocalUserId(e);\n }\n switch (e.type) {\n case 'mode_changed':\n this.#modeId = e.modeId;\n break;\n case 'model_changed':\n this.#modelId = e.modelId;\n break;\n case 'thread_changed':\n this.#threadId = e.threadId;\n break;\n case 'thread_created':\n this.#threadId = e.thread.id;\n break;\n case 'state_changed':\n this.#state = { ...this.#state, ...e.state };\n break;\n case 'display_state_changed':\n this.#displayState = e.displayState;\n break;\n case 'agent_start':\n this.#running = true;\n break;\n case 'agent_end':\n this.#running = false;\n break;\n }\n for (const listener of this.#listeners) {\n void Promise.resolve(listener(event)).catch(() => {});\n }\n }\n\n subscribe(listener: AgentControllerEventListener): () => void {\n this.#listeners.add(listener);\n return () => this.#listeners.delete(listener);\n }\n\n onBeforeAgentEnd(): () => void {\n return () => {};\n }\n\n readonly identity = {\n getResourceId: () => this.#resourceId,\n getDefaultResourceId: () => this.#resourceId,\n getId: () => this.#sessionId,\n getOwnerId: () => this.#sessionId,\n setResourceId: () => {},\n };\n\n readonly thread = (() => {\n const remote = () => this.#remote;\n const resourceId = () => this.#resourceId;\n return {\n getId: () => this.#threadId,\n isSet: () => this.#threadId !== null,\n requireId: () => {\n if (!this.#threadId) throw new Error('No active thread');\n return this.#threadId;\n },\n ensureId: async () => {\n if (!this.#threadId) await this.thread.create();\n return this.#threadId!;\n },\n list: async () => (await remote().listThreads()).map(t => toThread(t, resourceId())),\n getById: async ({ threadId }: { threadId: string }) =>\n (await this.thread.list()).find(t => t.id === threadId) ?? null,\n create: async ({ title }: { title?: string } = {}) => {\n const created = await remote().createThread(title);\n const thread = toThread(created as any, resourceId());\n this.#threadId = thread.id;\n return thread;\n },\n switch: async ({ threadId }: { threadId: string }) => {\n await remote().switchThread(threadId);\n this.#threadId = threadId;\n },\n clone: async ({ sourceThreadId, title }: { sourceThreadId?: string; title?: string } = {}) => {\n const created = await remote().cloneThread({ sourceThreadId, title });\n const thread = toThread(created as any, resourceId());\n this.#threadId = thread.id;\n return thread;\n },\n rename: async ({ title }: { title: string }) => {\n if (this.#threadId) await remote().renameThread(this.#threadId, title);\n },\n delete: async ({ threadId }: { threadId: string }) => {\n await remote().deleteThread(threadId);\n if (this.#threadId === threadId) this.#threadId = null;\n },\n listMessages: ({ threadId, limit }: { threadId: string; limit?: number }) =>\n remote().listMessages(threadId, limit),\n listActiveMessages: async ({ limit }: { limit?: number } = {}) =>\n this.#threadId ? remote().listMessages(this.#threadId, limit) : [],\n firstUserMessage: async () => null,\n firstUserMessages: async () => new Map(),\n // Thread settings live in the Factory; local overrides are not persisted.\n getSetting: async () => undefined,\n getSettingOn: async () => undefined,\n setSetting: async () => {},\n setSettingOn: async () => {},\n deleteSetting: async () => {},\n cleanupSubscription: () => {},\n detachFromCurrent: () => {},\n clear: () => {\n this.#threadId = null;\n },\n clearAndReleaseLock: async () => {},\n cloneToCurrentResource: async ({ sourceThreadId }: { sourceThreadId: string }) =>\n this.thread.clone({ sourceThreadId }),\n loadMetadata: async () => {},\n };\n })();\n\n readonly mode = {\n get: () => this.#modeId,\n hasSelection: () => Boolean(this.#modeId),\n resolve: () => this.#modes.find(mode => mode.id === this.#modeId) ?? { id: this.#modeId, name: this.#modeId },\n set: ({ modeId }: { modeId: string }) => {\n this.#modeId = modeId;\n },\n switch: async ({ modeId }: { modeId: string }) => {\n await this.#remote.switchMode(modeId);\n this.#modeId = modeId;\n },\n };\n\n readonly model = {\n get: () => this.#modelId,\n hasSelection: () => Boolean(this.#modelId),\n displayName: () => this.#modelId,\n set: ({ modelId }: { modelId: string }) => {\n this.#modelId = modelId;\n },\n switch: async (modelId: string, options: { thinkingLevel?: any } = {}) => {\n await this.#remote.switchModel(modelId, options);\n this.#modelId = modelId;\n },\n syncFromPersisted: async () => {},\n };\n\n readonly state = {\n get: () => this.#state,\n set: async (updates: Record<string, unknown>) => {\n this.#state = { ...this.#state, ...updates };\n await this.#remote.setState(updates);\n },\n setIf: async (updates: Record<string, unknown>, shouldApply: () => boolean) => {\n if (!shouldApply()) return false;\n await this.state.set(updates);\n return true;\n },\n update: async <TResult>(updater: AgentControllerRequestStateUpdater<any, TResult>): Promise<TResult> => {\n const { updates, result } = await updater(this.#state);\n if (updates) await this.state.set(updates);\n return result;\n },\n };\n\n readonly displayState = {\n get: () => this.#displayState,\n clearModifiedFiles: () => {\n this.#displayState = { ...this.#displayState, modifiedFiles: new Map() };\n },\n restoreTasks: () => {},\n clearPendingSuspensions: () => {},\n deletePendingSuspension: () => {},\n clearPendingApprovals: () => {},\n resetThread: () => {},\n apply: () => {},\n };\n\n readonly run = {\n isRunning: () => this.#running,\n getRunId: () => null,\n getTraceId: () => null,\n };\n\n readonly stream = { isActive: () => this.#running, isOpen: () => this.#running };\n\n readonly suspensions = {\n hasPending: () => this.#displayState.pendingSuspensions.size > 0,\n get: ({ toolCallId }: { toolCallId: string }) => this.#displayState.pendingSuspensions.get(toolCallId),\n has: ({ toolCallId }: { toolCallId: string }) => this.#displayState.pendingSuspensions.has(toolCallId),\n delete: () => {},\n };\n\n readonly permissions = {\n getRules: () => this.#permissionRules,\n setForCategory: async ({ category, policy }: { category: any; policy: any }) => {\n await this.#remote.setPermissionForCategory(category, policy);\n await this.#refreshPermissions();\n },\n setForTool: async ({ toolName, policy }: { toolName: string; policy: any }) => {\n await this.#remote.setPermissionForTool(toolName, policy);\n await this.#refreshPermissions();\n },\n };\n\n getCurrentRunId(): string | null {\n return null;\n }\n\n getTags(): Record<string, string> {\n return {};\n }\n\n getTokenUsage() {\n return this.#displayState.tokenUsage;\n }\n\n getGrants() {\n return { categories: [], tools: [] };\n }\n\n async sendMessage({ content, files }: { content: string; files?: Files }) {\n await this.#remote.sendMessage({ content, files });\n }\n\n async queueMessage({ content }: { content: string }) {\n await this.#remote.followUp(content);\n }\n\n sendSignal(input: Parameters<Session['sendSignal']>[0]) {\n const content: unknown = 'content' in input ? input.content : undefined;\n const text =\n typeof content === 'string'\n ? content\n : Array.isArray(content)\n ? content.map(part => (typeof part === 'string' ? part : ((part as { text?: string })?.text ?? ''))).join('')\n : JSON.stringify(content);\n const files = 'files' in input ? (input.files as Files | undefined) : undefined;\n const message = files?.length ? { content: text, files } : text;\n const id = randomUUID();\n this.#pendingUserIds.push(id);\n const sent = this.#running ? this.#remote.followUp(message) : this.#remote.sendMessage(message);\n sent.catch(() => {\n this.#pendingUserIds = this.#pendingUserIds.filter(pending => pending !== id);\n });\n const accepted = sent.then(() => ({ accepted: true as const }));\n accepted.catch(() => {});\n return { id, type: 'user' as const, accepted, persisted: sent };\n }\n\n async steer({ content }: { content: string }) {\n await this.#remote.steer(content);\n }\n\n async followUp({ content }: { content: string }) {\n await this.#remote.followUp(content);\n }\n\n respondToToolApproval({\n decision,\n toolCallId,\n }: {\n decision: 'approve' | 'decline' | 'always_allow_category';\n toolCallId: string;\n }) {\n void (async () => {\n const ack = await this.#remote.approveTool(toolCallId, decision !== 'decline', {\n alwaysAllowCategory: decision === 'always_allow_category',\n });\n if (ack && 'accepted' in ack && ack.accepted === false) {\n this.#emitError(`Tool approval was not applied: ${(ack as { reason?: string }).reason ?? 'rejected'}`);\n }\n })().catch(error =>\n this.#emitError(`Tool approval failed: ${error instanceof Error ? error.message : String(error)}`),\n );\n return { accepted: true as const };\n }\n\n async respondToToolSuspension({ resumeData, toolCallId }: { resumeData: unknown; toolCallId?: string }) {\n const id = toolCallId ?? this.#displayState.pendingSuspensions.keys().next().value;\n if (!id) throw new Error('No pending tool suspension to respond to');\n await this.#remote.respondToToolSuspension(\n id,\n resumeData as Parameters<RemoteHandle['respondToToolSuspension']>[1],\n );\n }\n\n claimToolResponse(): boolean {\n return true;\n }\n\n releaseToolResponse(): void {}\n\n abort(): void {\n void this.#remote\n .abort()\n .catch(error => this.#emitError(`Abort failed: ${error instanceof Error ? error.message : String(error)}`));\n }\n\n abortRun(): void {\n this.abort();\n }\n}\n"],"mappings":";;;;AAqBA,MAAM,aAAa;CACjB;CACA;CACA;CACA;CACA;CACA;AACF;;AAGA,SAAS,oBAAoB,MAA4D;CACvF,MAAM,QAAiC;EAAE,GAAG,oBAAoB;EAAG,GAAG;CAAK;CAC3E,KAAK,MAAM,SAAS,YAAY;EAC9B,MAAM,QAAQ,KAAK;EACnB,MAAM,SAAS,iBAAiB,MAAM,QAAQ,IAAI,IAAI,OAAO,QAAS,SAAgC,CAAC,CAAC,CAAC;CAC3G;CACA,OAAO;AACT;;AAGA,SAAS,aAAa,OAA0C;CAC9D,MAAM,MAA+B,EAAE,GAAG,MAAM;CAChD,IAAI,IAAI,gBAAgB,OAAO,IAAI,iBAAiB,UAClD,IAAI,eAAe,oBAAoB,IAAI,YAAuC;CAEpF,MAAM,QAAQ,IAAI;CAClB,IAAI,SAAS,OAAO,UAAU,YAAY,EAAE,iBAAiB,QAC3D,IAAI,QAAQ,OAAO,OAAO,IAAI,MAAM,MAAM,WAAW,cAAc,GAAG,EAAE,MAAM,MAAM,QAAQ,QAAQ,CAAC;CAEvG,OAAO;AACT;AAEA,SAAS,SACP,QACA,YACuB;CACvB,MAAM,QAAQ,UAAoB,QAAQ,IAAI,KAAK,KAAe,oBAAI,IAAI,KAAK;CAC/E,OAAO;EACL,IAAI,OAAO;EACX,YAAY,OAAO,cAAc;EACjC,OAAO,OAAO,SAAS;EACvB,WAAW,KAAK,OAAO,SAAS;EAChC,WAAW,KAAK,OAAO,SAAS;CAClC;AACF;;;;;;;;;AAUA,IAAa,gBAAb,MAA2B;CACzB;CACA;;CAEA;CACA,6BAAsB,IAAI,IAAkC;CAC5D;CACA,YAA2B;CAC3B,UAAU;CACV,WAAW;CACX,WAAW;CACX,SAAkC,CAAC;CACnC,gBAA6C,oBAAoB;CACjE,SAAuB,CAAC;CACxB,mBAAoC;EAAE,YAAY,CAAC;EAAG,OAAO,CAAC;CAAE;CAChE,kBAA4B,CAAC;CAC7B,6BAAsB,IAAI,IAAoB;CAE9C,YAAY,SAAyB,WAAmB,QAAuB;EAC7E,KAAK,aAAa;EAClB,KAAK,cAAc;EACnB,KAAK,UAAU,UAAU,oBAAoB,OAAO,CAAC,CAAC,mBAAA,MAAwC,CAAC,CAAC,QAAQ,SAAS;CACnH;;CAGA,MAAM,UAAyB;EAC7B,MAAM,UAAU,MAAM,KAAK,QAAQ,OAAO;EAC1C,IAAI,SAAS,YAAY,KAAK,cAAc,QAAQ;EACpD,MAAM,QAAQ,IAAI,CAAC,KAAK,SAAS,GAAG,KAAK,oBAAoB,CAAC,CAAC;EAC/D,MAAM,eAAe,MAAM,KAAK,QAAQ,UAAU;GAChD,WAAW;GACX,UAAS,UAAS,KAAK,SAAS,aAAa,KAAK,CAAC;GAEnD,mBAAmB,KAAK,KAAK,SAAS,CAAC,CAAC,YAAY,CAAC,CAAC;EACxD,CAAC;EACD,KAAK,eAAe,aAAa;CACnC;;CAGA,SAAS,OAA2B;EAClC,KAAK,SAAS;CAChB;CAEA,aAAmB;EACjB,KAAK,eAAe;EACpB,KAAK,eAAe,KAAA;CACtB;CAEA,MAAM,WAA0B;EAC9B,MAAM,WAAW,MAAM,KAAK,QAAQ,MAAM;EAC1C,KAAK,YAAY,SAAS,YAAY;EACtC,KAAK,UAAU,SAAS;EACxB,KAAK,WAAW,SAAS;EACzB,KAAK,WAAW,QAAQ,SAAS,OAAO;EAGxC,IAAI,SAAS,UAAU,KAAK,SAAS;GAAE,GAAG,KAAK;GAAQ,GAAG,SAAS;EAAS;EAC5E,KAAK,gBAAgB;GACnB,GAAG,KAAK;GACR,GAAI,SAAS,QAAQ,EAAE,OAAO,SAAS,MAAM,IAAI,CAAC;GAClD,GAAI,SAAS,aAAa,EAAE,YAAY,SAAS,WAAW,IAAI,CAAC;EACnE;CACF;CAEA,MAAM,sBAAqC;EACzC,MAAM,QAAQ,MAAM,KAAK,QAAQ,eAAe;EAChD,KAAK,mBAAmB;GACtB,YAAa,MAAM,cAAc,CAAC;GAClC,OAAO,MAAM,SAAS,CAAC;EACzB;CACF;;CAGA,WAAW,SAAuB;EAChC,KAAK,SAAS;GAAE,MAAM;GAAS,OAAO,IAAI,MAAM,OAAO;EAAE,CAAyB;CACpF;;;;;;CAOA,kBAAkB,GAA8B;EAC9C,MAAM,UAAU,EAAE;EAGlB,MAAM,SACJ,SAAS,SAAS,UAAW,SAAS,SAAS,YAAY,QAAQ,SAAS,UAAU,QAAQ,SAAS;EACzG,IAAI,CAAC,SAAS,MAAM,CAAC,QAAQ;EAC7B,IAAI,UAAU,KAAK,WAAW,IAAI,QAAQ,EAAE;EAC5C,IAAI,CAAC,WAAW,EAAE,SAAS,iBAAiB;GAC1C,UAAU,KAAK,gBAAgB,MAAM;GACrC,IAAI,SAAS,KAAK,WAAW,IAAI,QAAQ,IAAI,OAAO;EACtD;EACA,IAAI,SAAS,EAAE,UAAU;GAAE,GAAG;GAAS,IAAI;EAAQ;CACrD;CAEA,SAAS,OAAmC;EAC1C,MAAM,IAAI;EACV,IAAI,EAAE,SAAS,mBAAmB,EAAE,SAAS,oBAAoB,EAAE,SAAS,eAC1E,KAAK,kBAAkB,CAAC;EAE1B,QAAQ,EAAE,MAAV;GACE,KAAK;IACH,KAAK,UAAU,EAAE;IACjB;GACF,KAAK;IACH,KAAK,WAAW,EAAE;IAClB;GACF,KAAK;IACH,KAAK,YAAY,EAAE;IACnB;GACF,KAAK;IACH,KAAK,YAAY,EAAE,OAAO;IAC1B;GACF,KAAK;IACH,KAAK,SAAS;KAAE,GAAG,KAAK;KAAQ,GAAG,EAAE;IAAM;IAC3C;GACF,KAAK;IACH,KAAK,gBAAgB,EAAE;IACvB;GACF,KAAK;IACH,KAAK,WAAW;IAChB;GACF,KAAK,aACH,KAAK,WAAW;EAEpB;EACA,KAAK,MAAM,YAAY,KAAK,YAC1B,QAAa,QAAQ,SAAS,KAAK,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC;CAExD;CAEA,UAAU,UAAoD;EAC5D,KAAK,WAAW,IAAI,QAAQ;EAC5B,aAAa,KAAK,WAAW,OAAO,QAAQ;CAC9C;CAEA,mBAA+B;EAC7B,aAAa,CAAC;CAChB;CAEA,WAAoB;EAClB,qBAAqB,KAAK;EAC1B,4BAA4B,KAAK;EACjC,aAAa,KAAK;EAClB,kBAAkB,KAAK;EACvB,qBAAqB,CAAC;CACxB;CAEA,gBAAyB;EACvB,MAAM,eAAe,KAAK;EAC1B,MAAM,mBAAmB,KAAK;EAC9B,OAAO;GACL,aAAa,KAAK;GAClB,aAAa,KAAK,cAAc;GAChC,iBAAiB;IACf,IAAI,CAAC,KAAK,WAAW,MAAM,IAAI,MAAM,kBAAkB;IACvD,OAAO,KAAK;GACd;GACA,UAAU,YAAY;IACpB,IAAI,CAAC,KAAK,WAAW,MAAM,KAAK,OAAO,OAAO;IAC9C,OAAO,KAAK;GACd;GACA,MAAM,aAAa,MAAM,OAAO,CAAC,CAAC,YAAY,EAAA,CAAG,KAAI,MAAK,SAAS,GAAG,WAAW,CAAC,CAAC;GACnF,SAAS,OAAO,EAAE,gBACf,MAAM,KAAK,OAAO,KAAK,EAAA,CAAG,MAAK,MAAK,EAAE,OAAO,QAAQ,KAAK;GAC7D,QAAQ,OAAO,EAAE,UAA8B,CAAC,MAAM;IAEpD,MAAM,SAAS,SAAS,MADF,OAAO,CAAC,CAAC,aAAa,KAAK,GACT,WAAW,CAAC;IACpD,KAAK,YAAY,OAAO;IACxB,OAAO;GACT;GACA,QAAQ,OAAO,EAAE,eAAqC;IACpD,MAAM,OAAO,CAAC,CAAC,aAAa,QAAQ;IACpC,KAAK,YAAY;GACnB;GACA,OAAO,OAAO,EAAE,gBAAgB,UAAuD,CAAC,MAAM;IAE5F,MAAM,SAAS,SAAS,MADF,OAAO,CAAC,CAAC,YAAY;KAAE;KAAgB;IAAM,CAAC,GAC5B,WAAW,CAAC;IACpD,KAAK,YAAY,OAAO;IACxB,OAAO;GACT;GACA,QAAQ,OAAO,EAAE,YAA+B;IAC9C,IAAI,KAAK,WAAW,MAAM,OAAO,CAAC,CAAC,aAAa,KAAK,WAAW,KAAK;GACvE;GACA,QAAQ,OAAO,EAAE,eAAqC;IACpD,MAAM,OAAO,CAAC,CAAC,aAAa,QAAQ;IACpC,IAAI,KAAK,cAAc,UAAU,KAAK,YAAY;GACpD;GACA,eAAe,EAAE,UAAU,YACzB,OAAO,CAAC,CAAC,aAAa,UAAU,KAAK;GACvC,oBAAoB,OAAO,EAAE,UAA8B,CAAC,MAC1D,KAAK,YAAY,OAAO,CAAC,CAAC,aAAa,KAAK,WAAW,KAAK,IAAI,CAAC;GACnE,kBAAkB,YAAY;GAC9B,mBAAmB,4BAAY,IAAI,IAAI;GAEvC,YAAY,YAAY,KAAA;GACxB,cAAc,YAAY,KAAA;GAC1B,YAAY,YAAY,CAAC;GACzB,cAAc,YAAY,CAAC;GAC3B,eAAe,YAAY,CAAC;GAC5B,2BAA2B,CAAC;GAC5B,yBAAyB,CAAC;GAC1B,aAAa;IACX,KAAK,YAAY;GACnB;GACA,qBAAqB,YAAY,CAAC;GAClC,wBAAwB,OAAO,EAAE,qBAC/B,KAAK,OAAO,MAAM,EAAE,eAAe,CAAC;GACtC,cAAc,YAAY,CAAC;EAC7B;CACF,EAAA,CAAG;CAEH,OAAgB;EACd,WAAW,KAAK;EAChB,oBAAoB,QAAQ,KAAK,OAAO;EACxC,eAAe,KAAK,OAAO,MAAK,SAAQ,KAAK,OAAO,KAAK,OAAO,KAAK;GAAE,IAAI,KAAK;GAAS,MAAM,KAAK;EAAQ;EAC5G,MAAM,EAAE,aAAiC;GACvC,KAAK,UAAU;EACjB;EACA,QAAQ,OAAO,EAAE,aAAiC;GAChD,MAAM,KAAK,QAAQ,WAAW,MAAM;GACpC,KAAK,UAAU;EACjB;CACF;CAEA,QAAiB;EACf,WAAW,KAAK;EAChB,oBAAoB,QAAQ,KAAK,QAAQ;EACzC,mBAAmB,KAAK;EACxB,MAAM,EAAE,cAAmC;GACzC,KAAK,WAAW;EAClB;EACA,QAAQ,OAAO,SAAiB,UAAmC,CAAC,MAAM;GACxE,MAAM,KAAK,QAAQ,YAAY,SAAS,OAAO;GAC/C,KAAK,WAAW;EAClB;EACA,mBAAmB,YAAY,CAAC;CAClC;CAEA,QAAiB;EACf,WAAW,KAAK;EAChB,KAAK,OAAO,YAAqC;GAC/C,KAAK,SAAS;IAAE,GAAG,KAAK;IAAQ,GAAG;GAAQ;GAC3C,MAAM,KAAK,QAAQ,SAAS,OAAO;EACrC;EACA,OAAO,OAAO,SAAkC,gBAA+B;GAC7E,IAAI,CAAC,YAAY,GAAG,OAAO;GAC3B,MAAM,KAAK,MAAM,IAAI,OAAO;GAC5B,OAAO;EACT;EACA,QAAQ,OAAgB,YAAgF;GACtG,MAAM,EAAE,SAAS,WAAW,MAAM,QAAQ,KAAK,MAAM;GACrD,IAAI,SAAS,MAAM,KAAK,MAAM,IAAI,OAAO;GACzC,OAAO;EACT;CACF;CAEA,eAAwB;EACtB,WAAW,KAAK;EAChB,0BAA0B;GACxB,KAAK,gBAAgB;IAAE,GAAG,KAAK;IAAe,+BAAe,IAAI,IAAI;GAAE;EACzE;EACA,oBAAoB,CAAC;EACrB,+BAA+B,CAAC;EAChC,+BAA+B,CAAC;EAChC,6BAA6B,CAAC;EAC9B,mBAAmB,CAAC;EACpB,aAAa,CAAC;CAChB;CAEA,MAAe;EACb,iBAAiB,KAAK;EACtB,gBAAgB;EAChB,kBAAkB;CACpB;CAEA,SAAkB;EAAE,gBAAgB,KAAK;EAAU,cAAc,KAAK;CAAS;CAE/E,cAAuB;EACrB,kBAAkB,KAAK,cAAc,mBAAmB,OAAO;EAC/D,MAAM,EAAE,iBAAyC,KAAK,cAAc,mBAAmB,IAAI,UAAU;EACrG,MAAM,EAAE,iBAAyC,KAAK,cAAc,mBAAmB,IAAI,UAAU;EACrG,cAAc,CAAC;CACjB;CAEA,cAAuB;EACrB,gBAAgB,KAAK;EACrB,gBAAgB,OAAO,EAAE,UAAU,aAA6C;GAC9E,MAAM,KAAK,QAAQ,yBAAyB,UAAU,MAAM;GAC5D,MAAM,KAAK,oBAAoB;EACjC;EACA,YAAY,OAAO,EAAE,UAAU,aAAgD;GAC7E,MAAM,KAAK,QAAQ,qBAAqB,UAAU,MAAM;GACxD,MAAM,KAAK,oBAAoB;EACjC;CACF;CAEA,kBAAiC;EAC/B,OAAO;CACT;CAEA,UAAkC;EAChC,OAAO,CAAC;CACV;CAEA,gBAAgB;EACd,OAAO,KAAK,cAAc;CAC5B;CAEA,YAAY;EACV,OAAO;GAAE,YAAY,CAAC;GAAG,OAAO,CAAC;EAAE;CACrC;CAEA,MAAM,YAAY,EAAE,SAAS,SAA6C;EACxE,MAAM,KAAK,QAAQ,YAAY;GAAE;GAAS;EAAM,CAAC;CACnD;CAEA,MAAM,aAAa,EAAE,WAAgC;EACnD,MAAM,KAAK,QAAQ,SAAS,OAAO;CACrC;CAEA,WAAW,OAA6C;EACtD,MAAM,UAAmB,aAAa,QAAQ,MAAM,UAAU,KAAA;EAC9D,MAAM,OACJ,OAAO,YAAY,WACf,UACA,MAAM,QAAQ,OAAO,IACnB,QAAQ,KAAI,SAAS,OAAO,SAAS,WAAW,OAAS,MAA4B,QAAQ,EAAI,CAAC,CAAC,KAAK,EAAE,IAC1G,KAAK,UAAU,OAAO;EAC9B,MAAM,QAAQ,WAAW,QAAS,MAAM,QAA8B,KAAA;EACtE,MAAM,UAAU,OAAO,SAAS;GAAE,SAAS;GAAM;EAAM,IAAI;EAC3D,MAAM,KAAK,WAAW;EACtB,KAAK,gBAAgB,KAAK,EAAE;EAC5B,MAAM,OAAO,KAAK,WAAW,KAAK,QAAQ,SAAS,OAAO,IAAI,KAAK,QAAQ,YAAY,OAAO;EAC9F,KAAK,YAAY;GACf,KAAK,kBAAkB,KAAK,gBAAgB,QAAO,YAAW,YAAY,EAAE;EAC9E,CAAC;EACD,MAAM,WAAW,KAAK,YAAY,EAAE,UAAU,KAAc,EAAE;EAC9D,SAAS,YAAY,CAAC,CAAC;EACvB,OAAO;GAAE;GAAI,MAAM;GAAiB;GAAU,WAAW;EAAK;CAChE;CAEA,MAAM,MAAM,EAAE,WAAgC;EAC5C,MAAM,KAAK,QAAQ,MAAM,OAAO;CAClC;CAEA,MAAM,SAAS,EAAE,WAAgC;EAC/C,MAAM,KAAK,QAAQ,SAAS,OAAO;CACrC;CAEA,sBAAsB,EACpB,UACA,cAIC;EACD,CAAM,YAAY;GAChB,MAAM,MAAM,MAAM,KAAK,QAAQ,YAAY,YAAY,aAAa,WAAW,EAC7E,qBAAqB,aAAa,wBACpC,CAAC;GACD,IAAI,OAAO,cAAc,OAAO,IAAI,aAAa,OAC/C,KAAK,WAAW,kCAAmC,IAA4B,UAAU,YAAY;EAEzG,EAAA,CAAG,CAAC,CAAC,OAAM,UACT,KAAK,WAAW,yBAAyB,iBAAiB,QAAQ,MAAM,UAAU,OAAO,KAAK,GAAG,CACnG;EACA,OAAO,EAAE,UAAU,KAAc;CACnC;CAEA,MAAM,wBAAwB,EAAE,YAAY,cAA4D;EACtG,MAAM,KAAK,cAAc,KAAK,cAAc,mBAAmB,KAAK,CAAC,CAAC,KAAK,CAAC,CAAC;EAC7E,IAAI,CAAC,IAAI,MAAM,IAAI,MAAM,0CAA0C;EACnE,MAAM,KAAK,QAAQ,wBACjB,IACA,UACF;CACF;CAEA,oBAA6B;EAC3B,OAAO;CACT;CAEA,sBAA4B,CAAC;CAE7B,QAAc;EACZ,KAAU,QACP,MAAM,CAAC,CACP,OAAM,UAAS,KAAK,WAAW,iBAAiB,iBAAiB,QAAQ,MAAM,UAAU,OAAO,KAAK,GAAG,CAAC;CAC9G;CAEA,WAAiB;EACf,KAAK,MAAM;CACb;AACF"}
|
package/dist/headless/cli.js
CHANGED
|
@@ -159,7 +159,8 @@ async function runMCCli(predrainedInput, options) {
|
|
|
159
159
|
}
|
|
160
160
|
packId = resolved.packId;
|
|
161
161
|
}
|
|
162
|
-
const
|
|
162
|
+
const diagnosticsSetup = createProcessMemoryDiagnosticsFromEnvironment(process.env);
|
|
163
|
+
const processMemoryDiagnostics = await startConfiguredProcessMemoryDiagnostics(diagnosticsSetup, (warning) => {
|
|
163
164
|
process.stderr.write(`Warning: ${warning}\n`);
|
|
164
165
|
});
|
|
165
166
|
let boot;
|
package/dist/headless/cli.js.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"cli.js","names":[],"sources":["../../src/headless/cli.ts"],"sourcesContent":["/**\n * CLI adapter for headless MastraCode runs.\n *\n * This is the only headless layer that touches the process: it parses argv,\n * reads stdin, bootstraps MastraCode via `createMastraCode`, drives `runMC`,\n * renders events/results to stdout/stderr through the pure formatters, maps the\n * result to an exit code, and owns teardown + `process.exit`.\n */\nimport { existsSync } from 'node:fs';\nimport type { Writable } from 'node:stream';\nimport { parseArgs } from 'node:util';\n\nimport { createMastraCode } from '../index.js';\nimport { activateModePackInSettings, resolveModePackArg, resolvePackSelection } from '../onboarding/pack-activation.js';\nimport { listBuiltinModePacks } from '../onboarding/packs.js';\nimport { loadSettings, resolveModelDefaults, saveSettings } from '../onboarding/settings.js';\nimport {\n createProcessMemoryDiagnosticsFromEnvironment,\n startConfiguredProcessMemoryDiagnostics,\n stopProcessMemoryDiagnosticsWithTimeout,\n} from '../process-memory-diagnostics.js';\nimport { setupDebugLogging } from '../utils/debug-log.js';\nimport { releaseAllThreadLocks } from '../utils/thread-lock.js';\n\nimport { buildParseArgsOptions, FLAGS, renderFlagUsage } from './flags.js';\nimport { createHumanFormatState, formatHuman, formatJsonl, isJsonlEvent, renderJsonResult } from './format.js';\nimport { permissionModeToPolicy } from './policy.js';\nimport { runMC } from './run-mc.js';\nimport type { PermissionMode, RunMode, ThinkingLevel } from './types.js';\n\n/** Consolidated output mode (replaces the old `--format` + `--output-format`). */\nexport type OutputMode = 'human' | 'json' | 'jsonl';\n\nexport interface HeadlessArgs {\n prompt?: string;\n /** Timeout in seconds (CLI surface); converted to ms before `runMC`. */\n timeout?: number;\n output: OutputMode;\n continue_: boolean;\n model?: string;\n pack?: string;\n mode?: RunMode;\n thinkingLevel?: ThinkingLevel;\n settings?: string;\n thread?: string;\n title?: string;\n cloneThread: boolean;\n resourceId?: string;\n /** Max agentic turns before the run aborts with exit code 1. */\n maxTurns?: number;\n /** Named permission mode resolving to a built-in policy. Defaults to `auto`. */\n permissionMode?: PermissionMode;\n}\n\nconst parseArgsOptions = buildParseArgsOptions();\nconst STOP_NOTIFICATION_DISPATCH_TIMEOUT_MS = 2_000;\n\n/**\n * Returns true if `argv` selects headless mode. This must agree with what\n * {@link parseHeadlessArgs} (and `runMCCli`) accept as a prompt: `--prompt`/`-p`\n * or a bare positional prompt (e.g. `mastracode \"Fix the bug\"`). Note that a\n * prompt piped via stdin without a flag is handled separately by the caller.\n */\nexport function hasHeadlessFlag(argv: string[]): boolean {\n if (argv.some(a => a === '--prompt' || a === '-p')) return true;\n try {\n const { values, positionals } = parseArgs({\n args: argv.slice(2),\n options: parseArgsOptions,\n strict: false,\n allowPositionals: true,\n });\n // A positional prompt only counts when not asking for help.\n return positionals.length > 0 && !values.help;\n } catch {\n return false;\n }\n}\n\n/**\n * Parse CLI arguments for headless mode. The flag table in `flags.ts` is the\n * single source of truth: each flag carries its own coercion/validation, so this\n * function just walks {@link FLAGS} and assembles the typed {@link HeadlessArgs}.\n */\nexport function parseHeadlessArgs(argv: string[]): HeadlessArgs {\n const { values, positionals } = parseArgs({\n args: argv.slice(2),\n options: parseArgsOptions,\n strict: false,\n allowPositionals: true,\n });\n\n // Seed defaults; per-flag values below override these.\n const args: HeadlessArgs = {\n output: 'human',\n continue_: false,\n cloneThread: false,\n };\n const sink = args as unknown as Record<string, unknown>;\n\n for (const flag of FLAGS) {\n if (!flag.field) continue; // e.g. --help, handled by the caller\n const raw = values[flag.key];\n if (raw === undefined) continue;\n\n if (flag.type === 'boolean') {\n sink[flag.field] = Boolean(raw);\n } else if (typeof raw === 'string') {\n sink[flag.field] = flag.coerce ? flag.coerce(raw) : raw;\n }\n }\n\n // A bare positional acts as the prompt when --prompt/-p is absent.\n if (args.prompt === undefined && positionals[0] !== undefined) {\n args.prompt = positionals[0];\n }\n\n if (args.continue_ && args.thread) {\n throw new Error('--continue and --thread cannot be used together');\n }\n\n return args;\n}\n\nexport function printHeadlessUsage(): void {\n process.stdout.write(`\nUsage: mastracode --prompt <text> [options]\n\nHeadless (non-interactive) mode options:\n${renderFlagUsage()}\n\nThread behavior:\n By default, a new thread is created for each run.\n Use --continue to resume the most recent thread, or --thread to target a specific one.\n Use --clone-thread to branch off a copy before running.\n\nSettings file:\n Uses the same settings.json as the interactive TUI. Pass --settings to use\n a custom settings file (e.g., settings-ci.json for CI). All model, pack,\n subagent, and OM configuration is resolved from settings at startup.\n\nExit codes:\n 0 Agent completed successfully\n 1 Error, aborted, or max turns reached\n 2 Timeout\n\nExamples:\n mastracode --prompt \"Fix the bug in auth.ts\"\n mastracode --prompt \"Add tests\" --timeout 300 --output json\n mastracode --prompt \"Refactor\" --output jsonl\n mastracode --prompt \"Review this PR\" --permission-mode deny --max-turns 10\n mastracode --settings ./settings-ci.json --prompt \"Run tests\"\n mastracode -c --prompt \"Continue where you left off\"\n echo \"Summarize the repo\" | mastracode --prompt -\n`);\n}\n\n/**\n * Wait until everything written to `stream` has been handed to the OS, since\n * `process.exit()` drops writes a pipe has not accepted yet. Resolves `false`\n * if the stream fails or closes first; never rejects or hangs on a gone reader.\n */\nexport function flushWritable(stream: Writable): Promise<boolean> {\n if (stream.destroyed || stream.errored) return Promise.resolve(false);\n if (stream.writableLength === 0) return Promise.resolve(true);\n return new Promise(resolve => {\n // Left attached: a stream can emit `error` after failing pending callbacks.\n stream.once('error', () => resolve(false));\n stream.once('close', () => resolve(false));\n stream.write('', error => resolve(!error));\n });\n}\n\n/**\n * Headless CLI entry point: parse arguments, read stdin, initialize MastraCode,\n * run via `runMC`, render output, and exit with the mapped code.\n */\nexport async function runMCCli(\n predrainedInput?: string | null,\n options?: { coAuthor?: { name?: string; email?: string } },\n): Promise<never> {\n if (process.argv.includes('--help') || process.argv.includes('-h')) {\n printHeadlessUsage();\n process.exit(0);\n }\n\n let args: HeadlessArgs;\n try {\n args = parseHeadlessArgs(process.argv);\n } catch (e) {\n process.stderr.write(`Error: ${(e as Error).message}\\n`);\n process.exit(1);\n }\n\n let prompt = args.prompt;\n if (predrainedInput !== undefined) {\n prompt = predrainedInput ?? '';\n } else if (prompt === '-' || (!prompt && !process.stdin.isTTY)) {\n const chunks: Buffer[] = [];\n for await (const chunk of process.stdin) {\n chunks.push(chunk as Buffer);\n }\n prompt = Buffer.concat(chunks).toString('utf-8').trim();\n }\n\n if (!prompt) {\n printHeadlessUsage();\n process.stderr.write('Error: --prompt is required (or pipe via stdin)\\n');\n process.exit(1);\n }\n\n if (args.settings && !existsSync(args.settings)) {\n process.stderr.write(`Error: Settings file not found: ${args.settings}\\n`);\n process.exit(1);\n }\n\n // Resolve --pack up front: an unknown pack fails fast with the valid list,\n // before MastraCode boots. The selection itself is resolved inside runMC\n // after thread selection, because a resumed thread restores its persisted\n // mode — the mode that decides whether the pack has a model for this run.\n // Activation (which mirrors /models → Activate and becomes the global\n // default pack in settings) is saved via onApplied only once the pack has\n // actually been applied to the session — a rejected pack must not change\n // the persisted default on its way out.\n const settings = loadSettings(args.settings);\n let packId: string | undefined;\n if (args.pack !== undefined) {\n const resolved = resolveModePackArg(settings, args.pack);\n if ('error' in resolved) {\n process.stderr.write(`Error: ${resolved.error}\\n`);\n process.exit(1);\n }\n packId = resolved.packId;\n }\n\n const diagnosticsSetup = createProcessMemoryDiagnosticsFromEnvironment(process.env);\n const processMemoryDiagnostics = await startConfiguredProcessMemoryDiagnostics(diagnosticsSetup, warning => {\n process.stderr.write(`Warning: ${warning}\\n`);\n });\n\n let boot: Awaited<ReturnType<typeof createMastraCode>> | undefined;\n // Default to a non-zero exit so an unexpected throw before the run resolves\n // still surfaces as a failure to the caller / CI.\n let exitCode = 1;\n try {\n boot = await createMastraCode({ settingsPath: args.settings, coAuthor: options?.coAuthor });\n const { controller, session, mcpManager } = boot;\n\n const effectiveDefaults = resolveModelDefaults(settings, listBuiltinModePacks());\n\n if (mcpManager?.hasServers()) {\n try {\n await mcpManager.initInBackground();\n } catch (err) {\n process.stderr.write(`Warning: MCP server initialization failed: ${(err as Error).message ?? err}\\n`);\n }\n }\n\n setupDebugLogging();\n\n const humanState = createHumanFormatState();\n const run = runMC({\n controller,\n session,\n prompt,\n model: args.model,\n mode: args.mode,\n modeDefaults: effectiveDefaults,\n ...(packId\n ? {\n pack: {\n id: packId,\n resolve: (modeId: string) => resolvePackSelection(settings, packId, modeId),\n onApplied: () => {\n activateModePackInSettings(settings, packId);\n saveSettings(settings, args.settings);\n },\n },\n }\n : {}),\n thinkingLevel: args.thinkingLevel,\n thread: { id: args.thread, continueLatest: args.continue_, clone: args.cloneThread },\n resourceId: args.resourceId,\n title: args.title,\n timeoutMs: args.timeout ? args.timeout * 1000 : undefined,\n maxTurns: args.maxTurns,\n policy: args.permissionMode ? permissionModeToPolicy(args.permissionMode) : undefined,\n });\n\n // Stream live events for human + jsonl modes. (json mode prints only the final object.)\n for await (const event of run) {\n if (args.output === 'human') {\n const out = formatHuman(event, humanState);\n if (out.stdout) process.stdout.write(out.stdout);\n if (out.stderr) process.stderr.write(out.stderr);\n } else if (args.output === 'jsonl' && isJsonlEvent(event)) {\n process.stdout.write(JSON.stringify(formatJsonl(event)) + '\\n');\n }\n }\n\n const result = await run.result;\n exitCode = result.exitCode;\n\n if (args.output === 'json') {\n process.stdout.write(renderJsonResult(result));\n } else if (args.output === 'jsonl') {\n process.stdout.write(JSON.stringify({ type: 'result', ...result }) + '\\n');\n }\n\n if (result.status === 'timeout') {\n process.stderr.write(`\\nTimeout elapsed. Aborted.\\n`);\n } else if (result.error && args.output === 'human') {\n process.stderr.write(`Error: ${result.error.message}\\n`);\n }\n } catch (err) {\n process.stderr.write(`Error: ${(err as Error).message ?? err}\\n`);\n exitCode = 1;\n } finally {\n // --- Teardown (always runs, even on a thrown error) ---\n releaseAllThreadLocks();\n if (boot) {\n // Stop plugin-contributed signal providers (and the plugin reload listener)\n // before quiescing workers: a provider that keeps polling past this point\n // could dispatch into a controller that is shutting down. Release thread\n // claims first so a restarted process need not wait out this teardown.\n try {\n boot.releaseThreadClaims();\n } catch {\n // Best-effort — the process is exiting.\n }\n try {\n boot.stopPluginSignalProviders();\n } catch {\n // Best-effort — the process is exiting.\n }\n const { controller, mcpManager } = boot;\n // Release notification dispatch leases before the pubsub that holds them\n // closes. Headless has no shutdown deadline, so don't let a stalled\n // in-flight dispatch hold up exit.\n await Promise.race([\n boot.stopNotificationDispatch?.().catch(() => {}),\n new Promise<void>(resolve => setTimeout(resolve, STOP_NOTIFICATION_DISPATCH_TIMEOUT_MS).unref()),\n ]);\n await Promise.allSettled([\n mcpManager?.disconnect(),\n controller.getMastra()?.stopWorkers(),\n controller.stopIntervals(),\n ]);\n // The signals pubsub is Mastra's event bus, so close it after the workers\n // stop. Call close() on the object; a detached method loses `this` and\n // rejects silently, leaving socket files behind.\n const signalsPubSub = boot.signalsPubSub as { close?: () => Promise<void> | void } | undefined;\n await Promise.allSettled([signalsPubSub?.close?.()]);\n }\n await stopProcessMemoryDiagnosticsWithTimeout(processMemoryDiagnostics, warning => {\n process.stderr.write(`Warning: ${warning}\\n`);\n });\n }\n\n // Let piped output (the final result line in particular) reach the reader\n // before exiting. Output that never arrived must not count as success.\n const [stdoutFlushed] = await Promise.all([flushWritable(process.stdout), flushWritable(process.stderr)]);\n if (!stdoutFlushed && exitCode === 0) exitCode = 1;\n process.exit(exitCode);\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;AAsDA,MAAM,mBAAmB,sBAAsB;AAC/C,MAAM,wCAAwC;;;;;;;AAQ9C,SAAgB,gBAAgB,MAAyB;CACvD,IAAI,KAAK,MAAK,MAAK,MAAM,cAAc,MAAM,IAAI,GAAG,OAAO;CAC3D,IAAI;EACF,MAAM,EAAE,QAAQ,gBAAgB,UAAU;GACxC,MAAM,KAAK,MAAM,CAAC;GAClB,SAAS;GACT,QAAQ;GACR,kBAAkB;EACpB,CAAC;EAED,OAAO,YAAY,SAAS,KAAK,CAAC,OAAO;CAC3C,QAAQ;EACN,OAAO;CACT;AACF;;;;;;AAOA,SAAgB,kBAAkB,MAA8B;CAC9D,MAAM,EAAE,QAAQ,gBAAgB,UAAU;EACxC,MAAM,KAAK,MAAM,CAAC;EAClB,SAAS;EACT,QAAQ;EACR,kBAAkB;CACpB,CAAC;CAGD,MAAM,OAAqB;EACzB,QAAQ;EACR,WAAW;EACX,aAAa;CACf;CACA,MAAM,OAAO;CAEb,KAAK,MAAM,QAAQ,OAAO;EACxB,IAAI,CAAC,KAAK,OAAO;EACjB,MAAM,MAAM,OAAO,KAAK;EACxB,IAAI,QAAQ,KAAA,GAAW;EAEvB,IAAI,KAAK,SAAS,WAChB,KAAK,KAAK,SAAS,QAAQ,GAAG;OACzB,IAAI,OAAO,QAAQ,UACxB,KAAK,KAAK,SAAS,KAAK,SAAS,KAAK,OAAO,GAAG,IAAI;CAExD;CAGA,IAAI,KAAK,WAAW,KAAA,KAAa,YAAY,OAAO,KAAA,GAClD,KAAK,SAAS,YAAY;CAG5B,IAAI,KAAK,aAAa,KAAK,QACzB,MAAM,IAAI,MAAM,iDAAiD;CAGnE,OAAO;AACT;AAEA,SAAgB,qBAA2B;CACzC,QAAQ,OAAO,MAAM;;;;EAIrB,gBAAgB,EAAE;;;;;;;;;;;;;;;;;;;;;;;;;CAyBnB;AACD;;;;;;AAOA,SAAgB,cAAc,QAAoC;CAChE,IAAI,OAAO,aAAa,OAAO,SAAS,OAAO,QAAQ,QAAQ,KAAK;CACpE,IAAI,OAAO,mBAAmB,GAAG,OAAO,QAAQ,QAAQ,IAAI;CAC5D,OAAO,IAAI,SAAQ,YAAW;EAE5B,OAAO,KAAK,eAAe,QAAQ,KAAK,CAAC;EACzC,OAAO,KAAK,eAAe,QAAQ,KAAK,CAAC;EACzC,OAAO,MAAM,KAAI,UAAS,QAAQ,CAAC,KAAK,CAAC;CAC3C,CAAC;AACH;;;;;AAMA,eAAsB,SACpB,iBACA,SACgB;CAChB,IAAI,QAAQ,KAAK,SAAS,QAAQ,KAAK,QAAQ,KAAK,SAAS,IAAI,GAAG;EAClE,mBAAmB;EACnB,QAAQ,KAAK,CAAC;CAChB;CAEA,IAAI;CACJ,IAAI;EACF,OAAO,kBAAkB,QAAQ,IAAI;CACvC,SAAS,GAAG;EACV,QAAQ,OAAO,MAAM,UAAW,EAAY,QAAQ,GAAG;EACvD,QAAQ,KAAK,CAAC;CAChB;CAEA,IAAI,SAAS,KAAK;CAClB,IAAI,oBAAoB,KAAA,GACtB,SAAS,mBAAmB;MACvB,IAAI,WAAW,OAAQ,CAAC,UAAU,CAAC,QAAQ,MAAM,OAAQ;EAC9D,MAAM,SAAmB,CAAC;EAC1B,WAAW,MAAM,SAAS,QAAQ,OAChC,OAAO,KAAK,KAAe;EAE7B,SAAS,OAAO,OAAO,MAAM,CAAC,CAAC,SAAS,OAAO,CAAC,CAAC,KAAK;CACxD;CAEA,IAAI,CAAC,QAAQ;EACX,mBAAmB;EACnB,QAAQ,OAAO,MAAM,mDAAmD;EACxE,QAAQ,KAAK,CAAC;CAChB;CAEA,IAAI,KAAK,YAAY,CAAC,WAAW,KAAK,QAAQ,GAAG;EAC/C,QAAQ,OAAO,MAAM,mCAAmC,KAAK,SAAS,GAAG;EACzE,QAAQ,KAAK,CAAC;CAChB;CAUA,MAAM,WAAW,aAAa,KAAK,QAAQ;CAC3C,IAAI;CACJ,IAAI,KAAK,SAAS,KAAA,GAAW;EAC3B,MAAM,WAAW,mBAAmB,UAAU,KAAK,IAAI;EACvD,IAAI,WAAW,UAAU;GACvB,QAAQ,OAAO,MAAM,UAAU,SAAS,MAAM,GAAG;GACjD,QAAQ,KAAK,CAAC;EAChB;EACA,SAAS,SAAS;CACpB;CAGA,MAAM,2BAA2B,MAAM,wCADd,8CAA8C,QAAQ,GACe,IAAG,YAAW;EAC1G,QAAQ,OAAO,MAAM,YAAY,QAAQ,GAAG;CAC9C,CAAC;CAED,IAAI;CAGJ,IAAI,WAAW;CACf,IAAI;EACF,OAAO,MAAM,iBAAiB;GAAE,cAAc,KAAK;GAAU,UAAU,SAAS;EAAS,CAAC;EAC1F,MAAM,EAAE,YAAY,SAAS,eAAe;EAE5C,MAAM,oBAAoB,qBAAqB,UAAU,qBAAqB,CAAC;EAE/E,IAAI,YAAY,WAAW,GACzB,IAAI;GACF,MAAM,WAAW,iBAAiB;EACpC,SAAS,KAAK;GACZ,QAAQ,OAAO,MAAM,8CAA+C,IAAc,WAAW,IAAI,GAAG;EACtG;EAGF,kBAAkB;EAElB,MAAM,aAAa,uBAAuB;EAC1C,MAAM,MAAM,MAAM;GAChB;GACA;GACA;GACA,OAAO,KAAK;GACZ,MAAM,KAAK;GACX,cAAc;GACd,GAAI,SACA,EACE,MAAM;IACJ,IAAI;IACJ,UAAU,WAAmB,qBAAqB,UAAU,QAAQ,MAAM;IAC1E,iBAAiB;KACf,2BAA2B,UAAU,MAAM;KAC3C,aAAa,UAAU,KAAK,QAAQ;IACtC;GACF,EACF,IACA,CAAC;GACL,eAAe,KAAK;GACpB,QAAQ;IAAE,IAAI,KAAK;IAAQ,gBAAgB,KAAK;IAAW,OAAO,KAAK;GAAY;GACnF,YAAY,KAAK;GACjB,OAAO,KAAK;GACZ,WAAW,KAAK,UAAU,KAAK,UAAU,MAAO,KAAA;GAChD,UAAU,KAAK;GACf,QAAQ,KAAK,iBAAiB,uBAAuB,KAAK,cAAc,IAAI,KAAA;EAC9E,CAAC;EAGD,WAAW,MAAM,SAAS,KACxB,IAAI,KAAK,WAAW,SAAS;GAC3B,MAAM,MAAM,YAAY,OAAO,UAAU;GACzC,IAAI,IAAI,QAAQ,QAAQ,OAAO,MAAM,IAAI,MAAM;GAC/C,IAAI,IAAI,QAAQ,QAAQ,OAAO,MAAM,IAAI,MAAM;EACjD,OAAO,IAAI,KAAK,WAAW,WAAW,aAAa,KAAK,GACtD,QAAQ,OAAO,MAAM,KAAK,UAAU,YAAY,KAAK,CAAC,IAAI,IAAI;EAIlE,MAAM,SAAS,MAAM,IAAI;EACzB,WAAW,OAAO;EAElB,IAAI,KAAK,WAAW,QAClB,QAAQ,OAAO,MAAM,iBAAiB,MAAM,CAAC;OACxC,IAAI,KAAK,WAAW,SACzB,QAAQ,OAAO,MAAM,KAAK,UAAU;GAAE,MAAM;GAAU,GAAG;EAAO,CAAC,IAAI,IAAI;EAG3E,IAAI,OAAO,WAAW,WACpB,QAAQ,OAAO,MAAM,+BAA+B;OAC/C,IAAI,OAAO,SAAS,KAAK,WAAW,SACzC,QAAQ,OAAO,MAAM,UAAU,OAAO,MAAM,QAAQ,GAAG;CAE3D,SAAS,KAAK;EACZ,QAAQ,OAAO,MAAM,UAAW,IAAc,WAAW,IAAI,GAAG;EAChE,WAAW;CACb,UAAU;EAER,sBAAsB;EACtB,IAAI,MAAM;GAKR,IAAI;IACF,KAAK,oBAAoB;GAC3B,QAAQ,CAER;GACA,IAAI;IACF,KAAK,0BAA0B;GACjC,QAAQ,CAER;GACA,MAAM,EAAE,YAAY,eAAe;GAInC,MAAM,QAAQ,KAAK,CACjB,KAAK,2BAA2B,CAAC,CAAC,YAAY,CAAC,CAAC,GAChD,IAAI,SAAc,YAAW,WAAW,SAAS,qCAAqC,CAAC,CAAC,MAAM,CAAC,CACjG,CAAC;GACD,MAAM,QAAQ,WAAW;IACvB,YAAY,WAAW;IACvB,WAAW,UAAU,CAAC,EAAE,YAAY;IACpC,WAAW,cAAc;GAC3B,CAAC;GAID,MAAM,gBAAgB,KAAK;GAC3B,MAAM,QAAQ,WAAW,CAAC,eAAe,QAAQ,CAAC,CAAC;EACrD;EACA,MAAM,wCAAwC,2BAA0B,YAAW;GACjF,QAAQ,OAAO,MAAM,YAAY,QAAQ,GAAG;EAC9C,CAAC;CACH;CAIA,MAAM,CAAC,iBAAiB,MAAM,QAAQ,IAAI,CAAC,cAAc,QAAQ,MAAM,GAAG,cAAc,QAAQ,MAAM,CAAC,CAAC;CACxG,IAAI,CAAC,iBAAiB,aAAa,GAAG,WAAW;CACjD,QAAQ,KAAK,QAAQ;AACvB"}
|
|
1
|
+
{"version":3,"file":"cli.js","names":[],"sources":["../../src/headless/cli.ts"],"sourcesContent":["/**\n * CLI adapter for headless MastraCode runs.\n *\n * This is the only headless layer that touches the process: it parses argv,\n * reads stdin, bootstraps MastraCode via `createMastraCode`, drives `runMC`,\n * renders events/results to stdout/stderr through the pure formatters, maps the\n * result to an exit code, and owns teardown + `process.exit`.\n */\nimport { existsSync } from 'node:fs';\nimport type { Writable } from 'node:stream';\nimport { parseArgs } from 'node:util';\n\nimport { createMastraCode } from '../index.js';\nimport { activateModePackInSettings, resolveModePackArg, resolvePackSelection } from '../onboarding/pack-activation.js';\nimport { listBuiltinModePacks } from '../onboarding/packs.js';\nimport { loadSettings, resolveModelDefaults, saveSettings } from '../onboarding/settings.js';\nimport {\n createProcessMemoryDiagnosticsFromEnvironment,\n startConfiguredProcessMemoryDiagnostics,\n stopProcessMemoryDiagnosticsWithTimeout,\n} from '../process-memory-diagnostics.js';\nimport { setupDebugLogging } from '../utils/debug-log.js';\nimport { releaseAllThreadLocks } from '../utils/thread-lock.js';\n\nimport { buildParseArgsOptions, FLAGS, renderFlagUsage } from './flags.js';\nimport { createHumanFormatState, formatHuman, formatJsonl, isJsonlEvent, renderJsonResult } from './format.js';\nimport { permissionModeToPolicy } from './policy.js';\nimport { runMC } from './run-mc.js';\nimport type { PermissionMode, RunMode, ThinkingLevel } from './types.js';\n\n/** Consolidated output mode (replaces the old `--format` + `--output-format`). */\nexport type OutputMode = 'human' | 'json' | 'jsonl';\n\nexport interface HeadlessArgs {\n prompt?: string;\n /** Timeout in seconds (CLI surface); converted to ms before `runMC`. */\n timeout?: number;\n output: OutputMode;\n continue_: boolean;\n model?: string;\n pack?: string;\n mode?: RunMode;\n thinkingLevel?: ThinkingLevel;\n settings?: string;\n thread?: string;\n title?: string;\n cloneThread: boolean;\n resourceId?: string;\n /** Max agentic turns before the run aborts with exit code 1. */\n maxTurns?: number;\n /** Named permission mode resolving to a built-in policy. Defaults to `auto`. */\n permissionMode?: PermissionMode;\n}\n\nconst parseArgsOptions = buildParseArgsOptions();\nconst STOP_NOTIFICATION_DISPATCH_TIMEOUT_MS = 2_000;\n\n/**\n * Returns true if `argv` selects headless mode. This must agree with what\n * {@link parseHeadlessArgs} (and `runMCCli`) accept as a prompt: `--prompt`/`-p`\n * or a bare positional prompt (e.g. `mastracode \"Fix the bug\"`). Note that a\n * prompt piped via stdin without a flag is handled separately by the caller.\n */\nexport function hasHeadlessFlag(argv: string[]): boolean {\n if (argv.some(a => a === '--prompt' || a === '-p')) return true;\n try {\n const { values, positionals } = parseArgs({\n args: argv.slice(2),\n options: parseArgsOptions,\n strict: false,\n allowPositionals: true,\n });\n // A positional prompt only counts when not asking for help.\n return positionals.length > 0 && !values.help;\n } catch {\n return false;\n }\n}\n\n/**\n * Parse CLI arguments for headless mode. The flag table in `flags.ts` is the\n * single source of truth: each flag carries its own coercion/validation, so this\n * function just walks {@link FLAGS} and assembles the typed {@link HeadlessArgs}.\n */\nexport function parseHeadlessArgs(argv: string[]): HeadlessArgs {\n const { values, positionals } = parseArgs({\n args: argv.slice(2),\n options: parseArgsOptions,\n strict: false,\n allowPositionals: true,\n });\n\n // Seed defaults; per-flag values below override these.\n const args: HeadlessArgs = {\n output: 'human',\n continue_: false,\n cloneThread: false,\n };\n const sink = args as unknown as Record<string, unknown>;\n\n for (const flag of FLAGS) {\n if (!flag.field) continue; // e.g. --help, handled by the caller\n const raw = values[flag.key];\n if (raw === undefined) continue;\n\n if (flag.type === 'boolean') {\n sink[flag.field] = Boolean(raw);\n } else if (typeof raw === 'string') {\n sink[flag.field] = flag.coerce ? flag.coerce(raw) : raw;\n }\n }\n\n // A bare positional acts as the prompt when --prompt/-p is absent.\n if (args.prompt === undefined && positionals[0] !== undefined) {\n args.prompt = positionals[0];\n }\n\n if (args.continue_ && args.thread) {\n throw new Error('--continue and --thread cannot be used together');\n }\n\n return args;\n}\n\nexport function printHeadlessUsage(): void {\n process.stdout.write(`\nUsage: mastracode --prompt <text> [options]\n\nHeadless (non-interactive) mode options:\n${renderFlagUsage()}\n\nThread behavior:\n By default, a new thread is created for each run.\n Use --continue to resume the most recent thread, or --thread to target a specific one.\n Use --clone-thread to branch off a copy before running.\n\nSettings file:\n Uses the same settings.json as the interactive TUI. Pass --settings to use\n a custom settings file (e.g., settings-ci.json for CI). All model, pack,\n subagent, and OM configuration is resolved from settings at startup.\n\nExit codes:\n 0 Agent completed successfully\n 1 Error, aborted, or max turns reached\n 2 Timeout\n\nExamples:\n mastracode --prompt \"Fix the bug in auth.ts\"\n mastracode --prompt \"Add tests\" --timeout 300 --output json\n mastracode --prompt \"Refactor\" --output jsonl\n mastracode --prompt \"Review this PR\" --permission-mode deny --max-turns 10\n mastracode --settings ./settings-ci.json --prompt \"Run tests\"\n mastracode -c --prompt \"Continue where you left off\"\n echo \"Summarize the repo\" | mastracode --prompt -\n`);\n}\n\n/**\n * Wait until everything written to `stream` has been handed to the OS, since\n * `process.exit()` drops writes a pipe has not accepted yet. Resolves `false`\n * if the stream fails or closes first; never rejects or hangs on a gone reader.\n */\nexport function flushWritable(stream: Writable): Promise<boolean> {\n if (stream.destroyed || stream.errored) return Promise.resolve(false);\n if (stream.writableLength === 0) return Promise.resolve(true);\n return new Promise(resolve => {\n // Left attached: a stream can emit `error` after failing pending callbacks.\n stream.once('error', () => resolve(false));\n stream.once('close', () => resolve(false));\n stream.write('', error => resolve(!error));\n });\n}\n\n/**\n * Headless CLI entry point: parse arguments, read stdin, initialize MastraCode,\n * run via `runMC`, render output, and exit with the mapped code.\n */\nexport async function runMCCli(\n predrainedInput?: string | null,\n options?: { coAuthor?: { name?: string; email?: string } },\n): Promise<never> {\n if (process.argv.includes('--help') || process.argv.includes('-h')) {\n printHeadlessUsage();\n process.exit(0);\n }\n\n let args: HeadlessArgs;\n try {\n args = parseHeadlessArgs(process.argv);\n } catch (e) {\n process.stderr.write(`Error: ${(e as Error).message}\\n`);\n process.exit(1);\n }\n\n let prompt = args.prompt;\n if (predrainedInput !== undefined) {\n prompt = predrainedInput ?? '';\n } else if (prompt === '-' || (!prompt && !process.stdin.isTTY)) {\n const chunks: Buffer[] = [];\n for await (const chunk of process.stdin) {\n chunks.push(chunk as Buffer);\n }\n prompt = Buffer.concat(chunks).toString('utf-8').trim();\n }\n\n if (!prompt) {\n printHeadlessUsage();\n process.stderr.write('Error: --prompt is required (or pipe via stdin)\\n');\n process.exit(1);\n }\n\n if (args.settings && !existsSync(args.settings)) {\n process.stderr.write(`Error: Settings file not found: ${args.settings}\\n`);\n process.exit(1);\n }\n\n // Resolve --pack up front: an unknown pack fails fast with the valid list,\n // before MastraCode boots. The selection itself is resolved inside runMC\n // after thread selection, because a resumed thread restores its persisted\n // mode — the mode that decides whether the pack has a model for this run.\n // Activation (which mirrors /models → Activate and becomes the global\n // default pack in settings) is saved via onApplied only once the pack has\n // actually been applied to the session — a rejected pack must not change\n // the persisted default on its way out.\n const settings = loadSettings(args.settings);\n let packId: string | undefined;\n if (args.pack !== undefined) {\n const resolved = resolveModePackArg(settings, args.pack);\n if ('error' in resolved) {\n process.stderr.write(`Error: ${resolved.error}\\n`);\n process.exit(1);\n }\n packId = resolved.packId;\n }\n\n const diagnosticsSetup = createProcessMemoryDiagnosticsFromEnvironment(process.env);\n const processMemoryDiagnostics = await startConfiguredProcessMemoryDiagnostics(diagnosticsSetup, warning => {\n process.stderr.write(`Warning: ${warning}\\n`);\n });\n\n let boot: Awaited<ReturnType<typeof createMastraCode>> | undefined;\n // Default to a non-zero exit so an unexpected throw before the run resolves\n // still surfaces as a failure to the caller / CI.\n let exitCode = 1;\n try {\n boot = await createMastraCode({ settingsPath: args.settings, coAuthor: options?.coAuthor });\n const { controller, session, mcpManager } = boot;\n\n const effectiveDefaults = resolveModelDefaults(settings, listBuiltinModePacks());\n\n if (mcpManager?.hasServers()) {\n try {\n await mcpManager.initInBackground();\n } catch (err) {\n process.stderr.write(`Warning: MCP server initialization failed: ${(err as Error).message ?? err}\\n`);\n }\n }\n\n setupDebugLogging();\n\n const humanState = createHumanFormatState();\n const run = runMC({\n controller,\n session,\n prompt,\n model: args.model,\n mode: args.mode,\n modeDefaults: effectiveDefaults,\n ...(packId\n ? {\n pack: {\n id: packId,\n resolve: (modeId: string) => resolvePackSelection(settings, packId, modeId),\n onApplied: () => {\n activateModePackInSettings(settings, packId);\n saveSettings(settings, args.settings);\n },\n },\n }\n : {}),\n thinkingLevel: args.thinkingLevel,\n thread: { id: args.thread, continueLatest: args.continue_, clone: args.cloneThread },\n resourceId: args.resourceId,\n title: args.title,\n timeoutMs: args.timeout ? args.timeout * 1000 : undefined,\n maxTurns: args.maxTurns,\n policy: args.permissionMode ? permissionModeToPolicy(args.permissionMode) : undefined,\n });\n\n // Stream live events for human + jsonl modes. (json mode prints only the final object.)\n for await (const event of run) {\n if (args.output === 'human') {\n const out = formatHuman(event, humanState);\n if (out.stdout) process.stdout.write(out.stdout);\n if (out.stderr) process.stderr.write(out.stderr);\n } else if (args.output === 'jsonl' && isJsonlEvent(event)) {\n process.stdout.write(JSON.stringify(formatJsonl(event)) + '\\n');\n }\n }\n\n const result = await run.result;\n exitCode = result.exitCode;\n\n if (args.output === 'json') {\n process.stdout.write(renderJsonResult(result));\n } else if (args.output === 'jsonl') {\n process.stdout.write(JSON.stringify({ type: 'result', ...result }) + '\\n');\n }\n\n if (result.status === 'timeout') {\n process.stderr.write(`\\nTimeout elapsed. Aborted.\\n`);\n } else if (result.error && args.output === 'human') {\n process.stderr.write(`Error: ${result.error.message}\\n`);\n }\n } catch (err) {\n process.stderr.write(`Error: ${(err as Error).message ?? err}\\n`);\n exitCode = 1;\n } finally {\n // --- Teardown (always runs, even on a thrown error) ---\n releaseAllThreadLocks();\n if (boot) {\n // Stop plugin-contributed signal providers (and the plugin reload listener)\n // before quiescing workers: a provider that keeps polling past this point\n // could dispatch into a controller that is shutting down. Release thread\n // claims first so a restarted process need not wait out this teardown.\n try {\n boot.releaseThreadClaims();\n } catch {\n // Best-effort — the process is exiting.\n }\n try {\n boot.stopPluginSignalProviders();\n } catch {\n // Best-effort — the process is exiting.\n }\n const { controller, mcpManager } = boot;\n // Release notification dispatch leases before the pubsub that holds them\n // closes. Headless has no shutdown deadline, so don't let a stalled\n // in-flight dispatch hold up exit.\n await Promise.race([\n boot.stopNotificationDispatch?.().catch(() => {}),\n new Promise<void>(resolve => setTimeout(resolve, STOP_NOTIFICATION_DISPATCH_TIMEOUT_MS).unref()),\n ]);\n await Promise.allSettled([\n mcpManager?.disconnect(),\n controller.getMastra()?.stopWorkers(),\n controller.stopIntervals(),\n ]);\n // The signals pubsub is Mastra's event bus, so close it after the workers\n // stop. Call close() on the object; a detached method loses `this` and\n // rejects silently, leaving socket files behind.\n const signalsPubSub = boot.signalsPubSub as { close?: () => Promise<void> | void } | undefined;\n await Promise.allSettled([signalsPubSub?.close?.()]);\n }\n await stopProcessMemoryDiagnosticsWithTimeout(processMemoryDiagnostics, warning => {\n process.stderr.write(`Warning: ${warning}\\n`);\n });\n }\n\n // Let piped output (the final result line in particular) reach the reader\n // before exiting. Output that never arrived must not count as success.\n const [stdoutFlushed] = await Promise.all([flushWritable(process.stdout), flushWritable(process.stderr)]);\n if (!stdoutFlushed && exitCode === 0) exitCode = 1;\n process.exit(exitCode);\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;AAsDA,MAAM,mBAAmB,sBAAsB;AAC/C,MAAM,wCAAwC;;;;;;;AAQ9C,SAAgB,gBAAgB,MAAyB;CACvD,IAAI,KAAK,MAAK,MAAK,MAAM,cAAc,MAAM,IAAI,GAAG,OAAO;CAC3D,IAAI;EACF,MAAM,EAAE,QAAQ,gBAAgB,UAAU;GACxC,MAAM,KAAK,MAAM,CAAC;GAClB,SAAS;GACT,QAAQ;GACR,kBAAkB;EACpB,CAAC;EAED,OAAO,YAAY,SAAS,KAAK,CAAC,OAAO;CAC3C,QAAQ;EACN,OAAO;CACT;AACF;;;;;;AAOA,SAAgB,kBAAkB,MAA8B;CAC9D,MAAM,EAAE,QAAQ,gBAAgB,UAAU;EACxC,MAAM,KAAK,MAAM,CAAC;EAClB,SAAS;EACT,QAAQ;EACR,kBAAkB;CACpB,CAAC;CAGD,MAAM,OAAqB;EACzB,QAAQ;EACR,WAAW;EACX,aAAa;CACf;CACA,MAAM,OAAO;CAEb,KAAK,MAAM,QAAQ,OAAO;EACxB,IAAI,CAAC,KAAK,OAAO;EACjB,MAAM,MAAM,OAAO,KAAK;EACxB,IAAI,QAAQ,KAAA,GAAW;EAEvB,IAAI,KAAK,SAAS,WAChB,KAAK,KAAK,SAAS,QAAQ,GAAG;OACzB,IAAI,OAAO,QAAQ,UACxB,KAAK,KAAK,SAAS,KAAK,SAAS,KAAK,OAAO,GAAG,IAAI;CAExD;CAGA,IAAI,KAAK,WAAW,KAAA,KAAa,YAAY,OAAO,KAAA,GAClD,KAAK,SAAS,YAAY;CAG5B,IAAI,KAAK,aAAa,KAAK,QACzB,MAAM,IAAI,MAAM,iDAAiD;CAGnE,OAAO;AACT;AAEA,SAAgB,qBAA2B;CACzC,QAAQ,OAAO,MAAM;;;;EAIrB,gBAAgB,EAAE;;;;;;;;;;;;;;;;;;;;;;;;;CAyBnB;AACD;;;;;;AAOA,SAAgB,cAAc,QAAoC;CAChE,IAAI,OAAO,aAAa,OAAO,SAAS,OAAO,QAAQ,QAAQ,KAAK;CACpE,IAAI,OAAO,mBAAmB,GAAG,OAAO,QAAQ,QAAQ,IAAI;CAC5D,OAAO,IAAI,SAAQ,YAAW;EAE5B,OAAO,KAAK,eAAe,QAAQ,KAAK,CAAC;EACzC,OAAO,KAAK,eAAe,QAAQ,KAAK,CAAC;EACzC,OAAO,MAAM,KAAI,UAAS,QAAQ,CAAC,KAAK,CAAC;CAC3C,CAAC;AACH;;;;;AAMA,eAAsB,SACpB,iBACA,SACgB;CAChB,IAAI,QAAQ,KAAK,SAAS,QAAQ,KAAK,QAAQ,KAAK,SAAS,IAAI,GAAG;EAClE,mBAAmB;EACnB,QAAQ,KAAK,CAAC;CAChB;CAEA,IAAI;CACJ,IAAI;EACF,OAAO,kBAAkB,QAAQ,IAAI;CACvC,SAAS,GAAG;EACV,QAAQ,OAAO,MAAM,UAAW,EAAY,QAAQ,GAAG;EACvD,QAAQ,KAAK,CAAC;CAChB;CAEA,IAAI,SAAS,KAAK;CAClB,IAAI,oBAAoB,KAAA,GACtB,SAAS,mBAAmB;MACvB,IAAI,WAAW,OAAQ,CAAC,UAAU,CAAC,QAAQ,MAAM,OAAQ;EAC9D,MAAM,SAAmB,CAAC;EAC1B,WAAW,MAAM,SAAS,QAAQ,OAChC,OAAO,KAAK,KAAe;EAE7B,SAAS,OAAO,OAAO,MAAM,CAAC,CAAC,SAAS,OAAO,CAAC,CAAC,KAAK;CACxD;CAEA,IAAI,CAAC,QAAQ;EACX,mBAAmB;EACnB,QAAQ,OAAO,MAAM,mDAAmD;EACxE,QAAQ,KAAK,CAAC;CAChB;CAEA,IAAI,KAAK,YAAY,CAAC,WAAW,KAAK,QAAQ,GAAG;EAC/C,QAAQ,OAAO,MAAM,mCAAmC,KAAK,SAAS,GAAG;EACzE,QAAQ,KAAK,CAAC;CAChB;CAUA,MAAM,WAAW,aAAa,KAAK,QAAQ;CAC3C,IAAI;CACJ,IAAI,KAAK,SAAS,KAAA,GAAW;EAC3B,MAAM,WAAW,mBAAmB,UAAU,KAAK,IAAI;EACvD,IAAI,WAAW,UAAU;GACvB,QAAQ,OAAO,MAAM,UAAU,SAAS,MAAM,GAAG;GACjD,QAAQ,KAAK,CAAC;EAChB;EACA,SAAS,SAAS;CACpB;CAEA,MAAM,mBAAmB,8CAA8C,QAAQ,GAAG;CAClF,MAAM,2BAA2B,MAAM,wCAAwC,mBAAkB,YAAW;EAC1G,QAAQ,OAAO,MAAM,YAAY,QAAQ,GAAG;CAC9C,CAAC;CAED,IAAI;CAGJ,IAAI,WAAW;CACf,IAAI;EACF,OAAO,MAAM,iBAAiB;GAAE,cAAc,KAAK;GAAU,UAAU,SAAS;EAAS,CAAC;EAC1F,MAAM,EAAE,YAAY,SAAS,eAAe;EAE5C,MAAM,oBAAoB,qBAAqB,UAAU,qBAAqB,CAAC;EAE/E,IAAI,YAAY,WAAW,GACzB,IAAI;GACF,MAAM,WAAW,iBAAiB;EACpC,SAAS,KAAK;GACZ,QAAQ,OAAO,MAAM,8CAA+C,IAAc,WAAW,IAAI,GAAG;EACtG;EAGF,kBAAkB;EAElB,MAAM,aAAa,uBAAuB;EAC1C,MAAM,MAAM,MAAM;GAChB;GACA;GACA;GACA,OAAO,KAAK;GACZ,MAAM,KAAK;GACX,cAAc;GACd,GAAI,SACA,EACE,MAAM;IACJ,IAAI;IACJ,UAAU,WAAmB,qBAAqB,UAAU,QAAQ,MAAM;IAC1E,iBAAiB;KACf,2BAA2B,UAAU,MAAM;KAC3C,aAAa,UAAU,KAAK,QAAQ;IACtC;GACF,EACF,IACA,CAAC;GACL,eAAe,KAAK;GACpB,QAAQ;IAAE,IAAI,KAAK;IAAQ,gBAAgB,KAAK;IAAW,OAAO,KAAK;GAAY;GACnF,YAAY,KAAK;GACjB,OAAO,KAAK;GACZ,WAAW,KAAK,UAAU,KAAK,UAAU,MAAO,KAAA;GAChD,UAAU,KAAK;GACf,QAAQ,KAAK,iBAAiB,uBAAuB,KAAK,cAAc,IAAI,KAAA;EAC9E,CAAC;EAGD,WAAW,MAAM,SAAS,KACxB,IAAI,KAAK,WAAW,SAAS;GAC3B,MAAM,MAAM,YAAY,OAAO,UAAU;GACzC,IAAI,IAAI,QAAQ,QAAQ,OAAO,MAAM,IAAI,MAAM;GAC/C,IAAI,IAAI,QAAQ,QAAQ,OAAO,MAAM,IAAI,MAAM;EACjD,OAAO,IAAI,KAAK,WAAW,WAAW,aAAa,KAAK,GACtD,QAAQ,OAAO,MAAM,KAAK,UAAU,YAAY,KAAK,CAAC,IAAI,IAAI;EAIlE,MAAM,SAAS,MAAM,IAAI;EACzB,WAAW,OAAO;EAElB,IAAI,KAAK,WAAW,QAClB,QAAQ,OAAO,MAAM,iBAAiB,MAAM,CAAC;OACxC,IAAI,KAAK,WAAW,SACzB,QAAQ,OAAO,MAAM,KAAK,UAAU;GAAE,MAAM;GAAU,GAAG;EAAO,CAAC,IAAI,IAAI;EAG3E,IAAI,OAAO,WAAW,WACpB,QAAQ,OAAO,MAAM,+BAA+B;OAC/C,IAAI,OAAO,SAAS,KAAK,WAAW,SACzC,QAAQ,OAAO,MAAM,UAAU,OAAO,MAAM,QAAQ,GAAG;CAE3D,SAAS,KAAK;EACZ,QAAQ,OAAO,MAAM,UAAW,IAAc,WAAW,IAAI,GAAG;EAChE,WAAW;CACb,UAAU;EAER,sBAAsB;EACtB,IAAI,MAAM;GAKR,IAAI;IACF,KAAK,oBAAoB;GAC3B,QAAQ,CAER;GACA,IAAI;IACF,KAAK,0BAA0B;GACjC,QAAQ,CAER;GACA,MAAM,EAAE,YAAY,eAAe;GAInC,MAAM,QAAQ,KAAK,CACjB,KAAK,2BAA2B,CAAC,CAAC,YAAY,CAAC,CAAC,GAChD,IAAI,SAAc,YAAW,WAAW,SAAS,qCAAqC,CAAC,CAAC,MAAM,CAAC,CACjG,CAAC;GACD,MAAM,QAAQ,WAAW;IACvB,YAAY,WAAW;IACvB,WAAW,UAAU,CAAC,EAAE,YAAY;IACpC,WAAW,cAAc;GAC3B,CAAC;GAID,MAAM,gBAAgB,KAAK;GAC3B,MAAM,QAAQ,WAAW,CAAC,eAAe,QAAQ,CAAC,CAAC;EACrD;EACA,MAAM,wCAAwC,2BAA0B,YAAW;GACjF,QAAQ,OAAO,MAAM,YAAY,QAAQ,GAAG;EAC9C,CAAC;CACH;CAIA,MAAM,CAAC,iBAAiB,MAAM,QAAQ,IAAI,CAAC,cAAc,QAAQ,MAAM,GAAG,cAAc,QAAQ,MAAM,CAAC,CAAC;CACxG,IAAI,CAAC,iBAAiB,aAAa,GAAG,WAAW;CACjD,QAAQ,KAAK,QAAQ;AACvB"}
|
package/dist/headless/run-mc.js
CHANGED
|
@@ -60,13 +60,11 @@ function aggregate(event, acc) {
|
|
|
60
60
|
totalTokens: event.usage.totalTokens
|
|
61
61
|
};
|
|
62
62
|
break;
|
|
63
|
-
case "error":
|
|
64
|
-
|
|
65
|
-
|
|
66
|
-
|
|
67
|
-
|
|
68
|
-
};
|
|
69
|
-
break;
|
|
63
|
+
case "error": acc.error = {
|
|
64
|
+
name: event.error.name,
|
|
65
|
+
message: event.error.message,
|
|
66
|
+
stack: event.error.stack
|
|
67
|
+
};
|
|
70
68
|
}
|
|
71
69
|
}
|
|
72
70
|
/** Resolve a thread by its exact ID. Titles are not unique, so we don't match on them. */
|
|
@@ -228,8 +226,10 @@ function runMC(options) {
|
|
|
228
226
|
aborted = true;
|
|
229
227
|
session.abort();
|
|
230
228
|
}
|
|
231
|
-
if (options.signal)
|
|
232
|
-
|
|
229
|
+
if (options.signal) {
|
|
230
|
+
if (options.signal.aborted) queueMicrotask(() => abort());
|
|
231
|
+
else options.signal.addEventListener("abort", () => abort(), { once: true });
|
|
232
|
+
}
|
|
233
233
|
function armTimeout() {
|
|
234
234
|
if (!options.timeoutMs || settled) return;
|
|
235
235
|
if (timeoutId) clearTimeout(timeoutId);
|