pi-crew 0.1.45 → 0.1.49

package/src/runtime/live-agent-manager.ts

@@ -1,85 +1,179 @@
-import type { CrewAgentRecord } from "./crew-agent-runtime.ts";
-
-type LiveSessionHandle = {
-	steer?: (text: string) => Promise<void>;
-	prompt?: (text: string, options?: Record<string, unknown>) => Promise<void>;
-	abort?: () => Promise<void> | void;
-};
-
-export interface LiveAgentHandle {
-	agentId: string;
-	taskId: string;
-	runId: string;
-	session: LiveSessionHandle;
-	createdAt: string;
-	updatedAt: string;
-	status: CrewAgentRecord["status"];
-	pendingSteers: string[];
-}
-
-const liveAgents = new Map<string, LiveAgentHandle>();
-
-export function registerLiveAgent(input: Omit<LiveAgentHandle, "createdAt" | "updatedAt" | "pendingSteers">): LiveAgentHandle {
-	const now = new Date().toISOString();
-	const existing = liveAgents.get(input.agentId);
-	const handle: LiveAgentHandle = { ...input, createdAt: existing?.createdAt ?? now, updatedAt: now, pendingSteers: existing?.pendingSteers ?? [] };
-	liveAgents.set(input.agentId, handle);
-	if (handle.pendingSteers.length && typeof handle.session.steer === "function") {
-		const pending = [...handle.pendingSteers];
-		handle.pendingSteers.length = 0;
-		for (const message of pending) void handle.session.steer(message).catch(() => {});
-	}
-	return handle;
-}
-
-export function updateLiveAgentStatus(agentId: string, status: CrewAgentRecord["status"]): void {
-	const handle = liveAgents.get(agentId);
-	if (!handle) return;
-	handle.status = status;
-	handle.updatedAt = new Date().toISOString();
-}
-
-export function getLiveAgent(agentIdOrTaskId: string): LiveAgentHandle | undefined {
-	return liveAgents.get(agentIdOrTaskId) ?? [...liveAgents.values()].find((entry) => entry.taskId === agentIdOrTaskId);
-}
-
-export function listLiveAgents(): LiveAgentHandle[] {
-	return [...liveAgents.values()].sort((a, b) => b.updatedAt.localeCompare(a.updatedAt));
-}
-
-export async function steerLiveAgent(agentIdOrTaskId: string, message: string): Promise<LiveAgentHandle> {
-	const handle = getLiveAgent(agentIdOrTaskId);
-	if (!handle) throw new Error(`Live agent '${agentIdOrTaskId}' is not registered in this process.`);
-	if (typeof handle.session.steer !== "function") {
-		handle.pendingSteers.push(message);
-		return handle;
-	}
-	await handle.session.steer(message);
-	handle.updatedAt = new Date().toISOString();
-	return handle;
-}
-
-export async function stopLiveAgent(agentIdOrTaskId: string): Promise<LiveAgentHandle> {
-	const handle = getLiveAgent(agentIdOrTaskId);
-	if (!handle) throw new Error(`Live agent '${agentIdOrTaskId}' is not registered in this process.`);
-	if (typeof handle.session.abort !== "function") throw new Error(`Live agent '${agentIdOrTaskId}' does not expose abort().`);
-	await handle.session.abort();
-	handle.status = "stopped";
-	handle.updatedAt = new Date().toISOString();
-	return handle;
-}
-
-export async function resumeLiveAgent(agentIdOrTaskId: string, prompt: string): Promise<LiveAgentHandle> {
-	const handle = getLiveAgent(agentIdOrTaskId);
-	if (!handle) throw new Error(`Live agent '${agentIdOrTaskId}' is not registered in this process.`);
-	if (typeof handle.session.prompt !== "function") throw new Error(`Live agent '${agentIdOrTaskId}' does not expose prompt().`);
-	handle.status = "running";
-	await handle.session.prompt(prompt, { source: "api", expandPromptTemplates: false });
-	handle.status = "completed";
-	handle.updatedAt = new Date().toISOString();
-	return handle;
-}
-
-export function clearLiveAgentsForTest(): void {
-	liveAgents.clear();
-}
+import type { CrewAgentRecord } from "./crew-agent-runtime.ts";
+import type { IrcMessage } from "./live-irc.ts";
+
+type LiveSessionHandle = {
+	steer?: (text: string) => Promise<void>;
+	prompt?: (text: string, options?: Record<string, unknown>) => Promise<void>;
+	abort?: () => Promise<void> | void;
+};
+
+export interface LiveAgentHandle {
+	agentId: string;
+	taskId: string;
+	runId: string;
+	session: LiveSessionHandle;
+	createdAt: string;
+	updatedAt: string;
+	status: CrewAgentRecord["status"];
+	pendingSteers: string[];
+	pendingFollowUps: string[];
+	/** Phase 7: Pending IRC messages for this agent. */
+	pendingMessages: IrcMessage[];
+}
+
+const liveAgents = new Map<string, LiveAgentHandle>();
+
+export function registerLiveAgent(input: Omit<LiveAgentHandle, "createdAt" | "updatedAt" | "pendingSteers" | "pendingFollowUps" | "pendingMessages">): LiveAgentHandle {
+	const now = new Date().toISOString();
+	const existing = liveAgents.get(input.agentId);
+	const handle: LiveAgentHandle = { ...input, createdAt: existing?.createdAt ?? now, updatedAt: now, pendingSteers: existing?.pendingSteers ?? [], pendingFollowUps: existing?.pendingFollowUps ?? [], pendingMessages: existing?.pendingMessages ?? [] };
+	liveAgents.set(input.agentId, handle);
+	if (handle.pendingSteers.length && typeof handle.session.steer === "function") {
+		const pending = [...handle.pendingSteers];
+		handle.pendingSteers.length = 0;
+		for (const message of pending) void handle.session.steer(message).catch(() => {});
+	}
+	if (handle.pendingFollowUps.length && typeof handle.session.prompt === "function") {
+		const pending = [...handle.pendingFollowUps];
+		handle.pendingFollowUps.length = 0;
+		for (const message of pending) void handle.session.prompt(message, { source: "api", expandPromptTemplates: false }).catch(() => {});
+	}
+	return handle;
+}
+
+export function updateLiveAgentStatus(agentId: string, status: CrewAgentRecord["status"]): void {
+	const handle = liveAgents.get(agentId);
+	if (!handle) return;
+	handle.status = status;
+	handle.updatedAt = new Date().toISOString();
+}
+
+export function getLiveAgent(agentIdOrTaskId: string): LiveAgentHandle | undefined {
+	return liveAgents.get(agentIdOrTaskId) ?? [...liveAgents.values()].find((entry) => entry.taskId === agentIdOrTaskId);
+}
+
+export function listLiveAgents(): LiveAgentHandle[] {
+	return [...liveAgents.values()].sort((a, b) => b.updatedAt.localeCompare(a.updatedAt));
+}
+
+export async function steerLiveAgent(agentIdOrTaskId: string, message: string): Promise<LiveAgentHandle> {
+	const handle = getLiveAgent(agentIdOrTaskId);
+	if (!handle) throw new Error(`Live agent '${agentIdOrTaskId}' is not registered in this process.`);
+	if (typeof handle.session.steer !== "function") {
+		handle.pendingSteers.push(message);
+		return handle;
+	}
+	await handle.session.steer(message);
+	handle.updatedAt = new Date().toISOString();
+	return handle;
+}
+
+export async function followUpLiveAgent(agentIdOrTaskId: string, prompt: string): Promise<LiveAgentHandle> {
+	const handle = getLiveAgent(agentIdOrTaskId);
+	if (!handle) throw new Error(`Live agent '${agentIdOrTaskId}' is not registered in this process.`);
+	if (typeof handle.session.prompt !== "function") {
+		handle.pendingFollowUps.push(prompt);
+		return handle;
+	}
+	await handle.session.prompt(prompt, { source: "api", expandPromptTemplates: false });
+	handle.updatedAt = new Date().toISOString();
+	return handle;
+}
+
+export async function stopLiveAgent(agentIdOrTaskId: string): Promise<LiveAgentHandle> {
+	const handle = getLiveAgent(agentIdOrTaskId);
+	if (!handle) throw new Error(`Live agent '${agentIdOrTaskId}' is not registered in this process.`);
+	if (typeof handle.session.abort !== "function") throw new Error(`Live agent '${agentIdOrTaskId}' does not expose abort().`);
+	await handle.session.abort();
+	handle.status = "stopped";
+	handle.updatedAt = new Date().toISOString();
+	return handle;
+}
+
+export async function resumeLiveAgent(agentIdOrTaskId: string, prompt: string): Promise<LiveAgentHandle> {
+	const handle = getLiveAgent(agentIdOrTaskId);
+	if (!handle) throw new Error(`Live agent '${agentIdOrTaskId}' is not registered in this process.`);
+	if (typeof handle.session.prompt !== "function") throw new Error(`Live agent '${agentIdOrTaskId}' does not expose prompt().`);
+	handle.status = "running";
+	await handle.session.prompt(prompt, { source: "api", expandPromptTemplates: false });
+	handle.status = "completed";
+	handle.updatedAt = new Date().toISOString();
+	return handle;
+}
+
+export function clearLiveAgentsForTest(): void {
+	liveAgents.clear();
+}
+
+/** Phase 7/G4: Send an IRC message to a specific live agent (DM).
+ * Uses non-blocking delivery via sendCustomMessage when available.
+ * Falls back to session.prompt (blocking) when not.
+ */
+export function sendIrcMessage(targetAgentId: string, message: IrcMessage): void {
+	const handle = getLiveAgent(targetAgentId);
+	if (!handle) return;
+	handle.pendingMessages.push(message);
+	handle.updatedAt = new Date().toISOString();
+	// G4: Try non-blocking delivery via sendCustomMessage
+	const session = handle.session as Record<string, unknown>;
+	if (typeof session.sendCustomMessage === "function") {
+		try {
+			(session.sendCustomMessage as (msg: unknown, opts?: unknown) => void)(
+				{ customType: "irc", content: `[DM from ${message.from}] ${message.content}`, display: "collapsed" },
+				{ deliverAs: "followUp", triggerTurn: false },
+			);
+			return;
+		} catch {
+			// Fall through to prompt-based delivery
+		}
+	}
+	// Fallback: inject as prompt (blocking)
+	if (typeof handle.session.prompt === "function") {
+		const ircPrompt = `[Message from ${message.from}] ${message.content}`;
+		void handle.session.prompt(ircPrompt, { source: "api", expandPromptTemplates: false }).catch(() => {});
+	}
+}
+
+/** Phase 7/G4: Broadcast an IRC message to all live agents except the sender.
+ * Uses non-blocking delivery via sendCustomMessage when available.
+ * Returns recipient IDs.
+ */
+export function broadcastIrcMessage(fromAgentId: string, message: IrcMessage): string[] {
+	const recipients: string[] = [];
+	for (const handle of liveAgents.values()) {
+		if (handle.agentId === fromAgentId) continue;
+		if (handle.status !== "running" && handle.status !== "queued") continue;
+		handle.pendingMessages.push(message);
+		handle.updatedAt = new Date().toISOString();
+		// G4: Try non-blocking delivery
+		const session = handle.session as Record<string, unknown>;
+		if (typeof session.sendCustomMessage === "function") {
+			try {
+				(session.sendCustomMessage as (msg: unknown, opts?: unknown) => void)(
+					{ customType: "irc", content: `[Broadcast from ${message.from}] ${message.content}`, display: "collapsed" },
+					{ deliverAs: "followUp", triggerTurn: false },
+				);
+				recipients.push(handle.agentId);
+				continue;
+			} catch {
+				// Fall through to prompt-based delivery
+			}
+		}
+		// Fallback: inject as prompt
+		if (typeof handle.session.prompt === "function") {
+			const ircPrompt = `[Broadcast from ${message.from}] ${message.content}`;
+			void handle.session.prompt(ircPrompt, { source: "api", expandPromptTemplates: false }).catch(() => {});
+		}
+		recipients.push(handle.agentId);
+	}
+	return recipients;
+}
+
+/** Phase 7: Get pending IRC messages for an agent (and clear them). */
+export function drainIrcMessages(agentIdOrTaskId: string): IrcMessage[] {
+	const handle = getLiveAgent(agentIdOrTaskId);
+	if (!handle) return [];
+	const messages = [...handle.pendingMessages];
+	handle.pendingMessages.length = 0;
+	return messages;
+}

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

@@ -28,7 +28,7 @@ export function parseLiveControlRealtimeMessage(raw: unknown): LiveAgentControlR
 	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 === "stop" || request.operation === "resume") && typeof request.createdAt === "string" ? request as LiveAgentControlRequest : undefined;
+	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 {

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

@@ -0,0 +1,150 @@
+/**
+ * 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
+ */
+
+import type { YieldResult } from "./yield-handler.ts";
+
+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,92 @@
+/**
+ * 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.
+ */
+export 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.
+ */
+export 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,100 @@
+/**
+ * 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");
+}