pulse-coder-engine 0.0.1-alpha.13 → 0.0.1-alpha.14

package/dist/index.js

@@ -391,6 +391,9 @@ async function loop(context, options) {
   const loopHooks = options?.hooks ?? {};
   while (true) {
     try {
+      if (options?.abortSignal?.aborted) {
+        return "Request aborted.";
+      }
       if (compactionAttempts < MAX_COMPACTION_ATTEMPTS) {
         const { didCompact, newMessages } = await maybeCompactContext(context, {
           provider: options?.provider,
@@ -404,6 +407,9 @@ async function loop(context, options) {
           continue;
         }
       }
+      if (options?.abortSignal?.aborted) {
+        return "Request aborted.";
+      }
       let tools = options?.tools || {};
       let systemPrompt = options?.systemPrompt;
       if (loopHooks.beforeLLMCall?.length) {
@@ -428,6 +434,9 @@ async function loop(context, options) {
         onClarificationRequest: options?.onClarificationRequest,
         abortSignal: options?.abortSignal
       };
+      if (options?.abortSignal?.aborted) {
+        return "Request aborted.";
+      }
       const result = streamTextAI(context.messages, tools, {
         abortSignal: options?.abortSignal,
         toolExecutionContext,
@@ -466,6 +475,9 @@ async function loop(context, options) {
           await hook({ context, finishReason, text });
         }
       }
+      if (options?.abortSignal?.aborted) {
+        return "Request aborted.";
+      }
       if (finishReason === "stop") {
         if (!text) {
           continue;
@@ -515,7 +527,7 @@ async function loop(context, options) {
       }
       if (isRetryableError(error)) {
         const delay = Math.min(2e3 * Math.pow(2, errorCount - 1), 3e4);
-        await sleep(delay);
+        await sleep(delay, options?.abortSignal);
         continue;
       }
       return `Error: ${error?.message ?? String(error)}`;
@@ -526,8 +538,34 @@ function isRetryableError(error) {
   const status = error?.status ?? error?.statusCode;
   return status === 429 || status === 500 || status === 502 || status === 503;
 }
-function sleep(ms) {
-  return new Promise((resolve) => setTimeout(resolve, ms));
+function sleep(ms, abortSignal) {
+  return new Promise((resolve, reject) => {
+    if (abortSignal?.aborted) {
+      const abortError = new Error("Aborted");
+      abortError.name = "AbortError";
+      reject(abortError);
+      return;
+    }
+    let timeoutId;
+    const onAbort = () => {
+      clearTimeout(timeoutId);
+      if (abortSignal) {
+        abortSignal.removeEventListener("abort", onAbort);
+      }
+      const abortError = new Error("Aborted");
+      abortError.name = "AbortError";
+      reject(abortError);
+    };
+    timeoutId = setTimeout(() => {
+      if (abortSignal) {
+        abortSignal.removeEventListener("abort", onAbort);
+      }
+      resolve();
+    }, ms);
+    if (abortSignal) {
+      abortSignal.addEventListener("abort", onAbort, { once: true });
+    }
+  });
 }
 
 // src/tools/read.ts