@nudge-ai/cli 0.0.1-beta.3 → 0.1.0

package/dist/bin.mjs

@@ -1,26 +1,902 @@
 #!/usr/bin/env node
-import { t as generate } from "./src-DG37IBZ6.mjs";
+import { n as generate, r as improve, t as evaluate } from "./src-B7X5IQ4U.mjs";
 import * as fs from "fs";
 import * as path from "path";
+import { Box, Text, render, useApp } from "ink";
+import React, { useEffect, useState } from "react";
+import { Fragment, jsx, jsxs } from "react/jsx-runtime";
+import InkSpinner from "ink-spinner";
+import SelectInput from "ink-select-input";
+import TextInput from "ink-text-input";
 
-//#region src/bin.ts
+//#region src/components/Header.tsx
+function Header({ title, subtitle }) {
+	return /* @__PURE__ */ jsxs(Box, {
+		flexDirection: "column",
+		marginBottom: 1,
+		children: [/* @__PURE__ */ jsx(Text, {
+			bold: true,
+			color: "magenta",
+			children: title
+		}), subtitle && /* @__PURE__ */ jsxs(Text, {
+			dimColor: true,
+			children: [" ", subtitle]
+		})]
+	});
+}
+
+//#endregion
+//#region src/components/Spinner.tsx
+function Spinner({ label, color = "cyan" }) {
+	return /* @__PURE__ */ jsxs(Box, { children: [/* @__PURE__ */ jsx(Text, {
+		color,
+		children: /* @__PURE__ */ jsx(InkSpinner, { type: "dots" })
+	}), /* @__PURE__ */ jsxs(Text, { children: [" ", label] })] });
+}
+
+//#endregion
+//#region src/components/StatusMessage.tsx
+const STATUS_ICONS = {
+	success: "✓",
+	error: "✗",
+	warning: "⚠",
+	info: "ℹ"
+};
+const STATUS_COLORS = {
+	success: "green",
+	error: "red",
+	warning: "yellow",
+	info: "blue"
+};
+function StatusMessage({ type, children }) {
+	return /* @__PURE__ */ jsxs(Box, { children: [/* @__PURE__ */ jsxs(Text, {
+		color: STATUS_COLORS[type],
+		children: [STATUS_ICONS[type], " "]
+	}), /* @__PURE__ */ jsx(Text, { children })] });
+}
+function Success({ children }) {
+	return /* @__PURE__ */ jsx(StatusMessage, {
+		type: "success",
+		children
+	});
+}
+function Error({ children }) {
+	return /* @__PURE__ */ jsx(StatusMessage, {
+		type: "error",
+		children
+	});
+}
+function Warning({ children }) {
+	return /* @__PURE__ */ jsx(StatusMessage, {
+		type: "warning",
+		children
+	});
+}
+
+//#endregion
+//#region src/commands/Eval.tsx
+function EvalCommand({ verbose = false, judge = false }) {
+	const { exit } = useApp();
+	const [status, setStatus] = useState("loading");
+	const [error, setError] = useState(null);
+	const [evaluations, setEvaluations] = useState([]);
+	const [currentPrompt, setCurrentPrompt] = useState(null);
+	useEffect(() => {
+		async function run() {
+			const cwd = process.cwd();
+			const configPath = path.join(cwd, "nudge.config.json");
+			let config = {};
+			if (fs.existsSync(configPath)) config = JSON.parse(fs.readFileSync(configPath, "utf-8"));
+			const outputPath = config.generatedFile ? path.join(cwd, config.generatedFile) : path.join(cwd, "src", "prompts.gen.ts");
+			const targetDir = path.dirname(outputPath);
+			setStatus("evaluating");
+			try {
+				await evaluate(targetDir, outputPath, {
+					promptFilenamePattern: config.promptFilenamePattern,
+					aiConfig: config.ai,
+					verbose,
+					judge,
+					onVariantStart: (promptId, variantName) => {
+						setCurrentPrompt(`${promptId}:${variantName}`);
+					},
+					onVariantDone: (evaluation) => {
+						setEvaluations((prev) => [...prev, evaluation]);
+						setCurrentPrompt(null);
+					}
+				});
+				setStatus("done");
+				setTimeout(() => exit(), 100);
+			} catch (err) {
+				setError(String(err));
+				setStatus("error");
+				setTimeout(() => exit(), 100);
+			}
+		}
+		run();
+	}, [
+		verbose,
+		judge,
+		exit
+	]);
+	const totalPassed = evaluations.reduce((sum, e) => sum + e.passed, 0);
+	const totalTests = totalPassed + evaluations.reduce((sum, e) => sum + e.failed, 0);
+	const overallRate = totalTests > 0 ? Math.round(totalPassed / totalTests * 100) : 0;
+	return /* @__PURE__ */ jsxs(Box, {
+		flexDirection: "column",
+		padding: 1,
+		children: [
+			/* @__PURE__ */ jsx(Header, {
+				title: "Nudge Eval",
+				subtitle: "Running tests against your prompts"
+			}),
+			status === "loading" && /* @__PURE__ */ jsx(Spinner, { label: "Loading configuration..." }),
+			status === "evaluating" && /* @__PURE__ */ jsxs(Box, {
+				flexDirection: "column",
+				marginTop: 1,
+				children: [evaluations.map((evaluation) => /* @__PURE__ */ jsx(EvaluationResult, {
+					evaluation,
+					verbose
+				}, `${evaluation.promptId}-${evaluation.variantName}`)), currentPrompt && /* @__PURE__ */ jsx(Box, { children: /* @__PURE__ */ jsx(Spinner, { label: `Evaluating ${currentPrompt}...` }) })]
+			}),
+			status === "done" && /* @__PURE__ */ jsxs(Box, {
+				flexDirection: "column",
+				marginTop: 1,
+				children: [evaluations.map((evaluation) => /* @__PURE__ */ jsx(EvaluationResult, {
+					evaluation,
+					verbose
+				}, `${evaluation.promptId}-${evaluation.variantName}`)), /* @__PURE__ */ jsxs(Box, {
+					marginTop: 1,
+					flexDirection: "column",
+					children: [/* @__PURE__ */ jsx(Text, {
+						bold: true,
+						children: "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━"
+					}), /* @__PURE__ */ jsxs(Box, { children: [/* @__PURE__ */ jsx(Text, {
+						bold: true,
+						children: "Summary: "
+					}), /* @__PURE__ */ jsxs(Text, {
+						color: overallRate >= 80 ? "green" : overallRate >= 50 ? "yellow" : "red",
+						children: [
+							totalPassed,
+							"/",
+							totalTests,
+							" tests passed (",
+							overallRate,
+							"%)"
+						]
+					})] })]
+				})]
+			}),
+			status === "error" && /* @__PURE__ */ jsx(Box, {
+				marginTop: 1,
+				children: /* @__PURE__ */ jsx(Error, { children: error })
+			})
+		]
+	});
+}
+function EvaluationResult({ evaluation, verbose }) {
+	const allPassed = evaluation.failed === 0;
+	const rateColor = evaluation.successRate >= 80 ? "green" : evaluation.successRate >= 50 ? "yellow" : "red";
+	return /* @__PURE__ */ jsxs(Box, {
+		flexDirection: "column",
+		marginBottom: 1,
+		children: [/* @__PURE__ */ jsxs(Box, { children: [
+			/* @__PURE__ */ jsxs(Text, {
+				color: allPassed ? "green" : "red",
+				children: [allPassed ? "✓" : "✗", " "]
+			}),
+			/* @__PURE__ */ jsx(Text, {
+				bold: true,
+				children: evaluation.promptId
+			}),
+			evaluation.variantName !== "default" && /* @__PURE__ */ jsxs(Text, {
+				dimColor: true,
+				children: [
+					" (",
+					evaluation.variantName,
+					")"
+				]
+			}),
+			/* @__PURE__ */ jsx(Text, { children: " - " }),
+			/* @__PURE__ */ jsxs(Text, {
+				color: rateColor,
+				children: [
+					evaluation.passed,
+					"/",
+					evaluation.total,
+					" passed (",
+					Math.round(evaluation.successRate),
+					"%)"
+				]
+			})
+		] }), verbose && evaluation.results.map((result, i) => /* @__PURE__ */ jsxs(Box, {
+			marginLeft: 2,
+			children: [
+				/* @__PURE__ */ jsxs(Text, {
+					color: result.passed ? "green" : "red",
+					children: [result.passed ? "✓" : "✗", " "]
+				}),
+				/* @__PURE__ */ jsxs(Text, {
+					dimColor: true,
+					children: [
+						"\"",
+						result.input.slice(0, 40),
+						result.input.length > 40 ? "..." : "",
+						"\""
+					]
+				}),
+				!result.passed && result.reason && /* @__PURE__ */ jsxs(Text, {
+					color: "red",
+					children: [" - ", result.reason]
+				})
+			]
+		}, i))]
+	});
+}
+
+//#endregion
+//#region src/commands/Generate.tsx
+function GenerateCommand({ force = false }) {
+	const { exit } = useApp();
+	const [status, setStatus] = useState("loading");
+	const [error, setError] = useState(null);
+	const [prompts, setPrompts] = useState([]);
+	const [outputPath, setOutputPath] = useState("");
+	useEffect(() => {
+		async function run() {
+			const cwd = process.cwd();
+			const configPath = path.join(cwd, "nudge.config.json");
+			let config = {};
+			if (fs.existsSync(configPath)) config = JSON.parse(fs.readFileSync(configPath, "utf-8"));
+			const output = config.generatedFile ? path.join(cwd, config.generatedFile) : path.join(cwd, "src", "prompts.gen.ts");
+			setOutputPath(output);
+			const targetDir = path.dirname(output);
+			setStatus("generating");
+			try {
+				await generate(targetDir, output, {
+					promptFilenamePattern: config.promptFilenamePattern,
+					aiConfig: config.ai,
+					noCache: force,
+					onPromptStart: (id, variantCount) => {
+						setPrompts((prev) => [...prev, {
+							id,
+							status: "generating",
+							variantCount
+						}]);
+					},
+					onPromptCached: (id) => {
+						setPrompts((prev) => [...prev, {
+							id,
+							status: "cached"
+						}]);
+					},
+					onPromptDone: (id, variantCount) => {
+						setPrompts((prev) => prev.map((p) => p.id === id ? {
+							...p,
+							status: "done",
+							variantCount
+						} : p));
+					},
+					onPromptError: (id, err) => {
+						setPrompts((prev) => prev.map((p) => p.id === id ? {
+							...p,
+							status: "error"
+						} : p));
+					}
+				});
+				setStatus("done");
+				setTimeout(() => exit(), 100);
+			} catch (err) {
+				setError(String(err));
+				setStatus("error");
+				setTimeout(() => exit(), 100);
+			}
+		}
+		run();
+	}, [force, exit]);
+	return /* @__PURE__ */ jsxs(Box, {
+		flexDirection: "column",
+		padding: 1,
+		children: [
+			/* @__PURE__ */ jsx(Header, {
+				title: "Nudge Generate",
+				subtitle: "Generating type-safe prompt files"
+			}),
+			status === "loading" && /* @__PURE__ */ jsx(Spinner, { label: "Loading configuration..." }),
+			(status === "generating" || status === "done") && /* @__PURE__ */ jsx(Box, {
+				flexDirection: "column",
+				marginTop: 1,
+				children: prompts.map((prompt) => /* @__PURE__ */ jsxs(Box, {
+					gap: 1,
+					children: [
+						prompt.status === "generating" && /* @__PURE__ */ jsxs(Fragment, { children: [/* @__PURE__ */ jsx(Text, {
+							color: "cyan",
+							children: "◐"
+						}), /* @__PURE__ */ jsxs(Text, { children: [
+							"\"",
+							prompt.id,
+							"\" generating",
+							prompt.variantCount && prompt.variantCount > 1 ? ` ${prompt.variantCount} variant(s)` : "",
+							"..."
+						] })] }),
+						prompt.status === "cached" && /* @__PURE__ */ jsxs(Fragment, { children: [/* @__PURE__ */ jsx(Text, {
+							color: "gray",
+							children: "✓"
+						}), /* @__PURE__ */ jsxs(Text, {
+							dimColor: true,
+							children: [
+								"\"",
+								prompt.id,
+								"\" (cached)"
+							]
+						})] }),
+						prompt.status === "done" && /* @__PURE__ */ jsxs(Fragment, { children: [/* @__PURE__ */ jsx(Text, {
+							color: "green",
+							children: "✓"
+						}), /* @__PURE__ */ jsxs(Text, { children: [
+							"\"",
+							prompt.id,
+							"\" generated",
+							prompt.variantCount && prompt.variantCount > 1 ? ` ${prompt.variantCount} variant(s)` : ""
+						] })] }),
+						prompt.status === "error" && /* @__PURE__ */ jsxs(Fragment, { children: [/* @__PURE__ */ jsx(Text, {
+							color: "red",
+							children: "✗"
+						}), /* @__PURE__ */ jsxs(Text, { children: [
+							"\"",
+							prompt.id,
+							"\" failed"
+						] })] })
+					]
+				}, prompt.id))
+			}),
+			status === "done" && /* @__PURE__ */ jsx(Box, {
+				marginTop: 1,
+				children: /* @__PURE__ */ jsxs(Success, { children: [
+					"Generated ",
+					outputPath,
+					" with ",
+					prompts.length,
+					" prompt(s)"
+				] })
+			}),
+			status === "error" && /* @__PURE__ */ jsx(Box, {
+				marginTop: 1,
+				children: /* @__PURE__ */ jsx(Error, { children: error })
+			})
+		]
+	});
+}
+
+//#endregion
+//#region src/commands/Improve.tsx
+function ImproveCommand({ maxIterations = 3, promptIds, verbose = false, judge = false }) {
+	const { exit } = useApp();
+	const [status, setStatus] = useState("loading");
+	const [error, setError] = useState(null);
+	const [progress, setProgress] = useState([]);
+	const [currentAction, setCurrentAction] = useState(null);
+	const [results, setResults] = useState([]);
+	useEffect(() => {
+		async function run() {
+			const cwd = process.cwd();
+			const configPath = path.join(cwd, "nudge.config.json");
+			let config = {};
+			if (fs.existsSync(configPath)) config = JSON.parse(fs.readFileSync(configPath, "utf-8"));
+			if (!config.ai) {
+				setError("AI config is required in nudge.config.json");
+				setStatus("error");
+				setTimeout(() => exit(), 100);
+				return;
+			}
+			const outputPath = config.generatedFile ? path.join(cwd, config.generatedFile) : path.join(cwd, "src", "prompts.gen.ts");
+			const targetDir = path.dirname(outputPath);
+			setStatus("improving");
+			try {
+				setResults(await improve(targetDir, outputPath, {
+					maxIterations,
+					promptIds: promptIds ? promptIds.split(",").map((id) => id.trim()) : void 0,
+					verbose,
+					judge,
+					aiConfig: config.ai,
+					promptFilenamePattern: config.promptFilenamePattern,
+					onStatus: (promptId, variantName, statusMessage) => {
+						setProgress((prev) => {
+							if (prev.find((p) => p.promptId === promptId && p.variantName === variantName)) return prev.map((p) => p.promptId === promptId && p.variantName === variantName ? {
+								...p,
+								currentStatus: statusMessage
+							} : p);
+							return [...prev, {
+								promptId,
+								variantName,
+								iteration: 0,
+								maxIterations,
+								status: "improving",
+								currentStatus: statusMessage
+							}];
+						});
+					},
+					onIterationStart: (promptId, variantName, iteration) => {
+						setProgress((prev) => {
+							if (prev.find((p) => p.promptId === promptId && p.variantName === variantName)) return prev.map((p) => p.promptId === promptId && p.variantName === variantName ? {
+								...p,
+								iteration,
+								status: "improving"
+							} : p);
+							return [...prev, {
+								promptId,
+								variantName,
+								iteration,
+								maxIterations,
+								status: "improving"
+							}];
+						});
+						setCurrentAction(`Improving ${promptId}:${variantName} (iteration ${iteration}/${maxIterations})`);
+					},
+					onIterationDone: (promptId, variantName, result) => {
+						setProgress((prev) => prev.map((p) => p.promptId === promptId && p.variantName === variantName ? {
+							...p,
+							status: result.status,
+							initialFailures: result.initialFailures,
+							finalFailures: result.finalFailures,
+							sourceHints: result.sourceHints
+						} : p));
+						setCurrentAction(null);
+					}
+				}));
+				setStatus("done");
+				setTimeout(() => exit(), 100);
+			} catch (err) {
+				setError(String(err));
+				setStatus("error");
+				setTimeout(() => exit(), 100);
+			}
+		}
+		run();
+	}, [
+		maxIterations,
+		promptIds,
+		verbose,
+		judge,
+		exit
+	]);
+	const improved = results.filter((r) => r.status === "improved").length;
+	const plateau = results.filter((r) => r.status === "plateau").length;
+	const maxedOut = results.filter((r) => r.status === "max_iterations").length;
+	return /* @__PURE__ */ jsxs(Box, {
+		flexDirection: "column",
+		padding: 1,
+		children: [
+			/* @__PURE__ */ jsx(Header, {
+				title: "Nudge Improve",
+				subtitle: "Iteratively improving prompts based on failing tests"
+			}),
+			status === "loading" && /* @__PURE__ */ jsx(Spinner, { label: "Loading configuration..." }),
+			(status === "improving" || status === "done") && /* @__PURE__ */ jsxs(Box, {
+				flexDirection: "column",
+				marginTop: 1,
+				children: [progress.map((p) => /* @__PURE__ */ jsx(Box, {
+					marginBottom: 1,
+					children: /* @__PURE__ */ jsx(ImprovementStatus, { progress: p })
+				}, `${p.promptId}-${p.variantName}`)), currentAction && /* @__PURE__ */ jsx(Box, {
+					marginTop: 1,
+					children: /* @__PURE__ */ jsx(Spinner, { label: currentAction })
+				})]
+			}),
+			status === "done" && results.length > 0 && /* @__PURE__ */ jsxs(Box, {
+				marginTop: 1,
+				flexDirection: "column",
+				children: [/* @__PURE__ */ jsx(Text, {
+					bold: true,
+					children: "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━"
+				}), /* @__PURE__ */ jsxs(Box, {
+					flexDirection: "column",
+					children: [
+						/* @__PURE__ */ jsx(Text, {
+							bold: true,
+							children: "Summary:"
+						}),
+						improved > 0 && /* @__PURE__ */ jsxs(Success, { children: [improved, " prompt(s) improved"] }),
+						plateau > 0 && /* @__PURE__ */ jsxs(Warning, { children: [plateau, " prompt(s) reached plateau (no more improvements possible)"] }),
+						maxedOut > 0 && /* @__PURE__ */ jsxs(Warning, { children: [maxedOut, " prompt(s) hit max iterations"] })
+					]
+				})]
+			}),
+			status === "done" && results.length === 0 && /* @__PURE__ */ jsx(Box, {
+				marginTop: 1,
+				children: /* @__PURE__ */ jsx(Success, { children: "All prompts are passing their tests!" })
+			}),
+			status === "error" && /* @__PURE__ */ jsx(Box, {
+				marginTop: 1,
+				children: /* @__PURE__ */ jsx(Error, { children: error })
+			})
+		]
+	});
+}
+function ImprovementStatus({ progress }) {
+	const icon = progress.status === "improved" ? "✓" : progress.status === "improving" ? "◐" : progress.status === "plateau" ? "○" : "⚠";
+	return /* @__PURE__ */ jsxs(Box, {
+		flexDirection: "column",
+		children: [/* @__PURE__ */ jsxs(Box, { children: [
+			/* @__PURE__ */ jsxs(Text, {
+				color: progress.status === "improved" ? "green" : progress.status === "improving" ? "cyan" : "yellow",
+				children: [icon, " "]
+			}),
+			/* @__PURE__ */ jsx(Text, {
+				bold: true,
+				children: progress.promptId
+			}),
+			progress.variantName !== "default" && /* @__PURE__ */ jsxs(Text, {
+				dimColor: true,
+				children: [
+					" (",
+					progress.variantName,
+					")"
+				]
+			}),
+			progress.status === "improving" && progress.iteration > 0 && /* @__PURE__ */ jsxs(Text, {
+				dimColor: true,
+				children: [
+					" ",
+					"- iteration ",
+					progress.iteration,
+					"/",
+					progress.maxIterations
+				]
+			}),
+			progress.status === "improving" && progress.currentStatus && /* @__PURE__ */ jsxs(Text, {
+				dimColor: true,
+				children: [" - ", progress.currentStatus]
+			}),
+			progress.status === "improved" && /* @__PURE__ */ jsxs(Text, {
+				color: "green",
+				children: [
+					" ",
+					"- improved (",
+					progress.initialFailures,
+					" → ",
+					progress.finalFailures,
+					" ",
+					"failures)"
+				]
+			}),
+			progress.status === "plateau" && /* @__PURE__ */ jsx(Text, {
+				color: "yellow",
+				children: " - plateau reached"
+			}),
+			progress.status === "max_iterations" && /* @__PURE__ */ jsxs(Text, {
+				color: "yellow",
+				children: [
+					" ",
+					"- max iterations (",
+					progress.finalFailures,
+					" failures remaining)"
+				]
+			})
+		] }), progress.sourceHints && progress.sourceHints.length > 0 && /* @__PURE__ */ jsxs(Box, {
+			flexDirection: "column",
+			marginLeft: 2,
+			children: [/* @__PURE__ */ jsx(Text, {
+				dimColor: true,
+				children: "  💡 Suggested source changes:"
+			}), progress.sourceHints.map((hint, i) => /* @__PURE__ */ jsxs(Box, {
+				flexDirection: "column",
+				marginLeft: 2,
+				children: [/* @__PURE__ */ jsxs(Text, {
+					dimColor: true,
+					children: [
+						hint.action,
+						" ",
+						hint.stepType,
+						": ",
+						hint.suggestion
+					]
+				}), /* @__PURE__ */ jsxs(Text, {
+					dimColor: true,
+					italic: true,
+					children: ["└ ", hint.reason]
+				})]
+			}, i))]
+		})]
+	});
+}
+
+//#endregion
+//#region src/commands/Init.tsx
+const PROVIDER_CHOICES = [
+	{
+		label: "OpenAI",
+		value: "openai"
+	},
+	{
+		label: "OpenRouter",
+		value: "openrouter"
+	},
+	{
+		label: "Local (custom endpoint)",
+		value: "local"
+	}
+];
+const DEFAULT_API_KEYS = {
+	openai: "OPENAI_API_KEY",
+	openrouter: "OPENROUTER_API_KEY",
+	local: "API_KEY"
+};
+const DEFAULT_MODELS = {
+	openai: "gpt-4o",
+	openrouter: "x-ai/grok-code-fast-1",
+	local: "llama2"
+};
+function InitCommand() {
+	const { exit } = useApp();
+	const cwd = process.cwd();
+	const configPath = path.join(cwd, "nudge.config.json");
+	const [step, setStep] = useState("check-existing");
+	const [provider, setProvider] = useState("openai");
+	const [apiKeyEnvVar, setApiKeyEnvVar] = useState("");
+	const [baseUrl, setBaseUrl] = useState("");
+	const [model, setModel] = useState("");
+	const [error, setError] = useState(null);
+	const [configExists, setConfigExists] = useState(false);
+	useEffect(() => {
+		const exists = fs.existsSync(configPath);
+		setConfigExists(exists);
+		if (exists) setStep("confirm-overwrite");
+		else setStep("select-provider");
+	}, [configPath]);
+	useEffect(() => {
+		if (step === "done") {
+			const config = { ai: {
+				provider,
+				apiKeyEnvVar,
+				model,
+				...baseUrl && { baseUrl }
+			} };
+			try {
+				fs.writeFileSync(configPath, JSON.stringify(config, null, 2) + "\n");
+			} catch (err) {
+				setError(`Failed to write config: ${err}`);
+				return;
+			}
+			setTimeout(() => exit(), 100);
+		}
+	}, [
+		step,
+		provider,
+		apiKeyEnvVar,
+		model,
+		baseUrl,
+		configPath,
+		exit
+	]);
+	const handleProviderSelect = (item) => {
+		setProvider(item.value);
+		setApiKeyEnvVar(DEFAULT_API_KEYS[item.value]);
+		setModel(DEFAULT_MODELS[item.value]);
+		setStep("enter-api-key-env");
+	};
+	const handleApiKeySubmit = () => {
+		if (provider === "local") {
+			setBaseUrl("http://localhost:11434/v1");
+			setStep("enter-base-url");
+		} else setStep("enter-model");
+	};
+	const handleBaseUrlSubmit = () => {
+		setStep("enter-model");
+	};
+	const handleModelSubmit = () => {
+		setStep("done");
+	};
+	const handleOverwriteSelect = (item) => {
+		if (item.value) setStep("select-provider");
+		else exit();
+	};
+	if (error) return /* @__PURE__ */ jsx(Box, {
+		flexDirection: "column",
+		children: /* @__PURE__ */ jsxs(Text, {
+			color: "red",
+			children: ["Error: ", error]
+		})
+	});
+	return /* @__PURE__ */ jsxs(Box, {
+		flexDirection: "column",
+		padding: 1,
+		children: [
+			/* @__PURE__ */ jsx(Header, {
+				title: "Nudge Init",
+				subtitle: "Let's set up your configuration!"
+			}),
+			step === "confirm-overwrite" && /* @__PURE__ */ jsxs(Box, {
+				flexDirection: "column",
+				children: [
+					/* @__PURE__ */ jsx(Warning, { children: "nudge.config.json already exists!" }),
+					/* @__PURE__ */ jsx(Box, {
+						marginTop: 1,
+						children: /* @__PURE__ */ jsx(Text, { children: "Do you want to overwrite it?" })
+					}),
+					/* @__PURE__ */ jsx(Box, {
+						marginTop: 1,
+						children: /* @__PURE__ */ jsx(SelectInput, {
+							items: [{
+								label: "No",
+								value: false
+							}, {
+								label: "Yes",
+								value: true
+							}],
+							onSelect: handleOverwriteSelect
+						})
+					})
+				]
+			}),
+			step === "select-provider" && /* @__PURE__ */ jsxs(Box, {
+				flexDirection: "column",
+				children: [/* @__PURE__ */ jsx(Text, { children: "Choose your AI provider:" }), /* @__PURE__ */ jsx(Box, {
+					marginTop: 1,
+					children: /* @__PURE__ */ jsx(SelectInput, {
+						items: PROVIDER_CHOICES,
+						onSelect: handleProviderSelect
+					})
+				})]
+			}),
+			step === "enter-api-key-env" && /* @__PURE__ */ jsxs(Box, {
+				flexDirection: "column",
+				children: [/* @__PURE__ */ jsx(Text, { children: "API key environment variable name:" }), /* @__PURE__ */ jsxs(Box, {
+					marginTop: 1,
+					children: [/* @__PURE__ */ jsx(Text, {
+						color: "cyan",
+						children: "❯ "
+					}), /* @__PURE__ */ jsx(TextInput, {
+						value: apiKeyEnvVar,
+						onChange: setApiKeyEnvVar,
+						onSubmit: handleApiKeySubmit
+					})]
+				})]
+			}),
+			step === "enter-base-url" && /* @__PURE__ */ jsxs(Box, {
+				flexDirection: "column",
+				children: [/* @__PURE__ */ jsx(Text, { children: "Base URL for your local provider:" }), /* @__PURE__ */ jsxs(Box, {
+					marginTop: 1,
+					children: [/* @__PURE__ */ jsx(Text, {
+						color: "cyan",
+						children: "❯ "
+					}), /* @__PURE__ */ jsx(TextInput, {
+						value: baseUrl,
+						onChange: setBaseUrl,
+						onSubmit: handleBaseUrlSubmit
+					})]
+				})]
+			}),
+			step === "enter-model" && /* @__PURE__ */ jsxs(Box, {
+				flexDirection: "column",
+				children: [/* @__PURE__ */ jsx(Text, { children: "Model name:" }), /* @__PURE__ */ jsxs(Box, {
+					marginTop: 1,
+					children: [/* @__PURE__ */ jsx(Text, {
+						color: "cyan",
+						children: "❯ "
+					}), /* @__PURE__ */ jsx(TextInput, {
+						value: model,
+						onChange: setModel,
+						onSubmit: handleModelSubmit
+					})]
+				})]
+			}),
+			step === "done" && /* @__PURE__ */ jsxs(Box, {
+				flexDirection: "column",
+				marginTop: 1,
+				children: [/* @__PURE__ */ jsx(Success, { children: "Configuration saved to nudge.config.json" }), /* @__PURE__ */ jsxs(Box, {
+					marginTop: 1,
+					flexDirection: "column",
+					children: [
+						/* @__PURE__ */ jsx(Text, {
+							color: "yellow",
+							children: "💡 Next steps:"
+						}),
+						/* @__PURE__ */ jsxs(Text, { children: [
+							" 1. Set your ",
+							apiKeyEnvVar,
+							" environment variable"
+						] }),
+						/* @__PURE__ */ jsxs(Text, { children: [" ", "2. Create your prompt files (e.g., src/prompts/example.prompt.ts)"] }),
+						/* @__PURE__ */ jsxs(Text, { children: [" ", "3. Run 'npx @nudge-ai/cli generate' to compile your prompts"] }),
+						/* @__PURE__ */ jsxs(Text, { children: [" ", "4. Import './prompts.gen' once at your app's entry point"] })
+					]
+				})]
+			})
+		]
+	});
+}
+
+//#endregion
+//#region src/bin.tsx
 const args = process.argv.slice(2);
-if (args[0] !== "generate") {
-	console.error(`Usage: nudge generate [--no-cache]`);
-	process.exit(1);
+const command = args[0];
+function parseArgs(args$1) {
+	const result = {};
+	for (let i = 0; i < args$1.length; i++) {
+		const arg = args$1[i];
+		if (arg.startsWith("--")) {
+			const key = arg.slice(2);
+			const next = args$1[i + 1];
+			if (next && !next.startsWith("--")) {
+				result[key] = next;
+				i++;
+			} else result[key] = true;
+		}
+	}
+	return result;
+}
+function showHelp() {
+	console.log(`
+Nudge CLI - Generate type-safe prompt files for AI applications
+
+Usage:
+  nudge <command> [options]
+
+Commands:
+  init        Initialize a new Nudge configuration
+  generate    Generate type-safe prompt files from your prompt templates
+  eval        Run tests defined in prompts to evaluate quality
+  improve     Iteratively improve prompts based on failing tests
+
+Options:
+  generate:
+    --force           Skip cache and regenerate all prompts
+
+  eval:
+    --verbose         Show detailed test results
+    --judge           Use LLM to evaluate string assertions
+
+  improve:
+    --max-iterations  Maximum improvement iterations (default: 3)
+    --prompt-ids      Comma-separated list of specific prompt IDs to improve
+    --verbose         Show detailed improvement steps
+    --judge           Use LLM to evaluate string assertions
+
+Examples:
+  nudge init
+  nudge generate
+  nudge generate --force
+  nudge eval --verbose
+  nudge eval --judge
+  nudge improve --max-iterations 5
+  nudge improve --prompt-ids summarizer,translator
+`);
+}
+async function main() {
+	if (!command || command === "--help" || command === "-h") {
+		showHelp();
+		return;
+	}
+	const options = parseArgs(args.slice(1));
+	switch (command) {
+		case "init":
+			render(React.createElement(InitCommand));
+			break;
+		case "generate":
+			render(React.createElement(GenerateCommand, { force: options.force === true }));
+			break;
+		case "eval":
+			render(React.createElement(EvalCommand, {
+				verbose: options.verbose === true,
+				judge: options.judge === true
+			}));
+			break;
+		case "improve":
+			render(React.createElement(ImproveCommand, {
+				maxIterations: typeof options["max-iterations"] === "string" ? parseInt(options["max-iterations"], 10) : 3,
+				promptIds: typeof options["prompt-ids"] === "string" ? options["prompt-ids"] : void 0,
+				verbose: options.verbose === true,
+				judge: options.judge === true
+			}));
+			break;
+		default:
+			console.error(`Unknown command: ${command}`);
+			console.log("Run \"nudge --help\" for usage information.");
+			process.exit(1);
+	}
 }
-const noCache = args.includes("--no-cache");
-const cwd = process.cwd();
-const configPath = path.join(cwd, "nudge.config.json");
-let config = {};
-if (fs.existsSync(configPath)) config = JSON.parse(fs.readFileSync(configPath, "utf-8"));
-const outputPath = config.generatedFile ? path.join(cwd, config.generatedFile) : path.join(cwd, "src", "prompts.gen.ts");
-generate(path.dirname(outputPath), outputPath, {
-	promptFilenamePattern: config.promptFilenamePattern,
-	aiConfig: config.ai,
-	noCache
-}).catch((error) => {
-	console.error("Error generating prompts:", error);
+main().catch((err) => {
+	console.error("Error:", err);
 	process.exit(1);
 });