@townco/agent 0.1.87 → 0.1.98

package/dist/runner/session-context.js

@@ -1,6 +1,60 @@
 import { AsyncLocalStorage } from "node:async_hooks";
 import * as path from "node:path";
 const sessionStorage = new AsyncLocalStorage();
+/**
+ * Abort signal context for cancellation support.
+ * Tools can access this to check if the current operation has been cancelled.
+ */
+const abortSignalStorage = new AsyncLocalStorage();
+/**
+ * Run a function with an abort signal available via AsyncLocalStorage.
+ * Tools can access the signal using getAbortSignal().
+ */
+export function runWithAbortSignal(signal, fn) {
+    return abortSignalStorage.run(signal, fn);
+}
+/**
+ * Bind an async generator to an abort signal context so that every iteration
+ * runs with the abort signal available.
+ */
+export function bindGeneratorToAbortSignal(signal, generator) {
+    const boundNext = (value) => abortSignalStorage.run(signal, () => generator.next(value));
+    const boundReturn = generator.return
+        ? (value) => abortSignalStorage.run(signal, () => generator.return?.(value))
+        : undefined;
+    const boundThrow = generator.throw
+        ? (e) => abortSignalStorage.run(signal, () => generator.throw?.(e))
+        : undefined;
+    const boundGenerator = {
+        next: boundNext,
+        return: boundReturn,
+        throw: boundThrow,
+        [Symbol.asyncIterator]() {
+            return this;
+        },
+        [Symbol.asyncDispose]: async () => {
+            if (Symbol.asyncDispose in generator) {
+                await generator[Symbol.asyncDispose]();
+            }
+        },
+    };
+    return boundGenerator;
+}
+/**
+ * Get the current abort signal.
+ * Returns undefined if called outside of an abort signal context.
+ */
+export function getAbortSignal() {
+    return abortSignalStorage.getStore();
+}
+/**
+ * Check if the current operation has been aborted.
+ * Returns false if no abort signal is available.
+ */
+export function isAborted() {
+    const signal = abortSignalStorage.getStore();
+    return signal?.aborted ?? false;
+}
 /**
  * Run a function with session context available via AsyncLocalStorage.
  * Tools can access the context using getSessionContext().

package/dist/runner/tools.d.ts

@@ -1,6 +1,6 @@
 import { z } from "zod";
 /** Built-in tool types. */
-export declare const zBuiltInToolType: z.ZodUnion<readonly [z.ZodLiteral<"artifacts">, z.ZodLiteral<"todo_write">, z.ZodLiteral<"get_weather">, z.ZodLiteral<"web_search">, z.ZodLiteral<"town_web_search">, z.ZodLiteral<"filesystem">, z.ZodLiteral<"generate_image">, z.ZodLiteral<"town_generate_image">, z.ZodLiteral<"browser">, z.ZodLiteral<"document_extract">]>;
+export declare const zBuiltInToolType: z.ZodUnion<readonly [z.ZodLiteral<"artifacts">, z.ZodLiteral<"todo_write">, z.ZodLiteral<"get_weather">, z.ZodLiteral<"web_search">, z.ZodLiteral<"town_web_search">, z.ZodLiteral<"filesystem">, z.ZodLiteral<"generate_image">, z.ZodLiteral<"town_generate_image">, z.ZodLiteral<"browser">, z.ZodLiteral<"document_extract">, z.ZodLiteral<"town_e2b">]>;
 /** Subagent configuration schema for Task tools. */
 export declare const zSubagentConfig: z.ZodObject<{
     agentName: z.ZodString;
@@ -23,7 +23,7 @@ declare const zDirectTool: z.ZodObject<{
     }, z.core.$strip>>>;
 }, z.core.$strip>;
 /** Tool type - can be a built-in tool string or custom tool object. */
-export declare const zToolType: z.ZodUnion<readonly [z.ZodUnion<readonly [z.ZodLiteral<"artifacts">, z.ZodLiteral<"todo_write">, z.ZodLiteral<"get_weather">, z.ZodLiteral<"web_search">, z.ZodLiteral<"town_web_search">, z.ZodLiteral<"filesystem">, z.ZodLiteral<"generate_image">, z.ZodLiteral<"town_generate_image">, z.ZodLiteral<"browser">, z.ZodLiteral<"document_extract">]>, z.ZodObject<{
+export declare const zToolType: z.ZodUnion<readonly [z.ZodUnion<readonly [z.ZodLiteral<"artifacts">, z.ZodLiteral<"todo_write">, z.ZodLiteral<"get_weather">, z.ZodLiteral<"web_search">, z.ZodLiteral<"town_web_search">, z.ZodLiteral<"filesystem">, z.ZodLiteral<"generate_image">, z.ZodLiteral<"town_generate_image">, z.ZodLiteral<"browser">, z.ZodLiteral<"document_extract">, z.ZodLiteral<"town_e2b">]>, z.ZodObject<{
     type: z.ZodLiteral<"custom">;
     modulePath: z.ZodString;
 }, z.core.$strip>, z.ZodObject<{

package/dist/runner/tools.js

@@ -11,6 +11,7 @@ export const zBuiltInToolType = z.union([
     z.literal("town_generate_image"),
     z.literal("browser"),
     z.literal("document_extract"),
+    z.literal("town_e2b"),
 ]);
 /** Custom tool schema (loaded from module path). */
 const zCustomTool = z.object({