@townco/agent 0.1.50 → 0.1.52

package/dist/runner/langchain/index.js

@@ -161,11 +161,103 @@ export class LangchainAgent {
             if ((this.definition.mcps?.length ?? 0) > 0) {
                 enabledTools.push(...(await makeMcpToolsClient(this.definition.mcps).getTools()));
             }
+            // Wrap tools with response compaction if hook is configured
+            const hooks = this.definition.hooks ?? [];
+            const hasToolResponseHook = hooks.some((h) => h.type === "tool_response");
+            const noSession = req.sessionMeta?.[SUBAGENT_MODE_KEY] === true; // Subagents don't have session storage
+            // Track cumulative tool output tokens in this turn for proper context calculation
+            let cumulativeToolOutputTokens = 0;
+            let wrappedTools = enabledTools;
+            if (hasToolResponseHook && !noSession) {
+                const { countToolResultTokens } = await import("../../utils/token-counter.js");
+                const { toolResponseCompactor } = await import("../hooks/predefined/tool-response-compactor.js");
+                wrappedTools = enabledTools.map((originalTool) => {
+                    const wrappedFunc = async (input) => {
+                        // Execute the original tool
+                        const result = await originalTool.invoke(input);
+                        // Check if result should be compacted
+                        const resultStr = typeof result === "string" ? result : JSON.stringify(result);
+                        const rawOutput = { content: resultStr };
+                        const outputTokens = countToolResultTokens(rawOutput);
+                        // Skip compaction for small results (under 10k tokens)
+                        if (outputTokens < 10000) {
+                            // Still track this in cumulative total
+                            cumulativeToolOutputTokens += outputTokens;
+                            return result;
+                        }
+                        _logger.info("Tool wrapper: compacting large tool result", {
+                            toolName: originalTool.name,
+                            originalTokens: outputTokens,
+                            cumulativeToolOutputTokens,
+                        });
+                        // Calculate current context including all tool outputs so far in this turn
+                        // This ensures we account for multiple large tool calls in the same turn
+                        const baseContextTokens = turnTokenUsage.inputTokens || 10000;
+                        const currentTokens = baseContextTokens + cumulativeToolOutputTokens;
+                        const maxTokens = 200000; // Claude's limit
+                        // Build proper hook context with all required fields
+                        const hookContext = {
+                            session: {
+                                messages: req.contextMessages || [],
+                                context: [],
+                                requestParams: {
+                                    hookSettings: hooks.find((h) => h.type === "tool_response")
+                                        ?.setting,
+                                },
+                            },
+                            currentTokens,
+                            maxTokens,
+                            percentage: (currentTokens / maxTokens) * 100,
+                            model: this.definition.model,
+                            toolResponse: {
+                                toolCallId: "pending",
+                                toolName: originalTool.name,
+                                toolInput: input,
+                                rawOutput,
+                                outputTokens,
+                            },
+                        };
+                        // Call the tool response compactor directly
+                        const hookResult = await toolResponseCompactor(hookContext);
+                        // Extract modified output from metadata
+                        if (hookResult?.metadata?.modifiedOutput) {
+                            const modifiedOutput = hookResult.metadata
+                                .modifiedOutput;
+                            const compactedTokens = countToolResultTokens(modifiedOutput);
+                            // Update cumulative total with the compacted size (not original!)
+                            cumulativeToolOutputTokens += compactedTokens;
+                            _logger.info("Tool wrapper: compaction complete", {
+                                toolName: originalTool.name,
+                                originalTokens: outputTokens,
+                                compactedTokens,
+                                reduction: `${((1 - compactedTokens / outputTokens) * 100).toFixed(1)}%`,
+                                totalCumulativeTokens: cumulativeToolOutputTokens,
+                            });
+                            return typeof result === "string"
+                                ? modifiedOutput.content
+                                : JSON.stringify(modifiedOutput);
+                        }
+                        // No compaction happened, count original size
+                        cumulativeToolOutputTokens += outputTokens;
+                        return result;
+                    };
+                    // Create new tool with wrapped function
+                    const wrappedTool = tool(wrappedFunc, {
+                        name: originalTool.name,
+                        description: originalTool.description,
+                        schema: originalTool.schema,
+                    });
+                    // Preserve metadata
+                    wrappedTool.prettyName = originalTool.prettyName;
+                    wrappedTool.icon = originalTool.icon;
+                    return wrappedTool;
+                });
+            }
             // 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;
+                ? wrappedTools.filter((t) => t.name !== TODO_WRITE_TOOL_NAME && t.name !== TASK_TOOL_NAME)
+                : wrappedTools;
             // Create the model instance using the factory
             // This detects the provider from the model string:
             // - "gemini-2.0-flash" → Google Generative AI
@@ -187,79 +279,6 @@ export class LangchainAgent {
             const agent = createAgent(agentConfig);
             // Add logging callbacks for model requests
             const provider = detectProvider(this.definition.model);
-            const loggingCallback = {
-                handleChatModelStart: async (_llm, messages, runId, parentRunId, extraParams) => {
-                    _logger.info("Model request started", {
-                        provider,
-                        model: this.definition.model,
-                        runId,
-                        parentRunId,
-                        messageCount: messages.length,
-                        extraParams,
-                    });
-                },
-                handleLLMEnd: async (output, runId, parentRunId, tags, extraParams) => {
-                    // Extract token usage from output
-                    const llmResult = output;
-                    _logger.info("Model request completed", {
-                        provider,
-                        model: this.definition.model,
-                        runId,
-                        parentRunId,
-                        tags,
-                        tokenUsage: llmResult.llmOutput?.tokenUsage,
-                        generationCount: llmResult.generations?.length,
-                        extraParams,
-                    });
-                },
-                handleLLMError: async (error, runId, parentRunId, tags) => {
-                    _logger.error("Model request failed", {
-                        provider,
-                        model: this.definition.model,
-                        runId,
-                        parentRunId,
-                        tags,
-                        error: error.message,
-                        stack: error.stack,
-                    });
-                },
-                handleToolStart: async (_tool, input, runId, parentRunId, tags, metadata, runName) => {
-                    if (process.env.DEBUG_TELEMETRY === "true") {
-                        console.log(`[handleToolStart] runId=${runId}, runName=${runName}, parentRunId=${parentRunId}`);
-                        console.log(`[handleToolStart] Active context span:`, trace.getSpan(context.active())?.spanContext());
-                    }
-                    _logger.info("Tool started", {
-                        runId,
-                        parentRunId,
-                        runName,
-                        tags,
-                        metadata,
-                        input: input.substring(0, 200), // Truncate for logging
-                    });
-                },
-                handleToolEnd: async (_output, runId, parentRunId, tags) => {
-                    if (process.env.DEBUG_TELEMETRY === "true") {
-                        console.log(`[handleToolEnd] runId=${runId}, parentRunId=${parentRunId}`);
-                    }
-                    _logger.info("Tool completed", {
-                        runId,
-                        parentRunId,
-                        tags,
-                    });
-                },
-                handleToolError: async (error, runId, parentRunId, tags) => {
-                    if (process.env.DEBUG_TELEMETRY === "true") {
-                        console.log(`[handleToolError] runId=${runId}, error=${error.message}`);
-                    }
-                    _logger.error("Tool failed", {
-                        runId,
-                        parentRunId,
-                        tags,
-                        error: error.message,
-                        stack: error.stack,
-                    });
-                },
-            };
             // Build messages from context history if available, otherwise use just the prompt
             let messages;
             if (req.contextMessages && req.contextMessages.length > 0) {
@@ -303,7 +322,7 @@ export class LangchainAgent {
             const stream = context.with(invocationContext, () => agent.stream({ messages }, {
                 streamMode: ["updates", "messages"],
                 recursionLimit: 200,
-                callbacks: [loggingCallback, otelCallbacks],
+                callbacks: [otelCallbacks],
             }));
             for await (const streamItem of await stream) {
                 const [streamMode, chunk] = streamItem;
@@ -345,9 +364,12 @@ export class LangchainAgent {
                             }
                             // Create tool span within the invocation context
                             // This makes the tool span a child of the invocation span
+                            const toolInputJson = JSON.stringify(toolCall.args);
                             const toolSpan = context.with(invocationContext, () => telemetry.startSpan("agent.tool_call", {
                                 "tool.name": toolCall.name,
                                 "tool.id": toolCall.id,
+                                "tool.input": toolInputJson,
+                                "agent.session_id": req.sessionId,
                             }));
                             this.toolSpans.set(toolCall.id, toolSpan);
                             telemetry.log("info", `Tool call started: ${toolCall.name}`, {
@@ -534,6 +556,10 @@ export class LangchainAgent {
                             // End telemetry span for this tool call
                             const toolSpan = this.toolSpans.get(aiMessage.tool_call_id);
                             if (toolSpan) {
+                                // Add tool output to span before ending
+                                telemetry.setSpanAttributes(toolSpan, {
+                                    "tool.output": aiMessage.content,
+                                });
                                 telemetry.log("info", "Tool call completed", {
                                     toolCallId: aiMessage.tool_call_id,
                                 });

package/dist/runner/langchain/otel-callbacks.js

@@ -1,4 +1,4 @@
-import { context } from "@opentelemetry/api";
+import { context, trace } from "@opentelemetry/api";
 import { telemetry } from "../../telemetry/index.js";
 /**
  * OpenTelemetry callback handler for LangChain LLM calls.
@@ -45,6 +45,38 @@ function extractSystemPrompt(messages) {
         return undefined;
     }
 }
+/**
+ * Serializes LLM output to a string for logging.
+ * Preserves the raw provider format (content blocks, tool_calls, etc.)
+ */
+function serializeOutput(output) {
+    try {
+        const generations = output.generations.flat();
+        const serialized = generations.map((gen) => {
+            // ChatGeneration has a message property with the full AIMessage
+            const chatGen = gen;
+            if (chatGen.message) {
+                const msg = chatGen.message;
+                const result = {
+                    role: msg._getType?.() ?? "assistant",
+                    content: msg.content, // Keep as-is: string or ContentBlock[]
+                };
+                // Include tool_calls if present (LangChain's normalized format)
+                const aiMsg = msg;
+                if (aiMsg.tool_calls && aiMsg.tool_calls.length > 0) {
+                    result.tool_calls = aiMsg.tool_calls;
+                }
+                return result;
+            }
+            // Fallback for non-chat generations
+            return { text: gen.text };
+        });
+        return JSON.stringify(serialized);
+    }
+    catch (error) {
+        return `[Error serializing output: ${error}]`;
+    }
+}
 /**
  * Creates OpenTelemetry callback handlers for LangChain LLM calls.
  * These handlers instrument model invocations with OTEL spans and record token usage.
@@ -84,6 +116,18 @@ export function makeOtelCallbacks(opts) {
             }));
             if (span) {
                 spansByRunId.set(runId, span);
+                // Emit log for LLM request with trace context
+                const spanContext = span.spanContext();
+                telemetry.log("info", "LLM Request", {
+                    "gen_ai.operation.name": "chat",
+                    "gen_ai.provider.name": opts.provider,
+                    "gen_ai.request.model": opts.model,
+                    "gen_ai.input.messages": serializedMessages,
+                    "langchain.run_id": runId,
+                    // Include trace context for correlation
+                    trace_id: spanContext.traceId,
+                    span_id: spanContext.spanId,
+                });
             }
         },
         /**
@@ -105,6 +149,28 @@ export function makeOtelCallbacks(opts) {
                         : 0);
                 telemetry.recordTokenUsage(inputTokens, outputTokens, span);
             }
+            // Serialize output and attach to span
+            const serializedOutput = serializeOutput(output);
+            telemetry.setSpanAttributes(span, {
+                "gen_ai.output.messages": serializedOutput,
+            });
+            // Emit log for LLM response with trace context
+            const spanContext = span.spanContext();
+            telemetry.log("info", "LLM Response", {
+                "gen_ai.operation.name": "chat",
+                "gen_ai.output.messages": serializedOutput,
+                "langchain.run_id": runId,
+                // Include token usage in log
+                ...(tokenUsage
+                    ? {
+                        "gen_ai.usage.input_tokens": tokenUsage.inputTokens ?? 0,
+                        "gen_ai.usage.output_tokens": tokenUsage.outputTokens ?? 0,
+                    }
+                    : {}),
+                // Include trace context for correlation
+                trace_id: spanContext.traceId,
+                span_id: spanContext.spanId,
+            });
             telemetry.endSpan(span);
             spansByRunId.delete(runId);
         },

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

@@ -0,0 +1,14 @@
+import { z } from "zod";
+export declare function makeBashTool(workingDirectory?: string): import("langchain").DynamicStructuredTool<z.ZodObject<{
+    command: z.ZodString;
+    timeout: z.ZodOptional<z.ZodNumber>;
+    description: z.ZodOptional<z.ZodString>;
+}, z.core.$strip>, {
+    command: string;
+    timeout?: number | undefined;
+    description?: string | undefined;
+}, {
+    command: string;
+    timeout?: number | undefined;
+    description?: string | undefined;
+}, unknown>;

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

@@ -0,0 +1,135 @@
+import { spawn } from "node:child_process";
+import { once } from "node:events";
+import * as path from "node:path";
+import { SandboxManager, } from "@anthropic-ai/sandbox-runtime";
+import { tool } from "langchain";
+import { z } from "zod";
+/**
+ * Lazily initialize Sandbox Runtime with write access limited to workingDirectory.
+ */
+let initialized = false;
+async function ensureSandbox(workingDirectory) {
+    if (initialized)
+        return;
+    const cfg = {
+        network: {
+            // No outbound network needed for basic bash commands; block by default.
+            allowedDomains: [],
+            deniedDomains: [],
+        },
+        filesystem: {
+            // Allow writes only within the configured sandbox directory.
+            allowWrite: [workingDirectory],
+            denyWrite: [],
+            // Optional: harden reads a bit (deny common sensitive dirs)
+            denyRead: ["~/.ssh", "~/.gnupg", "/etc/ssh"],
+        },
+    };
+    await SandboxManager.initialize(cfg);
+    initialized = true;
+}
+/** Run a command string inside the sandbox, returning { stdout, stderr, code, timedOut }. */
+async function runSandboxed(cmd, cwd, timeoutMs) {
+    const wrapped = await SandboxManager.wrapWithSandbox(cmd);
+    const child = spawn(wrapped, { shell: true, cwd });
+    const stdout = [];
+    const stderr = [];
+    child.stdout?.on("data", (d) => stdout.push(Buffer.from(d)));
+    child.stderr?.on("data", (d) => stderr.push(Buffer.from(d)));
+    let timedOut = false;
+    let timeoutHandle;
+    // Set up timeout if specified
+    if (timeoutMs !== undefined && timeoutMs > 0) {
+        timeoutHandle = setTimeout(() => {
+            timedOut = true;
+            child.kill("SIGTERM");
+            // If process doesn't exit after SIGTERM, force kill after 1 second
+            setTimeout(() => {
+                if (!child.killed) {
+                    child.kill("SIGKILL");
+                }
+            }, 1000);
+        }, timeoutMs);
+    }
+    const [code] = (await once(child, "exit"));
+    if (timeoutHandle) {
+        clearTimeout(timeoutHandle);
+    }
+    return {
+        stdout: Buffer.concat(stdout),
+        stderr: Buffer.concat(stderr),
+        code: code ?? 1,
+        timedOut,
+    };
+}
+/** Truncate output if it exceeds the maximum length. */
+function truncateOutput(output, maxLength = 30000) {
+    if (output.length <= maxLength) {
+        return output;
+    }
+    return `${output.slice(0, maxLength)}\n\n... (output truncated, ${output.length - maxLength} characters omitted)`;
+}
+export function makeBashTool(workingDirectory) {
+    const resolvedWd = path.resolve(workingDirectory ?? process.cwd());
+    const bash = tool(async ({ command, timeout, description }) => {
+        await ensureSandbox(resolvedWd);
+        // Validate timeout
+        const DEFAULT_TIMEOUT = 120000; // 2 minutes
+        const MAX_TIMEOUT = 600000; // 10 minutes
+        let timeoutMs = timeout ?? DEFAULT_TIMEOUT;
+        if (timeoutMs > MAX_TIMEOUT) {
+            throw new Error(`Timeout cannot exceed ${MAX_TIMEOUT}ms (10 minutes), got ${timeoutMs}ms`);
+        }
+        if (timeoutMs < 0) {
+            throw new Error(`Timeout must be positive, got ${timeoutMs}ms`);
+        }
+        // Execute the command
+        const startTime = Date.now();
+        const { stdout, stderr, code, timedOut } = await runSandboxed(command, resolvedWd, timeoutMs);
+        const executionTime = Date.now() - startTime;
+        // Convert buffers to strings
+        const stdoutStr = stdout.toString("utf8");
+        const stderrStr = stderr.toString("utf8");
+        // Truncate outputs if necessary
+        const truncatedStdout = truncateOutput(stdoutStr);
+        const truncatedStderr = truncateOutput(stderrStr);
+        // Build result message
+        let result = "";
+        if (timedOut) {
+            result += `Command timed out after ${timeoutMs}ms\n\n`;
+        }
+        if (truncatedStdout) {
+            result += truncatedStdout;
+        }
+        if (truncatedStderr) {
+            if (result)
+                result += "\n\n";
+            result += `STDERR:\n${truncatedStderr}`;
+        }
+        if (!truncatedStdout && !truncatedStderr) {
+            result = "(no output)";
+        }
+        // Add exit code info
+        result += `\n\nExit code: ${code}`;
+        if (description) {
+            result += `\nDescription: ${description}`;
+        }
+        result += `\nExecution time: ${executionTime}ms`;
+        return result;
+    }, {
+        name: "Bash",
+        description: 'Executes a given bash command in a persistent shell session with optional timeout, ensuring proper handling and security measures.\n\nBefore executing the command, please follow these steps:\n\n1. Directory Verification:\n   - If the command will create new directories or files, first use `ls` to verify the parent directory exists and is the correct location\n   - For example, before running "mkdir foo/bar", first use `ls foo` to check that "foo" exists and is the intended parent directory\n\n2. Command Execution:\n   - Always quote file paths that contain spaces with double quotes (e.g., cd "path with spaces/file.txt")\n   - Examples of proper quoting:\n     - cd "/Users/name/My Documents" (correct)\n     - cd /Users/name/My Documents (incorrect - will fail)\n     - python "/path/with spaces/script.py" (correct)\n     - python /path/with spaces/script.py (incorrect - will fail)\n   - After ensuring proper quoting, execute the command.\n   - Capture the output of the command.\n\nUsage notes:\n  - The command argument is required.\n  - You can specify an optional timeout in milliseconds (up to 600000ms / 10 minutes). If not specified, commands will timeout after 120000ms (2 minutes).\n  - It is very helpful if you write a clear, concise description of what this command does in 5-10 words.\n  - If the output exceeds 30000 characters, output will be truncated before being returned to you.\n  - You can use the `run_in_background` parameter to run the command in the background, which allows you to continue working while the command runs. You can monitor the output using the Bash tool as it becomes available. Never use `run_in_background` to run \'sleep\' as it will return immediately. You do not need to use \'&\' at the end of the command when using this parameter.\n  - VERY IMPORTANT: You MUST avoid using search commands like `find` and `grep`. Instead use Grep, Glob, or Task to search. You MUST avoid read tools like `cat`, `head`, and `tail`, and use Read to read files.\n - If you _still_ need to run `grep`, STOP. ALWAYS USE ripgrep at `rg` first, which all Claude Code users have pre-installed.\n  - When issuing multiple commands, use the \';\' or \'&&\' operator to separate them. DO NOT use newlines (newlines are ok in quoted strings).\n  - Try to maintain your current working directory throughout the session by using absolute paths and avoiding usage of `cd`. You may use `cd` if the User explicitly requests it.\n    <good-example>\n    pytest /foo/bar/tests\n    </good-example>\n    <bad-example>\n    cd /foo/bar && pytest tests\n    </bad-example>\n\n# Committing changes with git\n\nWhen the user asks you to create a new git commit, follow these steps carefully:\n\n1. You have the capability to call multiple tools in a single response. When multiple independent pieces of information are requested, batch your tool calls together for optimal performance. ALWAYS run the following bash commands in parallel, each using the Bash tool:\n  - Run a git status command to see all untracked files.\n  - Run a git diff command to see both staged and unstaged changes that will be committed.\n  - Run a git log command to see recent commit messages, so that you can follow this repository\'s commit message style.\n2. Analyze all staged changes (both previously staged and newly added) and draft a commit message:\n  - Summarize the nature of the changes (eg. new feature, enhancement to an existing feature, bug fix, refactoring, test, docs, etc.). Ensure the message accurately reflects the changes and their purpose (i.e. "add" means a wholly new feature, "update" means an enhancement to an existing feature, "fix" means a bug fix, etc.).\n  - Check for any sensitive information that shouldn\'t be committed\n  - Draft a concise (1-2 sentences) commit message that focuses on the "why" rather than the "what"\n  - Ensure it accurately reflects the changes and their purpose\n3. You have the capability to call multiple tools in a single response. When multiple independent pieces of information are requested, batch your tool calls together for optimal performance. ALWAYS run the following commands in parallel:\n   - Add relevant untracked files to the staging area.\n   - Create the commit with a message ending with:\n   🤖 Generated with [Claude Code](https://claude.ai/code)\n\n   Co-Authored-By: Claude <noreply@anthropic.com>\n   - Run git status to make sure the commit succeeded.\n4. If the commit fails due to pre-commit hook changes, retry the commit ONCE to include these automated changes. If it fails again, it usually means a pre-commit hook is preventing the commit. If the commit succeeds but you notice that files were modified by the pre-commit hook, you MUST amend your commit to include them.\n\nImportant notes:\n- NEVER update the git config\n- NEVER run additional commands to read or explore code, besides git bash commands\n- NEVER use the TodoWrite or Task tools\n- DO NOT push to the remote repository unless the user explicitly asks you to do so\n- IMPORTANT: Never use git commands with the -i flag (like git rebase -i or git add -i) since they require interactive input which is not supported.\n- If there are no changes to commit (i.e., no untracked files and no modifications), do not create an empty commit\n- In order to ensure good formatting, ALWAYS pass the commit message via a HEREDOC, a la this example:\n<example>\ngit commit -m "$(cat <<\'EOF\'\n   Commit message here.\n\n   🤖 Generated with [Claude Code](https://claude.ai/code)\n\n   Co-Authored-By: Claude <noreply@anthropic.com>\n   EOF\n   )"\n</example>\n\n# Creating pull requests\nUse the gh command via the Bash tool for ALL GitHub-related tasks including working with issues, pull requests, checks, and releases. If given a Github URL use the gh command to get the information needed.\n\nIMPORTANT: When the user asks you to create a pull request, follow these steps carefully:\n\n1. You have the capability to call multiple tools in a single response. When multiple independent pieces of information are requested, batch your tool calls together for optimal performance. ALWAYS run the following bash commands in parallel using the Bash tool, in order to understand the current state of the branch since it diverged from the main branch:\n   - Run a git status command to see all untracked files\n   - Run a git diff command to see both staged and unstaged changes that will be committed\n   - Check if the current branch tracks a remote branch and is up to date with the remote, so you know if you need to push to the remote\n   - Run a git log command and `git diff [base-branch]...HEAD` to understand the full commit history for the current branch (from the time it diverged from the base branch)\n2. Analyze all changes that will be included in the pull request, making sure to look at all relevant commits (NOT just the latest commit, but ALL commits that will be included in the pull request!!!), and draft a pull request summary\n3. You have the capability to call multiple tools in a single response. When multiple independent pieces of information are requested, batch your tool calls together for optimal performance. ALWAYS run the following commands in parallel:\n   - Create new branch if needed\n   - Push to remote with -u flag if needed\n   - Create PR using gh pr create with the format below. Use a HEREDOC to pass the body to ensure correct formatting.\n<example>\ngh pr create --title "the pr title" --body "$(cat <<\'EOF\'\n## Summary\n<1-3 bullet points>\n\n## Test plan\n[Checklist of TODOs for testing the pull request...]\n\n🤖 Generated with [Claude Code](https://claude.ai/code)\nEOF\n)"\n</example>\n\nImportant:\n- NEVER update the git config\n- DO NOT use the TodoWrite or Task tools\n- Return the PR URL when you\'re done, so the user can see it\n\n# Other common operations\n- View comments on a Github PR: gh api repos/foo/bar/pulls/123/comments',
+        schema: z.object({
+            command: z.string().describe("The command to execute"),
+            timeout: z
+                .number()
+                .optional()
+                .describe("Optional timeout in milliseconds (max 600000)"),
+            description: z
+                .string()
+                .optional()
+                .describe("Clear, concise description of what this command does in 5-10 words, in active voice. Examples:\nInput: ls\nOutput: List files in current directory\n\nInput: git status\nOutput: Show working tree status\n\nInput: npm install\nOutput: Install package dependencies\n\nInput: mkdir foo\nOutput: Create directory 'foo'"),
+        }),
+    });
+    return bash;
+}

package/dist/scaffold/link-local.d.ts

@@ -0,0 +1 @@
+export declare function linkLocalPackages(projectPath: string): Promise<void>;

package/dist/scaffold/link-local.js

@@ -0,0 +1,54 @@
+import { exists } from "node:fs/promises";
+import { join } from "node:path";
+import { $ } from "bun";
+const PACKAGE_PATHS = {
+    "@townco/ui": "packages/ui",
+    "@townco/core": "packages/core",
+    "@townco/tsconfig": "packages/tsconfig",
+    "@townco/tui-template": "apps/tui",
+    "@townco/gui-template": "apps/gui",
+    "@townco/secret": "packages/secret",
+    "@townco/agent": "packages/agent",
+    "@townco/cli": "apps/cli",
+};
+async function getMonorepoRoot() {
+    try {
+        // 1. Get git repo root
+        const result = await $ `git rev-parse --show-toplevel`.quiet();
+        const repoRoot = result.text().trim();
+        // 2. Check package.json name === "town"
+        const pkgJsonPath = join(repoRoot, "package.json");
+        const pkgJson = await Bun.file(pkgJsonPath).json();
+        if (pkgJson.name !== "town")
+            return null;
+        // 3. Check packages/agent and packages/ui exist
+        const agentExists = await exists(join(repoRoot, "packages/agent"));
+        const uiExists = await exists(join(repoRoot, "packages/ui"));
+        if (!agentExists || !uiExists)
+            return null;
+        return repoRoot;
+    }
+    catch {
+        return null;
+    }
+}
+export async function linkLocalPackages(projectPath) {
+    const repoRoot = await getMonorepoRoot();
+    if (!repoRoot)
+        return; // Not in monorepo, no-op
+    console.log("Detected town monorepo, linking local packages...");
+    // 1. Register each local package globally
+    for (const [, localPath] of Object.entries(PACKAGE_PATHS)) {
+        const pkgPath = join(repoRoot, localPath);
+        await $ `bun link`.cwd(pkgPath).quiet();
+    }
+    // 2. Parse project's package.json for @townco/* deps
+    const pkgJson = await Bun.file(join(projectPath, "package.json")).json();
+    const deps = { ...pkgJson.dependencies, ...pkgJson.devDependencies };
+    const towncoPackages = Object.keys(deps).filter((name) => name.startsWith("@townco/"));
+    // 3. Link each package in the project
+    for (const pkgName of towncoPackages) {
+        await $ `bun link ${pkgName}`.cwd(projectPath);
+        console.log(`Linked ${pkgName}`);
+    }
+}

package/dist/scaffold/project-scaffold.js

@@ -21,6 +21,7 @@ function generateProjectPackageJson() {
             "@radix-ui/react-select": "^2.2.6",
             "@radix-ui/react-slot": "^1.2.4",
             "@radix-ui/react-tabs": "^1.1.13",
+            "@townco/core": "~0.0.23",
             "@townco/ui": "^0.1.0",
             "@townco/agent": "^0.1.20",
             "class-variance-authority": "^0.7.1",

package/dist/telemetry/setup.d.ts

@@ -1,7 +1,8 @@
 /**
  * OpenTelemetry provider setup for @townco/agent
- * Initializes the trace provider, exporter, and propagator
+ * Initializes the trace provider, log provider, exporters, and propagator
  */
+import { LoggerProvider } from "@opentelemetry/sdk-logs";
 import { NodeTracerProvider } from "@opentelemetry/sdk-trace-node";
 export interface TelemetrySetupOptions {
     serviceName?: string;
@@ -14,6 +15,7 @@ export interface TelemetrySetupOptions {
  */
 export declare function initializeOpenTelemetry(options?: TelemetrySetupOptions): {
     provider: NodeTracerProvider;
+    loggerProvider: LoggerProvider;
     shutdown: () => Promise<void>;
 };
 /**

package/dist/telemetry/setup.js

@@ -1,11 +1,14 @@
 /**
  * OpenTelemetry provider setup for @townco/agent
- * Initializes the trace provider, exporter, and propagator
+ * Initializes the trace provider, log provider, exporters, and propagator
  */
 import { propagation } from "@opentelemetry/api";
+import { logs } from "@opentelemetry/api-logs";
 import { W3CTraceContextPropagator } from "@opentelemetry/core";
+import { OTLPLogExporter } from "@opentelemetry/exporter-logs-otlp-http";
 import { OTLPTraceExporter } from "@opentelemetry/exporter-trace-otlp-http";
 import { Resource } from "@opentelemetry/resources";
+import { BatchLogRecordProcessor, LoggerProvider, } from "@opentelemetry/sdk-logs";
 import { BatchSpanProcessor } from "@opentelemetry/sdk-trace-base";
 import { NodeTracerProvider } from "@opentelemetry/sdk-trace-node";
 import { ATTR_SERVICE_NAME } from "@opentelemetry/semantic-conventions";
@@ -83,13 +86,38 @@ export function initializeOpenTelemetry(options = {}) {
     const batchProcessor = new BatchSpanProcessor(loggingExporter, {
         maxQueueSize: 100,
         maxExportBatchSize: 10,
-        scheduledDelayMillis: 5000, // Export every 5 seconds (default)
+        scheduledDelayMillis: 2000, // Export every 2 seconds
     });
     provider.addSpanProcessor(batchProcessor);
     // Register the provider globally
     provider.register();
     // Configure W3C Trace Context propagator for cross-process traces
     propagation.setGlobalPropagator(new W3CTraceContextPropagator());
+    // Set up LoggerProvider for OTLP log export
+    const logUrl = otlpEndpoint.endsWith("/")
+        ? `${otlpEndpoint}v1/logs`
+        : `${otlpEndpoint}/v1/logs`;
+    if (debug) {
+        console.log(`OTLP log URL: ${logUrl}`);
+    }
+    const logExporter = new OTLPLogExporter({
+        url: logUrl,
+    });
+    const loggerProvider = new LoggerProvider({
+        resource: new Resource({
+            [ATTR_SERVICE_NAME]: serviceName,
+        }),
+    });
+    loggerProvider.addLogRecordProcessor(new BatchLogRecordProcessor(logExporter, {
+        maxQueueSize: 100,
+        maxExportBatchSize: 10,
+        scheduledDelayMillis: 2000,
+    }));
+    // Register the logger provider globally
+    logs.setGlobalLoggerProvider(loggerProvider);
+    if (debug) {
+        console.log("✓ Log exporter configured");
+    }
     // Now configure our telemetry wrapper
     configureTelemetry({
         enabled: true,
@@ -106,6 +134,8 @@ export function initializeOpenTelemetry(options = {}) {
         try {
             await provider.forceFlush();
             await provider.shutdown();
+            await loggerProvider.forceFlush();
+            await loggerProvider.shutdown();
             if (debug) {
                 console.log("✓ Telemetry flushed");
             }
@@ -114,7 +144,7 @@ export function initializeOpenTelemetry(options = {}) {
             console.error("Error flushing telemetry:", error);
         }
     };
-    return { provider, shutdown };
+    return { provider, loggerProvider, shutdown };
 }
 /**
  * Initialize OpenTelemetry from environment variables and register shutdown handlers

package/dist/templates/index.d.ts

@@ -23,6 +23,13 @@ export interface TemplateVars {
             threshold: number;
         } | undefined;
         callback: string;
+    } | {
+        type: "tool_response";
+        setting?: {
+            maxContextThreshold?: number | undefined;
+            responseTruncationThreshold?: number | undefined;
+        } | undefined;
+        callback: string;
     }> | undefined;
 }
 export declare function getTemplateVars(name: string, definition: AgentDefinition): TemplateVars;

package/dist/test-telemetry.d.ts

@@ -0,0 +1,5 @@
+/**
+ * Simple test script to verify OpenTelemetry is working
+ * Run with: ENABLE_TELEMETRY=true DEBUG_TELEMETRY=true bun test-telemetry.ts
+ */
+export {};