@poncho-ai/harness 0.16.1 → 0.18.0

package/dist/index.js

@@ -466,7 +466,7 @@ var createWriteTool = (workingDir) => defineTool({
 
 // src/harness.ts
 import { randomUUID as randomUUID3 } from "crypto";
-import { getTextContent } from "@poncho-ai/sdk";
+import { getTextContent as getTextContent2 } from "@poncho-ai/sdk";
 
 // src/upload-store.ts
 import { createHash as createHash2 } from "crypto";
@@ -1306,16 +1306,19 @@ var StreamableHttpMcpRpcClient = class {
   endpoint;
   timeoutMs;
   bearerToken;
+  customHeaders;
   idCounter = 1;
   initialized = false;
   sessionId;
-  constructor(endpoint, timeoutMs = 1e4, bearerToken) {
+  constructor(endpoint, timeoutMs = 1e4, bearerToken, customHeaders) {
     this.endpoint = endpoint;
     this.timeoutMs = timeoutMs;
     this.bearerToken = bearerToken;
+    this.customHeaders = customHeaders ?? {};
   }
   buildHeaders(accept) {
     const headers = {
+      ...this.customHeaders,
       "Content-Type": "application/json",
       Accept: accept
     };
@@ -1596,7 +1599,8 @@ var LocalMcpBridge = class {
         new StreamableHttpMcpRpcClient(
           server.url,
           server.timeoutMs ?? 1e4,
-          server.auth?.tokenEnv ? process.env[server.auth.tokenEnv] : void 0
+          server.auth?.tokenEnv ? process.env[server.auth.tokenEnv] : void 0,
+          server.headers
         )
       );
     }
@@ -2447,6 +2451,138 @@ var extractRunnableFunction = (value) => {
   return void 0;
 };
 
+// src/subagent-tools.ts
+import { defineTool as defineTool4, getTextContent } from "@poncho-ai/sdk";
+var LAST_MESSAGES_TO_RETURN = 10;
+var summarizeResult = (r) => {
+  const summary = {
+    subagentId: r.subagentId,
+    status: r.status
+  };
+  if (r.result) {
+    summary.result = {
+      status: r.result.status,
+      response: r.result.response,
+      steps: r.result.steps,
+      duration: r.result.duration
+    };
+  }
+  if (r.error) {
+    summary.error = r.error;
+  }
+  if (r.latestMessages && r.latestMessages.length > 0) {
+    summary.latestMessages = r.latestMessages.slice(-LAST_MESSAGES_TO_RETURN).map((m) => ({
+      role: m.role,
+      content: getTextContent(m).slice(0, 2e3)
+    }));
+  }
+  return summary;
+};
+var createSubagentTools = (manager, getConversationId, getOwnerId) => [
+  defineTool4({
+    name: "spawn_subagent",
+    description: "Spawn a subagent to work on a task and wait for it to finish. The subagent is a full copy of yourself running in its own conversation context with access to the same tools (except memory writes). This call blocks until the subagent completes and returns its result.\n\nGuidelines:\n- Use subagents to parallelize work: call spawn_subagent multiple times in one response for independent sub-tasks -- they run concurrently.\n- Prefer doing work yourself for simple or quick tasks. Spawn subagents for substantial, self-contained work.\n- The subagent has no memory of your conversation -- write thorough, self-contained instructions in the task.",
+    inputSchema: {
+      type: "object",
+      properties: {
+        task: {
+          type: "string",
+          description: "Thorough, self-contained instructions for the subagent. Include all relevant context, goals, and constraints -- the subagent starts with zero prior conversation history."
+        }
+      },
+      required: ["task"],
+      additionalProperties: false
+    },
+    handler: async (input) => {
+      const task = typeof input.task === "string" ? input.task : "";
+      if (!task.trim()) {
+        return { error: "task is required" };
+      }
+      const conversationId = getConversationId();
+      if (!conversationId) {
+        return { error: "no active conversation to spawn subagent from" };
+      }
+      const result = await manager.spawn({
+        task: task.trim(),
+        parentConversationId: conversationId,
+        ownerId: getOwnerId()
+      });
+      return summarizeResult(result);
+    }
+  }),
+  defineTool4({
+    name: "message_subagent",
+    description: "Send a follow-up message to a completed or stopped subagent and wait for it to finish. This restarts the subagent with the new message and blocks until it completes. Only works when the subagent is not currently running.",
+    inputSchema: {
+      type: "object",
+      properties: {
+        subagent_id: {
+          type: "string",
+          description: "The subagent ID (from spawn_subagent result or list_subagents)."
+        },
+        message: {
+          type: "string",
+          description: "The follow-up instructions or message to send."
+        }
+      },
+      required: ["subagent_id", "message"],
+      additionalProperties: false
+    },
+    handler: async (input) => {
+      const subagentId = typeof input.subagent_id === "string" ? input.subagent_id : "";
+      const message = typeof input.message === "string" ? input.message : "";
+      if (!subagentId || !message.trim()) {
+        return { error: "subagent_id and message are required" };
+      }
+      const result = await manager.sendMessage(subagentId, message.trim());
+      return summarizeResult(result);
+    }
+  }),
+  defineTool4({
+    name: "stop_subagent",
+    description: "Stop a running subagent. The subagent's conversation is preserved but it will stop processing. Use this to cancel work that is no longer needed.",
+    inputSchema: {
+      type: "object",
+      properties: {
+        subagent_id: {
+          type: "string",
+          description: "The subagent ID (from spawn_subagent result or list_subagents)."
+        }
+      },
+      required: ["subagent_id"],
+      additionalProperties: false
+    },
+    handler: async (input) => {
+      const subagentId = typeof input.subagent_id === "string" ? input.subagent_id : "";
+      if (!subagentId) {
+        return { error: "subagent_id is required" };
+      }
+      await manager.stop(subagentId);
+      return { message: `Subagent "${subagentId}" has been stopped.` };
+    }
+  }),
+  defineTool4({
+    name: "list_subagents",
+    description: "List all subagents that have been spawned in this conversation. Returns each subagent's ID, original task, current status, and message count. Use this to look up subagent IDs before calling message_subagent or stop_subagent.",
+    inputSchema: {
+      type: "object",
+      properties: {},
+      additionalProperties: false
+    },
+    handler: async () => {
+      const conversationId = getConversationId();
+      if (!conversationId) {
+        return { error: "no active conversation" };
+      }
+      const subagents = await manager.list(conversationId);
+      if (subagents.length === 0) {
+        return { message: "No subagents have been spawned in this conversation." };
+      }
+      return { subagents };
+    }
+  })
+];
+
 // src/harness.ts
 import { LatitudeTelemetry } from "@latitude-data/telemetry";
 import { diag, DiagLogLevel } from "@opentelemetry/api";
@@ -2928,6 +3064,7 @@ var AgentHarness = class {
   _browserMod;
   parsedAgent;
   mcpBridge;
+  subagentManager;
   resolveToolAccess(toolName) {
     const tools = this.loadedConfig?.tools;
     if (!tools) return true;
@@ -2964,6 +3101,19 @@ var AgentHarness = class {
       this.dispatcher.register(tool);
     }
   }
+  unregisterTools(names) {
+    this.dispatcher.unregisterMany(names);
+  }
+  setSubagentManager(manager) {
+    this.subagentManager = manager;
+    this.dispatcher.registerMany(
+      createSubagentTools(
+        manager,
+        () => this._currentRunConversationId,
+        () => this._currentRunOwnerId ?? "anonymous"
+      )
+    );
+  }
   registerConfiguredBuiltInTools(config) {
     for (const tool of createDefaultTools(this.workingDir)) {
       if (this.isToolEnabled(tool.name)) {
@@ -3418,6 +3568,8 @@ var AgentHarness = class {
   }
   /** Conversation ID of the currently executing run (set during run, cleared after). */
   _currentRunConversationId;
+  /** Owner ID of the currently executing run (used by subagent tools). */
+  _currentRunOwnerId;
   get browserSession() {
     return this._browserSession;
   }
@@ -3515,6 +3667,10 @@ var AgentHarness = class {
     }
     await this.refreshSkillsIfChanged();
     this._currentRunConversationId = input.conversationId;
+    const ownerParam = input.parameters?.__ownerId;
+    if (typeof ownerParam === "string") {
+      this._currentRunOwnerId = ownerParam;
+    }
     const agent = this.parsedAgent;
     const runId = `run_${randomUUID3()}`;
     const start = now();
@@ -3715,7 +3871,7 @@ ${boundedMainMemory.trim()}` : "";
             if (rich && rich.length > 0) {
               return [{ role: "tool", content: rich }];
             }
-            const textContent = typeof msg.content === "string" ? msg.content : getTextContent(msg);
+            const textContent = typeof msg.content === "string" ? msg.content : getTextContent2(msg);
             try {
               const parsed = JSON.parse(textContent);
               if (!Array.isArray(parsed)) {
@@ -3765,7 +3921,7 @@ ${boundedMainMemory.trim()}` : "";
             }
           }
           if (msg.role === "assistant") {
-            const assistantText = typeof msg.content === "string" ? msg.content : getTextContent(msg);
+            const assistantText = typeof msg.content === "string" ? msg.content : getTextContent2(msg);
             try {
               const parsed = JSON.parse(assistantText);
               if (typeof parsed === "object" && parsed !== null) {
@@ -3804,7 +3960,7 @@ ${boundedMainMemory.trim()}` : "";
           if (msg.role === "system") {
             return [{
               role: "system",
-              content: typeof msg.content === "string" ? msg.content : getTextContent(msg)
+              content: typeof msg.content === "string" ? msg.content : getTextContent2(msg)
             }];
           }
           if (msg.role === "user") {
@@ -3911,7 +4067,7 @@ ${textContent}` };
         let chunkCount = 0;
         const hasRunTimeout = timeoutMs > 0;
         const streamDeadline = hasRunTimeout ? start + timeoutMs : 0;
-        const textIterator = result.textStream[Symbol.asyncIterator]();
+        const fullStreamIterator = result.fullStream[Symbol.asyncIterator]();
         try {
           while (true) {
             if (isCancelled()) {
@@ -3937,20 +4093,20 @@ ${textContent}` };
             }
             const remaining = hasRunTimeout ? streamDeadline - now() : Infinity;
             const timeout = chunkCount === 0 ? Math.min(remaining, FIRST_CHUNK_TIMEOUT_MS) : hasRunTimeout ? remaining : 0;
-            let nextChunk;
+            let nextPart;
             if (timeout <= 0 && chunkCount > 0) {
-              nextChunk = await textIterator.next();
+              nextPart = await fullStreamIterator.next();
             } else {
               let timer;
-              nextChunk = await Promise.race([
-                textIterator.next(),
+              nextPart = await Promise.race([
+                fullStreamIterator.next(),
                 new Promise((resolve10) => {
                   timer = setTimeout(() => resolve10(null), timeout);
                 })
               ]);
               clearTimeout(timer);
             }
-            if (nextChunk === null) {
+            if (nextPart === null) {
               const isFirstChunk = chunkCount === 0;
               console.error(
                 `[poncho][harness] Stream timeout waiting for ${isFirstChunk ? "first" : "next"} chunk: model="${modelName}", step=${step}, chunks=${chunkCount}, elapsed=${now() - start}ms`
@@ -3968,13 +4124,19 @@ ${textContent}` };
               });
               return;
             }
-            if (nextChunk.done) break;
-            chunkCount += 1;
-            fullText += nextChunk.value;
-            yield pushEvent({ type: "model:chunk", content: nextChunk.value });
+            if (nextPart.done) break;
+            const part = nextPart.value;
+            if (part.type === "text-delta") {
+              chunkCount += 1;
+              fullText += part.text;
+              yield pushEvent({ type: "model:chunk", content: part.text });
+            } else if (part.type === "tool-input-start") {
+              chunkCount += 1;
+              yield pushEvent({ type: "tool:generating", tool: part.toolName, toolCallId: part.id });
+            }
           }
         } finally {
-          textIterator.return?.(void 0)?.catch?.(() => {
+          fullStreamIterator.return?.(void 0)?.catch?.(() => {
           });
         }
         if (isCancelled()) {
@@ -4074,7 +4236,8 @@ ${textContent}` };
           step,
           workingDir: this.workingDir,
           parameters: input.parameters ?? {},
-          abortSignal: input.abortSignal
+          abortSignal: input.abortSignal,
+          conversationId: input.conversationId
         };
         const toolResultsForModel = [];
         const richToolResults = [];
@@ -4179,11 +4342,14 @@ ${textContent}` };
             });
           } else {
             span?.end({ result: { value: result2.output ?? null, isError: false } });
+            const serialized = JSON.stringify(result2.output ?? null);
+            const outputTokenEstimate = Math.ceil(serialized.length / 4);
             yield pushEvent({
               type: "tool:completed",
               tool: result2.tool,
               output: result2.output,
-              duration: now() - batchStart
+              duration: now() - batchStart,
+              outputTokenEstimate
             });
             const { mediaItems, strippedOutput } = extractMediaFromToolOutput(result2.output);
             toolResultsForModel.push({
@@ -4537,9 +4703,22 @@ var InMemoryConversationStore = class {
       }
     }
   }
-  async list(ownerId = DEFAULT_OWNER) {
+  async list(ownerId) {
     this.purgeExpired();
-    return Array.from(this.conversations.values()).filter((conversation) => conversation.ownerId === ownerId).sort((a, b) => b.updatedAt - a.updatedAt);
+    return Array.from(this.conversations.values()).filter((conversation) => !ownerId || conversation.ownerId === ownerId).sort((a, b) => b.updatedAt - a.updatedAt);
+  }
+  async listSummaries(ownerId) {
+    this.purgeExpired();
+    return Array.from(this.conversations.values()).filter((c) => !ownerId || c.ownerId === ownerId).sort((a, b) => b.updatedAt - a.updatedAt).map((c) => ({
+      conversationId: c.conversationId,
+      title: c.title,
+      updatedAt: c.updatedAt,
+      createdAt: c.createdAt,
+      ownerId: c.ownerId,
+      parentConversationId: c.parentConversationId,
+      messageCount: c.messages.length,
+      hasPendingApprovals: Array.isArray(c.pendingApprovals) && c.pendingApprovals.length > 0
+    }));
   }
   async get(conversationId) {
     this.purgeExpired();
@@ -4642,8 +4821,12 @@ var FileConversationStore = class {
           conversationId: conversation.conversationId,
           title: conversation.title,
           updatedAt: conversation.updatedAt,
+          createdAt: conversation.createdAt,
           ownerId: conversation.ownerId,
-          fileName: entry.name
+          fileName: entry.name,
+          parentConversationId: conversation.parentConversationId,
+          messageCount: conversation.messages.length,
+          hasPendingApprovals: Array.isArray(conversation.pendingApprovals) && conversation.pendingApprovals.length > 0
         });
       }
     } catch {
@@ -4666,6 +4849,16 @@ var FileConversationStore = class {
       for (const conversation of parsed.conversations ?? []) {
         this.conversations.set(conversation.conversationId, conversation);
       }
+      let needsRebuild = false;
+      for (const entry of this.conversations.values()) {
+        if (entry.messageCount === void 0) {
+          needsRebuild = true;
+          break;
+        }
+      }
+      if (needsRebuild) {
+        await this.rebuildIndexFromFiles();
+      }
     } catch {
       await this.rebuildIndexFromFiles();
     }
@@ -4681,16 +4874,20 @@ var FileConversationStore = class {
         conversationId: conversation.conversationId,
         title: conversation.title,
         updatedAt: conversation.updatedAt,
+        createdAt: conversation.createdAt,
         ownerId: conversation.ownerId,
-        fileName
+        fileName,
+        parentConversationId: conversation.parentConversationId,
+        messageCount: conversation.messages.length,
+        hasPendingApprovals: Array.isArray(conversation.pendingApprovals) && conversation.pendingApprovals.length > 0
       });
       await this.writeIndex();
     });
     await this.writing;
   }
-  async list(ownerId = DEFAULT_OWNER) {
+  async list(ownerId) {
     await this.ensureLoaded();
-    const summaries = Array.from(this.conversations.values()).filter((conversation) => conversation.ownerId === ownerId).sort((a, b) => b.updatedAt - a.updatedAt);
+    const summaries = Array.from(this.conversations.values()).filter((conversation) => !ownerId || conversation.ownerId === ownerId).sort((a, b) => b.updatedAt - a.updatedAt);
     const conversations = [];
     for (const summary of summaries) {
       const loaded = await this.readConversationFile(summary.fileName);
@@ -4700,6 +4897,19 @@ var FileConversationStore = class {
     }
     return conversations;
   }
+  async listSummaries(ownerId) {
+    await this.ensureLoaded();
+    return Array.from(this.conversations.values()).filter((c) => !ownerId || c.ownerId === ownerId).sort((a, b) => b.updatedAt - a.updatedAt).map((c) => ({
+      conversationId: c.conversationId,
+      title: c.title,
+      updatedAt: c.updatedAt,
+      createdAt: c.createdAt,
+      ownerId: c.ownerId,
+      parentConversationId: c.parentConversationId,
+      messageCount: c.messageCount,
+      hasPendingApprovals: c.hasPendingApprovals
+    }));
+  }
   async get(conversationId) {
     await this.ensureLoaded();
     const summary = this.conversations.get(conversationId);
@@ -4915,11 +5125,14 @@ var KeyValueConversationStoreBase = class {
       return void 0;
     }
   }
-  async list(ownerId = DEFAULT_OWNER) {
+  async list(ownerId) {
     const kv = await this.client();
     if (!kv) {
       return await this.memoryFallback.list(ownerId);
     }
+    if (!ownerId) {
+      return [];
+    }
     const ids = await this.getOwnerConversationIds(ownerId);
     const conversations = [];
     for (const id of ids) {
@@ -4934,6 +5147,33 @@ var KeyValueConversationStoreBase = class {
     }
     return conversations.sort((a, b) => b.updatedAt - a.updatedAt);
   }
+  async listSummaries(ownerId) {
+    const kv = await this.client();
+    if (!kv) {
+      return await this.memoryFallback.listSummaries(ownerId);
+    }
+    if (!ownerId) {
+      return [];
+    }
+    const ids = await this.getOwnerConversationIds(ownerId);
+    const summaries = [];
+    for (const id of ids) {
+      const meta = await this.getConversationMeta(id);
+      if (meta && meta.ownerId === ownerId) {
+        summaries.push({
+          conversationId: meta.conversationId,
+          title: meta.title,
+          updatedAt: meta.updatedAt,
+          createdAt: meta.createdAt,
+          ownerId: meta.ownerId,
+          parentConversationId: meta.parentConversationId,
+          messageCount: meta.messageCount,
+          hasPendingApprovals: meta.hasPendingApprovals
+        });
+      }
+    }
+    return summaries.sort((a, b) => b.updatedAt - a.updatedAt);
+  }
   async get(conversationId) {
     const kv = await this.client();
     if (!kv) {
@@ -4983,7 +5223,11 @@ var KeyValueConversationStoreBase = class {
         conversationId: nextConversation.conversationId,
         title: nextConversation.title,
         updatedAt: nextConversation.updatedAt,
-        ownerId: nextConversation.ownerId
+        createdAt: nextConversation.createdAt,
+        ownerId: nextConversation.ownerId,
+        parentConversationId: nextConversation.parentConversationId,
+        messageCount: nextConversation.messages.length,
+        hasPendingApprovals: Array.isArray(nextConversation.pendingApprovals) && nextConversation.pendingApprovals.length > 0
       }),
       this.ttl
     );
@@ -5449,7 +5693,7 @@ var TelemetryEmitter = class {
 };
 
 // src/index.ts
-import { defineTool as defineTool4 } from "@poncho-ai/sdk";
+import { defineTool as defineTool5 } from "@poncho-ai/sdk";
 export {
   AgentHarness,
   InMemoryConversationStore,
@@ -5472,9 +5716,10 @@ export {
   createModelProvider,
   createSkillTools,
   createStateStore,
+  createSubagentTools,
   createUploadStore,
   createWriteTool,
-  defineTool4 as defineTool,
+  defineTool5 as defineTool,
   deriveUploadKey,
   ensureAgentIdentity,
   generateAgentId,

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@poncho-ai/harness",
-  "version": "0.16.1",
+  "version": "0.18.0",
   "description": "Agent execution runtime - conversation loop, tool dispatch, streaming",
   "repository": {
     "type": "git",
@@ -31,7 +31,7 @@
     "redis": "^5.10.0",
     "yaml": "^2.4.0",
     "zod": "^3.22.0",
-    "@poncho-ai/sdk": "1.1.1"
+    "@poncho-ai/sdk": "1.3.0"
   },
   "devDependencies": {
     "@types/mustache": "^4.2.6",