pi-crew 0.9.53 → 0.9.55

package/src/runtime/live-control-realtime.ts

@@ -0,0 +1,56 @@
+import type { LiveAgentControlRequest } from "./live-agent-control.ts";
+
+export interface LiveControlRealtimeMessage {
+	type: "live-control";
+	version: 1;
+	request: LiveAgentControlRequest;
+}
+
+type Listener = (request: LiveAgentControlRequest) => void | Promise<void>;
+
+const listeners = new Set<Listener>();
+
+export function publishLiveControlRealtime(request: LiveAgentControlRequest): void {
+	for (const listener of [...listeners]) void listener(request);
+}
+
+export function subscribeLiveControlRealtime(listener: Listener): () => void {
+	listeners.add(listener);
+	return () => listeners.delete(listener);
+}
+
+export function liveControlRealtimeMessage(request: LiveAgentControlRequest): LiveControlRealtimeMessage {
+	return { type: "live-control", version: 1, request };
+}
+
+export function parseLiveControlRealtimeMessage(raw: unknown): LiveAgentControlRequest | undefined {
+	if (!raw || typeof raw !== "object" || Array.isArray(raw)) return undefined;
+	const message = raw as {
+		type?: unknown;
+		version?: unknown;
+		request?: unknown;
+	};
+	if (
+		message.type !== "live-control" ||
+		message.version !== 1 ||
+		!message.request ||
+		typeof message.request !== "object" ||
+		Array.isArray(message.request)
+	)
+		return undefined;
+	const request = message.request as Partial<LiveAgentControlRequest>;
+	return typeof request.id === "string" &&
+		typeof request.runId === "string" &&
+		typeof request.taskId === "string" &&
+		(request.operation === "steer" ||
+			request.operation === "follow-up" ||
+			request.operation === "stop" ||
+			request.operation === "resume") &&
+		typeof request.createdAt === "string"
+		? (request as LiveAgentControlRequest)
+		: undefined;
+}
+
+export function clearLiveControlRealtimeForTest(): void {
+	listeners.clear();
+}

package/src/runtime/live-extension-bridge.ts

@@ -0,0 +1,148 @@
+/**
+ * G5: Extension runner bridge for live-session workers.
+ *
+ * Bridges pi-crew's extension lifecycle with the Pi SDK session's
+ * extension runner. Verified against actual SDK API surface:
+ *
+ * Session methods:
+ *   - sendCustomMessage(message, options?)
+ *   - sendUserMessage(content, options?)
+ *   - getActiveToolNames() / setActiveToolsByName()
+ *   - getAllTools() / getToolDefinition()
+ *   - steer(text) / prompt(text, options?)
+ *   - abort()
+ *   - getContextUsage()
+ *   - bindExtensions()
+ *   - compact()
+ *   - getSessionStats()
+ *
+ * ExtensionRunner methods:
+ *   - initialize(apis, host)
+ *   - emit(event)
+ *   - hasHandlers(eventType)
+ *   - getAllRegisteredTools()
+ *   - onError(listener)
+ *   - shutdown()
+ *
+ * ExtensionContext actions (via registerTool):
+ *   - sendMessage / sendUserMessage / appendEntry / setLabel
+ *   - setActiveTools / getActiveTools / getAllTools
+ */
+
+export interface ExtensionBridgeApis {
+	sendMessage: (message: unknown, options?: Record<string, unknown>) => void;
+	sendUserMessage: (content: unknown, options?: Record<string, unknown>) => void;
+	appendEntry: (customType: string, data: unknown) => void;
+	setLabel: (targetId: string, label: string) => void;
+	getActiveTools: () => string[];
+	getAllTools: () => string[];
+	setActiveTools: (toolNames: string[]) => void;
+}
+
+export interface ExtensionHostApis {
+	getModel: () => unknown;
+	isIdle: () => boolean;
+	abort: () => void;
+	hasPendingMessages: () => boolean;
+	shutdown: () => void;
+	getContextUsage: () => unknown;
+	getSystemPrompt: () => string;
+}
+
+/**
+ * Pi SDK session-like object with the methods we need.
+ * Verified against actual `createAgentSession().session` prototype.
+ */
+interface PiSdkSession {
+	sendCustomMessage: (message: unknown, options?: Record<string, unknown>) => void;
+	sendUserMessage: (content: unknown, options?: Record<string, unknown>) => void;
+	getActiveToolNames: () => string[];
+	getAllTools: () => string[];
+	setActiveToolsByName: (toolNames: string[]) => void;
+	steer: (text: string) => Promise<void>;
+	prompt: (text: string, options?: Record<string, unknown>) => Promise<void>;
+	abort: () => void | Promise<void>;
+	getContextUsage: () => unknown;
+	subscribe: (listener: (event: unknown) => void) => () => void;
+	bindExtensions: (bindings?: Record<string, unknown>) => Promise<void>;
+	compact: (options?: unknown) => void;
+	getSessionStats: () => unknown;
+	isStreaming?: boolean;
+	model?: unknown;
+	systemPrompt?: string;
+	pendingMessageCount?: number;
+}
+
+/**
+ * Build extension bridge APIs from a Pi SDK session.
+ * Returns null if the session doesn't support extension running.
+ */
+export function buildExtensionBridge(session: PiSdkSession): { apis: ExtensionBridgeApis; host: ExtensionHostApis } | null {
+	if (typeof session.sendCustomMessage !== "function") return null;
+
+	return {
+		apis: {
+			sendMessage: (message, options) => {
+				try {
+					session.sendCustomMessage(message, options);
+				} catch {
+					/* non-blocking */
+				}
+			},
+			sendUserMessage: (content, options) => {
+				try {
+					session.sendUserMessage(content, options);
+				} catch {
+					/* non-blocking */
+				}
+			},
+			appendEntry: () => {
+				// appendEntry requires sessionManager access which isn't directly on session
+				// This is a no-op placeholder; extensions that rely on it will gracefully degrade
+			},
+			setLabel: () => {
+				// setLabel requires sessionManager access — no-op placeholder
+			},
+			getActiveTools: () => {
+				try {
+					return session.getActiveToolNames();
+				} catch {
+					return [];
+				}
+			},
+			getAllTools: () => {
+				try {
+					return session.getAllTools();
+				} catch {
+					return session.getActiveToolNames();
+				}
+			},
+			setActiveTools: (toolNames) => {
+				try {
+					session.setActiveToolsByName(toolNames);
+				} catch {
+					/* ignore */
+				}
+			},
+		},
+		host: {
+			getModel: () => session.model,
+			isIdle: () => !session.isStreaming,
+			abort: () => {
+				void session.abort();
+			},
+			hasPendingMessages: () => (session.pendingMessageCount ?? 0) > 0,
+			shutdown: () => {
+				/* no-op for live-session — caller manages session lifecycle */
+			},
+			getContextUsage: () => {
+				try {
+					return session.getContextUsage();
+				} catch {
+					return undefined;
+				}
+			},
+			getSystemPrompt: () => session.systemPrompt ?? "",
+		},
+	};
+}

package/src/runtime/live-irc.ts

@@ -0,0 +1,97 @@
+/**
+ * Phase 7: Inter-agent communication for live-session workers.
+ *
+ * Provides IRC-like messaging between live-session workers, adapted from
+ * oh-my-pi's IrcTool pattern. Uses the existing LiveAgentHandle manager
+ * for message routing.
+ *
+ * Features:
+ * - DM: Send a message to a specific agent
+ * - Broadcast: Send a message to all live agents
+ * - Side-channel: Non-blocking message injection (via pendingFollowUps)
+ *
+ * For child-process workers, messages fall back to file-based mailbox.
+ */
+
+export type IrcOperation = "send" | "list";
+
+export interface IrcMessage {
+	from: string;
+	to: string;
+	content: string;
+	timestamp: string;
+	/** Whether the sender expects a reply. */
+	awaitReply?: boolean;
+}
+
+export interface IrcSendMessage {
+	op: IrcOperation;
+	/** Target agent ID or "all" for broadcast. */
+	to: string;
+	/** Message content. */
+	message: string;
+	/** Whether to wait for a reply (default: true for DM, false for broadcast). */
+	awaitReply?: boolean;
+}
+
+export interface IrcListResult {
+	peers: Array<{ id: string; name: string; status: string }>;
+}
+
+/**
+ * Build IRC peer roster for injection into system prompt.
+ * Lists all currently live agents except the caller.
+ */
+export function renderIrcPeerRoster(selfId: string, peers: Array<{ agentId: string; status: string }>): string {
+	const visible = peers.filter((p) => p.agentId !== selfId && (p.status === "running" || p.status === "idle"));
+	if (visible.length === 0) return "- (no other live agents)";
+	return visible.map((peer) => `- \`${peer.agentId}\` (${peer.status})`).join("\n");
+}
+
+/**
+ * Build the IRC system prompt section for a live-session worker.
+ */
+/** @internal */
+function buildIrcSystemSection(selfId: string, peers: Array<{ agentId: string; status: string }>): string {
+	const roster = renderIrcPeerRoster(selfId, peers);
+	return [
+		"## Inter-Agent Communication",
+		`Your agent ID: \`${selfId}\``,
+		"You can send messages to other live agents via the `irc` tool.",
+		"Available peers:",
+		roster,
+	].join("\n");
+}
+
+/**
+ * Route an IRC message to the appropriate agent(s).
+ * Returns the list of agent IDs that received the message.
+ */
+/** @internal */
+function routeIrcMessage(
+	message: IrcSendMessage,
+	selfId: string,
+	routing: {
+		sendDm: (agentId: string, content: string) => void;
+		broadcast: (content: string, excludeId: string) => string[];
+	},
+): { deliveredTo: string[]; error?: string } {
+	if (!message.to || !message.message?.trim()) {
+		return {
+			deliveredTo: [],
+			error: "Missing 'to' (agent ID or 'all') and 'message' fields.",
+		};
+	}
+	if (message.to === selfId) {
+		return { deliveredTo: [], error: "Cannot send a message to yourself." };
+	}
+
+	if (message.to === "all") {
+		const recipients = routing.broadcast(message.message, selfId);
+		return { deliveredTo: recipients };
+	}
+
+	// DM to specific agent
+	routing.sendDm(message.to, message.message);
+	return { deliveredTo: [message.to] };
+}

package/src/runtime/live-session-health.ts

@@ -0,0 +1,108 @@
+/**
+ * Phase 8: Monitoring and observability for live-session workers.
+ *
+ * Provides health checks, metrics collection, and diagnostics
+ * for live-session workers running in-process.
+ */
+
+export interface LiveSessionHealth {
+	/** Total number of registered live agents. */
+	totalAgents: number;
+	/** Number of agents currently running. */
+	runningAgents: number;
+	/** Number of agents in idle state. */
+	idleAgents: number;
+	/** Number of agents that have completed. */
+	completedAgents: number;
+	/** Number of agents that have failed. */
+	failedAgents: number;
+	/** Total tokens consumed across all live sessions. */
+	totalTokens: number;
+	/** Timestamp of this health snapshot. */
+	timestamp: string;
+}
+
+export interface LiveSessionMetrics {
+	agentId: string;
+	taskId: string;
+	status: string;
+	/** Accumulated usage from session stats. */
+	usage?: {
+		input?: number;
+		output?: number;
+		cacheRead?: number;
+		cacheWrite?: number;
+		cost?: number;
+		turns?: number;
+	};
+	/** Session duration in milliseconds. */
+	durationMs?: number;
+	/** Number of IRC messages received. */
+	ircMessagesReceived?: number;
+	/** Number of yield reminders sent. */
+	yieldReminders?: number;
+	/** Whether yield was called. */
+	yieldCalled: boolean;
+}
+
+/**
+ * Collect health snapshot from live agent handles.
+ */
+export function collectLiveSessionHealth(
+	agents: Array<{ status: string }>,
+	getUsage: (agentId: string) => { input?: number; output?: number; turns?: number } | undefined,
+): LiveSessionHealth {
+	let running = 0;
+	let idle = 0;
+	let completed = 0;
+	let failed = 0;
+	let totalTokens = 0;
+
+	for (const agent of agents) {
+		switch (agent.status) {
+			case "running":
+				running++;
+				break;
+			case "idle":
+				idle++;
+				break;
+			case "completed":
+				completed++;
+				break;
+			case "failed":
+				failed++;
+				break;
+		}
+	}
+
+	// Sum tokens from usage data
+	for (const agent of agents) {
+		const agentAny = agent as Record<string, unknown>;
+		const agentId = agentAny.agentId as string | undefined;
+		if (agentId) {
+			const usage = getUsage(agentId);
+			if (usage) {
+				totalTokens += (usage.input ?? 0) + (usage.output ?? 0);
+			}
+		}
+	}
+
+	return {
+		totalAgents: agents.length,
+		runningAgents: running,
+		idleAgents: idle,
+		completedAgents: completed,
+		failedAgents: failed,
+		totalTokens,
+		timestamp: new Date().toISOString(),
+	};
+}
+
+/**
+ * Build a diagnostic summary string for logging.
+ */
+export function formatLiveSessionDiagnostics(health: LiveSessionHealth): string {
+	return [
+		`[Live-Session Health] agents=${health.totalAgents} running=${health.runningAgents} idle=${health.idleAgents} completed=${health.completedAgents} failed=${health.failedAgents} tokens=${health.totalTokens}`,
+	].join("\n");
+}