pi-crew 0.1.51 → 0.2.1

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

@@ -1,150 +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 ?? "",
-		},
-	};
-}
+/**
+ * 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

@@ -1,92 +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] };
-}
+/**
+ * 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] };
+}