@algolia/wizard 0.120.0 → 0.121.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.121.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",
@@ -5799,7 +5871,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 +5892,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 +5903,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 +6036,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 +6135,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 +6163,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", {
@@ -10501,7 +10584,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 +11218,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 +11245,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 +11659,7 @@ async function promptResumeOrStartNew(session, next) {
 // package.json with { type: 'json' }
 var package_default2 = {
   name: "@algolia/wizard",
-  version: "0.120.0",
+  version: "0.121.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.121.0",
   "description": "Magically implement Algolia functionality in your codebase",
   "type": "module",
   "engines": {