@townco/agent 0.1.28 → 0.1.29

package/dist/acp-server/adapter.d.ts

@@ -1,19 +1,26 @@
 import * as acp from "@agentclientprotocol/sdk";
 import type { AgentRunner } from "../runner";
+/**
+ * ACP extension key for subagent mode indicator
+ * Following ACP extensibility pattern with namespaced key
+ */
+export declare const SUBAGENT_MODE_KEY = "town.com/isSubagent";
+/**
+ * Extension metadata type for subagent mode indicator
+ */
+export interface SubagentModeExtension {
+    [SUBAGENT_MODE_KEY]?: boolean;
+}
 /** Adapts an Agent to speak the ACP protocol */
 export declare class AgentAcpAdapter implements acp.Agent {
-	private connection;
-	private sessions;
-	private agent;
-	constructor(agent: AgentRunner, connection: acp.AgentSideConnection);
-	initialize(_params: acp.InitializeRequest): Promise<acp.InitializeResponse>;
-	newSession(_params: acp.NewSessionRequest): Promise<acp.NewSessionResponse>;
-	authenticate(
-		_params: acp.AuthenticateRequest,
-	): Promise<acp.AuthenticateResponse | undefined>;
-	setSessionMode(
-		_params: acp.SetSessionModeRequest,
-	): Promise<acp.SetSessionModeResponse>;
-	prompt(params: acp.PromptRequest): Promise<acp.PromptResponse>;
-	cancel(params: acp.CancelNotification): Promise<void>;
+    private connection;
+    private sessions;
+    private agent;
+    constructor(agent: AgentRunner, connection: acp.AgentSideConnection);
+    initialize(_params: acp.InitializeRequest): Promise<acp.InitializeResponse>;
+    newSession(params: acp.NewSessionRequest): Promise<acp.NewSessionResponse>;
+    authenticate(_params: acp.AuthenticateRequest): Promise<acp.AuthenticateResponse | undefined>;
+    setSessionMode(_params: acp.SetSessionModeRequest): Promise<acp.SetSessionModeResponse>;
+    prompt(params: acp.PromptRequest): Promise<acp.PromptResponse>;
+    cancel(params: acp.CancelNotification): Promise<void>;
 }

package/dist/acp-server/adapter.js

@@ -1,4 +1,9 @@
 import * as acp from "@agentclientprotocol/sdk";
+/**
+ * ACP extension key for subagent mode indicator
+ * Following ACP extensibility pattern with namespaced key
+ */
+export const SUBAGENT_MODE_KEY = "town.com/isSubagent";
 /** Adapts an Agent to speak the ACP protocol */
 export class AgentAcpAdapter {
     connection;
@@ -17,11 +22,12 @@ export class AgentAcpAdapter {
             },
         };
     }
-    async newSession(_params) {
+    async newSession(params) {
         const sessionId = Math.random().toString(36).substring(2);
         this.sessions.set(sessionId, {
             pendingPrompt: null,
             messages: [],
+            requestParams: params,
         });
         return {
             sessionId,
@@ -43,6 +49,7 @@ export class AgentAcpAdapter {
             session = {
                 pendingPrompt: null,
                 messages: [],
+                requestParams: { cwd: process.cwd(), mcpServers: [] },
             };
             this.sessions.set(params.sessionId, session);
         }
@@ -51,11 +58,16 @@ export class AgentAcpAdapter {
         // Generate a unique messageId for this assistant response
         const messageId = Math.random().toString(36).substring(2);
         try {
-            for await (const msg of this.agent.invoke({
+            const invokeParams = {
                 prompt: params.prompt,
                 sessionId: params.sessionId,
                 messageId,
-            })) {
+            };
+            // Only add sessionMeta if it's defined
+            if (session.requestParams._meta) {
+                invokeParams.sessionMeta = session.requestParams._meta;
+            }
+            for await (const msg of this.agent.invoke(invokeParams)) {
                 // Debug: log if this is an agent_message_chunk with tokenUsage in _meta
                 if ("sessionUpdate" in msg &&
                     msg.sessionUpdate === "agent_message_chunk") {

package/dist/acp-server/cli.d.ts

@@ -1,5 +1,3 @@
 import type { AgentDefinition } from "../definition";
 import { type AgentRunner } from "../runner";
-export declare function makeStdioTransport(
-	agent: AgentRunner | AgentDefinition,
-): void;
+export declare function makeStdioTransport(agent: AgentRunner | AgentDefinition): void;

package/dist/acp-server/cli.js

@@ -3,13 +3,9 @@ import * as acp from "@agentclientprotocol/sdk";
 import { makeRunnerFromDefinition } from "../runner";
 import { AgentAcpAdapter } from "./adapter";
 export function makeStdioTransport(agent) {
-	const agentRunner =
-		"definition" in agent ? agent : makeRunnerFromDefinition(agent);
-	const input = Writable.toWeb(process.stdout);
-	const output = Readable.toWeb(process.stdin);
-	const stream = acp.ndJsonStream(input, output);
-	new acp.AgentSideConnection(
-		(conn) => new AgentAcpAdapter(agentRunner, conn),
-		stream,
-	);
+    const agentRunner = "definition" in agent ? agent : makeRunnerFromDefinition(agent);
+    const input = Writable.toWeb(process.stdout);
+    const output = Readable.toWeb(process.stdin);
+    const stream = acp.ndJsonStream(input, output);
+    new acp.AgentSideConnection((conn) => new AgentAcpAdapter(agentRunner, conn), stream);
 }

package/dist/acp-server/http.d.ts

@@ -1,5 +1,3 @@
 import type { AgentDefinition } from "../definition";
 import { type AgentRunner } from "../runner";
-export declare function makeHttpTransport(
-	agent: AgentRunner | AgentDefinition,
-): void;
+export declare function makeHttpTransport(agent: AgentRunner | AgentDefinition): void;

package/dist/acp-server/http.js

@@ -2,11 +2,11 @@ import { createHash } from "node:crypto";
 import { gzipSync } from "node:zlib";
 import * as acp from "@agentclientprotocol/sdk";
 import { PGlite } from "@electric-sql/pglite";
+import { createLogger } from "@townco/core";
 import { Hono } from "hono";
 import { cors } from "hono/cors";
 import { streamSSE } from "hono/streaming";
 import { makeRunnerFromDefinition } from "../runner";
-import { createLogger } from "../utils/logger.js";
 import { AgentAcpAdapter } from "./adapter";
 const logger = createLogger("agent");
 /**
@@ -103,7 +103,7 @@ export function makeHttpTransport(agent) {
                             requestId: rawMsg.id,
                             originalSize,
                             compressedSize,
-                            compressionRatio: ((1 - compressedSize / originalSize) * 100).toFixed(1) + "%",
+                            compressionRatio: `${((1 - compressedSize / originalSize) * 100).toFixed(1)}%`,
                         });
                     }
                     // Escape single quotes for PostgreSQL
@@ -214,7 +214,7 @@ export function makeHttpTransport(agent) {
                             messageType,
                             originalSize,
                             compressedSize,
-                            compressionRatio: ((1 - compressedSize / originalSize) * 100).toFixed(1) + "%",
+                            compressionRatio: `${((1 - compressedSize / originalSize) * 100).toFixed(1)}%`,
                         });
                     }
                     // Escape single quotes for PostgreSQL

package/dist/index.js

@@ -1,4 +1,5 @@
 import { makeHttpTransport, makeStdioTransport } from "./acp-server";
+import { makeSubagentsTool } from "./utils";
 const exampleAgent = {
     model: "claude-sonnet-4-5-20250929",
     systemPrompt: "You are a helpful assistant.",
@@ -10,6 +11,13 @@ const exampleAgent = {
             type: "filesystem",
             //working_directory: "/Users/<user>/town"
         },
+        makeSubagentsTool([
+            {
+                agentName: "example",
+                description: "An example subagent for demonstration",
+                path: import.meta.filename,
+            },
+        ]),
     ],
     mcps: [],
 };

package/dist/runner/agent-runner.d.ts

@@ -31,6 +31,7 @@ export declare const zAgentRunnerParams: z.ZodObject<{
 export type CreateAgentRunnerParams = z.infer<typeof zAgentRunnerParams>;
 export type InvokeRequest = Omit<PromptRequest, "_meta"> & {
     messageId: string;
+    sessionMeta?: Record<string, unknown>;
 };
 export interface TokenUsage {
     inputTokens?: number;

package/dist/runner/index.d.ts

@@ -1,6 +1,4 @@
 import type { AgentDefinition } from "../definition";
 import { type AgentRunner } from "./agent-runner";
 export type { AgentRunner };
-export declare const makeRunnerFromDefinition: (
-	definition: AgentDefinition,
-) => AgentRunner;
+export declare const makeRunnerFromDefinition: (definition: AgentDefinition) => AgentRunner;

package/dist/runner/index.js

@@ -1,22 +1,18 @@
 import { zAgentRunnerParams } from "./agent-runner";
 import { LangchainAgent } from "./langchain";
 export const makeRunnerFromDefinition = (definition) => {
-	const agentRunnerParams = zAgentRunnerParams.safeParse(definition);
-	if (!agentRunnerParams.success) {
-		throw new Error(
-			`Invalid agent definition: ${agentRunnerParams.error.message}`,
-		);
-	}
-	switch (definition.harnessImplementation) {
-		case undefined:
-		case "langchain": {
-			return new LangchainAgent(agentRunnerParams.data);
-		}
-		default: {
-			const _exhaustiveCheck = definition.harnessImplementation;
-			throw new Error(
-				`Unsupported harness implementation: ${definition.harnessImplementation}`,
-			);
-		}
-	}
+    const agentRunnerParams = zAgentRunnerParams.safeParse(definition);
+    if (!agentRunnerParams.success) {
+        throw new Error(`Invalid agent definition: ${agentRunnerParams.error.message}`);
+    }
+    switch (definition.harnessImplementation) {
+        case undefined:
+        case "langchain": {
+            return new LangchainAgent(agentRunnerParams.data);
+        }
+        default: {
+            const _exhaustiveCheck = definition.harnessImplementation;
+            throw new Error(`Unsupported harness implementation: ${definition.harnessImplementation}`);
+        }
+    }
 };

package/dist/runner/langchain/index.d.ts

@@ -1,7 +1,7 @@
 import type { PromptResponse } from "@agentclientprotocol/sdk";
 import { type DynamicStructuredTool, type Tool } from "langchain";
 import type { AgentRunner, CreateAgentRunnerParams, ExtendedSessionUpdate, InvokeRequest } from "../agent-runner";
-import type { BuiltInToolType } from "../tools.js";
+import type { BuiltInToolType } from "../tools";
 type LangchainTool = DynamicStructuredTool | Tool;
 /** Lazily-loaded langchain tools */
 type LazyLangchainTool = MakeLazy<LangchainTool>;
@@ -13,4 +13,4 @@ export declare class LangchainAgent implements AgentRunner {
     constructor(params: CreateAgentRunnerParams);
     invoke(req: InvokeRequest): AsyncGenerator<ExtendedSessionUpdate, PromptResponse, undefined>;
 }
-export {};
+export { makeSubagentsTool } from "./tools/subagent.js";

package/dist/runner/langchain/index.js

@@ -1,12 +1,14 @@
 import { MultiServerMCPClient } from "@langchain/mcp-adapters";
+import { createLogger } from "@townco/core";
 import { AIMessageChunk, createAgent, ToolMessage, tool, } from "langchain";
 import { z } from "zod";
-import { createLogger } from "../../utils/logger.js";
+import { SUBAGENT_MODE_KEY } from "../../acp-server/adapter";
 import { loadCustomToolModule, } from "../tool-loader.js";
-import { makeFilesystemTools } from "./tools/filesystem.js";
-import { todoItemSchema, todoWrite } from "./tools/todo";
+import { makeFilesystemTools } from "./tools/filesystem";
+import { TASK_TOOL_NAME } from "./tools/subagent";
+import { TODO_WRITE_TOOL_NAME, todoWrite } from "./tools/todo";
 import { makeWebSearchTools } from "./tools/web_search";
-const logger = createLogger("agent-runner");
+const _logger = createLogger("agent-runner");
 const getWeather = tool(({ city }) => `It's always sunny in ${city}!`, {
     name: "get_weather",
     description: "Get the weather for a given city",
@@ -113,12 +115,19 @@ export class LangchainAgent {
         if ((this.definition.mcps?.length ?? 0) > 0) {
             enabledTools.push(...(await makeMcpToolsClient(this.definition.mcps).getTools()));
         }
+        // Filter tools if running in subagent mode
+        const isSubagent = req.sessionMeta?.[SUBAGENT_MODE_KEY] === true;
+        const finalTools = isSubagent
+            ? enabledTools.filter((t) => t.name !== TODO_WRITE_TOOL_NAME && t.name !== TASK_TOOL_NAME)
+            : enabledTools;
         const agentConfig = {
             model: this.definition.model,
-            tools: enabledTools,
+            tools: finalTools,
         };
-        if (this.definition.systemPrompt) {
-            agentConfig.systemPrompt = this.definition.systemPrompt;
+        // Inject system prompt with optional TodoWrite instructions
+        const hasTodoWrite = builtInNames.includes("todo_write");
+        if (hasTodoWrite) {
+            agentConfig.systemPrompt = `${agentConfig.systemPrompt ?? ""}\n\n${TODO_WRITE_INSTRUCTIONS}`;
         }
         const agent = createAgent(agentConfig);
         const messages = req.prompt
@@ -391,3 +400,53 @@ const makeMcpToolsClient = (mcpConfigs) => {
     });
     return client;
 };
+// System prompt instructions to inject when TodoWrite tool is enabled
+const TODO_WRITE_INSTRUCTIONS = `
+# Task Management
+You have access to the TodoWrite tools to help you manage and plan tasks. Use these tools VERY frequently to ensure that you are tracking your tasks and giving the user visibility into your progress.
+These tools are also EXTREMELY helpful for planning tasks, and for breaking down larger complex tasks into smaller steps. If you do not use this tool when planning, you may forget to do important tasks - and that is unacceptable.
+
+It is critical that you mark todos as completed as soon as you are done with a task. Do not batch up multiple tasks before marking them as completed.
+
+Examples:
+
+<example>
+user: Run the build and fix any type errors
+assistant: I'm going to use the TodoWrite tool to write the following items to the todo list:
+- Run the build
+- Fix any type errors
+
+I'm now going to run the build using Bash.
+
+Looks like I found 10 type errors. I'm going to use the TodoWrite tool to write 10 items to the todo list.
+
+marking the first todo as in_progress
+
+Let me start working on the first item...
+
+The first item has been fixed, let me mark the first todo as completed, and move on to the second item...
+..
+..
+</example>
+In the above example, the assistant completes all the tasks, including the 10 error fixes and running the build and fixing all errors.
+
+<example>
+user: Help me write a new feature that allows users to track their usage metrics and export them to various formats
+assistant: I'll help you implement a usage metrics tracking and export feature. Let me first use the TodoWrite tool to plan this task.
+Adding the following todos to the todo list:
+1. Research existing metrics tracking in the codebase
+2. Design the metrics collection system
+3. Implement core metrics tracking functionality
+4. Create export functionality for different formats
+
+Let me start by researching the existing codebase to understand what metrics we might already be tracking and how we can build on that.
+
+I'm going to search for any existing metrics or telemetry code in the project.
+
+I've found some existing telemetry code. Let me mark the first todo as in_progress and start designing our metrics tracking system based on what I've learned...
+
+[Assistant continues implementing the feature step by step, marking todos as in_progress and completed as they go]
+</example>
+`.trim();
+// Re-export subagent tool utility
+export { makeSubagentsTool } from "./tools/subagent.js";

package/dist/runner/langchain/tools/subagent.d.ts

@@ -0,0 +1,43 @@
+import type { DirectTool } from "../../tools.js";
+/**
+ * Name of the Task tool created by makeSubagentsTool
+ */
+export declare const TASK_TOOL_NAME = "Task";
+/**
+ * Configuration for a single subagent - supports two variants:
+ * 1. Agent name with optional working directory
+ * 2. Direct path to agent's index.ts file
+ */
+type SubagentConfig = {
+    agentName: string;
+    description: string;
+    cwd?: string;
+} | {
+    agentName: string;
+    description: string;
+    path: string;
+};
+/**
+ * Creates a DirectTool that delegates work to one of multiple configured subagents.
+ *
+ * @param configs - Array of subagent configurations
+ * @returns A DirectTool named "Task" that can route to any configured subagent
+ *
+ * @example
+ * ```typescript
+ * import { makeSubagentsTool } from "@townco/agent/utils";
+ *
+ * const agent: AgentDefinition = {
+ *   model: "claude-sonnet-4-5-20250929",
+ *   systemPrompt: "You are a coordinator.",
+ *   tools: [
+ *     makeSubagentsTool([
+ *       { agentName: "researcher", description: "Use this agent to research topics", cwd: "/path/to/agents" },
+ *       { agentName: "writer", description: "Use this agent to write content", path: "/absolute/path/to/writer/index.ts" },
+ *     ]),
+ *   ]
+ * };
+ * ```
+ */
+export declare function makeSubagentsTool(configs: SubagentConfig[]): DirectTool;
+export {};