opencode-usage-coach 0.2.1 → 0.2.3

package/dist/index.js

@@ -3,7 +3,7 @@ import { mkdirSync, writeFileSync, appendFileSync, readFileSync, existsSync } fr
 import { spawn } from "child_process";
 import { createHash } from "crypto";
 import { homedir } from "os";
-import { join, resolve } from "path";
+import { join, resolve, dirname } from "path";
 import { tool } from "@opencode-ai/plugin";
 var PLUGIN_NAME = "opencode-usage-coach";
 var DEBUG = process.env.UC_DEBUG === "1";
@@ -38,19 +38,24 @@ function writeState(c) {
   } catch {
   }
 }
-function readHarness() {
+function harnessFile(sessionID) {
+  return join(STATE_DIR, sessionID || "_default", "harness.json");
+}
+function readHarness(sessionID) {
   try {
-    if (!existsSync(HARNESS_FILE)) return null;
-    return JSON.parse(readFileSync(HARNESS_FILE, "utf8"));
+    const f = harnessFile(sessionID);
+    if (!existsSync(f)) return null;
+    return JSON.parse(readFileSync(f, "utf8"));
   } catch {
     return null;
   }
 }
-function writeHarness(h) {
+function writeHarness(sessionID, h) {
   try {
-    mkdirSync(STATE_DIR, { recursive: true });
+    const f = harnessFile(sessionID);
+    mkdirSync(dirname(f), { recursive: true });
     h.updatedAt = (/* @__PURE__ */ new Date()).toISOString();
-    writeFileSync(HARNESS_FILE, JSON.stringify(h, null, 2));
+    writeFileSync(f, JSON.stringify(h, null, 2));
   } catch {
   }
 }
@@ -68,6 +73,7 @@ function readHarnessCfg(dir) {
   };
 }
 async function runModel(client, model, prompt, directory) {
+  const sleep = (ms) => new Promise((r) => setTimeout(r, ms));
   try {
     const slash = model.indexOf("/");
     const providerID = slash >= 0 ? model.slice(0, slash) : model;
@@ -75,11 +81,20 @@ async function runModel(client, model, prompt, directory) {
     const s = await client.session.create({ body: { title: "uc-harness-sub" }, query: { directory } });
     const id = s?.data?.info?.id;
     if (!id) return null;
-    const r = await client.session.prompt({
+    await client.session.prompt({
       path: { id },
       body: { model: { providerID, modelID }, parts: [{ type: "text", text: prompt }] }
     });
-    const parts = r?.data?.parts ?? [];
+    for (let i = 0; i < 600; i++) {
+      await sleep(1e3);
+      const st = await client.session.status({ path: { id } });
+      const status = st?.data?.[id]?.status ?? st?.data?.status;
+      if (status === "idle" || status === "completed" || !status) break;
+    }
+    const msgs = await client.session.messages({ path: { id } });
+    const all = msgs?.data ?? [];
+    const lastAssistant = all.filter((m) => m?.info?.role === "assistant").pop();
+    const parts = lastAssistant?.parts ?? [];
     const text = parts.filter((p) => p?.type === "text").map((p) => p?.text ?? "").join("");
     try {
       await client.session.remove?.({ path: { id } });
@@ -298,8 +313,8 @@ async function UsageCoachPlugin(input) {
         harness_start: tool({
           description: "Start the harness: register the total task count on the panel. Call once when the harness loop begins.",
           args: { name: tool.schema.string(), total: tool.schema.number() },
-          async execute(args) {
-            writeHarness({ name: args.name, total: args.total, current: 0, tasks: [], usage: {}, startedAt: (/* @__PURE__ */ new Date()).toISOString() });
+          async execute(args, ctx) {
+            writeHarness(ctx.sessionID, { name: args.name, total: args.total, current: 0, tasks: [], usage: {}, active: true, startedAt: (/* @__PURE__ */ new Date()).toISOString() });
             return `Harness '${args.name}' started (${args.total} tasks). Now report each task's status via task_update and run the loop.`;
           }
         }),
@@ -313,31 +328,31 @@ async function UsageCoachPlugin(input) {
             score: tool.schema.string().optional().describe("PASS | FAIL"),
             model: tool.schema.string().optional()
           },
-          async execute(args) {
-            const h = readHarness() ?? { name: "batch", total: 0, current: 0, tasks: [], usage: {} };
+          async execute(args, ctx) {
+            const h = readHarness(ctx.sessionID) ?? { name: "batch", total: 0, current: 0, tasks: [], usage: {}, active: true };
             h.tasks = h.tasks.filter((x) => x.id !== args.id);
             h.tasks.push({ id: args.id, title: args.title, status: args.status, model: args.model ?? "", revisions: args.revisions ?? 0, score: args.score ?? null });
             if (args.id > h.current) h.current = args.id;
-            writeHarness(h);
+            writeHarness(ctx.sessionID, h);
             return `task ${args.id} -> ${args.status}${args.score ? ` (${args.score})` : ""}`;
           }
         }),
         harness_done: tool({
           description: "Mark the harness as complete \u2014 call when the loop ends.",
           args: {},
-          async execute() {
-            const h = readHarness();
+          async execute(_args, ctx) {
+            const h = readHarness(ctx.sessionID);
             if (h) {
               h.current = h.total;
-              writeHarness(h);
+              h.active = false;
+              writeHarness(ctx.sessionID, h);
             }
             return "Harness complete.";
           }
         }),
-        // Per-role model execution (config-driven). Runs a NEW session with the configured model
-        // on the same server — enables multi-model (e.g. GLM generate + free mimo grade) in 1 terminal.
+        // Per-role model execution (config-driven, same server, no deadlock).
         generate: tool({
-          description: "Run the GENERATOR model (from harness.config.json) on a prompt. Returns the model's text response. The generator can use tools (write/edit files) in the directory. Use for the 'do the work' step.",
+          description: "Run the GENERATOR model on a prompt. The generator can use tools (write/edit files). Returns the model's text response.",
           args: { prompt: tool.schema.string() },
           async execute(args, ctx) {
             const cfg = readHarnessCfg(ctx.directory);
@@ -346,8 +361,21 @@ async function UsageCoachPlugin(input) {
             return out ?? `(generator ${model} produced no text)`;
           }
         }),
+        generate_batch: tool({
+          description: "Run the GENERATOR model on MULTIPLE tasks in PARALLEL. Pass an array of {id, prompt}. Returns all results at once. Use for INDEPENDENT tasks (much faster than sequential generate calls).",
+          args: { tasks: tool.schema.array(tool.schema.object({ id: tool.schema.number(), prompt: tool.schema.string() })) },
+          async execute(args, ctx) {
+            const cfg = readHarnessCfg(ctx.directory);
+            const model = cfg.generator ?? "zai-coding-plan/glm-5.1";
+            const results = await Promise.all(args.tasks.map(async (t) => {
+              const out = await runModel(input.client, model, t.prompt, ctx.directory);
+              return `[task ${t.id}] ${out ?? "(no output)"}`;
+            }));
+            return results.join("\n\n");
+          }
+        }),
         grade: tool({
-          description: "Run the GRADER model (from harness.config.json) on a prompt. Returns the model's text response (expect PASS/FAIL on the first line). Use for the 'evaluate the result' step. Falls back to the generator if the grader has no quota.",
+          description: "Run the GRADER model on a prompt. Returns PASS/FAIL on the first line. Falls back to generator if grader quota is out.",
           args: { prompt: tool.schema.string() },
           async execute(args, ctx) {
             const cfg = readHarnessCfg(ctx.directory);

package/dist/tui.js

@@ -132,7 +132,11 @@ function initializeTui(api, disposeRoot) {
     }
     let h = null;
     try {
-      h = getHarness();
+      const sid = ctx.session_id ?? "";
+      if (sid) {
+        const hf = join(STATE_DIR, sid, "harness.json");
+        if (existsSync(hf)) h = JSON.parse(readFileSync(hf, "utf8"));
+      }
     } catch {
       h = null;
     }

package/package.json

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