@poncho-ai/harness 0.8.0 → 0.9.0

package/.turbo/turbo-build.log

@@ -1,5 +1,5 @@
 
-> @poncho-ai/harness@0.8.0 build /Users/cesar/Dev/latitude/poncho-ai/packages/harness
+> @poncho-ai/harness@0.9.0 build /Users/cesar/Dev/latitude/poncho-ai/packages/harness
 > tsup src/index.ts --format esm --dts
 
 CLI Building entry: src/index.ts
@@ -7,8 +7,8 @@
 CLI tsup v8.5.1
 CLI Target: es2022
 ESM Build start
-ESM dist/index.js 130.60 KB
-ESM ⚡️ Build success in 39ms
+ESM dist/index.js 133.06 KB
+ESM ⚡️ Build success in 29ms
 DTS Build start
-DTS ⚡️ Build success in 3563ms
+DTS ⚡️ Build success in 3021ms
 DTS dist/index.d.ts 17.17 KB

package/dist/index.js

@@ -2032,6 +2032,13 @@ var ToolDispatcher = class {
     return this.tools.get(name);
   }
   async execute(call, context) {
+    if (context.abortSignal?.aborted) {
+      return {
+        callId: call.id,
+        tool: call.name,
+        error: "Tool execution cancelled"
+      };
+    }
     const definition = this.tools.get(call.name);
     if (!definition) {
       return {
@@ -2042,12 +2049,26 @@ var ToolDispatcher = class {
     }
     try {
       const output = await definition.handler(call.input, context);
+      if (context.abortSignal?.aborted) {
+        return {
+          callId: call.id,
+          tool: call.name,
+          error: "Tool execution cancelled"
+        };
+      }
       return {
         callId: call.id,
         tool: call.name,
         output
       };
     } catch (error) {
+      if (context.abortSignal?.aborted) {
+        return {
+          callId: call.id,
+          tool: call.name,
+          error: "Tool execution cancelled"
+        };
+      }
       return {
         callId: call.id,
         tool: call.name,
@@ -2056,7 +2077,19 @@ var ToolDispatcher = class {
     }
   }
   async executeBatch(calls, context) {
-    return Promise.all(calls.map(async (call) => this.execute(call, context)));
+    const results = [];
+    for (const call of calls) {
+      if (context.abortSignal?.aborted) {
+        results.push({
+          callId: call.id,
+          tool: call.name,
+          error: "Tool execution cancelled"
+        });
+        continue;
+      }
+      results.push(await this.execute(call, context));
+    }
+    return results;
   }
 };
 
@@ -2072,6 +2105,17 @@ var SKILL_TOOL_NAMES = [
   "run_skill_script"
 ];
 var trimMessageWindow = (messages) => messages.length <= MAX_CONTEXT_MESSAGES ? messages : messages.slice(messages.length - MAX_CONTEXT_MESSAGES);
+var isAbortError = (error) => {
+  if (!error || typeof error !== "object") {
+    return false;
+  }
+  const maybeName = "name" in error ? String(error.name ?? "") : "";
+  if (maybeName === "AbortError") {
+    return true;
+  }
+  const maybeMessage = "message" in error ? String(error.message ?? "") : "";
+  return maybeMessage.toLowerCase().includes("abort");
+};
 var MODEL_TOOL_NAME_PATTERN = /^[a-zA-Z0-9_-]{1,128}$/;
 var toProviderSafeToolName = (originalName, index, used) => {
   if (MODEL_TOOL_NAME_PATTERN.test(originalName) && !used.has(originalName)) {
@@ -2582,6 +2626,12 @@ ${boundedMainMemory.trim()}` : "";
       events.push(event);
       return event;
     };
+    const isCancelled = () => input.abortSignal?.aborted === true;
+    let cancellationEmitted = false;
+    const emitCancellation = () => {
+      cancellationEmitted = true;
+      return pushEvent({ type: "run:cancelled", runId });
+    };
     yield pushEvent({
       type: "run:started",
       runId,
@@ -2597,6 +2647,10 @@ ${boundedMainMemory.trim()}` : "";
     let totalOutputTokens = 0;
     for (let step = 1; step <= maxSteps; step += 1) {
       try {
+        if (isCancelled()) {
+          yield emitCancellation();
+          return;
+        }
         if (now() - start > timeoutMs) {
           yield pushEvent({
             type: "run:error",
@@ -2690,6 +2744,7 @@ ${boundedMainMemory.trim()}` : "";
           messages: coreMessages,
           tools,
           temperature,
+          abortSignal: input.abortSignal,
           ...typeof maxTokens === "number" ? { maxTokens } : {},
           experimental_telemetry: {
             isEnabled: telemetryEnabled && !!latitudeApiKey
@@ -2698,10 +2753,18 @@ ${boundedMainMemory.trim()}` : "";
         let streamedAnyChunk = false;
         let fullText = "";
         for await (const chunk of result.textStream) {
+          if (isCancelled()) {
+            yield emitCancellation();
+            return;
+          }
           streamedAnyChunk = true;
           fullText += chunk;
           yield pushEvent({ type: "model:chunk", content: chunk });
         }
+        if (isCancelled()) {
+          yield emitCancellation();
+          return;
+        }
         const fullResult = await result.response;
         const usage = await result.usage;
         const toolCallsResult = await result.toolCalls;
@@ -2746,11 +2809,16 @@ ${boundedMainMemory.trim()}` : "";
           agentId: agent.frontmatter.id ?? agent.frontmatter.name,
           step,
           workingDir: this.workingDir,
-          parameters: input.parameters ?? {}
+          parameters: input.parameters ?? {},
+          abortSignal: input.abortSignal
         };
         const toolResultsForModel = [];
         const approvedCalls = [];
         for (const call of toolCalls) {
+          if (isCancelled()) {
+            yield emitCancellation();
+            return;
+          }
           const runtimeToolName = exposedToolNames.get(call.name) ?? call.name;
           yield pushEvent({ type: "tool:started", tool: runtimeToolName, input: call.input });
           const requiresApproval = this.requiresApprovalForToolCall(
@@ -2772,6 +2840,10 @@ ${boundedMainMemory.trim()}` : "";
               step,
               approvalId
             }) : false;
+            if (isCancelled()) {
+              yield emitCancellation();
+              return;
+            }
             if (!approved) {
               yield pushEvent({
                 type: "tool:approval:denied",
@@ -2801,7 +2873,15 @@ ${boundedMainMemory.trim()}` : "";
           });
         }
         const batchStart = now();
+        if (isCancelled()) {
+          yield emitCancellation();
+          return;
+        }
         const batchResults = approvedCalls.length > 0 ? await this.dispatcher.executeBatch(approvedCalls, toolContext) : [];
+        if (isCancelled()) {
+          yield emitCancellation();
+          return;
+        }
         for (const result2 of batchResults) {
           if (result2.error) {
             yield pushEvent({
@@ -2855,6 +2935,12 @@ ${boundedMainMemory.trim()}` : "";
           duration: now() - stepStart
         });
       } catch (error) {
+        if (isCancelled() || isAbortError(error)) {
+          if (!cancellationEmitted) {
+            yield emitCancellation();
+          }
+          return;
+        }
         yield pushEvent({
           type: "run:error",
           runId,
@@ -2903,6 +2989,14 @@ ${boundedMainMemory.trim()}` : "";
           duration: 0
         };
       }
+      if (event.type === "run:cancelled") {
+        finalResult = {
+          status: "cancelled",
+          steps: 0,
+          tokens: { input: 0, output: 0, cached: 0 },
+          duration: 0
+        };
+      }
     }
     return {
       runId,

package/package.json

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

package/src/harness.ts

@@ -69,6 +69,18 @@ const trimMessageWindow = (messages: Message[]): Message[] =>
     ? messages
     : messages.slice(messages.length - MAX_CONTEXT_MESSAGES);
 
+const isAbortError = (error: unknown): boolean => {
+  if (!error || typeof error !== "object") {
+    return false;
+  }
+  const maybeName = "name" in error ? String(error.name ?? "") : "";
+  if (maybeName === "AbortError") {
+    return true;
+  }
+  const maybeMessage = "message" in error ? String(error.message ?? "") : "";
+  return maybeMessage.toLowerCase().includes("abort");
+};
+
 const MODEL_TOOL_NAME_PATTERN = /^[a-zA-Z0-9_-]{1,128}$/;
 
 const toProviderSafeToolName = (
@@ -658,6 +670,12 @@ ${boundedMainMemory.trim()}`
       events.push(event);
       return event;
     };
+    const isCancelled = (): boolean => input.abortSignal?.aborted === true;
+    let cancellationEmitted = false;
+    const emitCancellation = (): AgentEvent => {
+      cancellationEmitted = true;
+      return pushEvent({ type: "run:cancelled", runId });
+    };
 
     yield pushEvent({
       type: "run:started",
@@ -677,6 +695,10 @@ ${boundedMainMemory.trim()}`
 
     for (let step = 1; step <= maxSteps; step += 1) {
       try {
+        if (isCancelled()) {
+          yield emitCancellation();
+          return;
+        }
         if (now() - start > timeoutMs) {
           yield pushEvent({
             type: "run:error",
@@ -793,6 +815,7 @@ ${boundedMainMemory.trim()}`
           messages: coreMessages,
           tools,
           temperature,
+          abortSignal: input.abortSignal,
           ...(typeof maxTokens === "number" ? { maxTokens } : {}),
           experimental_telemetry: {
             isEnabled: telemetryEnabled && !!latitudeApiKey,
@@ -803,10 +826,19 @@ ${boundedMainMemory.trim()}`
         let streamedAnyChunk = false;
         let fullText = "";
         for await (const chunk of result.textStream) {
-        streamedAnyChunk = true;
-        fullText += chunk;
-        yield pushEvent({ type: "model:chunk", content: chunk });
-      }
+          if (isCancelled()) {
+            yield emitCancellation();
+            return;
+          }
+          streamedAnyChunk = true;
+          fullText += chunk;
+          yield pushEvent({ type: "model:chunk", content: chunk });
+        }
+
+        if (isCancelled()) {
+          yield emitCancellation();
+          return;
+        }
 
       // Get full response with usage and tool calls
       const fullResult = await result.response;
@@ -861,6 +893,7 @@ ${boundedMainMemory.trim()}`
         step,
         workingDir: this.workingDir,
         parameters: input.parameters ?? {},
+        abortSignal: input.abortSignal,
       };
 
       const toolResultsForModel: Array<{
@@ -877,6 +910,10 @@ ${boundedMainMemory.trim()}`
       }> = [];
 
       for (const call of toolCalls) {
+        if (isCancelled()) {
+          yield emitCancellation();
+          return;
+        }
         const runtimeToolName = exposedToolNames.get(call.name) ?? call.name;
         yield pushEvent({ type: "tool:started", tool: runtimeToolName, input: call.input });
         const requiresApproval = this.requiresApprovalForToolCall(
@@ -900,6 +937,10 @@ ${boundedMainMemory.trim()}`
                 approvalId,
               })
             : false;
+          if (isCancelled()) {
+            yield emitCancellation();
+            return;
+          }
           if (!approved) {
             yield pushEvent({
               type: "tool:approval:denied",
@@ -929,11 +970,20 @@ ${boundedMainMemory.trim()}`
         });
       }
       const batchStart = now();
+      if (isCancelled()) {
+        yield emitCancellation();
+        return;
+      }
       const batchResults =
         approvedCalls.length > 0
           ? await this.dispatcher.executeBatch(approvedCalls, toolContext)
           : [];
 
+      if (isCancelled()) {
+        yield emitCancellation();
+        return;
+      }
+
       for (const result of batchResults) {
         if (result.error) {
           yield pushEvent({
@@ -993,6 +1043,12 @@ ${boundedMainMemory.trim()}`
           duration: now() - stepStart,
         });
       } catch (error) {
+        if (isCancelled() || isAbortError(error)) {
+          if (!cancellationEmitted) {
+            yield emitCancellation();
+          }
+          return;
+        }
         yield pushEvent({
           type: "run:error",
           runId,
@@ -1044,6 +1100,14 @@ ${boundedMainMemory.trim()}`
           duration: 0,
         };
       }
+      if (event.type === "run:cancelled") {
+        finalResult = {
+          status: "cancelled",
+          steps: 0,
+          tokens: { input: 0, output: 0, cached: 0 },
+          duration: 0,
+        };
+      }
     }
 
     return {

package/src/tool-dispatcher.ts

@@ -45,6 +45,13 @@ export class ToolDispatcher {
   }
 
   async execute(call: ToolCall, context: ToolContext): Promise<ToolExecutionResult> {
+    if (context.abortSignal?.aborted) {
+      return {
+        callId: call.id,
+        tool: call.name,
+        error: "Tool execution cancelled",
+      };
+    }
     const definition = this.tools.get(call.name);
     if (!definition) {
       return {
@@ -56,12 +63,26 @@ export class ToolDispatcher {
 
     try {
       const output = await definition.handler(call.input, context);
+      if (context.abortSignal?.aborted) {
+        return {
+          callId: call.id,
+          tool: call.name,
+          error: "Tool execution cancelled",
+        };
+      }
       return {
         callId: call.id,
         tool: call.name,
         output,
       };
     } catch (error) {
+      if (context.abortSignal?.aborted) {
+        return {
+          callId: call.id,
+          tool: call.name,
+          error: "Tool execution cancelled",
+        };
+      }
       return {
         callId: call.id,
         tool: call.name,
@@ -74,6 +95,19 @@ export class ToolDispatcher {
     calls: ToolCall[],
     context: ToolContext,
   ): Promise<ToolExecutionResult[]> {
-    return Promise.all(calls.map(async (call) => this.execute(call, context)));
+    const results: ToolExecutionResult[] = [];
+    for (const call of calls) {
+      if (context.abortSignal?.aborted) {
+        results.push({
+          callId: call.id,
+          tool: call.name,
+          error: "Tool execution cancelled",
+        });
+        continue;
+      }
+      // eslint-disable-next-line no-await-in-loop
+      results.push(await this.execute(call, context));
+    }
+    return results;
   }
 }