@ycodium-ai/agent-fake 0.2.3976 → 0.2.3990

package/dist/index.js

@@ -2,7 +2,7 @@
 import { definePlugin } from "@ycodium-ai/plugin-api";
 
 // ../../packages/plugin-api/src/version.ts
-var PLUGIN_API_VERSION = 75;
+var PLUGIN_API_VERSION = 77;
 
 // src/card.ts
 var FAKE_SCRIPTS = [
@@ -101,38 +101,32 @@ var FAKE_CARD = {
 import * as NodeCrypto from "node:crypto";
 
 // src/fakeEventBus.ts
-var EventBus = class {
-  #queued = [];
-  #waiters = [];
-  #closed = false;
-  push(event) {
-    const waiter = this.#waiters.shift();
-    if (waiter) {
-      waiter({ done: false, value: event });
-      return;
-    }
-    this.#queued.push(event);
-  }
-  close() {
-    if (this.#closed) return;
-    this.#closed = true;
-    while (this.#waiters.length > 0) {
-      this.#waiters.shift()?.({ done: true, value: void 0 });
+function createFakeEmitter(input) {
+  return (type, threadId, payload, extra) => {
+    const context = input.thread(threadId);
+    if (!context) return;
+    if (type.startsWith("task.")) {
+      const task = payload;
+      if (type === "task.completed" || ["completed", "failed", "cancelled", "interrupted", "idle"].includes(task.status ?? ""))
+        context.tasks.delete(task.taskId);
+      else context.tasks.set(task.taskId, { ...context.tasks.get(task.taskId), ...task });
     }
-  }
-  [Symbol.asyncIterator]() {
-    return {
-      next: () => {
-        const queued = this.#queued.shift();
-        if (queued !== void 0) return Promise.resolve({ done: false, value: queued });
-        if (this.#closed) return Promise.resolve({ done: true, value: void 0 });
-        return new Promise((resolve) => {
-          this.#waiters.push(resolve);
-        });
-      }
-    };
-  }
-};
+    context.events.emit({
+      eventId: input.nextId(),
+      provider: "fake",
+      providerInstanceId: input.instanceId,
+      threadId,
+      createdAt: input.now(),
+      type,
+      payload,
+      ...extra
+    });
+  };
+}
+function settleFakeTasks(context, threadId, emit) {
+  for (const [taskId, task] of context.tasks)
+    emit("task.completed", threadId, { ...task, taskId, status: "stopped" });
+}
 
 // src/approvalToolScript.ts
 function createApprovalToolScript({
@@ -302,7 +296,6 @@ function createFakeSessionLifecycle(input) {
           if (result.status === "rejected") throw result.reason;
         }
         await input.disposePlaybook();
-        input.closeEvents();
       })().then(disposing.resolve, disposing.reject);
       return disposal;
     }
@@ -888,115 +881,8 @@ function createPlaybookMcp(input) {
   };
 }
 
-// src/playbook.ts
-var lineClaimsSession = (line, threadId, title) => (line.match.threadId === void 0 || line.match.threadId.test(threadId)) && (line.match.title === void 0 || title !== void 0 && title.includes(line.match.title));
+// src/playbookResolve.ts
 var isRecord2 = (value) => typeof value === "object" && value !== null && !Array.isArray(value);
-var isName = (value) => typeof value === "string" && value.length > 0;
-var onlyKeys = (value, keys) => Object.keys(value).every((key) => keys.includes(key));
-var parseTimeout = (value) => {
-  const timeoutMs = value === void 0 ? 3e4 : value;
-  return typeof timeoutMs === "number" && Number.isFinite(timeoutMs) && timeoutMs >= 0 ? timeoutMs : null;
-};
-function parseStep(value) {
-  if (!isRecord2(value)) return null;
-  if (Object.hasOwn(value, "expectInput"))
-    return onlyKeys(value, ["expectInput"]) && isName(value.expectInput) ? { expectInput: value.expectInput } : null;
-  if (Object.hasOwn(value, "awaitSteer")) {
-    const wait = value.awaitSteer;
-    if (!onlyKeys(value, ["awaitSteer"]) || !isRecord2(wait) || !onlyKeys(wait, ["timeoutMs", "continueOnTimeout"]) || Object.hasOwn(wait, "continueOnTimeout") && wait.continueOnTimeout !== true)
-      return null;
-    const timeoutMs = parseTimeout(wait.timeoutMs);
-    if (timeoutMs === null) return null;
-    const onTimeout = wait.continueOnTimeout === true ? { continueOnTimeout: true } : {};
-    return { awaitSteer: { timeoutMs, ...onTimeout } };
-  }
-  if (Object.hasOwn(value, "wait")) {
-    const wait = value.wait;
-    if (!onlyKeys(value, ["wait"]) || !isRecord2(wait) || !isName(wait.tool)) return null;
-    if (!onlyKeys(wait, ["server", "tool", "timeoutMs"]) || wait.server !== void 0 && !isName(wait.server))
-      return null;
-    const timeoutMs = parseTimeout(wait.timeoutMs);
-    return timeoutMs === null ? null : {
-      wait: {
-        tool: wait.tool,
-        timeoutMs,
-        ...wait.server === void 0 ? {} : { server: wait.server }
-      }
-    };
-  }
-  if (Object.hasOwn(value, "waitFor")) {
-    const wait = value.waitFor;
-    if (!onlyKeys(value, ["waitFor"]) || !isRecord2(wait) || !isName(wait.tool)) return null;
-    if (!onlyKeys(wait, ["id", "server", "tool", "arguments", "path", "atLeast", "timeoutMs"]) || !isName(wait.path) || wait.id !== void 0 && !isName(wait.id) || wait.arguments !== void 0 && !isRecord2(wait.arguments) || wait.server !== void 0 && !isName(wait.server))
-      return null;
-    if (typeof wait.atLeast !== "number" || !Number.isInteger(wait.atLeast) || wait.atLeast < 0)
-      return null;
-    const timeoutMs = parseTimeout(wait.timeoutMs);
-    return timeoutMs === null ? null : {
-      waitFor: {
-        tool: wait.tool,
-        path: wait.path,
-        atLeast: wait.atLeast,
-        timeoutMs,
-        ...wait.id === void 0 ? {} : { id: wait.id },
-        ...wait.server === void 0 ? {} : { server: wait.server },
-        ...wait.arguments === void 0 ? {} : { arguments: wait.arguments }
-      }
-    };
-  }
-  if (!onlyKeys(value, ["id", "server", "tool", "arguments"]) || !isName(value.tool) || value.server !== void 0 && !isName(value.server))
-    return null;
-  if (value.id !== void 0 && typeof value.id !== "string") return null;
-  if (value.arguments !== void 0 && !isRecord2(value.arguments)) return null;
-  return {
-    tool: value.tool,
-    ...value.server === void 0 ? {} : { server: value.server },
-    ...value.id === void 0 ? {} : { id: value.id },
-    ...value.arguments === void 0 ? {} : { arguments: value.arguments }
-  };
-}
-function parsePlaybook(text) {
-  try {
-    const value = JSON.parse(text);
-    if (!isRecord2(value) || value.version !== 1) return { ok: false, reason: "version must be 1" };
-    if (!onlyKeys(value, ["version", "threads"]) || !Array.isArray(value.threads))
-      return { ok: false, reason: "threads must be an array" };
-    const threads = [];
-    for (const [index, line] of value.threads.entries()) {
-      if (!isRecord2(line) || !onlyKeys(line, ["match", "turns"]) || !isRecord2(line.match))
-        return { ok: false, reason: `threads[${index}] must have match and turns` };
-      const match = line.match;
-      if (!onlyKeys(match, ["threadId", "title"]) || match.threadId === void 0 && match.title === void 0 || match.threadId !== void 0 && typeof match.threadId !== "string" || match.title !== void 0 && typeof match.title !== "string")
-        return { ok: false, reason: `threads[${index}].match needs threadId or title` };
-      if (!Array.isArray(line.turns))
-        return { ok: false, reason: `threads[${index}].turns must be an array` };
-      const turns = [];
-      for (const [turn, batch] of line.turns.entries()) {
-        if (!Array.isArray(batch))
-          return { ok: false, reason: `threads[${index}].turns[${turn}] must be an array` };
-        const steps = [];
-        for (const entry of batch) {
-          const step = parseStep(entry);
-          const last = steps.at(-1);
-          if (step === null || "expectInput" in step && steps.length > 0 || last !== void 0 && "awaitSteer" in last)
-            return { ok: false, reason: `threads[${index}].turns[${turn}] has an invalid step` };
-          steps.push(step);
-        }
-        turns.push(steps);
-      }
-      threads.push({
-        match: {
-          ...typeof match.threadId === "string" ? { threadId: new RegExp(match.threadId) } : {},
-          ...typeof match.title === "string" ? { title: match.title } : {}
-        },
-        turns
-      });
-    }
-    return { ok: true, threads };
-  } catch (error) {
-    return { ok: false, reason: String(error) };
-  }
-}
 var UnresolvedPlaceholder = class extends Error {
 };
 var splitPath = (path) => path.split(/\.(?![^\[]*\])/u);
@@ -1102,6 +988,127 @@ function stepResult(result) {
     return first.text;
   }
 }
+
+// src/playbook.ts
+var lineClaimsSession = (line, threadId, title) => (line.match.threadId === void 0 || line.match.threadId.test(threadId)) && (line.match.title === void 0 || title !== void 0 && title.includes(line.match.title));
+var isName = (value) => typeof value === "string" && value.length > 0;
+var onlyKeys = (value, keys) => Object.keys(value).every((key) => keys.includes(key));
+var parseTimeout = (value) => {
+  const timeoutMs = value === void 0 ? 3e4 : value;
+  return typeof timeoutMs === "number" && Number.isFinite(timeoutMs) && timeoutMs >= 0 ? timeoutMs : null;
+};
+function parseStep(value) {
+  if (!isRecord2(value)) return null;
+  if (Object.hasOwn(value, "expectInput"))
+    return onlyKeys(value, ["expectInput"]) && isName(value.expectInput) ? { expectInput: value.expectInput } : null;
+  if (Object.hasOwn(value, "awaitSteer")) {
+    const wait = value.awaitSteer;
+    if (!onlyKeys(value, ["awaitSteer"]) || !isRecord2(wait) || !onlyKeys(wait, ["timeoutMs", "continueOnTimeout"]) || Object.hasOwn(wait, "continueOnTimeout") && wait.continueOnTimeout !== true)
+      return null;
+    const timeoutMs = parseTimeout(wait.timeoutMs);
+    if (timeoutMs === null) return null;
+    const onTimeout = wait.continueOnTimeout === true ? { continueOnTimeout: true } : {};
+    return { awaitSteer: { timeoutMs, ...onTimeout } };
+  }
+  if (Object.hasOwn(value, "wait")) {
+    const wait = value.wait;
+    if (!onlyKeys(value, ["wait"]) || !isRecord2(wait) || !isName(wait.tool)) return null;
+    if (!onlyKeys(wait, ["server", "tool", "timeoutMs"]) || wait.server !== void 0 && !isName(wait.server))
+      return null;
+    const timeoutMs = parseTimeout(wait.timeoutMs);
+    return timeoutMs === null ? null : {
+      wait: {
+        tool: wait.tool,
+        timeoutMs,
+        ...wait.server === void 0 ? {} : { server: wait.server }
+      }
+    };
+  }
+  if (Object.hasOwn(value, "waitFor")) {
+    const wait = value.waitFor;
+    if (!onlyKeys(value, ["waitFor"]) || !isRecord2(wait) || !isName(wait.tool)) return null;
+    if (!onlyKeys(wait, [
+      "id",
+      "server",
+      "tool",
+      "arguments",
+      "path",
+      "atLeast",
+      "timeoutMs",
+      "turnGate",
+      "continueOnTimeout"
+    ]) || wait.continueOnTimeout !== void 0 && wait.continueOnTimeout !== true || wait.turnGate !== void 0 && wait.turnGate !== true || !isName(wait.path) || wait.id !== void 0 && !isName(wait.id) || wait.arguments !== void 0 && !isRecord2(wait.arguments) || wait.server !== void 0 && !isName(wait.server))
+      return null;
+    if (typeof wait.atLeast !== "number" || !Number.isInteger(wait.atLeast) || wait.atLeast < 0)
+      return null;
+    const timeoutMs = parseTimeout(wait.timeoutMs);
+    return timeoutMs === null ? null : {
+      waitFor: {
+        tool: wait.tool,
+        path: wait.path,
+        atLeast: wait.atLeast,
+        timeoutMs,
+        ...wait.id === void 0 ? {} : { id: wait.id },
+        ...wait.server === void 0 ? {} : { server: wait.server },
+        ...wait.arguments === void 0 ? {} : { arguments: wait.arguments },
+        ...wait.turnGate === true ? { turnGate: true } : {},
+        ...wait.continueOnTimeout === true ? { continueOnTimeout: true } : {}
+      }
+    };
+  }
+  if (!onlyKeys(value, ["id", "server", "tool", "arguments"]) || !isName(value.tool) || value.server !== void 0 && !isName(value.server))
+    return null;
+  if (value.id !== void 0 && typeof value.id !== "string") return null;
+  if (value.arguments !== void 0 && !isRecord2(value.arguments)) return null;
+  return {
+    tool: value.tool,
+    ...value.server === void 0 ? {} : { server: value.server },
+    ...value.id === void 0 ? {} : { id: value.id },
+    ...value.arguments === void 0 ? {} : { arguments: value.arguments }
+  };
+}
+function parsePlaybook(text) {
+  try {
+    const value = JSON.parse(text);
+    if (!isRecord2(value) || value.version !== 1) return { ok: false, reason: "version must be 1" };
+    if (!onlyKeys(value, ["version", "threads"]) || !Array.isArray(value.threads))
+      return { ok: false, reason: "threads must be an array" };
+    const threads = [];
+    for (const [index, line] of value.threads.entries()) {
+      if (!isRecord2(line) || !onlyKeys(line, ["match", "turns"]) || !isRecord2(line.match))
+        return { ok: false, reason: `threads[${index}] must have match and turns` };
+      const match = line.match;
+      if (!onlyKeys(match, ["threadId", "title"]) || match.threadId === void 0 && match.title === void 0 || match.threadId !== void 0 && typeof match.threadId !== "string" || match.title !== void 0 && typeof match.title !== "string")
+        return { ok: false, reason: `threads[${index}].match needs threadId or title` };
+      if (!Array.isArray(line.turns))
+        return { ok: false, reason: `threads[${index}].turns must be an array` };
+      const turns = [];
+      for (const [turn, batch] of line.turns.entries()) {
+        if (!Array.isArray(batch))
+          return { ok: false, reason: `threads[${index}].turns[${turn}] must be an array` };
+        const steps = [];
+        for (const entry of batch) {
+          const step = parseStep(entry);
+          const last = steps.at(-1);
+          if (step === null || "expectInput" in step && steps.length > 0 || last !== void 0 && "awaitSteer" in last)
+            return { ok: false, reason: `threads[${index}].turns[${turn}] has an invalid step` };
+          steps.push(step);
+        }
+        turns.push(steps);
+      }
+      threads.push({
+        match: {
+          ...typeof match.threadId === "string" ? { threadId: new RegExp(match.threadId) } : {},
+          ...typeof match.title === "string" ? { title: match.title } : {}
+        },
+        turns
+      });
+    }
+    return { ok: true, threads };
+  } catch (error) {
+    return { ok: false, reason: String(error) };
+  }
+}
 function pickBatch(line, consumed, input) {
   let earliest = true;
   for (const [index, batch] of line.turns.entries()) {
@@ -1143,25 +1150,38 @@ var waitForTool = (step, input, clients) => poll(step.wait.timeoutMs, input, asy
     (tool) => isRecord2(tool) && (tool.name === step.wait.tool || tool.name === `mcp__${step.wait.server ?? "ycodium"}__${step.wait.tool}`)
   );
 });
+async function readCount(step, args, clients, signal) {
+  const called = await clients.request(
+    step.waitFor.server,
+    "tools/call",
+    { name: step.waitFor.tool, arguments: args },
+    signal
+  );
+  if (isRecord2(called) && called.isError === true) return void 0;
+  const result = stepResult(called);
+  const value = valueAt(result, splitPath(step.waitFor.path));
+  const count = Array.isArray(value) ? value.length : typeof value === "number" ? value : value === void 0 || value === null ? 0 : 1;
+  return count < step.waitFor.atLeast ? void 0 : { result };
+}
 async function waitForCount(step, args, input, clients) {
   let ready;
   const ok = await poll(step.waitFor.timeoutMs, input, async (signal) => {
-    const called = await clients.request(
-      step.waitFor.server,
-      "tools/call",
-      { name: step.waitFor.tool, arguments: args },
-      signal
-    );
-    if (isRecord2(called) && called.isError === true) return false;
-    const result = stepResult(called);
-    const value = valueAt(result, splitPath(step.waitFor.path));
-    const count = Array.isArray(value) ? value.length : typeof value === "number" ? value : value === void 0 || value === null ? 0 : 1;
-    if (count < step.waitFor.atLeast) return false;
-    ready = { result };
-    return true;
+    ready = await readCount(step, args, clients, signal);
+    return ready !== void 0;
   });
   return ok ? ready : void 0;
 }
+var REQUIRED_WAIT_TOOLS = /* @__PURE__ */ new Set([
+  "work_history",
+  "work_list_comments",
+  "work_list_items"
+]);
+var GATE_PROBE_TIMEOUT_MS = 5e3;
+var REQUIRED_WAIT_FAILED = "playbook required wait failed:";
+function turnGateOf(batch) {
+  const first = batch.find((step) => !("expectInput" in step));
+  return first !== void 0 && "waitFor" in first && first.waitFor.turnGate === true ? first : void 0;
+}
 async function runCall(step, playback, input, clients) {
   let args = step.arguments ?? {};
   let failure;
@@ -1251,22 +1271,21 @@ function createPlaybookRunner(input) {
       if (playback === void 0) {
         const title = titles.get(turn.threadId);
         const index = parsed.threads.findIndex(
-          (line, index2) => !claimed.has(index2) && lineClaimsSession(line, turn.threadId, title)
+          (line2, index2) => !claimed.has(index2) && lineClaimsSession(line2, turn.threadId, title)
         );
         if (index >= 0) claimed.add(index);
         playback = {
           line: parsed.threads[index] ?? null,
           consumed: /* @__PURE__ */ new Set(),
+          owed: /* @__PURE__ */ new Set(),
           results: /* @__PURE__ */ new Map(),
           history: []
         };
         sessions.set(turn.threadId, playback);
       }
-      const picked = playback.line === null ? void 0 : pickBatch(playback.line, playback.consumed, turn.input ?? "");
-      const batch = picked === void 0 ? void 0 : playback.line?.turns[picked];
-      if (picked === void 0 || batch === void 0) return null;
-      playback.consumed.add(picked);
-      playback.current = picked;
+      const { line } = playback;
+      const picked = line === null ? void 0 : pickBatch(line, playback.consumed, turn.input ?? "");
+      if (picked === void 0 && playback.owed.size === 0) return null;
       let opened = clients.get(turn.threadId);
       if (opened === void 0) {
         opened = createPlaybookMcp({ fetch: turn.fetch, servers: turn.mcpServers });
@@ -1278,16 +1297,75 @@ function createPlaybookRunner(input) {
           turn.emit(args[0], args[1], redactMcpUrls(args[2], turn.mcpServers), args[3]);
         }
       };
+      const gateMet = async (gate) => {
+        try {
+          const args = resolveArguments(
+            gate.waitFor.arguments ?? {},
+            turn.threadId,
+            playback.results,
+            playback.history
+          );
+          const signal = AbortSignal.any([
+            turn.signal,
+            AbortSignal.timeout(Math.min(GATE_PROBE_TIMEOUT_MS, gate.waitFor.timeoutMs))
+          ]);
+          return await readCount(gate, args, opened, signal) !== void 0;
+        } catch {
+          return false;
+        }
+      };
       const lines = [];
-      const queue = [...batch];
-      for (let step = queue.shift(); step !== void 0; step = queue.shift()) {
+      const queue = [];
+      const pickedBatch = picked === void 0 ? void 0 : line?.turns[picked];
+      if (picked !== void 0 && pickedBatch !== void 0) {
+        const gate = turnGateOf(pickedBatch);
+        const aheadPending = [...line?.turns.keys() ?? []].some(
+          (index) => index < picked && !playback.consumed.has(index)
+        );
+        if (gate !== void 0 && aheadPending && !await gateMet(gate)) {
+          if (turn.signal.aborted) return "";
+          playback.owed.add(picked);
+          return `mcp-send deferred batch ${String(picked)}: ${gate.waitFor.tool} ${gate.waitFor.path} < ${String(gate.waitFor.atLeast)}`;
+        }
+        playback.consumed.add(picked);
+        playback.owed.delete(picked);
+        playback.current = picked;
+        queue.push(...pickedBatch);
+      }
+      let ran = pickedBatch !== void 0;
+      let halted = false;
+      const next = async () => {
+        while (queue.length === 0 && !halted && !turn.signal.aborted) {
+          const [owed] = [...playback.owed].sort((left, right) => left - right);
+          const batch = owed === void 0 ? void 0 : line?.turns[owed];
+          if (owed === void 0 || batch === void 0) break;
+          const gate = turnGateOf(batch);
+          if (gate !== void 0 && !await gateMet(gate)) break;
+          playback.owed.delete(owed);
+          playback.consumed.add(owed);
+          playback.current = owed;
+          queue.push(...batch);
+          ran = true;
+        }
+        return queue.shift();
+      };
+      const haltRequired = (step, reason) => {
+        const { tool, path: path2, atLeast } = step.waitFor;
+        const who = line?.match.title ?? line?.match.threadId?.source ?? "";
+        lines.push(
+          `${REQUIRED_WAIT_FAILED} session ${turn.threadId} line ${who} batch ${String(playback.current)}: ${tool} ${path2} >= ${String(atLeast)} ${reason}`
+        );
+        halted = true;
+        queue.length = 0;
+      };
+      for (let step = await next(); step !== void 0; step = await next()) {
         if (turn.signal.aborted) break;
         if ("expectInput" in step) continue;
         if ("awaitSteer" in step) {
           const ready = await turn.awaitSteer?.(step.awaitSteer.timeoutMs, turn.signal) ?? false;
           if (turn.signal.aborted) break;
           lines.push(`mcp-send awaitSteer${ready ? ": ready" : " failed: timeout"}`);
-          const after = (playback.current ?? picked) + 1;
+          const after = (playback.current ?? -1) + 1;
           const following = playback.line?.turns[after];
           if (!ready && step.awaitSteer.continueOnTimeout === true && following !== void 0 && !playback.consumed.has(after)) {
             playback.consumed.add(after);
@@ -1304,9 +1382,12 @@ function createPlaybookRunner(input) {
           if (turn.signal.aborted) break;
           lines.push(`mcp-send wait ${step.wait.tool}${ready ? ": ready" : " failed: timeout"}`);
         } else if ("waitFor" in step) {
+          const { id, tool, path: path2, atLeast } = step.waitFor;
+          const required = REQUIRED_WAIT_TOOLS.has(tool) && step.waitFor.continueOnTimeout !== true;
           const missing = opened.missing(step.waitFor.server);
           if (step.waitFor.server !== void 0 && missing !== void 0) {
-            lines.push(`mcp-send ${step.waitFor.tool} failed: ${missing}`);
+            lines.push(`mcp-send ${tool} failed: ${missing}`);
+            if (required) haltRequired(step, missing);
             continue;
           }
           let args;
@@ -1318,20 +1399,23 @@ function createPlaybookRunner(input) {
               playback.history
             );
           } catch (error) {
-            lines.push(`mcp-send ${step.waitFor.tool} failed: ${String(error)}`);
+            lines.push(`mcp-send ${tool} failed: ${String(error)}`);
+            if (required) haltRequired(step, String(error));
             continue;
           }
           const ready = await waitForCount(step, args, safeTurn, opened);
           if (turn.signal.aborted) break;
-          const { id, tool, path: path2, atLeast } = step.waitFor;
           if (ready !== void 0 && id !== void 0) playback.results.set(id, ready.result);
           lines.push(
             `mcp-send waitFor ${tool} ${path2} >= ${String(atLeast)}${ready ? ": ready" : " failed: timeout"}`
           );
+          if (ready === void 0 && required)
+            haltRequired(step, `not reached within ${String(step.waitFor.timeoutMs)} ms`);
         } else {
           lines.push(await runCall(step, playback, safeTurn, opened));
         }
       }
+      if (!ran) return null;
       return redactMcpUrls(lines.join("\n"), turn.mcpServers);
     }
   };
@@ -1580,7 +1664,6 @@ var FAKE_MODELS = [
   { slug: "fake-large", name: "Fake Large" }
 ];
 var createFakeInstance = (config, host) => {
-  const events = new EventBus();
   const threads = /* @__PURE__ */ new Map();
   const script = parseScript(config);
   const playbook = script === "mcp-send" ? createPlaybookRunner(config) : void 0;
@@ -1611,22 +1694,12 @@ var createFakeInstance = (config, host) => {
   let disposed = false;
   const nextId = (prefix) => `${prefix}-fake-${++seq}-${NodeCrypto.randomUUID()}`;
   const now = () => (/* @__PURE__ */ new Date()).toISOString();
-  const emit = (type, threadId, payload, extra) => {
-    events.push({
-      eventId: nextId("evt"),
-      provider: FAKE_DRIVER_KIND,
-      providerInstanceId: host.instanceId,
-      threadId,
-      createdAt: now(),
-      type,
-      payload,
-      ...extra?.turnId === void 0 ? {} : { turnId: extra.turnId },
-      ...extra?.itemId === void 0 ? {} : { itemId: extra.itemId },
-      ...extra?.requestId === void 0 ? {} : { requestId: extra.requestId },
-      ...extra?.originBindingGeneration === void 0 ? {} : { originBindingGeneration: extra.originBindingGeneration },
-      ...extra?.originResumeCursor === void 0 ? {} : { originResumeCursor: extra.originResumeCursor }
-    });
-  };
+  const emit = createFakeEmitter({
+    thread: (threadId) => threads.get(threadId),
+    nextId: () => nextId("evt"),
+    now,
+    instanceId: host.instanceId
+  });
   const lifecycle = createFakeSessionLifecycle({
     threads,
     abort: (threadId) => inflight.get(threadId)?.abort(),
@@ -1634,19 +1707,17 @@ var createFakeInstance = (config, host) => {
     // 代次与续接游标取被关闭的那份 session，不在发送时重查当前会话。
     exited: (session) => {
       if (stopWithoutExit) return;
+      const context = threads.get(session.threadId);
+      if (context) settleFakeTasks(context, session.threadId, emit);
       emit(
         "session.exited",
         session.threadId,
         { reason: "stopped" },
-        {
-          ...session.bindingGeneration === void 0 ? {} : { originBindingGeneration: session.bindingGeneration },
-          ...session.resumeCursor === void 0 ? {} : { originResumeCursor: session.resumeCursor }
-        }
+        session.resumeCursor === void 0 ? {} : { originResumeCursor: session.resumeCursor }
       );
     },
     forget: (threadId) => steeredInputs.delete(threadId),
-    disposePlaybook: () => playbook?.dispose(),
-    closeEvents: () => events.close()
+    disposePlaybook: () => playbook?.dispose()
   });
   const requireThread = (threadId) => {
     const booked = threads.get(threadId);
@@ -1869,6 +1940,8 @@ ${more ?? "playbook: steered input not handled"}`;
       if (signal?.aborted) throw new Error("aborted");
       if (threads.get(input.threadId)?.closing !== void 0) throw new Error("session closing");
       if (failStartSession !== void 0) throw new Error(failStartSession);
+      const previous = threads.get(input.threadId);
+      if (previous) await lifecycle.closeBooked(input.threadId, previous);
       const cwd = input.cwd ?? host.workspacePath(input.threadId);
       if (cwd.trim() === "") throw new Error("workspace path missing");
       const createdAt = now();
@@ -1884,7 +1957,13 @@ ${more ?? "playbook: steered input not handled"}`;
         ...input.discardResumeCursor === true || input.resumeCursor === void 0 ? {} : { resumeCursor: input.resumeCursor },
         ...input.bindingGeneration === void 0 ? {} : { bindingGeneration: input.bindingGeneration }
       };
-      threads.set(input.threadId, { session, turns: [], mcpServers: input.mcpServers });
+      threads.set(input.threadId, {
+        tasks: /* @__PURE__ */ new Map(),
+        events: input.events,
+        session,
+        turns: [],
+        mcpServers: input.mcpServers
+      });
       playbook?.startSession(input);
       emit(
         "session.started",
@@ -1974,7 +2053,6 @@ ${more ?? "playbook: steered input not handled"}`;
       const booked = requireThread(threadId);
       return { threadId, turns: booked.turns };
     },
-    events,
     usage: createFakeUsage({ instanceId: host.instanceId, script: parseUsageScript(config) }),
     dispose: async () => {
       disposed = true;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@ycodium-ai/agent-fake",
-  "version": "0.2.3976",
+  "version": "0.2.3990",
   "description": "Minimal fake executor for plugin authors and host orchestration tests.",
   "keywords": [
     "ycodium",
@@ -21,7 +21,7 @@
     "@types/node": "24.12.4",
     "esbuild": "0.28.1",
     "vite-plus": "0.2.2",
-    "@ycodium-ai/plugin-api": "0.2.3976",
+    "@ycodium-ai/plugin-api": "0.2.3990",
     "ycodium-executor-testing": "0.0.0"
   },
   "ycodium": {