@ai-devkit/agent-manager 0.32.0 → 0.33.0

package/src/providers/pi/durable/PiPrintAgentService.ts

@@ -1,69 +1,77 @@
-import type { DurableAgent, ProcessIdentity } from '../../../durable/DurableAgent.js';
-import { PiPrintError } from '../../../durable/DurableAgent.js';
-import { PiCliProbe } from './PiCliProbe.js';
-import { PiPrintRunner, type PiPrintRunResult } from './PiPrintRunner.js';
-import { DurableAgentRepository, type CreateDurableAgentInput } from '../../../durable/DurableAgentRepository.js';
-import { runDurableAgent, type DurableRunRepository } from '../../../durable/run.js';
+import type { DurableAgent, ProcessIdentity } from "../../../durable/DurableAgent.js";
+import { PiPrintError } from "../../../durable/DurableAgent.js";
+import { PiCliProbe } from "./PiCliProbe.js";
+import { PiPrintRunner, type PiPrintRunResult } from "./PiPrintRunner.js";
+import {
+  DurableAgentRepository,
+  type CreateDurableAgentInput,
+} from "../../../durable/DurableAgentRepository.js";
+import { runDurableAgent, type DurableRunRepository } from "../../../durable/run.js";
 
 interface RepositoryLike extends DurableRunRepository {
-    create(input: CreateDurableAgentInput): Promise<DurableAgent>;
-    list(): Promise<DurableAgent[]>;
-    recordProviderProcess(id: string, token: string, identity: ProcessIdentity): Promise<void>;
+  create(input: CreateDurableAgentInput): Promise<DurableAgent>;
+  list(): Promise<DurableAgent[]>;
+  recordProviderProcess(id: string, token: string, identity: ProcessIdentity): Promise<void>;
 }
 
-interface ProbeLike { validate(): Promise<{ executable: string; version: string }> }
-interface RunnerLike { run(request: Parameters<PiPrintRunner['run']>[0]): Promise<PiPrintRunResult> }
+interface ProbeLike {
+  validate(): Promise<{ executable: string; version: string }>;
+}
+interface RunnerLike {
+  run(request: Parameters<PiPrintRunner["run"]>[0]): Promise<PiPrintRunResult>;
+}
 
 export interface PiPrintAgentServiceOptions {
-    repository?: RepositoryLike;
-    probe?: ProbeLike;
-    runner?: RunnerLike;
-    executable?: string;
+  repository?: RepositoryLike;
+  probe?: ProbeLike;
+  runner?: RunnerLike;
+  executable?: string;
 }
 
 export interface PiPrintSendResult extends PiPrintRunResult {
-    agentId: string;
-    agentName: string;
+  agentId: string;
+  agentName: string;
 }
 
 export class PiPrintAgentService {
-    readonly repository: RepositoryLike;
-    private readonly probe: ProbeLike;
-    private readonly runner: RunnerLike;
-    private readonly executable?: string;
+  readonly repository: RepositoryLike;
+  private readonly probe: ProbeLike;
+  private readonly runner: RunnerLike;
+  private readonly executable?: string;
 
-    constructor(options: PiPrintAgentServiceOptions = {}) {
-        this.repository = options.repository ?? new DurableAgentRepository();
-        this.probe = options.probe ?? new PiCliProbe();
-        this.runner = options.runner ?? new PiPrintRunner();
-        this.executable = options.executable;
-    }
+  constructor(options: PiPrintAgentServiceOptions = {}) {
+    this.repository = options.repository ?? new DurableAgentRepository();
+    this.probe = options.probe ?? new PiCliProbe();
+    this.runner = options.runner ?? new PiPrintRunner();
+    this.executable = options.executable;
+  }
 
-    async create(input: Omit<CreateDurableAgentInput, 'provider'>): Promise<DurableAgent> {
-        await this.probe.validate();
-        return this.repository.create({ ...input, provider: 'pi' });
-    }
+  async create(input: Omit<CreateDurableAgentInput, "provider">): Promise<DurableAgent> {
+    await this.probe.validate();
+    return this.repository.create({ ...input, provider: "pi" });
+  }
 
-    async send(reference: string, prompt: string): Promise<PiPrintSendResult> {
-        const completed = await runDurableAgent({
-            reference,
-            provider: 'pi',
-            repository: this.repository,
-            ambiguousError: () => new PiPrintError('Multiple print agents match.', 'PI_UNSUPPORTED'),
-            providerError: () => new PiPrintError('Print agent provider is not Pi.', 'PI_UNSUPPORTED'),
-            execute: (agent, token) => this.runner.run({
-                agent,
-                prompt,
-                executable: this.executable,
-                onSpawn: (identity) => this.repository.recordProviderProcess(agent.id, token, identity),
-            }),
-            isSessionMismatch: (error) => error instanceof PiPrintError
-                && error.code === 'PI_SESSION_MISMATCH',
-        });
-        return {
-            ...completed.result,
-            agentId: completed.agent.id,
-            agentName: completed.agent.name,
-        };
-    }
+  async send(reference: string, prompt: string): Promise<PiPrintSendResult> {
+    const completed = await runDurableAgent({
+      reference,
+      provider: "pi",
+      repository: this.repository,
+      ambiguousError: () => new PiPrintError("Multiple print agents match.", "PI_UNSUPPORTED"),
+      providerError: () => new PiPrintError("Print agent provider is not Pi.", "PI_UNSUPPORTED"),
+      execute: (agent, token) =>
+        this.runner.run({
+          agent,
+          prompt,
+          executable: this.executable,
+          onSpawn: (identity) => this.repository.recordProviderProcess(agent.id, token, identity),
+        }),
+      isSessionMismatch: (error) =>
+        error instanceof PiPrintError && error.code === "PI_SESSION_MISMATCH",
+    });
+    return {
+      ...completed.result,
+      agentId: completed.agent.id,
+      agentName: completed.agent.name,
+    };
+  }
 }

package/src/providers/pi/durable/PiPrintProtocol.ts

@@ -1,39 +1,41 @@
-import { PiPrintError } from '../../../durable/DurableAgent.js';
-import { isUuid } from '../../../durable/utils.js';
+import { PiPrintError } from "../../../durable/DurableAgent.js";
+import { isUuid } from "../../../durable/utils.js";
 
 export function buildPiPrintArgs(providerSessionId: string, firstRun: boolean): string[] {
-    return firstRun
-        ? ['--mode', 'json', '--session-id', providerSessionId]
-        : ['--mode', 'json', '--session', providerSessionId];
+  return firstRun
+    ? ["--mode", "json", "--session-id", providerSessionId]
+    : ["--mode", "json", "--session", providerSessionId];
 }
 
 export function readPiSessionId(
-    event: Record<string, unknown>,
-    currentSessionId: string | null,
-    expectedSessionId: string,
+  event: Record<string, unknown>,
+  currentSessionId: string | null,
+  expectedSessionId: string,
 ): string {
-    if (currentSessionId !== null || typeof event.id !== 'string' || !isUuid(event.id)) {
-        throw new PiPrintError('Pi emitted an invalid session identity.', 'PI_PROTOCOL');
-    }
-    const sessionId = event.id;
-    if (expectedSessionId !== sessionId) {
-        throw new PiPrintError('Pi returned a different session identity.', 'PI_SESSION_MISMATCH');
-    }
-    return sessionId;
+  if (currentSessionId !== null || typeof event.id !== "string" || !isUuid(event.id)) {
+    throw new PiPrintError("Pi emitted an invalid session identity.", "PI_PROTOCOL");
+  }
+  const sessionId = event.id;
+  if (expectedSessionId !== sessionId) {
+    throw new PiPrintError("Pi returned a different session identity.", "PI_SESSION_MISMATCH");
+  }
+  return sessionId;
 }
 
 export function readPiAssistantText(event: Record<string, unknown>): string | null {
-    if (event.type !== 'message_end') return null;
-    const message = event.message;
-    if (!message || typeof message !== 'object' || Array.isArray(message)) return null;
-    const record = message as Record<string, unknown>;
-    if (record.role !== 'assistant') return null;
-    if (typeof record.content === 'string') return record.content.trim() ? record.content : null;
-    if (!Array.isArray(record.content)) return null;
-    const text = record.content.flatMap((part) => {
-        if (!part || typeof part !== 'object' || Array.isArray(part)) return [];
-        const block = part as Record<string, unknown>;
-        return block.type === 'text' && typeof block.text === 'string' ? [block.text] : [];
-    }).join('');
-    return text.trim() ? text : null;
+  if (event.type !== "message_end") return null;
+  const message = event.message;
+  if (!message || typeof message !== "object" || Array.isArray(message)) return null;
+  const record = message as Record<string, unknown>;
+  if (record.role !== "assistant") return null;
+  if (typeof record.content === "string") return record.content.trim() ? record.content : null;
+  if (!Array.isArray(record.content)) return null;
+  const text = record.content
+    .flatMap((part) => {
+      if (!part || typeof part !== "object" || Array.isArray(part)) return [];
+      const block = part as Record<string, unknown>;
+      return block.type === "text" && typeof block.text === "string" ? [block.text] : [];
+    })
+    .join("");
+  return text.trim() ? text : null;
 }

package/src/providers/pi/durable/PiPrintRunner.ts

@@ -1,88 +1,95 @@
-import { spawn as nodeSpawn, type ChildProcessWithoutNullStreams, type SpawnOptionsWithoutStdio } from 'child_process';
-import type { PiDurableAgent, ProcessIdentity } from '../../../durable/DurableAgent.js';
-import { PiPrintError } from '../../../durable/DurableAgent.js';
-import { LocalProcessInspector, type ProcessInspector } from '../../../durable/process.js';
-import { waitForChildClose } from '../../../durable/utils.js';
-import { buildPiPrintArgs } from './PiPrintProtocol.js';
-import { PiStreamParser } from './PiStreamParser.js';
+import {
+  spawn as nodeSpawn,
+  type ChildProcessWithoutNullStreams,
+  type SpawnOptionsWithoutStdio,
+} from "child_process";
+import type { PiDurableAgent, ProcessIdentity } from "../../../durable/DurableAgent.js";
+import { PiPrintError } from "../../../durable/DurableAgent.js";
+import { LocalProcessInspector, type ProcessInspector } from "../../../durable/process.js";
+import { waitForChildClose } from "../../../durable/utils.js";
+import { buildPiPrintArgs } from "./PiPrintProtocol.js";
+import { PiStreamParser } from "./PiStreamParser.js";
 
 type Spawn = (
-    command: string,
-    args: readonly string[],
-    options: SpawnOptionsWithoutStdio & { stdio: ['pipe', 'pipe', 'pipe'] },
+  command: string,
+  args: readonly string[],
+  options: SpawnOptionsWithoutStdio & { stdio: ["pipe", "pipe", "pipe"] },
 ) => ChildProcessWithoutNullStreams;
 
 export interface PiPrintRunRequest {
-    agent: PiDurableAgent;
-    prompt: string;
-    executable?: string;
-    onSpawn(identity: ProcessIdentity): Promise<void>;
+  agent: PiDurableAgent;
+  prompt: string;
+  executable?: string;
+  onSpawn(identity: ProcessIdentity): Promise<void>;
 }
 
 export interface PiPrintRunResult {
-    sessionId: string;
-    result: string;
-    messages: string[];
-    exitCode: number;
+  sessionId: string;
+  result: string;
+  messages: string[];
+  exitCode: number;
 }
 
 export interface PiPrintRunnerOptions {
-    spawn?: Spawn;
-    processInspector?: ProcessInspector;
-    maxLineBytes?: number;
+  spawn?: Spawn;
+  processInspector?: ProcessInspector;
+  maxLineBytes?: number;
 }
 
 export class PiPrintRunner {
-    private readonly spawn: Spawn;
-    private readonly processInspector: ProcessInspector;
-    private readonly maxLineBytes: number;
+  private readonly spawn: Spawn;
+  private readonly processInspector: ProcessInspector;
+  private readonly maxLineBytes: number;
 
-    constructor(options: PiPrintRunnerOptions = {}) {
-        this.spawn = options.spawn ?? (nodeSpawn as Spawn);
-        this.processInspector = options.processInspector ?? new LocalProcessInspector();
-        this.maxLineBytes = options.maxLineBytes ?? 1024 * 1024;
-    }
-
-    async run(request: PiPrintRunRequest): Promise<PiPrintRunResult> {
-        const args = buildPiPrintArgs(request.agent.providerSessionId, request.agent.sessionHealth === 'uninitialized');
-        const child = this.spawn(request.executable ?? 'pi', args, {
-            cwd: request.agent.cwd,
-            shell: false,
-            stdio: ['pipe', 'pipe', 'pipe'],
-        });
-        if (!child.pid) {
-            child.kill();
-            throw new PiPrintError('Pi process did not provide a PID.', 'PI_PROCESS');
-        }
-        const identity = this.processInspector.getIdentity(child.pid);
-        if (!identity) {
-            child.kill();
-            throw new PiPrintError('Cannot verify Pi process identity.', 'PI_PROCESS');
-        }
+  constructor(options: PiPrintRunnerOptions = {}) {
+    this.spawn = options.spawn ?? (nodeSpawn as Spawn);
+    this.processInspector = options.processInspector ?? new LocalProcessInspector();
+    this.maxLineBytes = options.maxLineBytes ?? 1024 * 1024;
+  }
 
-        const parser = new PiStreamParser(request.agent.providerSessionId, this.maxLineBytes);
-        child.stdout.on('data', (chunk: Buffer | string) => {
-            if (parser.hasFailed()) return;
-            try {
-                parser.accept(chunk);
-            } catch (error) {
-                parser.fail(error);
-            }
-        });
-        child.stderr.resume();
+  async run(request: PiPrintRunRequest): Promise<PiPrintRunResult> {
+    const args = buildPiPrintArgs(
+      request.agent.providerSessionId,
+      request.agent.sessionHealth === "uninitialized",
+    );
+    const child = this.spawn(request.executable ?? "pi", args, {
+      cwd: request.agent.cwd,
+      shell: false,
+      stdio: ["pipe", "pipe", "pipe"],
+    });
+    if (!child.pid) {
+      child.kill();
+      throw new PiPrintError("Pi process did not provide a PID.", "PI_PROCESS");
+    }
+    const identity = this.processInspector.getIdentity(child.pid);
+    if (!identity) {
+      child.kill();
+      throw new PiPrintError("Cannot verify Pi process identity.", "PI_PROCESS");
+    }
 
-        const closed = waitForChildClose(child);
-        try {
-            await request.onSpawn(identity);
-        } catch (error) {
-            child.kill();
-            throw error;
-        }
+    const parser = new PiStreamParser(request.agent.providerSessionId, this.maxLineBytes);
+    child.stdout.on("data", (chunk: Buffer | string) => {
+      if (parser.hasFailed()) return;
+      try {
+        parser.accept(chunk);
+      } catch (error) {
+        parser.fail(error);
+      }
+    });
+    child.stderr.resume();
 
-        child.stdin.end(request.prompt);
-        const close = await closed.catch(() => {
-            throw new PiPrintError('Pi process failed to start or communicate.', 'PI_PROCESS');
-        });
-        return parser.result(close);
+    const closed = waitForChildClose(child);
+    try {
+      await request.onSpawn(identity);
+    } catch (error) {
+      child.kill();
+      throw error;
     }
+
+    child.stdin.end(request.prompt);
+    const close = await closed.catch(() => {
+      throw new PiPrintError("Pi process failed to start or communicate.", "PI_PROCESS");
+    });
+    return parser.result(close);
+  }
 }

package/src/providers/pi/durable/PiStreamParser.ts

@@ -1,84 +1,94 @@
-import { PiPrintError } from '../../../durable/DurableAgent.js';
-import { LineBuffer, type ChildCloseResult } from '../../../durable/utils.js';
-import { readPiAssistantText, readPiSessionId } from './PiPrintProtocol.js';
+import { PiPrintError } from "../../../durable/DurableAgent.js";
+import { LineBuffer, type ChildCloseResult } from "../../../durable/utils.js";
+import { readPiAssistantText, readPiSessionId } from "./PiPrintProtocol.js";
 
 export class PiStreamParser {
-    private readonly lines: LineBuffer;
-    private sessionId: string | null = null;
-    private ended = false;
-    private readonly messages: string[] = [];
-    private failure: PiPrintError | null = null;
+  private readonly lines: LineBuffer;
+  private sessionId: string | null = null;
+  private ended = false;
+  private readonly messages: string[] = [];
+  private failure: PiPrintError | null = null;
 
-    constructor(
-        private readonly expectedSessionId: string,
-        maxLineBytes: number,
-    ) {
-        this.lines = new LineBuffer(
-            maxLineBytes,
-            () => new PiPrintError('Pi stream line exceeded the safety limit.', 'PI_PROTOCOL'),
-        );
-    }
+  constructor(
+    private readonly expectedSessionId: string,
+    maxLineBytes: number,
+  ) {
+    this.lines = new LineBuffer(
+      maxLineBytes,
+      () => new PiPrintError("Pi stream line exceeded the safety limit.", "PI_PROTOCOL"),
+    );
+  }
 
-    hasFailed(): boolean {
-        return this.failure !== null;
-    }
+  hasFailed(): boolean {
+    return this.failure !== null;
+  }
 
-    accept(chunk: Buffer | string): void {
-        for (const line of this.lines.accept(chunk)) {
-            if (line.length === 0) continue;
-            this.acceptLine(line);
-        }
+  accept(chunk: Buffer | string): void {
+    for (const line of this.lines.accept(chunk)) {
+      if (line.length === 0) continue;
+      this.acceptLine(line);
     }
+  }
 
-    fail(error: unknown): void {
-        this.failure = error instanceof PiPrintError
-            ? error
-            : new PiPrintError('Pi stream processing failed.', 'PI_PROTOCOL');
-    }
+  fail(error: unknown): void {
+    this.failure =
+      error instanceof PiPrintError
+        ? error
+        : new PiPrintError("Pi stream processing failed.", "PI_PROTOCOL");
+  }
 
-    result(close: ChildCloseResult): { sessionId: string; result: string; messages: string[]; exitCode: number } {
-        if (this.failure) throw this.failure;
-        this.lines.assertComplete(
-            () => new PiPrintError('Pi stream ended with incomplete JSON.', 'PI_PROTOCOL'),
-        );
-        if (close.code !== 0) {
-            throw new PiPrintError(`Pi print run failed${close.signal ? ` (${close.signal})` : '.'}`, 'PI_PROCESS');
-        }
-        if (this.sessionId === null) {
-            throw new PiPrintError('Pi stream ended without a session identity.', 'PI_PROTOCOL');
-        }
-        if (!this.ended) throw new PiPrintError('Pi stream ended before agent completion.', 'PI_PROTOCOL');
-        if (this.messages.length === 0) {
-            throw new PiPrintError('Pi stream ended without an assistant result.', 'PI_RESULT_MISSING');
-        }
-        return {
-            sessionId: this.sessionId,
-            result: this.messages.at(-1)!,
-            messages: this.messages,
-            exitCode: close.code,
-        };
+  result(close: ChildCloseResult): {
+    sessionId: string;
+    result: string;
+    messages: string[];
+    exitCode: number;
+  } {
+    if (this.failure) throw this.failure;
+    this.lines.assertComplete(
+      () => new PiPrintError("Pi stream ended with incomplete JSON.", "PI_PROTOCOL"),
+    );
+    if (close.code !== 0) {
+      throw new PiPrintError(
+        `Pi print run failed${close.signal ? ` (${close.signal})` : "."}`,
+        "PI_PROCESS",
+      );
+    }
+    if (this.sessionId === null) {
+      throw new PiPrintError("Pi stream ended without a session identity.", "PI_PROTOCOL");
     }
+    if (!this.ended)
+      throw new PiPrintError("Pi stream ended before agent completion.", "PI_PROTOCOL");
+    if (this.messages.length === 0) {
+      throw new PiPrintError("Pi stream ended without an assistant result.", "PI_RESULT_MISSING");
+    }
+    return {
+      sessionId: this.sessionId,
+      result: this.messages.at(-1)!,
+      messages: this.messages,
+      exitCode: close.code,
+    };
+  }
 
-    private acceptLine(line: Buffer): void {
-        let value: unknown;
-        try {
-            value = JSON.parse(line.toString('utf8'));
-        } catch {
-            throw new PiPrintError('Pi emitted malformed stream JSON.', 'PI_PROTOCOL');
-        }
-        if (!value || typeof value !== 'object' || Array.isArray(value)) {
-            throw new PiPrintError('Pi emitted a non-object stream message.', 'PI_PROTOCOL');
-        }
-        const event = value as Record<string, unknown>;
-        if (event.type === 'session') {
-            this.sessionId = readPiSessionId(event, this.sessionId, this.expectedSessionId);
-            return;
-        }
-        if (event.type === 'agent_end') {
-            this.ended = true;
-            return;
-        }
-        const text = readPiAssistantText(event);
-        if (text !== null) this.messages.push(text);
+  private acceptLine(line: Buffer): void {
+    let value: unknown;
+    try {
+      value = JSON.parse(line.toString("utf8"));
+    } catch {
+      throw new PiPrintError("Pi emitted malformed stream JSON.", "PI_PROTOCOL");
+    }
+    if (!value || typeof value !== "object" || Array.isArray(value)) {
+      throw new PiPrintError("Pi emitted a non-object stream message.", "PI_PROTOCOL");
+    }
+    const event = value as Record<string, unknown>;
+    if (event.type === "session") {
+      this.sessionId = readPiSessionId(event, this.sessionId, this.expectedSessionId);
+      return;
+    }
+    if (event.type === "agent_end") {
+      this.ended = true;
+      return;
     }
+    const text = readPiAssistantText(event);
+    if (text !== null) this.messages.push(text);
+  }
 }