@townco/agent 0.1.88 → 0.1.101

package/dist/runner/langchain/tools/web_search.js

@@ -30,6 +30,17 @@ function getTownExaClient() {
     _townExaClient = new Exa(shedAuth.accessToken, `${shedAuth.shedUrl}/api/exa`);
     return _townExaClient;
 }
+// Track citation counter for web search results
+// This is reset at the start of each session via resetWebSearchCitationCounter()
+let webSearchCitationCounter = 0;
+/** Reset the web search citation counter (call at start of each session) */
+export function resetWebSearchCitationCounter() {
+    webSearchCitationCounter = 0;
+}
+/** Get the current citation counter value */
+export function getWebSearchCitationCounter() {
+    return webSearchCitationCounter;
+}
 function makeWebSearchToolsInternal(getClient) {
     const webSearch = tool(async ({ query }) => {
         const client = getClient();
@@ -43,7 +54,25 @@ function makeWebSearchToolsInternal(getClient) {
         if (!result.results || result.results.length === 0) {
             return `No results found for query: ${query}`;
         }
-        return result;
+        // Format results with citation numbers so the LLM can reference them
+        // The adapter will extract sources using the same numbering
+        const formattedResults = result.results.map((r) => {
+            webSearchCitationCounter++;
+            return {
+                citationId: webSearchCitationCounter,
+                url: r.url,
+                title: r.title,
+                text: r.text,
+            };
+        });
+        // Return both the formatted results (for LLM) and raw results (for adapter extraction)
+        return {
+            results: result.results, // Raw results for adapter source extraction
+            formattedForCitation: formattedResults.map((r) => ({
+                ...r,
+                citationInstruction: `Use [[${r.citationId}]] when citing this source`,
+            })),
+        };
     }, {
         name: "WebSearch",
         description: "\n- Allows you to search the web and use the results to inform responses\n" +
@@ -51,11 +80,13 @@ function makeWebSearchToolsInternal(getClient) {
             "- Returns search result information formatted as search result blocks\n" +
             "- Use this tool for accessing information beyond your knowledge cutoff\n" +
             "- Searches are performed automatically within a single API call\n" +
+            "- Results include citation IDs - use [[N]] format to cite sources in your response\n" +
             "\n" +
             "Usage notes:\n" +
             "  - Domain filtering is supported to include or block specific websites\n" +
             "  - Web search is only available in the US\n" +
-            '  - Account for "Today\'s date" in <env>. For example, if <env> says "Today\'s date: 2025-07-01", and the user wants the latest docs, do not use 2024 in the search query. Use 2025.\n',
+            '  - Account for "Today\'s date" in <env>. For example, if <env> says "Today\'s date: 2025-07-01", and the user wants the latest docs, do not use 2024 in the search query. Use 2025.\n' +
+            "  - IMPORTANT: When using information from search results, cite the source using [[N]] format where N is the citationId\n",
         schema: z.object({
             query: z.string().describe("The search query to use"),
         }),

package/dist/runner/session-context.d.ts

@@ -10,6 +10,26 @@ export interface SessionContext {
     /** Artifacts directory: <agentDir>/.sessions/<sessionId>/artifacts */
     artifactsDir: string;
 }
+/**
+ * Run a function with an abort signal available via AsyncLocalStorage.
+ * Tools can access the signal using getAbortSignal().
+ */
+export declare function runWithAbortSignal<T>(signal: AbortSignal, fn: () => T): T;
+/**
+ * Bind an async generator to an abort signal context so that every iteration
+ * runs with the abort signal available.
+ */
+export declare function bindGeneratorToAbortSignal<T, R, N = unknown>(signal: AbortSignal, generator: AsyncGenerator<T, R, N>): AsyncGenerator<T, R, N>;
+/**
+ * Get the current abort signal.
+ * Returns undefined if called outside of an abort signal context.
+ */
+export declare function getAbortSignal(): AbortSignal | undefined;
+/**
+ * Check if the current operation has been aborted.
+ * Returns false if no abort signal is available.
+ */
+export declare function isAborted(): boolean;
 /**
  * Run a function with session context available via AsyncLocalStorage.
  * Tools can access the context using getSessionContext().

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({