opencode-usage-coach 0.8.0 → 0.8.1

package/dist/index.js

@@ -1682,7 +1682,7 @@ ${priorNotes}
           }
         }),
         generate_batch: tool({
-          description: "Run the GENERATOR model on MULTIPLE tasks. Quota-aware: GO = full parallel; THROTTLE = lighter model + concurrency capped at 2; STOP = refused. Use for INDEPENDENT tasks. Step-limited: each sub-session aborts after max_steps (default 30).",
+          description: "Run the GENERATOR model on MULTIPLE tasks. Quota-aware: GO = full parallel; THROTTLE = lighter model + concurrency capped at 2; STOP = refused. Use for INDEPENDENT tasks. Step-limited: each sub-session aborts after max_steps (default 30). Resilient: failed tasks are retried sequentially (once); if retry also fails, re-run them individually with generate().",
           args: { tasks: tool.schema.array(tool.schema.object({ id: tool.schema.number(), prompt: tool.schema.string() })), max_steps: tool.schema.number().optional().describe("Maximum sub-session steps per task before timeout (default 30).") },
           async execute(args, ctx) {
             const cfg = readHarnessCfg(ctx.directory);
@@ -1698,54 +1698,83 @@ ${priorNotes}
             const limit = decision === "THROTTLE" ? 2 : args.tasks.length;
             const rules = readRules();
             const priorNotes = readImplNotes(5);
-            const results = [];
-            for (let i = 0; i < args.tasks.length; i += limit) {
-              const batch = args.tasks.slice(i, i + limit);
-              const out = await Promise.all(batch.map(async (t) => {
-                let prefix = rules ? `Lessons learned from previous failures (apply where relevant):
+            const maxSteps = args.max_steps ?? DEFAULT_MAX_STEPS;
+            const runOne = async (t) => {
+              let prefix = rules ? `Lessons learned from previous failures (apply where relevant):
 ${rules}
 
 ---
 
 ` : "";
-                if (priorNotes) prefix = `Notes from previous runs (context for this task):
+              if (priorNotes) prefix = `Notes from previous runs (context for this task):
 ${priorNotes}
 
 ---
 
 ` + prefix;
-                prefix += IMPL_NOTE_INSTRUCTION;
-                const r = await runModel(
-                  input.client,
-                  model,
-                  prefix + t.prompt,
-                  ctx.directory,
-                  { sessionID: ctx.sessionID, taskId: t.id },
-                  args.max_steps ?? DEFAULT_MAX_STEPS
-                );
-                const isTimeoutOrError = r.startsWith("Task appears too large") || r.startsWith("ERROR:");
-                if (!isTimeoutOrError) {
-                  try {
-                    const { notes } = extractImplNotes(r);
-                    if (notes) {
-                      appendImplNotes(notes, t.prompt);
-                      const kw = extractKeywords(t.prompt);
-                      if (kw.length) {
-                        const noteNodeId = saveInvestigationResult(kw, notes, "impl-note", 0.5);
-                        if (noteNodeId) linkImplNoteToDomain(noteNodeId, notes);
-                      }
+              prefix += IMPL_NOTE_INSTRUCTION;
+              const r = await runModel(
+                input.client,
+                model,
+                prefix + t.prompt,
+                ctx.directory,
+                { sessionID: ctx.sessionID, taskId: t.id },
+                maxSteps
+              );
+              const isTimeoutOrError = r.startsWith("Task appears too large") || r.startsWith("ERROR:");
+              if (!isTimeoutOrError) {
+                try {
+                  const { notes } = extractImplNotes(r);
+                  if (notes) {
+                    appendImplNotes(notes, t.prompt);
+                    const kw = extractKeywords(t.prompt);
+                    if (kw.length) {
+                      const noteNodeId = saveInvestigationResult(kw, notes, "impl-note", 0.5);
+                      if (noteNodeId) linkImplNoteToDomain(noteNodeId, notes);
                     }
-                  } catch (e) {
-                    log(`generate_batch impl-notes extract err: ${String(e)}`);
                   }
+                } catch (e) {
+                  log(`generate_batch impl-notes extract err: ${String(e)}`);
+                }
+              }
+              return { id: t.id, result: r };
+            };
+            const results = [];
+            const failed = [];
+            for (let i = 0; i < args.tasks.length; i += limit) {
+              const batch = args.tasks.slice(i, i + limit);
+              const settled = await Promise.allSettled(batch.map((t) => runOne(t)));
+              for (let j = 0; j < settled.length; j++) {
+                const s = settled[j];
+                const t = batch[j];
+                if (s.status === "fulfilled") {
+                  results.push(`[task ${s.value.id}] ${s.value.result}`);
+                } else {
+                  const err = String(s.reason ?? "unknown rejection");
+                  log(`generate_batch task ${t.id} REJECTED: ${err}`);
+                  failed.push({ id: t.id, prompt: t.prompt, error: err });
                 }
-                return `[task ${t.id}] ${r}`;
-              }));
-              results.push(...out);
+              }
+            }
+            if (failed.length > 0) {
+              log(`generate_batch: retrying ${failed.length} failed task(s) sequentially`);
+              for (const f of failed) {
+                try {
+                  const r = await runOne(f);
+                  results.push(`[task ${r.id}] (retry) ${r.result}`);
+                } catch (e2) {
+                  const err2 = String(e2 ?? "unknown rejection on retry");
+                  log(`generate_batch task ${f.id} retry ALSO FAILED: ${err2}`);
+                  results.push(`[task ${f.id}] ERROR: task failed on both batch and retry. Error: ${err2}`);
+                }
+              }
             }
+            const failedCount = results.filter((r) => r.includes("ERROR:")).length;
             const note = throttle ? `
 [usage-coach] quota THROTTLE \u2014 lighter model ${cfg.lighterModel}, concurrency capped at ${limit}` : "";
-            return results.join("\n\n") + note;
+            const failNext = failedCount > 0 ? `
+[usage-coach NEXT] ${failedCount} task(s) failed even after retry. Re-run each failed task individually with generate() (sequential), NOT generate_batch. This isolates failures and avoids batch-level abort.` : "";
+            return results.join("\n\n") + note + failNext;
           }
         }),
         grade: tool({

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "opencode-usage-coach",
-  "version": "0.8.0",
+  "version": "0.8.1",
   "description": "opencode closed-loop usage coach — quota SENSE -> coaching DECIDE -> loop ACT + TUI integration",
   "type": "module",
   "main": "./dist/index.js",