Mario Zechner
7c553acd1e
- Hook discovery from ~/.pi/agent/hooks/, .pi/hooks/, --hook flag - Events: session_start, session_switch, agent_start/end, turn_start/end, tool_call, tool_result, branch - tool_call can block execution, tool_result can modify results - pi.send(text, attachments?) to inject messages from external sources - UI primitives: ctx.ui.select/confirm/input/notify - Context: ctx.exec(), ctx.cwd, ctx.sessionFile, ctx.hasUI - Docs shipped with npm package and binary builds - System prompt references docs folder
86 lines
2.6 KiB
TypeScript
86 lines
2.6 KiB
TypeScript
/**
|
|
* 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<void> {
|
|
// Hook runner already has no-op UI context by default (set in main.ts)
|
|
// Set up hooks for print mode (no UI, ephemeral session)
|
|
const hookRunner = session.hookRunner;
|
|
if (hookRunner) {
|
|
hookRunner.setSessionFile(null); // Print mode is ephemeral
|
|
hookRunner.onError((err) => {
|
|
console.error(`Hook error (${err.hookPath}): ${err.error}`);
|
|
});
|
|
// No-op send handler for print mode (single-shot, no async messages)
|
|
hookRunner.setSendHandler(() => {
|
|
console.error("Warning: pi.send() is not supported in print mode");
|
|
});
|
|
// Emit session_start event
|
|
await hookRunner.emit({ type: "session_start" });
|
|
}
|
|
|
|
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);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|