@townco/agent 0.1.113 → 0.1.115

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

@@ -22,7 +22,7 @@ export async function getTownE2BApiKey() {
         return _apiKeyFetchPromise;
     }
     _apiKeyFetchPromise = (async () => {
-        const shedAuth = getShedAuth();
+        const shedAuth = await getShedAuth();
         if (!shedAuth) {
             throw new Error("Not logged in. Run 'town login' or set SHED_API_KEY to use the code_sandbox tools.");
         }
@@ -329,7 +329,7 @@ function makeE2BToolsInternal(getSandbox) {
             const fileBuffer = Buffer.from(result.stdout.trim(), "base64");
             await fs.writeFile(outputPath, fileBuffer);
             // Step 2: Upload to Supabase Storage
-            const shedAuth = getShedAuth();
+            const shedAuth = await getShedAuth();
             if (!shedAuth) {
                 // Fallback to local URL if not authenticated
                 const port = process.env.PORT || "3100";

package/dist/runner/langchain/tools/subagent-connections.js

@@ -37,16 +37,6 @@ export function emitSubagentMessages(queryHash, messages, completed = false) {
         return;
     }
     if (toolCallId) {
-        const firstMessage = messages[0];
-        logger.info("✓ Emitting subagent messages for live streaming", {
-            queryHash,
-            toolCallId,
-            messageCount: messages.length,
-            hasContent: firstMessage ? firstMessage.content.length > 0 : false,
-            hasToolCalls: firstMessage ? firstMessage.toolCalls.length > 0 : false,
-            completed,
-            invocationId: invocationCtx.invocationId,
-        });
         // Emit to the parent's invocation-scoped EventEmitter
         invocationCtx.subagentEventEmitter.emit("messages", {
             toolCallId,

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

@@ -3,11 +3,13 @@ import * as fs from "node:fs/promises";
 import { mkdir } from "node:fs/promises";
 import * as path from "node:path";
 import { context, propagation, trace } from "@opentelemetry/api";
+import { createLogger } from "@townco/core";
 import { z } from "zod";
 import { SUBAGENT_MODE_KEY, } from "../../../acp-server/adapter.js";
 import { makeRunnerFromDefinition } from "../../index.js";
 import { bindGeneratorToSessionContext, getAbortSignal, } from "../../session-context.js";
 import { emitSubagentMessages, hashQuery, } from "./subagent-connections.js";
+const logger = createLogger("subagent-tool", "debug");
 /**
  * Name of the Task tool created by makeSubagentsTool
  */
@@ -217,6 +219,11 @@ async function querySubagent(agentName, agentPath, agentWorkingDirectory, query)
     };
     const toolCallMap = new Map();
     const queryHash = hashQuery(query);
+    logger.info("[DEBUG] Starting subagent generator loop", {
+        agentName,
+        queryHash,
+        sessionId: subagentSessionId,
+    });
     try {
         for await (const update of generator) {
             let shouldEmit = false;
@@ -291,23 +298,60 @@ async function querySubagent(agentName, agentPath, agentWorkingDirectory, query)
             }
             // Emit incremental update to parent (for live streaming)
             if (shouldEmit) {
+                logger.debug("[SUBAGENT-ACCUMULATION] Emitting incremental update", {
+                    agentName,
+                    queryHash,
+                    contentLength: currentMessage.content.length,
+                    contentBlocksCount: currentMessage.contentBlocks.length,
+                    toolCallsCount: currentMessage.toolCalls.length,
+                });
                 emitSubagentMessages(queryHash, [{ ...currentMessage }]);
             }
         }
+        logger.info("[DEBUG] Subagent generator loop finished", {
+            agentName,
+            queryHash,
+            sessionId: subagentSessionId,
+            contentLength: currentMessage.content.length,
+            toolCallCount: currentMessage.toolCalls.length,
+        });
         // Final emit to ensure everything is captured, with completion flag
         if (currentMessage.content || currentMessage.toolCalls.length > 0) {
+            logger.info("[DEBUG] Emitting final completion flag", {
+                agentName,
+                queryHash,
+                sessionId: subagentSessionId,
+                hasContent: true,
+            });
             emitSubagentMessages(queryHash, [currentMessage], true);
         }
         else {
             // Even if no messages, emit completion sentinel
+            logger.info("[DEBUG] Emitting empty completion flag", {
+                agentName,
+                queryHash,
+                sessionId: subagentSessionId,
+                hasContent: false,
+            });
             emitSubagentMessages(queryHash, [], true);
         }
+        logger.info("[DEBUG] Subagent querySubagent() returning result", {
+            agentName,
+            queryHash,
+            sessionId: subagentSessionId,
+        });
         return {
             text: responseText,
             sources: collectedSources,
         };
     }
     catch (error) {
+        logger.info("[DEBUG] Subagent querySubagent() caught error", {
+            agentName,
+            queryHash,
+            sessionId: subagentSessionId,
+            error: error instanceof Error ? error.message : String(error),
+        });
         // Emit completion sentinel even on error to prevent parent from hanging
         emitSubagentMessages(queryHash, [], true);
         if (parentAbortSignal?.aborted) {

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

@@ -19,11 +19,11 @@ function getDirectExaClient() {
     return _directExaClient;
 }
 /** Get Exa client using Town proxy with authenticated credentials */
-function getTownExaClient() {
+async function getTownExaClient() {
     if (_townExaClient) {
         return _townExaClient;
     }
-    const shedAuth = getShedAuth();
+    const shedAuth = await getShedAuth();
     if (!shedAuth) {
         throw new Error("Not logged in. Run 'town login' or set SHED_API_KEY to use the town_web_search tool.");
     }
@@ -43,7 +43,7 @@ export function getWebSearchCitationCounter() {
 }
 function makeWebSearchToolsInternal(getClient) {
     const webSearch = tool(async ({ query }) => {
-        const client = getClient();
+        const client = await getClient();
         const result = await client.searchAndContents(query, {
             numResults: 5,
             type: "auto",
@@ -102,7 +102,7 @@ function makeWebSearchToolsInternal(getClient) {
         paramKey: "query",
     };
     const webFetch = tool(async ({ url, prompt }) => {
-        const client = getClient();
+        const client = await getClient();
         try {
             const result = await client.getContents([url], {
                 text: true,