small improvements

Author: YousefED

packages/xl-ai/src/api/formats/base-tools/createAddBlocksTool.ts

@@ -9,6 +9,7 @@ import {
 import { updateToReplaceSteps } from "../../../prosemirror/changeset.js";
 import { RebaseTool } from "../../../prosemirror/rebaseTool.js";
 import { Result, streamTool } from "../../../streamTool/streamTool.js";
+import { AbortError } from "../../../util/AbortError.js";
 import { isEmptyParagraph } from "../../../util/emptyBlock.js";
 import { validateBlockArray } from "./util/validateBlockArray.js";
 
@@ -269,9 +270,7 @@ export function createAddBlocksTool<T>(config: {
 
               for (const step of agentSteps) {
                 if (abortSignal?.aborted) {
-                  const error = new Error("Operation was aborted");
-                  error.name = "AbortError";
-                  throw error;
+                  throw new AbortError("Operation was aborted");
                 }
                 if (options.withDelays) {
                   await delayAgentStep(step);

packages/xl-ai/src/api/formats/base-tools/createUpdateBlockTool.ts

@@ -9,6 +9,7 @@ import {
 import { updateToReplaceSteps } from "../../../prosemirror/changeset.js";
 import { RebaseTool } from "../../../prosemirror/rebaseTool.js";
 import { Result, streamTool } from "../../../streamTool/streamTool.js";
+import { AbortError } from "../../../util/AbortError.js";
 
 export type UpdateBlockToolCall<T> = {
   type: "update";
@@ -245,9 +246,7 @@ export function createUpdateBlockTool<T>(config: {
 
             for (const step of agentSteps) {
               if (abortSignal?.aborted) {
-                const error = new Error("Operation was aborted");
-                error.name = "AbortError";
-                throw error;
+                throw new AbortError("Operation was aborted");
               }
               if (options.withDelays) {
                 await delayAgentStep(step);

packages/xl-ai/src/util/AbortError.ts

@@ -0,0 +1,9 @@
+export class AbortError extends Error {
+  constructor(
+    message: string,
+    options?: { cause?: unknown; aborted?: boolean },
+  ) {
+    super(message, options);
+    this.name = "AbortError";
+  }
+}