@mindstudio-ai/agent 0.1.46 → 0.1.48

package/dist/index.js

@@ -42,9 +42,9 @@ async function requestWithRetry(config, method, url, body, attempt) {
     body: body != null ? JSON.stringify(body) : void 0
   });
   config.rateLimiter.updateFromHeaders(res.headers);
-  if (res.status === 429 && attempt < config.maxRetries) {
+  if (attempt < config.maxRetries && (res.status === 429 || res.status === 502 || res.status === 503 || res.status === 504)) {
     const retryAfter = res.headers.get("retry-after");
-    const waitMs = retryAfter ? parseFloat(retryAfter) * 1e3 : 1e3;
+    const waitMs = retryAfter ? parseFloat(retryAfter) * 1e3 : Math.min(1e3 * Math.pow(2, attempt), 1e4);
     await sleep(waitMs);
     return requestWithRetry(config, method, url, body, attempt + 1);
   }
@@ -1493,15 +1493,16 @@ var Table = class {
   // -------------------------------------------------------------------------
   // Internal helpers
   // -------------------------------------------------------------------------
-  /** @internal Throw if data includes a platform-managed email/phone column. */
+  /** @internal Throw if data includes a platform-managed read-only column. */
   _checkManagedColumns(data) {
     const mc = this._config.managedColumns;
     if (!mc) return;
     const keys = Object.keys(data);
     for (const key of keys) {
-      if (mc.email && key === mc.email || mc.phone && key === mc.phone) {
+      if (mc.email && key === mc.email || mc.phone && key === mc.phone || mc.apiKey && key === mc.apiKey) {
+        const friendly = key === mc.email ? "email" : key === mc.phone ? "phone" : "API key";
         throw new MindStudioError(
-          `Cannot write to "${key}" \u2014 this column is managed by auth. Use the auth API to change a user's ${key === mc.email ? "email" : "phone"}.`,
+          `Cannot write to "${key}" \u2014 this column is managed by auth. Use the auth API to change a user's ${friendly}.`,
           "managed_column_write",
           400
         );
@@ -2884,7 +2885,7 @@ function buildTaskRequestBody(options) {
     prompt: options.prompt,
     input: options.input,
     tools: mapTools(options.tools),
-    structuredOutputExample: options.structuredOutputExample,
+    structuredOutputExample: typeof options.structuredOutputExample === "string" ? options.structuredOutputExample : JSON.stringify(options.structuredOutputExample),
     model: options.model,
     ...options.maxTurns != null && { maxTurns: options.maxTurns },
     ...options.appId != null && { appId: options.appId },
@@ -2894,6 +2895,16 @@ function buildTaskRequestBody(options) {
 function sleep2(ms) {
   return new Promise((resolve) => setTimeout(resolve, ms));
 }
+function isDevMode() {
+  return !!(process.env.CALLBACK_TOKEN || getRequestContext()?.callbackToken);
+}
+function logTaskResult(result) {
+  if (!isDevMode()) return;
+  const toolSummary = result.toolCalls.map((tc) => `${tc.name} (${tc.durationMs}ms) ${tc.success ? "\u2713" : "\u2717"}`).join(", ");
+  console.log(
+    `[task] ${result.turns} turn${result.turns === 1 ? "" : "s"}` + (toolSummary ? `: ${toolSummary}` : "") + ` | ${result.parsedSuccessfully ? "output OK" : "\u26A0 output not valid JSON"} | cost: ${result.usage.totalBillingCost}`
+  );
+}
 async function runTaskPoll(httpConfig, body) {
   const { data } = await request(
     httpConfig,
@@ -2909,6 +2920,7 @@ async function runTaskPoll(httpConfig, body) {
     const res = await fetch(pollUrl, {
       headers: { "User-Agent": "@mindstudio-ai/agent" }
     });
+    if (res.status === 502 || res.status === 503 || res.status === 504) continue;
     if (res.status === 404) {
       throw new MindStudioError(
         "Task poll token not found or expired.",
@@ -2926,7 +2938,12 @@ async function runTaskPoll(httpConfig, body) {
       );
     }
     const poll = await res.json();
-    if (poll.status === "pending") continue;
+    if (poll.status === "pending") {
+      if (isDevMode() && poll.currentTurn != null) {
+        console.log(`[task] running... turn ${poll.currentTurn}/${poll.maxTurns ?? "?"}`);
+      }
+      continue;
+    }
     if (poll.status === "error") {
       throw new MindStudioError(
         poll.error ?? "Task execution failed.",
@@ -2934,11 +2951,16 @@ async function runTaskPoll(httpConfig, body) {
         500
       );
     }
-    return {
+    const result = {
       output: poll.output,
+      outputRaw: poll.outputRaw ?? "",
+      parsedSuccessfully: poll.parsedSuccessfully ?? true,
       turns: poll.turns ?? 0,
-      usage: poll.usage ?? { inputTokens: 0, outputTokens: 0, totalBillingCost: 0 }
+      usage: poll.usage ?? { inputTokens: 0, outputTokens: 0, totalBillingCost: 0 },
+      toolCalls: poll.toolCalls ?? []
     };
+    logTaskResult(result);
+    return result;
   }
 }
 async function runTaskStream(httpConfig, body, onEvent) {
@@ -2998,12 +3020,15 @@ async function runTaskStream(httpConfig, body, onEvent) {
         if (event.type === "done") {
           result = {
             output: event.output,
+            outputRaw: event.outputRaw ?? "",
+            parsedSuccessfully: event.parsedSuccessfully ?? true,
             turns: event.turns ?? 0,
             usage: event.usage ?? {
               inputTokens: 0,
               outputTokens: 0,
               totalBillingCost: 0
-            }
+            },
+            toolCalls: event.toolCalls ?? []
           };
         }
       } catch (err) {
@@ -3025,12 +3050,15 @@ async function runTaskStream(httpConfig, body, onEvent) {
       if (event.type === "done") {
         result = {
           output: event.output,
+          outputRaw: event.outputRaw ?? "",
+          parsedSuccessfully: event.parsedSuccessfully ?? true,
           turns: event.turns ?? 0,
           usage: event.usage ?? {
             inputTokens: 0,
             outputTokens: 0,
             totalBillingCost: 0
-          }
+          },
+          toolCalls: event.toolCalls ?? []
         };
       }
     } catch (err) {
@@ -3044,6 +3072,7 @@ async function runTaskStream(httpConfig, body, onEvent) {
       500
     );
   }
+  logTaskResult(result);
   return result;
 }
 
@@ -3644,48 +3673,80 @@ var MindStudioAgent = class {
       );
     }
     const threadId = options?.threadId ?? (this._reuseThreadId && !getRequestContext() ? this._threadId : void 0);
-    const { data, headers } = await request(this._httpConfig, "POST", `/steps/${stepType}/execute`, {
+    const { data: asyncData, headers } = await request(this._currentHttpConfig, "POST", `/steps/${stepType}/execute-async`, {
       step,
       ...options?.appId != null && { appId: options.appId },
       ...threadId != null && { threadId },
       ...this._currentStreamId != null && { streamId: this._currentStreamId }
     });
-    let output;
-    if (data.output != null) {
-      output = data.output;
-    } else if (data.outputUrl) {
-      const res = await fetch(data.outputUrl);
+    const remaining = headers.get("x-ratelimit-remaining");
+    const returnedThreadId = asyncData.threadId ?? "";
+    if (this._reuseThreadId && returnedThreadId && !getRequestContext()) {
+      this._threadId = returnedThreadId;
+    }
+    if (!this._appId && asyncData.appId && !getRequestContext()) {
+      this._appId = asyncData.appId;
+    }
+    const pollUrl = `${this._currentHttpConfig.baseUrl}/developer/v2/steps/${stepType}/execute-async/poll/${asyncData.executionToken}`;
+    let pollDelay = 100;
+    while (true) {
+      await sleep3(pollDelay);
+      pollDelay = Math.min(pollDelay * 2, 5e3);
+      const res = await fetch(pollUrl, {
+        headers: { "User-Agent": "@mindstudio-ai/agent" }
+      });
+      if (res.status === 502 || res.status === 503 || res.status === 504) continue;
+      if (res.status === 404) {
+        throw new MindStudioError(
+          `[${stepType}] Execution token expired.`,
+          "poll_token_expired",
+          404
+        );
+      }
       if (!res.ok) {
+        const errorBody = await res.json().catch(() => ({}));
         throw new MindStudioError(
-          `Failed to fetch ${stepType} output from S3: ${res.status} ${res.statusText}`,
-          "output_fetch_error",
-          res.status
+          errorBody.message ?? errorBody.error ?? `[${stepType}] Poll failed: ${res.status} ${res.statusText}`,
+          errorBody.code ?? "poll_error",
+          res.status,
+          errorBody
         );
       }
-      const envelope = await res.json();
-      output = envelope.value;
-    } else {
-      output = void 0;
-    }
-    const returnedThreadId = headers.get("x-mindstudio-thread-id") ?? "";
-    if (this._reuseThreadId && returnedThreadId && !getRequestContext()) {
-      this._threadId = returnedThreadId;
-    }
-    const returnedAppId = headers.get("x-mindstudio-app-id");
-    if (!this._appId && returnedAppId && !getRequestContext()) {
-      this._appId = returnedAppId;
+      const poll = await res.json();
+      if (poll.status === "pending") continue;
+      if (poll.status === "error") {
+        throw new MindStudioError(
+          `[${stepType}] ${poll.error ?? "Step execution failed."}`,
+          "step_error",
+          500
+        );
+      }
+      let output;
+      if (poll.output != null) {
+        output = poll.output;
+      } else if (poll.outputUrl) {
+        const s3Res = await fetch(poll.outputUrl);
+        if (!s3Res.ok) {
+          throw new MindStudioError(
+            `Failed to fetch ${stepType} output from S3: ${s3Res.status} ${s3Res.statusText}`,
+            "output_fetch_error",
+            s3Res.status
+          );
+        }
+        const envelope = await s3Res.json();
+        output = envelope.value;
+      } else {
+        output = void 0;
+      }
+      return {
+        ...output,
+        $appId: poll.appId ?? asyncData.appId ?? "",
+        $threadId: poll.threadId ?? returnedThreadId,
+        $rateLimitRemaining: remaining != null ? parseInt(remaining, 10) : void 0,
+        $billingCost: poll.billingCost,
+        $billingEvents: poll.billingEvents
+      };
     }
-    const remaining = headers.get("x-ratelimit-remaining");
-    const billingCost = headers.get("x-mindstudio-billing-cost");
-    const billingEvents = headers.get("x-mindstudio-billing-events");
-    return {
-      ...output,
-      $appId: headers.get("x-mindstudio-app-id") ?? "",
-      $threadId: returnedThreadId,
-      $rateLimitRemaining: remaining != null ? parseInt(remaining, 10) : void 0,
-      $billingCost: billingCost != null ? parseFloat(billingCost) : void 0,
-      $billingEvents: billingEvents != null ? JSON.parse(billingEvents) : void 0
-    };
   }
   /**
    * @internal Streaming step execution — sends `Accept: text/event-stream`
@@ -3884,6 +3945,7 @@ var MindStudioAgent = class {
       const res = await fetch(pollUrl, {
         headers: { "User-Agent": "@mindstudio-ai/agent" }
       });
+      if (res.status === 502 || res.status === 503 || res.status === 504) continue;
       if (res.status === 404) {
         throw new MindStudioError(
           "Batch poll token not found or expired.",
@@ -4047,6 +4109,7 @@ var MindStudioAgent = class {
       const res = await fetch(pollUrl, {
         headers: { "User-Agent": "@mindstudio-ai/agent" }
       });
+      if (res.status === 502 || res.status === 503 || res.status === 504) continue;
       if (res.status === 404) {
         throw new MindStudioError(
           "Poll token not found or expired.",