diff --git a/packages/coding-agent/docs/refactor.md b/packages/coding-agent/docs/refactor.md index 5a5be3f8..a1206ef1 100644 --- a/packages/coding-agent/docs/refactor.md +++ b/packages/coding-agent/docs/refactor.md @@ -1216,8 +1216,8 @@ export async function runPrintMode( 1. `npm run check` passes 2. Manual test: `pi -p "echo hello"` still works -- [ ] Create `src/modes/print-mode.ts` -- [ ] Verify with `npm run check` +- [x] Create `src/modes/print-mode.ts` +- [x] Verify with `npm run check` --- diff --git a/packages/coding-agent/src/modes/print-mode.ts b/packages/coding-agent/src/modes/print-mode.ts new file mode 100644 index 00000000..7130baae --- /dev/null +++ b/packages/coding-agent/src/modes/print-mode.ts @@ -0,0 +1,69 @@ +/** + * Print mode (single-shot): Send prompts, output result, exit. + * + * Used for: + * - `pi -p "prompt"` - text output + * - `pi --mode json "prompt"` - JSON event stream + */ + +import type { Attachment } from "@mariozechner/pi-agent-core"; +import type { AssistantMessage } from "@mariozechner/pi-ai"; +import type { AgentSession } from "../core/agent-session.js"; + +/** + * Run in print (single-shot) mode. + * Sends prompts to the agent and outputs the result. + * + * @param session The agent session + * @param mode Output mode: "text" for final response only, "json" for all events + * @param messages Array of prompts to send + * @param initialMessage Optional first message (may contain @file content) + * @param initialAttachments Optional attachments for the initial message + */ +export async function runPrintMode( + session: AgentSession, + mode: "text" | "json", + messages: string[], + initialMessage?: string, + initialAttachments?: Attachment[], +): Promise { + if (mode === "json") { + // Output all events as JSON + session.subscribe((event) => { + console.log(JSON.stringify(event)); + }); + } + + // Send initial message with attachments + if (initialMessage) { + await session.prompt(initialMessage, { attachments: initialAttachments }); + } + + // Send remaining messages + for (const message of messages) { + await session.prompt(message); + } + + // In text mode, output final response + if (mode === "text") { + const state = session.state; + const lastMessage = state.messages[state.messages.length - 1]; + + if (lastMessage?.role === "assistant") { + const assistantMsg = lastMessage as AssistantMessage; + + // Check for error/aborted + if (assistantMsg.stopReason === "error" || assistantMsg.stopReason === "aborted") { + console.error(assistantMsg.errorMessage || `Request ${assistantMsg.stopReason}`); + process.exit(1); + } + + // Output text content + for (const content of assistantMsg.content) { + if (content.type === "text") { + console.log(content.text); + } + } + } + } +}