@tt-a1i/openpi 0.4.0 → 0.5.0

package/extensions/subagents/src/backends/stub.ts

@@ -1,303 +0,0 @@
-/**
- * Scripted stub sessions, test-only: they stand in for the pi backend so the
- * manager can be driven without a real child session. A stub session:
- *
- * - streams a plausible turn (thinking deltas, one fake tool cycle, text
- *   deltas, usage ramp, a final assistant message, RunSettled) over a few
- *   seconds so streaming UI, wait, and the footer counters are observable;
- * - supports send() while running (queued-steer rendering) and while idle
- *   (fresh run);
- * - supports interrupt (RunSettled Interrupted -> status "error", matching v1);
- * - fails the run when the prompt starts with "FAIL:", and refuses to spawn
- *   at all when it starts with "SPAWNFAIL:" (error-path testing);
- * - hangs the run after RunStarted without any assistant event when the
- * prompt starts with "HANG:" (first-response watchdog testing);
- * - appends every event to a JSONL "session file" in tmpdir so the
- *   "full transcript in session file" pointers resolve.
- */
-
-import * as fs from "node:fs";
-import * as os from "node:os";
-import * as path from "node:path";
-import type { Cause, Scope } from "effect";
-import { Duration, Effect, Fiber, Queue, Ref, Stream } from "effect";
-import type { SubagentBackend, SubagentSession } from "../backend.ts";
-import type {
-  BackendName,
-  QueuedMessage,
-  SpawnTask,
-  SubagentEvent,
-  SubagentMeta,
-} from "../domain.ts";
-import { SendError, SpawnError } from "../domain.ts";
-
-export interface StubProfile {
-  readonly backend: BackendName;
-  readonly defaultModelLabel: string;
-  readonly contextWindow: number;
-  readonly toolName: string;
-  /** Delay between scripted events; varies per backend so streams differ. */
-  readonly cadenceMs: number;
-}
-
-const STUB_DIR = path.join(os.tmpdir(), "subagents-stub");
-let sessionCounter = 0;
-
-export function makeStubBackend(profile: StubProfile): SubagentBackend {
-  return {
-    name: profile.backend,
-    spawn: (task) =>
-      task.prompt.startsWith("SPAWNFAIL:")
-        ? Effect.fail(new SpawnError({ message: "stub refused to spawn" }))
-        : makeStubSession(profile, task),
-  };
-}
-
-function firstLine(text: string): string {
-  return (
-    text
-      .split("\n")
-      .find((line) => line.trim())
-      ?.trim() ?? ""
-  );
-}
-
-function chunked(text: string, size: number): string[] {
-  const chunks: string[] = [];
-  for (let i = 0; i < text.length; i += size)
-    chunks.push(text.slice(i, i + size));
-  return chunks;
-}
-
-const makeStubSession = (
-  profile: StubProfile,
-  task: SpawnTask,
-): Effect.Effect<SubagentSession, never, Scope.Scope> =>
-  Effect.gen(function* () {
-    const sessionId = `stub-${profile.backend}-${++sessionCounter}`;
-    const sessionFile = path.join(STUB_DIR, `${sessionId}.jsonl`);
-
-    const state = {
-      meta: {
-        backend: profile.backend,
-        modelLabel: task.model ?? profile.defaultModelLabel,
-        contextWindow: profile.contextWindow,
-        sessionFilePath: sessionFile,
-      } satisfies SubagentMeta as SubagentMeta,
-      pending: [] as string[],
-      turnCount: 0,
-      closed: false,
-      /** True between the driver dequeuing a prompt and registering its turn fiber. */
-      dispatching: false,
-    };
-
-    const events = yield* Queue.make<SubagentEvent, Cause.Done>();
-    const inbox = yield* Queue.make<string, Cause.Done>();
-    const activeTurn = yield* Ref.make<Fiber.Fiber<void> | undefined>(
-      undefined,
-    );
-
-    const emit = (event: SubagentEvent) =>
-      Effect.suspend(() => {
-        try {
-          fs.appendFileSync(sessionFile, `${JSON.stringify(event)}\n`);
-        } catch {
-          // The fake session file is best-effort.
-        }
-        if (event._tag === "MetaChanged") {
-          state.meta = { ...state.meta, ...event.meta };
-        }
-        return Queue.offer(events, event);
-      }).pipe(Effect.asVoid);
-
-    const pause = Effect.sleep(Duration.millis(profile.cadenceMs));
-
-    const runTurn = (userText: string, turn: number) =>
-      Effect.gen(function* () {
-        yield* emit({ _tag: "RunStarted" });
-        const failing = userText.trimStart().startsWith("FAIL:");
-        // Stand in for a provider that accepts the request but never emits
-        // its first assistant event.
-        if (userText.trimStart().startsWith("HANG:")) {
-          return yield* Effect.never;
-        }
-
-        const thinking = "Looking at the task and planning an approach...";
-        for (const delta of chunked(thinking, 16)) {
-          yield* emit({ _tag: "AssistantDelta", kind: "thinking", delta });
-          yield* pause;
-        }
-
-        const toolId = `${sessionId}-tool-${turn}`;
-        const argsPreview = `{"command":"ls ${task.cwd}"}`;
-        yield* emit({
-          _tag: "AssistantMessage",
-          parts: [
-            { type: "thinking", text: thinking },
-            {
-              type: "text",
-              text: `I'll run ${profile.toolName} to look around first.`,
-            },
-            { type: "toolCall", toolId, name: profile.toolName, argsPreview },
-          ],
-        });
-        yield* emit({
-          _tag: "ToolStart",
-          toolId,
-          name: profile.toolName,
-          argsPreview,
-        });
-        yield* pause;
-        yield* emit({
-          _tag: "ToolUpdate",
-          toolId,
-          outputPreview: "src docs package.json",
-        });
-        yield* pause;
-        yield* emit({
-          _tag: "ToolEnd",
-          toolId,
-          name: profile.toolName,
-          isError: false,
-          outputPreview: "src docs package.json",
-        });
-        yield* emit({
-          _tag: "UsageChanged",
-          tokens: Math.min(profile.contextWindow, 2400 * (turn + 1)),
-          contextWindow: profile.contextWindow,
-        });
-
-        if (failing) {
-          yield* pause;
-          yield* emit({
-            _tag: "RunSettled",
-            outcome: {
-              _tag: "Failed",
-              errorText: `[stub:${profile.backend}] task failed as requested by FAIL: prefix`,
-            },
-          });
-          return;
-        }
-
-        const finalText =
-          `[stub:${profile.backend}] completed: ${firstLine(userText).slice(0, 200)}\n\n` +
-          `This is a stubbed ${profile.backend} subagent turn ${turn + 1}. ` +
-          `The real backend integration will replace this scripted output.`;
-        for (const delta of chunked(finalText, 24)) {
-          yield* emit({ _tag: "AssistantDelta", kind: "text", delta });
-          yield* pause;
-        }
-        yield* emit({
-          _tag: "AssistantMessage",
-          parts: [{ type: "text", text: finalText }],
-        });
-        yield* emit({
-          _tag: "UsageChanged",
-          tokens: Math.min(profile.contextWindow, 2400 * (turn + 1) + 900),
-          contextWindow: profile.contextWindow,
-        });
-        yield* emit({
-          _tag: "RunSettled",
-          outcome: { _tag: "Completed", finalText },
-        });
-      });
-
-    const queuedView = (): ReadonlyArray<QueuedMessage> =>
-      state.pending.map((text) => ({ text, kind: "steer" as const }));
-
-    // Driver: one turn at a time, in submission order. Turns run as child
-    // fibers so interrupt() stops the turn without killing the driver.
-    const driver = Effect.gen(function* () {
-      while (true) {
-        const text = yield* Queue.take(inbox);
-        state.dispatching = true;
-        state.pending.shift();
-        yield* emit({ _tag: "QueueChanged", queued: queuedView() });
-        yield* emit({ _tag: "UserMessage", text });
-        const turn = state.turnCount++;
-        const fiber = yield* Effect.forkChild(
-          runTurn(text, turn).pipe(
-            Effect.onInterrupt(() =>
-              emit({
-                _tag: "RunSettled",
-                outcome: { _tag: "Interrupted" },
-              }).pipe(Effect.ignore),
-            ),
-          ),
-        );
-        yield* Ref.set(activeTurn, fiber);
-        state.dispatching = false;
-        yield* Fiber.await(fiber);
-        yield* Ref.set(activeTurn, undefined);
-      }
-    });
-    yield* Effect.forkScoped(driver.pipe(Effect.ignore));
-
-    yield* Effect.addFinalizer(() =>
-      Effect.gen(function* () {
-        state.closed = true;
-        yield* Queue.end(inbox).pipe(Effect.ignore);
-        yield* Queue.end(events).pipe(Effect.ignore);
-      }),
-    );
-
-    const submit = (text: string) =>
-      Effect.gen(function* () {
-        if (state.closed) {
-          return yield* new SendError({
-            message: "Subagent session is closed.",
-          });
-        }
-        state.pending.push(text);
-        const busy = (yield* Ref.get(activeTurn)) !== undefined;
-        if (busy) {
-          // Show the queued steer line until the driver picks it up.
-          yield* emit({ _tag: "QueueChanged", queued: queuedView() });
-        }
-        yield* Queue.offer(inbox, text);
-      });
-
-    // Announce metadata, then kick off the initial run.
-    yield* Effect.try(() => fs.mkdirSync(STUB_DIR, { recursive: true })).pipe(
-      Effect.ignore, // The fake session file directory is best-effort.
-    );
-    yield* emit({ _tag: "MetaChanged", meta: state.meta });
-    // The session cannot be closed yet, so the initial submit cannot fail.
-    yield* submit(task.prompt).pipe(Effect.orDie);
-
-    return {
-      meta: Effect.sync(() => state.meta),
-      events: Stream.fromQueue(events),
-      send: submit,
-      interrupt: Effect.gen(function* () {
-        // Drop queued prompts so interrupting cannot immediately start
-        // another turn, then stop the active turn. A prompt may be mid-flight
-        // between the driver dequeuing it and registering its fiber, so wait
-        // that window out instead of silently missing the turn.
-        const cleared = yield* Queue.clear(inbox).pipe(
-          Effect.orElseSucceed(() => []),
-        );
-        state.pending = [];
-        yield* emit({ _tag: "QueueChanged", queued: [] });
-        while (true) {
-          const fiber = yield* Ref.get(activeTurn);
-          if (fiber) {
-            yield* Fiber.interrupt(fiber);
-            return;
-          }
-          if (!state.dispatching) {
-            // No turn ever started. If we cancelled queued prompts, the run
-            // still needs a terminal event or it would look running forever.
-            if (cleared.length > 0) {
-              yield* emit({
-                _tag: "RunSettled",
-                outcome: { _tag: "Interrupted" },
-              });
-            }
-            return;
-          }
-          yield* Effect.sleep(Duration.millis(5));
-        }
-      }),
-    } satisfies SubagentSession;
-  });