@posthog/wizard 2.78.0 → 2.79.1

package/dist/{agent-bCZ3i3fh.js → agent-BWFhuLgo.js}

@@ -1,19 +1,19 @@
-import { t as __exportAll } from "./rolldown-runtime-B_-DWIq7.js";
-import { a as AgentSignals, d as ErrorCodes, i as AGENT_ERROR_CODE, l as WizardError, r as skillErrorCode, t as classifyRunFailure } from "./errors-CQ9kttJD.js";
-import { a as getLogFilePath, h as RUN_SURFACE, l as WIZARD_BENCHMARK_FILE, s as logToFile, t as configureLogFile, u as WIZARD_LOG_FILE, v as runtimeEnv } from "./debug-Ch4BDOtD.js";
-import { t as VERSION } from "./version-aXxpgRKx.js";
-import { B as WIZARD_CONTACT_EMAIL, G as WIZARD_ORCHESTRATOR_FLAG_KEY, J as WIZARD_PROVISIONING_SCOPES, K as WIZARD_ORCHESTRATOR_OVERRIDE_FLAG_KEY, L as SONNET_5_MODEL, R as Sequence, S as Harness, W as WIZARD_OAUTH_SCOPES, Z as WIZARD_SELF_DRIVING_USE_PI_HARNESS_FLAG_KEY, _ as GPT5_6_LUNA_MODEL, b as HAIKU_MODEL, l as DEFAULT_AGENT_MODEL, q as WIZARD_ORCHESTRATOR_SEEDED_TASKS_FLAG_KEY, v as GPT5_6_SOL_MODEL, y as GPT5_6_TERRA_MODEL } from "./constants-jFI2SaE1.js";
+import { U as installSkillById, et as fetchSkillMenu, i as runAgent$1, lt as createTriageLLMProvider, nt as LONGER_ASK_TIMEOUT_MS, r as initializeAgent, rt as createWizardAskBridge } from "./agent-interface-FkbaBJwz.js";
+import { n as ErrorCodes } from "./emit-BSetXzDV.js";
+import { a as getLogFilePath, h as RUN_SURFACE, l as WIZARD_BENCHMARK_FILE, s as logToFile, t as configureLogFile, u as WIZARD_LOG_FILE, v as runtimeEnv } from "./debug-kAHwWGg7.js";
+import { t as VERSION } from "./version-CHhJjz_H.js";
+import { B as WIZARD_CONTACT_EMAIL, G as WIZARD_ORCHESTRATOR_FLAG_KEY, J as WIZARD_PROVISIONING_SCOPES, K as WIZARD_ORCHESTRATOR_OVERRIDE_FLAG_KEY, L as SONNET_5_MODEL, R as Sequence, S as Harness, W as WIZARD_OAUTH_SCOPES, Z as WIZARD_SELF_DRIVING_USE_PI_HARNESS_FLAG_KEY, _ as GPT5_6_LUNA_MODEL, b as HAIKU_MODEL, l as DEFAULT_AGENT_MODEL, q as WIZARD_ORCHESTRATOR_SEEDED_TASKS_FLAG_KEY, v as GPT5_6_SOL_MODEL, y as GPT5_6_TERRA_MODEL } from "./constants-DHCBSfnx.js";
 import "./wizard-session-2w4Ru-Wz.js";
-import { a as ciExcludedTaskTypes, n as analytics } from "./analytics-CHCZ_qUR.js";
-import { d as mcpUrlFor, p as detectNodePackageManagers } from "./setup-utils-CmTCTPkX.js";
-import { n as currentAccessToken } from "./oauth-session-DJ8h6GFa.js";
-import { c as configureGatewayFromCIEnvironment, l as gatewayAuth } from "./gateway-CjoShdLa.js";
-import { a as requireKnownModel, r as isValidModel } from "./models-DDdBJKwT.js";
-import { U as installSkillById, et as fetchSkillMenu, i as runAgent$1, lt as createTriageLLMProvider, nt as LONGER_ASK_TIMEOUT_MS, r as initializeAgent, rt as createWizardAskBridge } from "./agent-interface-BAOIHuXS.js";
-import { a as piBackend, d as createEmitSpinner, f as createProgressCollector, l as createAioCapture, u as createEmitLog } from "./pi-D3h5Ok0i.js";
-import { a as flushScanReport, s as formatYaraAbortMessage } from "./yara-hooks-DCeMDd1o.js";
-import { _ as TaskStatus, d as uncoveredBySink, h as SkipReason, m as QueueStore, p as QUEUE_DIR_NAME, u as dependencyClosure } from "./queue-tools-9OOVx5TE.js";
-import { a as assembleTaskPrompt, d as taskModelSpec, i as assembleSeedPrompt, n as agentRunTools, o as loadAgentRegistry, s as promptModelFor, t as ASK_TOOL, u as resolveTask } from "./agent-prompt-loader-7Pd4CfZO.js";
+import { a as ciExcludedTaskTypes, n as analytics } from "./analytics-DXK1gc12.js";
+import { a as WizardError, r as skillErrorCode, t as classifyRunFailure } from "./errors-C2domSkK.js";
+import { g as detectNodePackageManagers, m as mcpUrlFor } from "./setup-utils-B26Lf2yF.js";
+import { n as currentAccessToken } from "./oauth-session-DYM2Hrl2.js";
+import { f as gatewayAuth, s as AgentSignals } from "./gateway-C9-GlLO9.js";
+import { a as requireKnownModel, o as triageModelFor, r as isValidModel } from "./models-WcwUnSKb.js";
+import { a as piBackend, d as createEmitSpinner, f as createProgressCollector, l as createAioCapture, u as createEmitLog } from "./pi-8KsYsPmM.js";
+import { a as flushScanReport, s as formatYaraAbortMessage } from "./yara-hooks-BnxejtXK.js";
+import { _ as TaskStatus, d as uncoveredBySink, h as SkipReason, m as QueueStore, p as QUEUE_DIR_NAME, u as dependencyClosure } from "./queue-tools-C8mXOh7e.js";
+import { a as assembleTaskPrompt, d as taskModelSpec, i as assembleSeedPrompt, n as agentRunTools, o as loadAgentRegistry, s as promptModelFor, t as ASK_TOOL, u as resolveTask } from "./agent-prompt-loader-D0weQ-7k.js";
 import fs, { cpSync, existsSync, mkdirSync, readdirSync, rmSync, writeFileSync } from "fs";
 import * as path$1 from "path";
 import path from "path";
@@ -96,12 +96,63 @@ async function prepareRun(config, input) {
 		}, config.binding.harness)
 	};
 }
+const ACTIVITY_LINE_CHARS = 100;
+function createTranscriptTail(emit) {
+	const collected = [];
+	let collectedChars = 0;
+	let resultText = "";
+	const collect = (text) => {
+		collected.push(text);
+		collectedChars += text.length;
+		while (collectedChars > 262144 && collected.length > 1) collectedChars -= collected.shift()?.length ?? 0;
+	};
+	const activity = (line) => emit({
+		kind: "activity",
+		line
+	});
+	return {
+		onMessage(message) {
+			if (message?.type === "assistant") {
+				const content = message.message?.content;
+				if (!Array.isArray(content)) return;
+				for (const block of content) if (block?.type === "text" && typeof block.text === "string") {
+					collect(block.text);
+					const line = block.text.trim();
+					if (line) activity(line.length > ACTIVITY_LINE_CHARS ? `${line.slice(0, ACTIVITY_LINE_CHARS)}…` : line);
+				} else if (block?.type === "tool_use") activity(formatToolUse(block));
+			} else if (message?.type === "result" && typeof message.result === "string") resultText = message.result;
+		},
+		finalize: () => void 0,
+		text: () => `${collected.join("\n")}\n${resultText}`
+	};
+}
+/** Put the tail in front of any other middleware, so both see every message. */
+function withTranscript(middleware, transcript) {
+	if (!transcript) return middleware;
+	if (!middleware) return transcript;
+	return {
+		onMessage(message) {
+			transcript.onMessage(message);
+			middleware.onMessage(message);
+		},
+		finalize(resultMessage, totalDurationMs) {
+			transcript.finalize(resultMessage, totalDurationMs);
+			return middleware.finalize(resultMessage, totalDurationMs);
+		}
+	};
+}
+function formatToolUse(block) {
+	const name = typeof block?.name === "string" ? block.name : "tool";
+	const input = block?.input ?? {};
+	const detail = input.file_path || input.pattern || input.path || "";
+	return detail ? `${name} ${detail}` : name;
+}
 //#endregion
 //#region src/agent/runner/harness/anthropic/index.ts
 const anthropicBackend = {
 	name: "anthropic",
 	async run(inputs) {
-		const { config: runConfig, input, boot, emit, prompt, spinner, askBridge, middleware, model } = inputs;
+		const { config: runConfig, input, boot, emit, prompt, spinner, askBridge, middleware, model, structured } = inputs;
 		const config = runConfig.run;
 		const { skillsBaseUrl, credentials, wizardFlags, wizardMetadata } = boot;
 		const { accessToken, host, projectApiKey } = credentials;
@@ -112,7 +163,7 @@ const anthropicBackend = {
 			apiHost: host.apiHost,
 			runTags: wizardMetadata
 		});
-		log.step("Initializing Claude agent...");
+		if (!structured) log.step("Initializing Claude agent...");
 		const agent = await initializeAgent({
 			workingDirectory: input.installDir,
 			posthogMcpUrl: host.mcpUrl,
@@ -129,19 +180,27 @@ const anthropicBackend = {
 			askBridge,
 			getPendingQuestion: askBridge?.getPendingQuestion,
 			askMaxQuestions: config.maxQuestions,
+			readOnly: config.readOnly,
 			allowedTools: runConfig.allowedTools,
 			disallowedTools: runConfig.disallowedTools,
 			modelOverride: model,
+			outputFormat: structured && {
+				type: "json_schema",
+				schema: structured.schema
+			},
 			capture,
 			emit
 		}, runOptions(input));
-		log.step(`Verbose logs: ${getLogFilePath()}`);
-		log.success("Agent initialized. Let's get cooking!");
+		if (!structured) {
+			log.step(`Verbose logs: ${getLogFilePath()}`);
+			log.success("Agent initialized. Let's get cooking!");
+		}
 		logToFile("[agent-runner] agent initialized");
 		return runAgent$1({
 			...agent,
 			signal: inputs.signal
 		}, prompt, runOptions(input), spinner, {
+			timeoutMs: structured?.timeoutMs,
 			estimatedDurationMinutes: config.estimatedDurationMinutes,
 			spinnerMessage: config.spinnerMessage,
 			successMessage: config.successMessage,
@@ -149,6 +208,7 @@ const anthropicBackend = {
 			abortCases: config.abortCases,
 			emitStepEvents: config.trackStepProgress ?? false,
 			resolveStepKey: config.resolveStepKey,
+			requestRemark: structured ? false : config.requestRemark,
 			triageProvider: boot.triageProvider
 		}, middleware);
 	},
@@ -1380,6 +1440,20 @@ function createBenchmarkPipeline(emit, spinner, options, configOverride) {
 	});
 }
 //#endregion
+//#region src/agent/error-map.ts
+const AGENT_ERROR_CODE = {
+	["WIZARD_AGENTIC_DETECTION_TIMEOUT"]: ErrorCodes.AgenticDetectionTimeout,
+	["WIZARD_MCP_MISSING"]: ErrorCodes.AgentMcpMissing,
+	["WIZARD_RESOURCE_MISSING"]: ErrorCodes.AgentResourceMissing,
+	["WIZARD_RATE_LIMIT"]: ErrorCodes.AgentRateLimit,
+	["WIZARD_API_ERROR"]: ErrorCodes.AgentApiError,
+	["WIZARD_YARA_VIOLATION"]: ErrorCodes.AgentYaraViolation,
+	["WIZARD_ABORT"]: ErrorCodes.AgentAbort,
+	["WIZARD_NO_PROGRESS"]: ErrorCodes.AgentNoProgress,
+	["WIZARD_INVALID_STRUCTURED_OUTPUT"]: ErrorCodes.AgentInvalidStructuredOutput,
+	["WIZARD_INCOMPLETE_TASKS"]: ErrorCodes.AgentIncompleteTasks
+};
+//#endregion
 //#region src/agent/agent-prompt.ts
 function defaultProjectPrompt(ctx) {
 	return `You have access to the PostHog MCP server.
@@ -1475,8 +1549,8 @@ async function runLinearProgram(context) {
 		controller.abort();
 	}
 }
-/** The host's `signal` decides the outcome; `runSignal` also ends with the run. */
-async function executeLinear({ config, input, boot, emit, interaction, signal }, runSignal) {
+/** The caller's `signal` decides the outcome; `runSignal` also ends with the run. */
+async function executeLinear({ config, input, boot, emit, interaction, signal, transcript }, runSignal) {
 	const { run, composed } = config;
 	const { skillsBaseUrl, credentials, project } = boot;
 	const { projectApiKey, host, projectId } = credentials;
@@ -1498,18 +1572,18 @@ async function executeLinear({ config, input, boot, emit, interaction, signal },
 		logToFile(`[agent-runner] skill installed at ${skillPath}`);
 	}
 	const spinner = createEmitSpinner(emit);
-	emit({
+	if (!run.structured) emit({
 		kind: "lifecycle",
 		phase: "started"
 	});
-	const ask = shouldDisableAsk(input.flags) && process.env.WIZARD_ASK_AUTODRIVE !== "1" ? void 0 : createAskBridge(interaction, {
+	const ask = !!run.structured || shouldDisableAsk(input.flags) && process.env.WIZARD_ASK_AUTODRIVE !== "1" ? void 0 : createAskBridge(interaction, {
 		getSource: () => input.skillId ?? run.integrationLabel,
 		richLinks: run.richLinks ?? false,
 		timeoutMs: run.askTimeoutMs,
 		signal: runSignal
 	});
-	const middleware = input.flags.benchmark ? createBenchmarkPipeline(emit, spinner, runOptions(input)) : void 0;
-	const prompt = assemblePrompt(run, {
+	const middleware = withTranscript(input.flags.benchmark ? createBenchmarkPipeline(emit, spinner, runOptions(input)) : void 0, transcript);
+	const promptContext = {
 		projectId,
 		projectApiKey,
 		host,
@@ -1520,7 +1594,8 @@ async function executeLinear({ config, input, boot, emit, interaction, signal },
 			exceptionAutocapture: project.autocapture_exceptions_opt_in ?? null,
 			surveys: project.surveys_opt_in ?? null
 		} : null
-	});
+	};
+	const prompt = run.prompt ? run.prompt(promptContext) : assemblePrompt(run, promptContext);
 	logToFile(`[agent-runner] prompt assembled (${prompt.length} chars)`);
 	if (signal?.aborted) return aborted();
 	const { harness, model, thinkingLevel } = config.binding;
@@ -1536,7 +1611,8 @@ async function executeLinear({ config, input, boot, emit, interaction, signal },
 		middleware,
 		model,
 		thinkingLevel,
-		signal: runSignal
+		signal: runSignal,
+		structured: run.structured
 	});
 	if (signal?.aborted && agentResult.kind === "success") return aborted();
 	if (agentResult.kind === "decided_failure") return failed(agentResult.failure);
@@ -1640,6 +1716,10 @@ Please try again, or check the documentation:\n${run.docsUrl}`,
 		message: agentResult.message ?? "Agent failed",
 		error: agentResult.error
 	});
+	if (run.structured) return {
+		outcome: "success",
+		structuredOutput: agentResult.structuredOutput
+	};
 	if (config.hooks?.postRun) {
 		await config.hooks.postRun(credentials);
 		if (signal?.aborted) return aborted();
@@ -2923,6 +3003,15 @@ function resolveBinding(ctx, role = "default") {
 		thinkingLevel
 	};
 }
+/** A scan's two attempts on triage models: the resolved harness, then the SDK as a second provider. */
+function resolveScanBindings(ctx) {
+	const scan = (harness) => ({
+		sequence: "linear",
+		harness,
+		model: triageModelFor(harness)
+	});
+	return [scan(resolveBinding(ctx).harness), scan("anthropic")];
+}
 //#endregion
 //#region src/agent/runner/index.ts
 /**
@@ -2934,9 +3023,16 @@ function resolveBinding(ctx, role = "default") {
 */
 async function runAgent(config, input, options = {}) {
 	let collector;
+	let transcript;
 	const snapshot = () => {
 		try {
-			if (collector) return collector.snapshot();
+			if (collector) {
+				const collected = collector.snapshot();
+				return transcript ? {
+					...collected,
+					transcriptTail: transcript.text()
+				} : collected;
+			}
 		} catch {}
 		return {
 			tasks: [],
@@ -2950,6 +3046,7 @@ async function runAgent(config, input, options = {}) {
 		};
 	};
 	const flushReport = () => {
+		if (config.scanReport === "defer") return;
 		try {
 			const report = flushScanReport({ yaraReport: input.flags.yaraReport });
 			if (report) collector?.emit({
@@ -2963,6 +3060,7 @@ async function runAgent(config, input, options = {}) {
 	try {
 		collector = createProgressCollector(options.onProgress);
 		const { emit } = collector;
+		if (config.run.collectTranscript) transcript = createTranscriptTail(emit);
 		const log = (message) => emit({
 			kind: "log",
 			level: "info",
@@ -2991,7 +3089,8 @@ async function runAgent(config, input, options = {}) {
 			boot,
 			emit,
 			interaction: options.interaction,
-			signal: options.signal
+			signal: options.signal,
+			transcript
 		});
 		result = {
 			...options.signal?.aborted && sequenceResult.outcome === "success" ? {
@@ -3065,21 +3164,15 @@ function safeErrorMessage(error) {
 }
 //#endregion
 //#region src/agent/index.ts
-var agent_exports = /* @__PURE__ */ __exportAll({
-	LONGER_ASK_TIMEOUT_MS: () => LONGER_ASK_TIMEOUT_MS,
-	TASK_OUTCOMES_KEY: () => TASK_OUTCOMES_KEY,
-	configureGatewayFromCIEnvironment: () => configureGatewayFromCIEnvironment,
-	runMcpPromptViaSdk: () => runMcpPromptViaSdk
-});
 /**
 * Leaves in C2. The TUI receives agent data through program state. Until
 * then the suggested-prompts screen streams through this wrapper, which loads
 * the streaming module on first call so the startup chunk does not grow.
 */
 async function* runMcpPromptViaSdk(args) {
-	yield* (await import("./mcp-prompt-streaming-I2U_X7BC.js")).runMcpPromptViaSdk(args);
+	yield* (await import("./mcp-prompt-streaming-DDWQ-FCA.js")).runMcpPromptViaSdk(args);
 }
 //#endregion
-export { formatCostUsd as a, computeTokenCostUsd as i, runAgent as n, formatTokenCount as o, resolveBinding as r, shouldDisableAsk as s, agent_exports as t };
+export { computeTokenCostUsd as a, shouldDisableAsk as c, resolveScanBindings as i, runAgent as n, formatCostUsd as o, resolveBinding as r, formatTokenCount as s, runMcpPromptViaSdk as t };
 
-//# sourceMappingURL=agent-bCZ3i3fh.js.map
+//# sourceMappingURL=agent-BWFhuLgo.js.map