@agntn/harnesses 0.0.1 → 0.1.0

package/packages/pi/extensions/harnesses.ts

@@ -38,6 +38,21 @@ function loadToolOperations(): Promise<typeof HarnessTools> {
   return toolOperationsPromise;
 }
 
+function prepareInfoArguments(args: unknown): unknown {
+  if (typeof args !== "object" || args === null || !("id" in args) || typeof args.id !== "string") {
+    return args;
+  }
+
+  try {
+    const id: unknown = JSON.parse(args.id);
+    return Array.isArray(id) && id.every((value) => typeof value === "string")
+      ? { ...args, id }
+      : args;
+  } catch {
+    return args;
+  }
+}
+
 function statusRenderers(tool: HarnessToolName) {
   return {
     renderCall(args: unknown, theme: StatusTheme, context: RenderOptions) {
@@ -87,6 +102,7 @@ export default function harnessesExtension(pi: ExtensionAPI): void {
       "Resolved paths are absolute for the current platform and home directory.",
     ],
     parameters: schemas.info,
+    prepareArguments: prepareInfoArguments,
     ...statusRenderers("harnesses_info"),
     async execute(
       _toolCallId,
@@ -114,20 +130,24 @@ export default function harnessesExtension(pi: ExtensionAPI): void {
     async execute(
       _toolCallId,
       params: HarnessSchemas.ModelsParams,
+      signal,
     ): Promise<AgentToolResult<HarnessTools.ModelsOutcome | HarnessTools.RunFailure>> {
       const { listHarnessModels } = await loadToolOperations();
       const { content, details, isError } = await listHarnessModels(params.id, {
         search: params.search,
         cwd: params.cwd,
         timeoutSeconds: params.timeoutSeconds,
+        signal,
       });
       if (isError) {
         const message =
           "error" in details
             ? details.error
-            : details.timedOut
-              ? `Harness ${details.id} model listing timed out`
-              : details.stderr || `Harness ${details.id} exited with code ${details.exitCode}`;
+            : details.aborted
+              ? `Harness ${details.id} model listing aborted`
+              : details.timedOut
+                ? `Harness ${details.id} model listing timed out`
+                : details.stderr || `Harness ${details.id} exited with code ${details.exitCode}`;
         throw new Error(message);
       }
       return { content, details };
@@ -152,23 +172,25 @@ export default function harnessesExtension(pi: ExtensionAPI): void {
     async execute(
       _toolCallId,
       params: HarnessSchemas.RunParams,
+      signal,
     ): Promise<AgentToolResult<HarnessTools.RunOutcome | HarnessTools.RunFailure>> {
       const { runHarness } = await loadToolOperations();
       const { content, details, isError } = await runHarness(params.id, params.prompt, {
         cwd: params.cwd,
         model: params.model,
         timeoutSeconds: params.timeoutSeconds,
+        signal,
         structured: params.structured,
         tools: params.tools,
         readOnly: params.readOnly,
       });
       if (isError) {
+        if ("error" in details) throw new Error(details.error);
         const message =
-          "error" in details
-            ? details.error
-            : details.timedOut
-              ? `Harness ${details.id} timed out`
-              : details.stderr || `Harness ${details.id} exited with code ${details.exitCode}`;
+          content[0]?.text ??
+          (details.timedOut
+            ? `Harness ${details.id} timed out`
+            : details.stderr || `Harness ${details.id} exited with code ${details.exitCode}`);
         throw new Error(message);
       }
       return { content, details };

package/packages/shared/tui.ts

@@ -222,7 +222,7 @@ function modelsMeta(details: ResultDetails): string[] {
   const models = listLength(details, "models");
   return compactStrings([
     id ? sanitizeTerminalText(id) : undefined,
-    models === undefined ? undefined : `${models} models`,
+    details.aborted === true ? "aborted" : models === undefined ? undefined : `${models} models`,
   ]);
 }
 
@@ -230,7 +230,9 @@ function runMeta(details: ResultDetails): string[] {
   const meta: string[] = [];
   const id = scalar(details, "id");
   if (id) meta.push(sanitizeTerminalText(id));
-  if (details.timedOut === true) {
+  if (details.aborted === true) {
+    meta.push("aborted");
+  } else if (details.timedOut === true) {
     meta.push("timed out");
   } else if (typeof details.exitCode === "number") {
     meta.push(`exit ${details.exitCode}`);