@algolia/wizard 0.105.0 → 0.107.0

package/dist/main.js

@@ -1706,7 +1706,9 @@ function tokenCount(value = NO_TOKENS) {
   return value;
 }
 function estimateModelCost(profileName, usage) {
-  const { prices } = getModelProfile(profileName);
+  return estimateCostFromPrices(getModelProfile(profileName).prices, usage);
+}
+function estimateCostFromPrices(prices, usage) {
   const inputDetails = Object.assign({}, usage.inputTokenDetails);
   const cacheRead = tokenCount(inputDetails.cacheReadTokens);
   const cacheWrite = tokenCount(inputDetails.cacheWriteTokens);
@@ -2467,7 +2469,7 @@ async function ensureApplication(resumeAppId) {
 // package.json
 var package_default = {
   name: "@algolia/wizard",
-  version: "0.105.0",
+  version: "0.107.0",
   description: "Magically implement Algolia functionality in your codebase",
   type: "module",
   engines: {
@@ -5469,11 +5471,25 @@ var ATTEMPT_NUMBER_OFFSET = 1;
 var INITIAL_ATTEMPT = 0;
 var PROVIDER_ERROR_USER_MESSAGE = "The AI service had trouble responding. Run the wizard again to retry this step.";
 var SESSION_TIMEOUT_USER_MESSAGE = "The wizard session timed out after a period of inactivity. Run the wizard again to retry this step.";
+var USER_BUDGET_USER_MESSAGE = "You reached the daily AI usage limit of the wizard. Try again tomorrow.";
+var GLOBAL_BUDGET_USER_MESSAGE = "The wizard reached its daily AI usage limit for all users. Try again tomorrow.";
 var PROXY_AUTH_REJECTED_HEADER2 = "x-wizard-proxy-auth";
 var PROXY_REJECTION_BODY = "You shall not pass!";
-function isProxyInactivityRejection(err) {
+var BUDGET_EXCEEDED_HEADER = "x-wizard-budget-exceeded";
+function lastCallError(err) {
   const cause = RetryError.isInstance(err) ? err.lastError : err;
-  return APICallError.isInstance(cause) && cause.statusCode === 401 && cause.responseBody === PROXY_REJECTION_BODY && cause.responseHeaders?.[PROXY_AUTH_REJECTED_HEADER2] === void 0;
+  return APICallError.isInstance(cause) ? cause : void 0;
+}
+function isProxyInactivityRejection(err) {
+  const cause = lastCallError(err);
+  return cause?.statusCode === 401 && cause.responseBody === PROXY_REJECTION_BODY && cause.responseHeaders?.[PROXY_AUTH_REJECTED_HEADER2] === void 0;
+}
+function budgetExceededMessage(err) {
+  const cause = lastCallError(err);
+  const scope = cause?.responseHeaders?.[BUDGET_EXCEEDED_HEADER];
+  if (scope === "user") return USER_BUDGET_USER_MESSAGE;
+  if (scope === "global") return GLOBAL_BUDGET_USER_MESSAGE;
+  return void 0;
 }
 var INCOMPLETE_CAUSES = {
   stepLimit: {
@@ -5726,6 +5742,8 @@ async function runAgentAttempt(req, attempt, retryReason, continuation) {
     if (isProxyInactivityRejection(streamError)) {
       throw new Error(SESSION_TIMEOUT_USER_MESSAGE, { cause: streamError });
     }
+    const budgetMessage = budgetExceededMessage(streamError);
+    if (budgetMessage) throw new Error(budgetMessage, { cause: streamError });
     throw streamError;
   }
   const end = Date.now();
@@ -7478,6 +7496,7 @@ async function implement(ctx, useCases = DEFAULT_IMPLEMENT_USE_CASES) {
         ingestCommand
       )
     });
+    const ingestStartedAt = Date.now();
     await runIngestionAgent();
     let outcome = ingestOutcome(executions, repoRoot, ingestCommand);
     let failureAcknowledged = false;
@@ -7575,7 +7594,7 @@ async function implement(ctx, useCases = DEFAULT_IMPLEMENT_USE_CASES) {
         track("AI Wizard Ingest Successful", {
           data_model,
           record_count: ingestRecordCount,
-          duration_ms: ingestDurationMs ?? 0,
+          duration_ms: Date.now() - ingestStartedAt,
           index_name: targetIndex,
           record_schema: recordSchema(sample, confirmed2?.[0]?.attributes ?? []),
           attempts: ingestRunAttempts
@@ -7615,6 +7634,7 @@ ${detail}` : ""}`
         error_type: "Ingest Records",
         product_area: "AI Wizard",
         attempts: ingestRunAttempts,
+        duration_ms: Date.now() - ingestStartedAt,
         ...ingestAttempt?.exitCode != null ? { error_code: String(ingestAttempt.exitCode) } : {}
       });
     }
@@ -10180,7 +10200,7 @@ async function resetProjectState() {
 // package.json with { type: 'json' }
 var package_default2 = {
   name: "@algolia/wizard",
-  version: "0.105.0",
+  version: "0.107.0",
   description: "Magically implement Algolia functionality in your codebase",
   type: "module",
   engines: {

package/package.json

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