@algolia/wizard 0.120.0 → 0.122.0

package/dist/main.js

@@ -325,6 +325,7 @@ var useWizard = create((set, get) => ({
   logs: [],
   _outputLogCount: 0,
   error: null,
+  exitRequested: false,
   inputReq: null,
   _resolve: null,
   settingUpAppId: null,
@@ -488,7 +489,15 @@ var useWizard = create((set, get) => ({
     get().logEnd(id, "success");
   },
   setDone: () => set({ phase: "done", loadingMessage: null }),
-  setError: (message) => set({ phase: "error", error: message, loadingMessage: null }),
+  setError: (message, options) => {
+    if (options?.exit) get()._clearNoticeQueue();
+    set({
+      phase: "error",
+      error: message,
+      loadingMessage: null,
+      exitRequested: options?.exit ?? false
+    });
+  },
   reset: () => {
     get()._clearNoticeQueue();
     get()._resolve?.(false);
@@ -508,6 +517,7 @@ var useWizard = create((set, get) => ({
       logs: [],
       _outputLogCount: 0,
       error: null,
+      exitRequested: false,
       inputReq: null,
       _resolve: null,
       settingUpAppId: null,
@@ -1460,12 +1470,13 @@ function EnterToContinuePrompt({
   ] });
 }
 function PromptInput() {
-  const { phase, inputReq, submitInput } = useWizard();
+  const { phase, inputReq, submitInput, exitRequested } = useWizard();
   const [draft, setDraft] = useState9("");
   const [submittedReq, setSubmittedReq] = useState9(null);
   const promptKey = inputReq && `${inputReq.promptType}:${inputReq.prompt}`;
   const submitting = inputReq != null && submittedReq === inputReq;
   if (phase === "done" || phase === "error") {
+    if (exitRequested) return null;
     return /* @__PURE__ */ jsx10(Box11, { marginTop: 1, children: /* @__PURE__ */ jsx10(Text12, { color: "gray", dimColor: true, children: "Press Enter or Esc to exit" }) });
   }
   if (phase !== "awaitingInput" || !inputReq) return null;
@@ -2544,10 +2555,24 @@ async function ensureApplication(resumeAppId) {
   }
 }
 
+// src/lib/rateLimit.ts
+var RateLimitError = class extends Error {
+  constructor(retryAfterSeconds, options) {
+    super(rateLimitMessage(retryAfterSeconds), options);
+    this.retryAfterSeconds = retryAfterSeconds;
+    this.name = "RateLimitError";
+  }
+  retryAfterSeconds;
+};
+function rateLimitMessage(retryAfterSeconds) {
+  const when = retryAfterSeconds === void 0 ? "in a moment" : `in ${retryAfterSeconds} ${retryAfterSeconds === 1 ? "second" : "seconds"}`;
+  return `You've hit the rate limit. Try again ${when}.`;
+}
+
 // package.json
 var package_default = {
   name: "@algolia/wizard",
-  version: "0.120.0",
+  version: "0.122.0",
   description: "Magically implement Algolia functionality in your codebase",
   type: "module",
   engines: {
@@ -2638,6 +2663,16 @@ var RestartWorkflowSignal = class extends Error {
   }
   appId;
 };
+var ExitWizardSignal = class extends Error {
+  constructor(original) {
+    super(original instanceof Error ? original.message : String(original), {
+      cause: original
+    });
+    this.original = original;
+    this.name = "ExitWizardSignal";
+  }
+  original;
+};
 function ensureExecutedStepCount(state) {
   if (state.executedStepCount == null) {
     state.executedStepCount = state.steps.filter(
@@ -2828,6 +2863,29 @@ async function rewindTo(state, targetIndex, fromIndex) {
   store.syncSteps([...state.steps], targetIndex);
   await persistState(state);
 }
+var RATE_LIMIT_RETRY_OPTION = "Retry";
+var RATE_LIMIT_EXIT_OPTION = "Exit";
+async function runStepWithRateLimitRecovery(state, step, ctx) {
+  for (; ; ) {
+    try {
+      return await step.run(ctx);
+    } catch (err) {
+      if (!(err instanceof RateLimitError)) throw err;
+      const choice = await useWizard.getState().requestUserInput({
+        prompt: "Retry the step, or exit the wizard?",
+        promptType: "multipleChoice",
+        options: [RATE_LIMIT_RETRY_OPTION, RATE_LIMIT_EXIT_OPTION],
+        alert: {
+          variant: "error",
+          title: "Too many requests",
+          message: err.message
+        }
+      });
+      if (choice !== RATE_LIMIT_RETRY_OPTION) throw new ExitWizardSignal(err);
+      ctx = await makeContext(state, true);
+    }
+  }
+}
 async function runStep(state, index, step) {
   const store = useWizard.getState();
   const record = state.steps[index];
@@ -2854,7 +2912,7 @@ async function runStep(state, index, step) {
   const stopSavingCost = saveCostDuringStep(state);
   let output;
   try {
-    const raw = await step.run(ctx);
+    const raw = await runStepWithRateLimitRecovery(state, step, ctx);
     output = step.outputSchema.parse(raw);
     await step.onStepComplete?.(output, ctx);
   } catch (err) {
@@ -3014,7 +3072,9 @@ async function runWorkflow(workflow, startWorkflowAppId, options) {
     await recordWorkflowRun(workflow.id, nowIso());
     await clearWorkflowState(workflow.id);
     store.setDone();
-  } catch (err) {
+  } catch (thrown) {
+    const exitRequested = thrown instanceof ExitWizardSignal;
+    const err = exitRequested ? thrown.original : thrown;
     const message = err instanceof Error ? err.message : String(err);
     const failed = await loadWorkflowState(workflow.id);
     const failedAppId = failed ? runAppId(failed) : startWorkflowAppId;
@@ -3063,7 +3123,7 @@ async function runWorkflow(workflow, startWorkflowAppId, options) {
       ...errorType ? { error_type: errorType } : {},
       ...errorCode2 ? { error_code: errorCode2 } : {}
     });
-    store.setError(message);
+    store.setError(message, { exit: exitRequested });
   }
 }
 
@@ -5559,6 +5619,7 @@ var GLOBAL_BUDGET_USER_MESSAGE = "The wizard reached its daily AI usage limit fo
 var PROXY_AUTH_REJECTED_HEADER2 = "x-wizard-proxy-auth";
 var PROXY_REJECTION_BODY = "You shall not pass!";
 var BUDGET_EXCEEDED_HEADER = "x-wizard-budget-exceeded";
+var RETRY_AFTER_HEADER = "x-wizard-retry-after";
 function lastCallError(err) {
   const cause = RetryError.isInstance(err) ? err.lastError : err;
   return APICallError.isInstance(cause) ? cause : void 0;
@@ -5574,6 +5635,17 @@ function budgetExceededMessage(err) {
   if (scope === "global") return GLOBAL_BUDGET_USER_MESSAGE;
   return void 0;
 }
+function rateLimitError(err) {
+  const cause = lastCallError(err);
+  if (cause?.statusCode !== 429) return void 0;
+  const headers = cause.responseHeaders;
+  const seconds = Number(
+    headers?.[RETRY_AFTER_HEADER] ?? headers?.["retry-after"]
+  );
+  return new RateLimitError(seconds > 0 ? Math.ceil(seconds) : void 0, {
+    cause: err
+  });
+}
 var INCOMPLETE_CAUSES = {
   stepLimit: {
     kind: "missingReportAtStepLimit",
@@ -5612,11 +5684,19 @@ function incompleteCauseOf(stepCount, maxSteps, finishReason) {
   if (finishReason !== "stop") return "stopped";
   return null;
 }
+var NO_FRONTEND_REASON = "no_frontend";
+var NO_FRONTEND_MESSAGE = "No frontend found in this folder. `cd` into the directory where your frontend lives (in a monorepo, the app folder, usually the one with that app's `package.json`), then rerun the Wizard.";
+function userFacingAgentFailure(reason) {
+  if (reason?.trim().toLowerCase() === NO_FRONTEND_REASON) {
+    return NO_FRONTEND_MESSAGE;
+  }
+  return `Agent reported failure: ${reason ?? "no reason given"}`;
+}
 var AgentReportedFailureError = class extends Error {
   reason;
   output;
   constructor(reason, output) {
-    super(`Agent reported failure: ${reason ?? "no reason given"}`);
+    super(userFacingAgentFailure(reason));
     this.reason = reason;
     this.output = output;
   }
@@ -5799,7 +5879,7 @@ async function runAgentAttempt(req, attempt, retryReason, continuation) {
     ...continuation.messages,
     { role: "user", content: ENDED_TURN_NUDGE }
   ] : [{ role: "user", content: INITIAL_PROMPT }];
-  const stream = await agent.stream({ prompt });
+  const stream = await agent.stream({ prompt, abortSignal: req.abortSignal });
   let chunks = [];
   const chunkLimit = 5;
   let streamError;
@@ -5820,6 +5900,10 @@ async function runAgentAttempt(req, attempt, retryReason, continuation) {
   if (chunks.length) {
     logger.debug(chunks.join(""));
   }
+  if (req.abortSignal?.aborted) {
+    logger.info({ operation: req.operation }, "runAgent aborted");
+    req.abortSignal.throwIfAborted();
+  }
   if (streamError !== void 0) {
     logger.error({ err: streamError }, "agent stream reported an error");
     if (isProxyInactivityRejection(streamError)) {
@@ -5827,7 +5911,7 @@ async function runAgentAttempt(req, attempt, retryReason, continuation) {
     }
     const budgetMessage = budgetExceededMessage(streamError);
     if (budgetMessage) throw new Error(budgetMessage, { cause: streamError });
-    throw streamError;
+    throw rateLimitError(streamError) ?? streamError;
   }
   const end = Date.now();
   const usage = await stream.totalUsage;
@@ -5960,9 +6044,10 @@ function prioritizeFrontendFrameworks(result) {
   );
   return { ...result, frameworks: [...frontend, ...remaining] };
 }
-var detectLanguage = async () => {
+var detectLanguage = async (abortSignal) => {
   const result = await runAgent({
     operation: "language-detection",
+    abortSignal,
     instructions: [
       "Analyze the codebase and determine the programming languages and frameworks used",
       "If a superset language is found, exclude the subset language (e.g. TypeScript over JavaScript).",
@@ -6058,20 +6143,21 @@ var MODE_CONFIG = {
     outputSchema: verificationSchema
   }
 };
-function runMode(mode, extraInstructions = []) {
+function runMode(mode, extraInstructions = [], abortSignal) {
   const { instructions, outputSchema } = MODE_CONFIG[mode];
   return runAgent({
     operation: `code-analysis-${mode}`,
     instructions: [...instructions, ...extraInstructions],
     tools: READONLY_TOOLS,
     outputSchema,
-    modelProfile: "analysis" /* analysis */
+    modelProfile: "analysis" /* analysis */,
+    abortSignal
   });
 }
-async function runAnalysis(mode, extraInstructions = []) {
+async function runAnalysis(mode, extraInstructions = [], abortSignal) {
   const modes = mode ? [mode] : ["ingestion", "searchImplementation", "verification"];
   const results = await Promise.all(
-    modes.map((m) => runMode(m, extraInstructions))
+    modes.map((m) => runMode(m, extraInstructions, abortSignal))
   );
   return Object.assign({}, ...results);
 }
@@ -6085,9 +6171,14 @@ var projectScanSchema = detectLanguageSchema.extend({
 });
 async function projectScan(ctx) {
   const startedAt = Date.now();
+  const controller = new AbortController();
+  const abortOthersOnFailure = (run2) => run2.catch((err) => {
+    controller.abort();
+    throw err;
+  });
   const [detected, analyses] = await Promise.all([
-    detectLanguage(),
-    runAnalysis()
+    abortOthersOnFailure(detectLanguage(controller.signal)),
+    abortOthersOnFailure(runAnalysis(void 0, [], controller.signal))
   ]);
   const entities = toEntitySummary(analyses.ingestionAnalysis ?? []);
   track("AI Wizard Scan Completed", {
@@ -6540,7 +6631,7 @@ async function assertGitRepoWithHead(repoRoot) {
     await git(["-C", repoRoot, "rev-parse", "HEAD"]);
   } catch {
     throw new Error(
-      "implement requires a git repository with at least one commit. Initialize git and commit before running the wizard."
+      "This folder must be a Git repository with at least one commit so you can review and revert Wizard changes. Run `git init`, commit your project files, then rerun."
     );
   }
 }
@@ -7341,7 +7432,7 @@ function searchInstructions(input) {
     "Match the application styles closely, including light, dark, and high-contrast themes the project already supports.",
     "Give the search input frame the same height, corner radius, border color and width, background, and focus style as the input it replaces. When it replaces nothing, copy them from the form controls beside it. Read these values from their source (the theme, the component props, or the CSS). Do not keep the widget theme defaults, for example a 44px height and 4px corners, and do not invent colors. When the app theme exists only in code, take the values from that theme object.",
     "When you add UI text to a localized dictionary or message file, add the same keys, translated, to every locale the project supports.",
-    "Before reporting, mount the component. Do not run a type check, build, or test command unless a later instruction allows it: the validation pass runs the type check, after the developer agrees to it. If a tool limit is reached or an error remains unresolved, keep the implementation, report status=success, and name the remaining issue in the summary so validation can drive the next pass. Report status=fail only when no search implementation can be written, with the concrete blocker as the reason.",
+    `Before reporting, mount the component. Do not run a type check, build, or test command unless a later instruction allows it: the validation pass runs the type check, after the developer agrees to it. If a tool limit is reached or an error remains unresolved, keep the implementation, report status=success, and name the remaining issue in the summary so validation can drive the next pass. Report status=fail only when no search implementation can be written. When this folder has no frontend to build search UI in (for example a monorepo root or a backend-only package), use reason exactly \`${NO_FRONTEND_REASON}\`. For any other blocker, use a short concrete reason.`,
     "The summary should be extremely concise; do not mention manual testing steps.",
     ...input.uiTemplate ? [
       "For the playbook recap, report uiFiles as every file you created or edited for this UI. Each entry needs a short action label and its repo-relative path. Report devCommand only when you found the project\u2019s existing command for starting the app locally. Do not invent a command, host, or port."
@@ -10501,7 +10592,7 @@ function Ribbon() {
 }
 
 // src/ui/App.tsx
-import { useState as useState14 } from "react";
+import { useEffect as useEffect5, useState as useState14 } from "react";
 
 // src/ui/Logs.tsx
 import { Box as Box21, Text as Text22, useInput as useInput9 } from "ink";
@@ -11135,9 +11226,13 @@ function App() {
     workflow,
     settingUpAppId,
     loadingMessage,
-    review: review2
+    review: review2,
+    exitRequested
   } = useWizard();
   const { exit } = useApp2();
+  useEffect5(() => {
+    if (exitRequested) exit();
+  }, [exitRequested, exit]);
   const { columns, rows } = useWindowSize8();
   useRepaintOnResize();
   const compactRun = isCompactTerminal(columns, rows);
@@ -11158,7 +11253,7 @@ function App() {
   const holdForTip = !compactRun && tipsActive && isAwaitingUserInput && tipState === "typing";
   const showTips = !compactRun && tipsActive && (phase === "running" && currentStep?.status === "running" && !loadingMessage && !settingUpAppId || holdForTip);
   const showNoticesInMain = !tipsActive;
-  const showNotices = !compactRun && (!isAwaitingUserInput || holdForTip) && !review2;
+  const showNotices = !compactRun && (!isAwaitingUserInput || holdForTip) && !review2 && !exitRequested;
   useInput10(
     (_input, key) => {
       if (key.return) {
@@ -11572,7 +11667,7 @@ async function promptResumeOrStartNew(session, next) {
 // package.json with { type: 'json' }
 var package_default2 = {
   name: "@algolia/wizard",
-  version: "0.120.0",
+  version: "0.122.0",
   description: "Magically implement Algolia functionality in your codebase",
   type: "module",
   engines: {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@algolia/wizard",
-  "version": "0.120.0",
+  "version": "0.122.0",
   "description": "Magically implement Algolia functionality in your codebase",
   "type": "module",
   "engines": {