@prometheus-ai/agent 0.5.4 → 0.5.8

package/src/commands/models.ts

@@ -0,0 +1,61 @@
+/**
+ * List, search, and refresh available models.
+ */
+import { APP_NAME } from "@prometheus-ai/utils";
+import { Args, Command, Flags } from "@prometheus-ai/utils/cli";
+import { resolveModelsArgs, runModelsCommand } from "../cli/models-cli";
+
+export default class Models extends Command {
+	static description = "List, search, and refresh available models";
+
+	static args = {
+		action: Args.string({
+			description: "ls (default) | find | refresh | canonical | <provider>",
+			required: false,
+		}),
+		pattern: Args.string({
+			description: "Filter/search substring, or provider name (required for find)",
+			required: false,
+		}),
+	};
+
+	static flags = {
+		json: Flags.boolean({ description: "Output JSON" }),
+		extension: Flags.string({
+			char: "e",
+			description: "Load an extension file before listing (repeatable)",
+			multiple: true,
+		}),
+		"no-extensions": Flags.boolean({
+			description: "Disable extension discovery (explicit -e paths still work)",
+		}),
+		config: Flags.string({
+			description: "Load an extra config.yml-style overlay for this run (repeatable)",
+			multiple: true,
+		}),
+	};
+
+	static examples = [
+		`# List every available model, grouped by provider\n  ${APP_NAME} models`,
+		`# List one provider's models (any provider name works)\n  ${APP_NAME} models openai-codex`,
+		`# Find models by substring\n  ${APP_NAME} models find minimax`,
+		`# Force a fresh catalog fetch (replaces rm -rf ~/.prometheus/models.db)\n  ${APP_NAME} models refresh`,
+		`# Show the coalesced canonical model view\n  ${APP_NAME} models canonical`,
+		`# Machine-readable output\n  ${APP_NAME} models --json`,
+	];
+
+	async run(): Promise<void> {
+		const { args, flags } = await this.parse(Models);
+		const { action, pattern } = resolveModelsArgs(args.action, args.pattern);
+		await runModelsCommand({
+			action,
+			pattern,
+			flags: {
+				json: flags.json,
+				extensions: flags.extension,
+				noExtensions: flags["no-extensions"],
+				config: flags.config,
+			},
+		});
+	}
+}

package/src/commands/read.ts

@@ -1,16 +1,17 @@
 /**
- * Show what the read tool will return for a given path.
+ * Show what the read tool will return for a path, URL, or internal URI.
  */
 import { Args, Command } from "@prometheus-ai/utils/cli";
 import { type ReadCommandArgs, runReadCommand } from "../cli/read-cli";
 import { initTheme } from "../modes/theme/theme";
 
 export default class Read extends Command {
-	static description = "Show what the read tool will return for a path or URL";
+	static description = "Show what the read tool will return for a path, URL, or internal URI";
 
 	static args = {
 		path: Args.string({
-			description: "Path or URL to read (append :sel for line ranges or raw mode, e.g. src/foo.ts:50-100)",
+			description:
+				"Path, URL, or internal URI to read (append :sel for line ranges or raw mode, e.g. src/foo.ts:50-100)",
 			required: true,
 		}),
 	};
@@ -20,6 +21,8 @@ export default class Read extends Command {
 		"prometheus read src/foo.ts:50-100",
 		"prometheus read src/foo.ts:raw",
 		"prometheus read https://example.com",
+		"prometheus read prometheus://",
+		"prometheus read issue://123",
 		"prometheus read path/to/archive.zip:dir/file.ts",
 		"prometheus read path/to/db.sqlite:users:42",
 	];

package/src/commands/say.ts

@@ -0,0 +1,102 @@
+/**
+ * Synthesize text with the local TTS engine and play it (or save it with --out).
+ *
+ * Demonstrates the on-device speech stack end to end: the first run downloads
+ * the configured local model, synthesis happens in the TTS worker subprocess,
+ * and the resulting WAV is either played through the speakers or written to disk.
+ */
+import * as os from "node:os";
+import * as path from "node:path";
+import { getProjectDir, Snowflake } from "@prometheus-ai/utils";
+import { Args, Command, Flags } from "@prometheus-ai/utils/cli";
+import chalk from "chalk";
+import { Settings, settings } from "../config/settings";
+import { playAudioFile, removeTempFile } from "../tts/player";
+import { shutdownTtsClient, ttsClient } from "../tts/tts-client";
+import { encodeWav } from "../tts/wav";
+
+export default class Say extends Command {
+	static description = "Synthesize text with the local TTS engine and play it through the speakers";
+
+	static args = {
+		text: Args.string({ required: true, description: "Text to speak" }),
+	};
+
+	static flags = {
+		voice: Flags.string({ description: "Voice id" }),
+		model: Flags.string({ description: "Local TTS model key" }),
+		out: Flags.string({ char: "o", description: "Write WAV to this path instead of playing" }),
+	};
+
+	static examples = [
+		'prometheus say "hello world"',
+		'prometheus say "hello world" --out /tmp/hello.wav',
+		'prometheus say "bonjour" --voice af_heart --model kokoro',
+	];
+
+	async run(): Promise<void> {
+		const { args, flags } = await this.parse(Say);
+		const text = args.text ?? "";
+
+		await Settings.init({ cwd: getProjectDir() });
+		const model = flags.model ?? settings.get("tts.localModel");
+		const voice = flags.voice ?? settings.get("tts.localVoice");
+
+		let exitCode = 0;
+		const unsubscribe = ttsClient.onProgress(event => {
+			if (event.status === "progress" && typeof event.progress === "number") {
+				process.stderr.write(
+					`\r${chalk.dim(`downloading ${event.file ?? model}: ${Math.round(event.progress)}%`)}`,
+				);
+			} else if (event.status === "done" || event.status === "ready") {
+				// Clear the progress line once the download finishes.
+				process.stderr.write("\r\x1b[K");
+			}
+		});
+
+		try {
+			const audio = await ttsClient.synthesize(model, text, { voice });
+			if (!audio) {
+				process.stderr.write(
+					chalk.red(
+						`error: could not synthesize with local TTS model "${model}". ` +
+							"Run `prometheus setup speech` to install it.\n",
+					),
+				);
+				exitCode = 1;
+				return;
+			}
+
+			const wav = encodeWav(audio.pcm, audio.sampleRate);
+			const durationSec = audio.pcm.length / audio.sampleRate;
+
+			if (flags.out) {
+				await Bun.write(flags.out, wav);
+				process.stdout.write(
+					`${chalk.green("saved")} ${flags.out} ` +
+						`${chalk.dim(`(${voice}, ${model}, ${durationSec.toFixed(1)}s, ${wav.byteLength} bytes)`)}\n`,
+				);
+				return;
+			}
+
+			const tmp = path.join(os.tmpdir(), `prometheus-say-${Snowflake.next()}.wav`);
+			await Bun.write(tmp, wav);
+			try {
+				await playAudioFile(tmp);
+				process.stdout.write(
+					`${chalk.green("spoke")} ${chalk.dim(`(${voice}, ${model}, ${durationSec.toFixed(1)}s)`)}\n`,
+				);
+			} finally {
+				await removeTempFile(tmp);
+			}
+		} catch (err) {
+			process.stderr.write(chalk.red(`error: ${err instanceof Error ? err.message : String(err)}\n`));
+			exitCode = 1;
+		} finally {
+			unsubscribe();
+			await shutdownTtsClient();
+		}
+
+		if (exitCode !== 0) process.exit(exitCode);
+	}
+}

package/src/commands/setup.ts

@@ -7,7 +7,7 @@ import { runSetupCommand, type SetupCommandArgs, type SetupComponent } from "../
 import { runRootCommand } from "../main";
 import { initTheme } from "../modes/theme/theme";
 
-const COMPONENTS: SetupComponent[] = ["python", "stt"];
+const COMPONENTS: SetupComponent[] = ["python", "speech"];
 
 export interface OnboardingSetupDependencies {
 	runRoot?: typeof runRootCommand;

package/src/commands/token.ts

@@ -0,0 +1,89 @@
+/**
+ * Get the API key or OAuth token for a provider.
+ */
+
+import { PROVIDER_REGISTRY } from "@prometheus-ai/ai";
+import { Args, Command, Flags } from "@prometheus-ai/utils/cli";
+import chalk from "chalk";
+import { isAuthenticated, ModelRegistry } from "../config/model-registry";
+import { discoverAuthStorage } from "../sdk";
+
+export default class Token extends Command {
+	static description = "Get the API key or OAuth token for a provider";
+
+	static args = {
+		provider: Args.string({
+			description: "Provider ID (e.g. anthropic, openai)",
+			required: true,
+		}),
+	};
+
+	static flags = {
+		raw: Flags.boolean({
+			description: "Output the raw credential value without parsing nested JSON structures",
+			default: false,
+		}),
+		"force-refresh": Flags.boolean({
+			description: "Force refresh the OAuth token even if it has not expired",
+			default: false,
+		}),
+	};
+
+	static examples = [
+		"# Get API key for Anthropic\n  prometheus token anthropic",
+		"# Get raw Copilot credential JSON\n  prometheus token github-copilot --raw",
+		"# Force refresh and get Gemini CLI token\n  prometheus token google-gemini-cli --force-refresh",
+	];
+
+	async run(): Promise<void> {
+		const { args, flags } = await this.parse(Token);
+		const providerName = args.provider ?? "";
+		const provider = providerName.toLowerCase();
+
+		const authStorage = await discoverAuthStorage();
+		const modelRegistry = new ModelRegistry(authStorage);
+
+		// Resolve the API key / token
+		const apiKey = await modelRegistry.getApiKeyForProvider(provider, undefined, {
+			forceRefresh: flags["force-refresh"],
+		});
+
+		if (!isAuthenticated(apiKey)) {
+			// Find all active/configured providers
+			const activeProviders = new Set<string>();
+			for (const p of PROVIDER_REGISTRY) {
+				if (authStorage.hasAuth(p.id)) {
+					activeProviders.add(p.id);
+				}
+			}
+			const all = authStorage.getAll();
+			for (const p in all) {
+				if (authStorage.hasAuth(p)) {
+					activeProviders.add(p);
+				}
+			}
+
+			const msg = `No active credential found for provider "${providerName}".`;
+			process.stderr.write(`${chalk.red(msg)}\n`);
+			if (activeProviders.size > 0) {
+				process.stderr.write(`Configured providers: ${Array.from(activeProviders).sort().join(", ")}\n`);
+			}
+			process.exitCode = 1;
+			return;
+		}
+
+		if (!flags.raw) {
+			try {
+				const parsed = JSON.parse(apiKey);
+				if (parsed && typeof parsed === "object" && typeof parsed.token === "string") {
+					process.stdout.write(`${parsed.token}\n`);
+					return;
+				}
+			} catch {
+				// Not a JSON string, print as-is
+			}
+		}
+
+		process.stdout.write(`${apiKey}\n`);
+	}
+}

package/src/commands/usage.ts

@@ -0,0 +1,43 @@
+/**
+ * Show provider usage limits for every authenticated account.
+ */
+import { Command, Flags } from "@prometheus-ai/utils/cli";
+import { runUsageCommand } from "../cli/usage-cli";
+
+export default class Usage extends Command {
+	static description = "Show provider usage limits for every authenticated account";
+
+	static flags = {
+		json: Flags.boolean({ char: "j", description: "Output usage reports as JSON", default: false }),
+		provider: Flags.string({ char: "p", description: "Only show usage for this provider id (e.g. anthropic)" }),
+		redact: Flags.boolean({
+			char: "r",
+			description: "Redact account emails/ids (shortest unique prefix) for sharing screenshots",
+			default: false,
+		}),
+		history: Flags.boolean({
+			description: "Show recorded usage-limit history (hourly snapshots) instead of a live snapshot",
+			default: false,
+		}),
+		days: Flags.integer({ char: "d", description: "History window in days (with --history)", default: 7 }),
+	};
+
+	static examples = [
+		"# Detailed per-account usage breakdown across all providers\n  prometheus usage",
+		"# Only Anthropic accounts\n  prometheus usage --provider anthropic",
+		"# Redact account identifiers for screenshots\n  prometheus usage --redact",
+		"# Machine-readable output\n  prometheus usage --json",
+		"# Usage-limit trend over the last 30 days\n  prometheus usage --history --days 30",
+	];
+
+	async run(): Promise<void> {
+		const { flags } = await this.parse(Usage);
+		await runUsageCommand({
+			json: flags.json,
+			provider: flags.provider,
+			redact: flags.redact,
+			history: flags.history,
+			days: flags.days,
+		});
+	}
+}

package/src/commit/agentic/agent.ts

@@ -170,6 +170,7 @@ export async function runCommitAgentSession(input: CommitAgentInput): Promise<Co
 			await session.prompt(reminder, {
 				attribution: "agent",
 				expandPromptTemplates: false,
+				synthetic: true,
 			});
 		}
 
@@ -212,7 +213,7 @@ function writeAssistantMessage(message: string): void {
 	}
 }
 
-function renderMarkdownLines(message: string): string[] {
+function renderMarkdownLines(message: string): readonly string[] {
 	const width = Math.max(40, process.stdout.columns ?? 100);
 	const markdown = new Markdown(message, 0, 0, getMarkdownTheme());
 	return markdown.render(width);

package/src/commit/agentic/tools/analyze-file.ts

@@ -1,5 +1,5 @@
 import { prompt } from "@prometheus-ai/utils";
-import * as z from "zod/v4";
+import { z } from "zod/v4";
 import analyzeFilePrompt from "../../../commit/agentic/prompts/analyze-file.md" with { type: "text" };
 import type { CommitAgentState } from "../../../commit/agentic/state";
 import type { NumstatEntry } from "../../../commit/types";
@@ -38,11 +38,17 @@ function buildToolSession(
 	return {
 		cwd: options.cwd,
 		hasUI: false,
+		// Programmatic fan-out: results feed the commit agent's evidence, not a
+		// model choosing further spawns, so the specialization nudge is noise here.
+		suppressSpawnAdvisory: true,
 		getSessionFile: () => ctx.sessionManager.getSessionFile() ?? null,
 		getSessionSpawns: () => options.spawns,
 		settings: options.settings,
 		authStorage: options.authStorage,
 		modelRegistry: options.modelRegistry,
+		// The task tool no longer takes a per-call schema; the inherited session
+		// schema drives structured output for every spawn from this session.
+		outputSchema: analyzeFileOutputSchema,
 	};
 }
 
@@ -59,29 +65,45 @@ export function createAnalyzeFileTool(options: {
 		label: "Analyze Files",
 		description: "Spawn quick_task agents to analyze files.",
 		parameters: analyzeFileSchema,
-		async execute(toolCallId, params, onUpdate, ctx, signal) {
+		async execute(toolCallId, params, _onUpdate, ctx, signal) {
 			const toolSession = buildToolSession(ctx, options);
+			// The hand-built ToolSession carries no asyncJobManager, so every
+			// execute() below takes the task tool's sync fallback and resolves
+			// with the subagent's result inline — exactly what this flow needs.
+			// The tool's session semaphore bounds the parallel fan-out.
 			const taskTool = await TaskTool.create(toolSession);
 			const numstat = options.state.overview?.numstat ?? [];
-			const tasks = params.files.map((file, index) => {
-				const relatedFiles = formatRelatedFiles(params.files, file, numstat);
-				const assignment = prompt.render(analyzeFilePrompt, {
-					file,
-					goal: params.goal,
-					related_files: relatedFiles,
-				});
-				return {
-					id: `AnalyzeFile${index + 1}`,
-					description: `Analyze ${file}`,
-					assignment,
-				};
-			});
-			const taskParams: TaskParams = {
-				agent: "quick_task",
-				schema: JSON.stringify(analyzeFileOutputSchema),
-				tasks,
+
+			const analyses = await Promise.all(
+				params.files.map((file, index) => {
+					const relatedFiles = formatRelatedFiles(params.files, file, numstat);
+					const assignment = prompt.render(analyzeFilePrompt, {
+						file,
+						goal: params.goal,
+						related_files: relatedFiles,
+					});
+					const taskParams: TaskParams = {
+						agent: "quick_task",
+						id: `AnalyzeFile${index + 1}`,
+						description: `Analyze ${file}`,
+						assignment,
+					};
+					return taskTool.execute(`${toolCallId}-${index + 1}`, taskParams, signal);
+				}),
+			);
+			const results = analyses.flatMap(analysis => analysis.details?.results ?? []);
+			const text = analyses
+				.map(analysis => analysis.content.find(part => part.type === "text")?.text ?? "")
+				.filter(Boolean)
+				.join("\n\n");
+			return {
+				content: [{ type: "text", text: text || "(no output)" }],
+				details: {
+					projectAgentsDir: null,
+					results,
+					totalDurationMs: analyses.reduce((sum, analysis) => sum + (analysis.details?.totalDurationMs ?? 0), 0),
+				},
 			};
-			return taskTool.execute(toolCallId, taskParams, signal, onUpdate);
 		},
 	};
 }

package/src/commit/agentic/tools/git-file-diff.ts

@@ -1,4 +1,4 @@
-import * as z from "zod/v4";
+import { z } from "zod/v4";
 import type { CommitAgentState } from "../../../commit/agentic/state";
 import type { CustomTool } from "../../../extensibility/custom-tools/types";
 import * as git from "../../../utils/git";

package/src/commit/agentic/tools/git-hunk.ts

@@ -1,4 +1,4 @@
-import * as z from "zod/v4";
+import { z } from "zod/v4";
 import type { DiffHunk, FileHunks } from "../../../commit/types";
 import type { CustomTool } from "../../../extensibility/custom-tools/types";
 import * as git from "../../../utils/git";

package/src/commit/agentic/tools/git-overview.ts

@@ -1,4 +1,4 @@
-import * as z from "zod/v4";
+import { z } from "zod/v4";
 import type { CommitAgentState, GitOverviewSnapshot } from "../../../commit/agentic/state";
 import { extractScopeCandidates } from "../../../commit/analysis/scope";
 import type { CustomTool } from "../../../extensibility/custom-tools/types";

package/src/commit/agentic/tools/propose-changelog.ts

@@ -1,4 +1,4 @@
-import * as z from "zod/v4";
+import { z } from "zod/v4";
 import type { CommitAgentState } from "../../../commit/agentic/state";
 import { CHANGELOG_CATEGORIES, type ChangelogCategory } from "../../../commit/types";
 import type { CustomTool } from "../../../extensibility/custom-tools/types";

package/src/commit/agentic/tools/propose-commit.ts

@@ -1,4 +1,4 @@
-import * as z from "zod/v4";
+import { z } from "zod/v4";
 import type { CommitAgentState } from "../../../commit/agentic/state";
 import {
 	capDetails,

package/src/commit/agentic/tools/recent-commits.ts

@@ -1,4 +1,4 @@
-import * as z from "zod/v4";
+import { z } from "zod/v4";
 import type { CustomTool } from "../../../extensibility/custom-tools/types";
 import * as git from "../../../utils/git";
 

package/src/commit/agentic/tools/schemas.ts

@@ -1,4 +1,4 @@
-import * as z from "zod/v4";
+import { z } from "zod/v4";
 
 export const commitTypeSchema = z.enum([
 	"feat",

package/src/commit/agentic/tools/split-commit.ts

@@ -1,4 +1,4 @@
-import * as z from "zod/v4";
+import { z } from "zod/v4";
 import type { CommitAgentState, SplitCommitGroup, SplitCommitPlan } from "../../../commit/agentic/state";
 import { computeDependencyOrder } from "../../../commit/agentic/topo-sort";
 import {
@@ -68,6 +68,7 @@ export function createSplitCommitTool(
 			const errors: string[] = [];
 			const warnings: string[] = [];
 			const diffText = await git.diff(cwd, { cached: true });
+			const validateHunksForDiff = git.createHunkSelectionValidator(diffText);
 
 			const commits: SplitCommitGroup[] = params.commits.map((commit, index) => {
 				const scope = commit.scope?.trim() || null;
@@ -102,7 +103,7 @@ export function createSplitCommitTool(
 				}
 				warnings.push(...summaryValidation.warnings.map(warning => `Commit ${index + 1}: ${warning}`));
 				warnings.push(...typeValidation.warnings.map(warning => `Commit ${index + 1}: ${warning}`));
-				const hunkValidation = validateHunkSelectors(index, changes, files);
+				const hunkValidation = validateHunkSelectors(index, changes, files, validateHunksForDiff);
 				warnings.push(...hunkValidation.warnings);
 				errors.push(...hunkValidation.errors);
 				errors.push(...validateDependencies(index, dependencies, params.commits.length));
@@ -186,6 +187,7 @@ function validateHunkSelectors(
 	commitIndex: number,
 	changes: SplitCommitGroup["changes"],
 	files: string[],
+	validateHunksForDiff: (changes: SplitCommitGroup["changes"]) => git.HunkSelectionValidationError[],
 ): { errors: string[]; warnings: string[] } {
 	const errors: string[] = [];
 	const warnings: string[] = [];
@@ -215,6 +217,11 @@ function validateHunkSelectors(
 			}
 		}
 	}
+	if (errors.length === 0) {
+		for (const error of validateHunksForDiff(changes)) {
+			errors.push(`${prefix}: ${error.message}`);
+		}
+	}
 	return { errors, warnings };
 }
 

package/src/commit/analysis/conventional.ts

@@ -1,5 +1,5 @@
 import type { ThinkingLevel } from "@prometheus-ai/agent-core";
-import type { Api, Model } from "@prometheus-ai/ai";
+import type { Api, ApiKey, Model } from "@prometheus-ai/ai";
 import { completeSimple } from "@prometheus-ai/ai";
 import { prompt } from "@prometheus-ai/utils";
 import analysisSystemPrompt from "../../commit/prompts/analysis-system.md" with { type: "text" };
@@ -14,7 +14,7 @@ const ConventionalAnalysisTool = createConventionalAnalysisTool(
 
 export interface ConventionalAnalysisInput {
 	model: Model<Api>;
-	apiKey: string;
+	apiKey: ApiKey;
 	thinkingLevel?: ThinkingLevel;
 	contextFiles?: Array<{ path: string; content: string }>;
 	userContext?: string;

package/src/commit/analysis/summary.ts

@@ -1,8 +1,8 @@
 import type { ThinkingLevel } from "@prometheus-ai/agent-core";
-import type { Api, AssistantMessage, Model } from "@prometheus-ai/ai";
+import type { Api, ApiKey, AssistantMessage, Model } from "@prometheus-ai/ai";
 import { completeSimple, validateToolCall } from "@prometheus-ai/ai";
 import { prompt } from "@prometheus-ai/utils";
-import * as z from "zod/v4";
+import { z } from "zod/v4";
 import summarySystemPrompt from "../../commit/prompts/summary-system.md" with { type: "text" };
 import summaryUserPrompt from "../../commit/prompts/summary-user.md" with { type: "text" };
 import type { CommitSummary } from "../../commit/types";
@@ -19,7 +19,7 @@ const SummaryTool = {
 
 export interface SummaryInput {
 	model: Model<Api>;
-	apiKey: string;
+	apiKey: ApiKey;
 	thinkingLevel?: ThinkingLevel;
 	commitType: string;
 	scope: string | null;

package/src/commit/changelog/generate.ts

@@ -1,8 +1,8 @@
 import type { ThinkingLevel } from "@prometheus-ai/agent-core";
-import type { Api, AssistantMessage, Model } from "@prometheus-ai/ai";
+import type { Api, ApiKey, AssistantMessage, Model } from "@prometheus-ai/ai";
 import { completeSimple, validateToolCall } from "@prometheus-ai/ai";
 import { prompt } from "@prometheus-ai/utils";
-import * as z from "zod/v4";
+import { z } from "zod/v4";
 import changelogSystemPrompt from "../../commit/prompts/changelog-system.md" with { type: "text" };
 import changelogUserPrompt from "../../commit/prompts/changelog-user.md" with { type: "text" };
 import { CHANGELOG_CATEGORIES, type ChangelogCategory, type ChangelogGenerationResult } from "../../commit/types";
@@ -25,7 +25,7 @@ export const changelogTool = {
 
 export interface ChangelogPromptInput {
 	model: Model<Api>;
-	apiKey: string;
+	apiKey: ApiKey;
 	thinkingLevel?: ThinkingLevel;
 	changelogPath: string;
 	isPackageChangelog: boolean;

package/src/commit/changelog/index.ts

@@ -1,6 +1,6 @@
 import * as path from "node:path";
 import type { ThinkingLevel } from "@prometheus-ai/agent-core";
-import type { Api, Model } from "@prometheus-ai/ai";
+import type { Api, ApiKey, Model } from "@prometheus-ai/ai";
 import { logger } from "@prometheus-ai/utils";
 import { CHANGELOG_CATEGORIES } from "../../commit/types";
 import * as git from "../../utils/git";
@@ -15,7 +15,7 @@ const DEFAULT_MAX_DIFF_CHARS = 120_000;
 export interface ChangelogFlowInput {
 	cwd: string;
 	model: Model<Api>;
-	apiKey: string;
+	apiKey: ApiKey;
 	thinkingLevel?: ThinkingLevel;
 	stagedFiles: string[];
 	dryRun: boolean;

package/src/commit/map-reduce/index.ts

@@ -1,5 +1,5 @@
 import type { ThinkingLevel } from "@prometheus-ai/agent-core";
-import type { Api, Model } from "@prometheus-ai/ai";
+import type { Api, ApiKey, Model } from "@prometheus-ai/ai";
 import { $env } from "@prometheus-ai/utils";
 import { parseFileDiffs } from "../../commit/git/diff";
 import type { ConventionalAnalysis } from "../../commit/types";
@@ -21,10 +21,10 @@ export interface MapReduceSettings {
 
 export interface MapReduceInput {
 	model: Model<Api>;
-	apiKey: string;
+	apiKey: ApiKey;
 	thinkingLevel?: ThinkingLevel;
 	smolModel: Model<Api>;
-	smolApiKey: string;
+	smolApiKey: ApiKey;
 	smolThinkingLevel?: ThinkingLevel;
 	diff: string;
 	stat: string;

package/src/commit/map-reduce/map-phase.ts

@@ -1,5 +1,5 @@
 import type { ThinkingLevel } from "@prometheus-ai/agent-core";
-import type { Api, AssistantMessage, Message, Model } from "@prometheus-ai/ai";
+import type { Api, ApiKey, AssistantMessage, Message, Model } from "@prometheus-ai/ai";
 import { completeSimple } from "@prometheus-ai/ai";
 import { prompt } from "@prometheus-ai/utils";
 import fileObserverSystemPrompt from "../../commit/prompts/file-observer-system.md" with { type: "text" };
@@ -18,7 +18,7 @@ const RETRY_BACKOFF_MS = 1000;
 
 export interface MapPhaseInput {
 	model: Model<Api>;
-	apiKey: string;
+	apiKey: ApiKey;
 	thinkingLevel?: ThinkingLevel;
 	files: FileDiff[];
 	config?: {