@townco/agent 0.1.44 → 0.1.45

package/dist/runner/langchain/index.js

@@ -147,15 +147,14 @@ export class LangchainAgent {
             agentConfig.systemPrompt = `${agentConfig.systemPrompt ?? ""}\n\n${TODO_WRITE_INSTRUCTIONS}`;
         }
         const agent = createAgent(agentConfig);
-        // Build messages from session history if available, otherwise use just the prompt
+        // Build messages from context history if available, otherwise use just the prompt
         let messages;
-        if (req.sessionMessages && req.sessionMessages.length > 0) {
-            // Use session message history - convert to LangChain format
-            // Only include messages BEFORE the current user message (exclude the last one since it's the current prompt)
-            const historyMessages = req.sessionMessages.slice(0, -1);
-            messages = historyMessages.map((msg) => ({
+        if (req.contextMessages && req.contextMessages.length > 0) {
+            // Use context messages (already resolved from context entries)
+            // Convert to LangChain format
+            messages = req.contextMessages.map((msg) => ({
                 type: msg.role === "user" ? "human" : "ai",
-                // Extract text from content blocks (ignore tool call blocks for LLM context)
+                // Extract text from content blocks
                 content: msg.content
                     .filter((block) => block.type === "text")
                     .map((block) => block.text)
@@ -172,7 +171,7 @@ export class LangchainAgent {
             });
         }
         else {
-            // Fallback: No session history, use just the prompt (legacy behavior)
+            // Fallback: No context history, use just the prompt
             messages = req.prompt
                 .filter((promptMsg) => promptMsg.type === "text")
                 .map((promptMsg) => ({
@@ -282,7 +281,12 @@ export class LangchainAgent {
                             : typeof aiMessage.content === "string"
                                 ? aiMessage.content.length
                                 : -1;
-                        console.log("DEBUG agent: messageTokenUsage:", JSON.stringify(messageTokenUsage), "contentType:", contentType, "isArray:", contentIsArray, "length:", contentLength);
+                        _logger.debug("messageTokenUsage", {
+                            messageTokenUsage,
+                            contentType,
+                            isArray: contentIsArray,
+                            length: contentLength,
+                        });
                     }
                     // If we have tokenUsage but no content, send a token-only chunk
                     if (messageTokenUsage &&
@@ -290,7 +294,9 @@ export class LangchainAgent {
                             ? aiMessage.content === ""
                             : Array.isArray(aiMessage.content) &&
                                 aiMessage.content.length === 0)) {
-                        console.log("DEBUG agent: sending token-only chunk:", JSON.stringify(messageTokenUsage));
+                        _logger.debug("sending token-only chunk", {
+                            messageTokenUsage,
+                        });
                         const msgToYield = {
                             sessionUpdate: "agent_message_chunk",
                             content: {

package/dist/runner/langchain/model-factory.js

@@ -1,6 +1,8 @@
 import { ChatAnthropic } from "@langchain/anthropic";
 import { ChatGoogleGenerativeAI } from "@langchain/google-genai";
 import { ChatVertexAI } from "@langchain/google-vertexai";
+import { createLogger } from "@townco/core";
+const logger = createLogger("model-factory");
 /**
  * Detects the provider from a model string and returns the appropriate
  * LangChain chat model instance.
@@ -93,7 +95,7 @@ export function createModelFromString(modelString) {
             });
         default:
             // Fallback to Anthropic for unknown providers (backward compatibility)
-            console.warn(`Unknown provider "${provider}" in model string "${modelString}". Defaulting to Anthropic.`);
+            logger.warn(`Unknown provider "${provider}" in model string "${modelString}". Defaulting to Anthropic.`);
             return new ChatAnthropic({
                 model: modelString,
             });

package/dist/scaffold/claude-scaffold.js

@@ -2,8 +2,10 @@ import { existsSync } from "node:fs";
 import { appendFile, readFile, writeFile } from "node:fs/promises";
 import { dirname, join } from "node:path";
 import { fileURLToPath } from "node:url";
+import { createLogger } from "@townco/core";
 const __filename = fileURLToPath(import.meta.url);
 const __dirname = dirname(__filename);
+const logger = createLogger("claude-scaffold");
 /**
  * Initialize Claude Code workspace integration by creating AGENTS.town.md
  * and adding references to it in CLAUDE.md and AGENTS.md.
@@ -29,30 +31,30 @@ export async function initForClaudeCode(targetPath) {
     const townContent = await readFile(appendTemplatePath, "utf-8");
     // Create AGENTS.town.md with the Town-specific content
     await writeFile(agentsTownMdPath, townContent, "utf-8");
-    console.log("✓ Created AGENTS.town.md");
+    logger.info("✓ Created AGENTS.town.md");
     // Add reference to AGENTS.town.md in both CLAUDE.md and AGENTS.md
     const referenceText = "\n\nThis repo uses the Town agent SDK -- please reference @AGENTS.town.md for more information\n";
     // Handle CLAUDE.md
     if (existsSync(claudeMdPath)) {
         // CLAUDE.md exists, append reference
         await appendFile(claudeMdPath, referenceText);
-        console.log("✓ Added Town SDK reference to CLAUDE.md");
+        logger.info("✓ Added Town SDK reference to CLAUDE.md");
     }
     else {
         // CLAUDE.md doesn't exist, create it with just the reference
         await writeFile(claudeMdPath, `${referenceText.trim()}\n`, "utf-8");
-        console.log("✓ Created CLAUDE.md with Town SDK reference");
+        logger.info("✓ Created CLAUDE.md with Town SDK reference");
     }
     // Handle AGENTS.md
     if (existsSync(agentsMdPath)) {
         // AGENTS.md exists, append reference
         await appendFile(agentsMdPath, referenceText);
-        console.log("✓ Added Town SDK reference to AGENTS.md");
+        logger.info("✓ Added Town SDK reference to AGENTS.md");
     }
     else {
         // AGENTS.md doesn't exist, create it with just the reference
         await writeFile(agentsMdPath, `${referenceText.trim()}\n`, "utf-8");
-        console.log("✓ Created AGENTS.md with Town SDK reference");
+        logger.info("✓ Created AGENTS.md with Town SDK reference");
     }
-    console.log("✓ Claude Code workspace initialized");
+    logger.info("✓ Claude Code workspace initialized");
 }

package/dist/storage/index.js

@@ -1,7 +1,9 @@
 import { mkdir, readdir, rm, stat } from "node:fs/promises";
 import { homedir } from "node:os";
 import { dirname, join } from "node:path";
+import { createLogger } from "@townco/core";
 const AGENTS_DIR = join(homedir(), ".config", "town", "agents");
+const logger = createLogger("storage");
 /**
  * Get the base directory where all agents are stored
  */
@@ -39,7 +41,7 @@ export async function listAgents() {
             .map((entry) => entry.name);
     }
     catch (error) {
-        console.error("Error listing agents:", error);
+        logger.error("Error listing agents", { error });
         return [];
     }
 }

package/dist/templates/index.d.ts

@@ -17,6 +17,13 @@ export interface TemplateVars {
     }>;
     systemPrompt: string | null;
     hasWebSearch: boolean;
+    hooks?: Array<{
+        type: "context_size";
+        setting?: {
+            threshold: number;
+        } | undefined;
+        callback: string;
+    }> | undefined;
 }
 export declare function getTemplateVars(name: string, definition: AgentDefinition): TemplateVars;
 export declare function generatePackageJson(vars: TemplateVars): string;

package/dist/templates/index.js

@@ -7,6 +7,7 @@ export function getTemplateVars(name, definition) {
         tools,
         systemPrompt: definition.systemPrompt,
         hasWebSearch: tools.some((tool) => typeof tool === "string" && tool === "web_search"),
+        hooks: definition.hooks,
     };
 }
 export function generatePackageJson(vars) {
@@ -47,10 +48,14 @@ export async function generateIndexTs(vars) {
         model: vars.model,
         systemPrompt: vars.systemPrompt,
         tools: vars.tools,
+        hooks: vars.hooks,
     };
     return prettier.format(`import { makeHttpTransport, makeStdioTransport } from "@townco/agent/acp-server";
 import type { AgentDefinition } from "@townco/agent/definition";
 import { basename } from "node:path";
+import { createLogger } from "@townco/core";
+
+const logger = createLogger("agent-index");
 
 // Load agent definition from JSON file
 const agent: AgentDefinition = ${JSON.stringify(agentDef)};
@@ -61,14 +66,14 @@ const transport = process.argv[2] || "stdio";
 const agentDir = process.cwd();
 const agentName = basename(agentDir);
 
-console.log("[agent] Configuration:", { transport, agentDir, agentName });
+logger.info("Configuration", { transport, agentDir, agentName });
 
 if (transport === "http") {
 	makeHttpTransport(agent, agentDir, agentName);
 } else if (transport === "stdio") {
 	makeStdioTransport(agent);
 } else {
-	console.error(\`Invalid transport: \${transport}\`);
+	logger.error(\`Invalid transport: \${transport}\`);
 	process.exit(1);
 }
 `, { parser: "typescript" });

package/dist/test-script.js

@@ -1,4 +1,6 @@
+import { createLogger } from "@townco/core";
 import { LangchainAgent } from "./runner/langchain";
+const logger = createLogger("test-script");
 const agent = new LangchainAgent({
     model: "claude-sonnet-4-5-20250929",
     systemPrompt: "You are a helpful assistant.",
@@ -14,5 +16,5 @@ for await (const event of agent.invoke({
     sessionId: "test-session",
     messageId: "test-message-1",
 })) {
-    console.log(event);
+    logger.info("Event received", { event });
 }