@tomsun28/pizza 0.1.0 → 0.1.1

package/dist/core/tools/read.js

@@ -0,0 +1,244 @@
+import { Text } from "@earendil-works/pi-tui";
+import { Type } from "@sinclair/typebox";
+import { constants } from "fs";
+import { access as fsAccess, readFile as fsReadFile } from "fs/promises";
+import { keyHint } from "../../modes/interactive/components/keybinding-hints.js";
+import { getLanguageFromPath, highlightCode } from "../../modes/interactive/theme/theme.js";
+import { formatDimensionNote, resizeImage } from "../../utils/image-resize.js";
+import { detectSupportedImageMimeTypeFromFile } from "../../utils/mime.js";
+import { annotateTextWithLineAnchors } from "./line-anchors.js";
+import { resolveReadPath } from "./path-utils.js";
+import { getTextOutput, invalidArgText, replaceTabs, shortenPath, str } from "./render-utils.js";
+import { wrapToolDefinition } from "./tool-definition-wrapper.js";
+import { DEFAULT_MAX_BYTES, DEFAULT_MAX_LINES, formatSize, truncateHead } from "./truncate.js";
+const readSchema = Type.Object({
+    path: Type.String({ description: "Path to the file to read (relative or absolute)" }),
+    offset: Type.Optional(Type.Number({ description: "Line number to start reading from (1-indexed)" })),
+    limit: Type.Optional(Type.Number({ description: "Maximum number of lines to read" })),
+    anchors: Type.Optional(Type.Union([Type.Literal("line"), Type.Literal("none")], {
+        description: 'Whether to include hashline anchors in text output. "line" prefixes each line with <line>#<2-hex-hash> for use with edit range. Default: "line". Use "none" only when raw file text is required.',
+    })),
+});
+const defaultReadOperations = {
+    readFile: (path) => fsReadFile(path),
+    access: (path) => fsAccess(path, constants.R_OK),
+    detectImageMimeType: detectSupportedImageMimeTypeFromFile,
+};
+function formatReadCall(args, theme) {
+    const rawPath = str(args?.file_path ?? args?.path);
+    const path = rawPath !== null ? shortenPath(rawPath) : null;
+    const offset = args?.offset;
+    const limit = args?.limit;
+    const invalidArg = invalidArgText(theme);
+    let pathDisplay = path === null ? invalidArg : path ? theme.fg("accent", path) : theme.fg("toolOutput", "...");
+    if (offset !== undefined || limit !== undefined) {
+        const startLine = offset ?? 1;
+        const endLine = limit !== undefined ? startLine + limit - 1 : "";
+        pathDisplay += theme.fg("warning", `:${startLine}${endLine ? `-${endLine}` : ""}`);
+    }
+    return `${theme.fg("toolTitle", theme.bold("read"))} ${pathDisplay}`;
+}
+function trimTrailingEmptyLines(lines) {
+    let end = lines.length;
+    while (end > 0 && lines[end - 1] === "") {
+        end--;
+    }
+    return lines.slice(0, end);
+}
+function getNonVisionImageNote(model) {
+    if (!model || model.input.includes("image")) {
+        return undefined;
+    }
+    return "[Current model does not support images. The image will be omitted from this request.]";
+}
+function formatTextReadOutput(text, startLine, anchors) {
+    return anchors === "line" ? annotateTextWithLineAnchors(text, startLine) : text;
+}
+function normalizeTextForAnchors(text, anchors) {
+    return anchors === "line" ? text.replace(/\r\n/g, "\n").replace(/\r/g, "\n") : text;
+}
+function formatReadResult(args, result, options, theme, showImages) {
+    const rawPath = str(args?.file_path ?? args?.path);
+    const output = getTextOutput(result, showImages);
+    const lang = rawPath ? getLanguageFromPath(rawPath) : undefined;
+    const renderedLines = lang ? highlightCode(replaceTabs(output), lang) : output.split("\n");
+    const lines = trimTrailingEmptyLines(renderedLines);
+    const maxLines = options.expanded ? lines.length : 10;
+    const displayLines = lines.slice(0, maxLines);
+    const remaining = lines.length - maxLines;
+    let text = `\n${displayLines.map((line) => (lang ? replaceTabs(line) : theme.fg("toolOutput", replaceTabs(line)))).join("\n")}`;
+    if (remaining > 0) {
+        text += `${theme.fg("muted", `\n... (${remaining} more lines,`)} ${keyHint("app.tools.expand", "to expand")})`;
+    }
+    const truncation = result.details?.truncation;
+    if (truncation?.truncated) {
+        if (truncation.firstLineExceedsLimit) {
+            text += `\n${theme.fg("warning", `[First line exceeds ${formatSize(truncation.maxBytes ?? DEFAULT_MAX_BYTES)} limit]`)}`;
+        }
+        else if (truncation.truncatedBy === "lines") {
+            text += `\n${theme.fg("warning", `[Truncated: showing ${truncation.outputLines} of ${truncation.totalLines} lines (${truncation.maxLines ?? DEFAULT_MAX_LINES} line limit)]`)}`;
+        }
+        else {
+            text += `\n${theme.fg("warning", `[Truncated: ${truncation.outputLines} lines shown (${formatSize(truncation.maxBytes ?? DEFAULT_MAX_BYTES)} limit)]`)}`;
+        }
+    }
+    return text;
+}
+export function createReadToolDefinition(cwd, options) {
+    const autoResizeImages = options?.autoResizeImages ?? true;
+    const ops = options?.operations ?? defaultReadOperations;
+    return {
+        name: "read",
+        label: "read",
+        description: `Read the contents of a file. Supports text files and images (jpg, png, gif, webp). Images are sent as attachments. For text files, output is truncated to ${DEFAULT_MAX_LINES} lines or ${DEFAULT_MAX_BYTES / 1024}KB (whichever is hit first). Text output includes 2-hex hashline anchors by default (<line>#<hash> | content); pass anchors="none" only when raw text is required. Use offset/limit for large files. When you need the full file, continue with offset until complete.`,
+        promptSnippet: "Read file contents",
+        promptGuidelines: [
+            "Use read to examine files instead of cat or sed.",
+            "Use read line anchors as edit edits[].range for whole-line edits.",
+        ],
+        parameters: readSchema,
+        async execute(_toolCallId, { path, offset, limit, anchors = "line" }, signal, _onUpdate, ctx) {
+            const absolutePath = resolveReadPath(path, cwd);
+            return new Promise((resolve, reject) => {
+                if (signal?.aborted) {
+                    reject(new Error("Operation aborted"));
+                    return;
+                }
+                let aborted = false;
+                const onAbort = () => {
+                    aborted = true;
+                    reject(new Error("Operation aborted"));
+                };
+                signal?.addEventListener("abort", onAbort, { once: true });
+                (async () => {
+                    try {
+                        // Check if file exists and is readable.
+                        await ops.access(absolutePath);
+                        if (aborted)
+                            return;
+                        const mimeType = ops.detectImageMimeType ? await ops.detectImageMimeType(absolutePath) : undefined;
+                        let content;
+                        let details;
+                        const nonVisionImageNote = getNonVisionImageNote(ctx?.model);
+                        if (mimeType) {
+                            // Read image as binary.
+                            const buffer = await ops.readFile(absolutePath);
+                            const base64 = buffer.toString("base64");
+                            if (autoResizeImages) {
+                                // Resize image if needed before sending it back to the model.
+                                const resized = await resizeImage({ type: "image", data: base64, mimeType });
+                                if (!resized) {
+                                    let textNote = `Read image file [${mimeType}]\n[Image omitted: could not be resized below the inline image size limit.]`;
+                                    if (nonVisionImageNote)
+                                        textNote += `\n${nonVisionImageNote}`;
+                                    content = [{ type: "text", text: textNote }];
+                                }
+                                else {
+                                    const dimensionNote = formatDimensionNote(resized);
+                                    let textNote = `Read image file [${resized.mimeType}]`;
+                                    if (dimensionNote)
+                                        textNote += `\n${dimensionNote}`;
+                                    if (nonVisionImageNote)
+                                        textNote += `\n${nonVisionImageNote}`;
+                                    content = [
+                                        { type: "text", text: textNote },
+                                        { type: "image", data: resized.data, mimeType: resized.mimeType },
+                                    ];
+                                }
+                            }
+                            else {
+                                let textNote = `Read image file [${mimeType}]`;
+                                if (nonVisionImageNote)
+                                    textNote += `\n${nonVisionImageNote}`;
+                                content = [
+                                    { type: "text", text: textNote },
+                                    { type: "image", data: base64, mimeType },
+                                ];
+                            }
+                        }
+                        else {
+                            // Read text content.
+                            const buffer = await ops.readFile(absolutePath);
+                            const textContent = normalizeTextForAnchors(buffer.toString("utf-8"), anchors);
+                            const allLines = textContent.split("\n");
+                            const totalFileLines = allLines.length;
+                            // Apply offset if specified. Convert from 1-indexed input to 0-indexed array access.
+                            const startLine = offset ? Math.max(0, offset - 1) : 0;
+                            const startLineDisplay = startLine + 1;
+                            // Check if offset is out of bounds.
+                            if (startLine >= allLines.length) {
+                                throw new Error(`Offset ${offset} is beyond end of file (${allLines.length} lines total)`);
+                            }
+                            let selectedContent;
+                            let userLimitedLines;
+                            // If limit is specified by the user, honor it first. Otherwise truncateHead decides.
+                            if (limit !== undefined) {
+                                const endLine = Math.min(startLine + limit, allLines.length);
+                                selectedContent = allLines.slice(startLine, endLine).join("\n");
+                                userLimitedLines = endLine - startLine;
+                            }
+                            else {
+                                selectedContent = allLines.slice(startLine).join("\n");
+                            }
+                            // Apply truncation, respecting both line and byte limits.
+                            const truncation = truncateHead(selectedContent);
+                            let outputText;
+                            if (truncation.firstLineExceedsLimit) {
+                                // First line alone exceeds the byte limit. Point the model at a bash fallback.
+                                const firstLineSize = formatSize(Buffer.byteLength(allLines[startLine], "utf-8"));
+                                outputText = `[Line ${startLineDisplay} is ${firstLineSize}, exceeds ${formatSize(DEFAULT_MAX_BYTES)} limit. Use bash: sed -n '${startLineDisplay}p' ${path} | head -c ${DEFAULT_MAX_BYTES}]`;
+                                details = { truncation };
+                            }
+                            else if (truncation.truncated) {
+                                // Truncation occurred. Build an actionable continuation notice.
+                                const endLineDisplay = startLineDisplay + truncation.outputLines - 1;
+                                const nextOffset = endLineDisplay + 1;
+                                outputText = formatTextReadOutput(truncation.content, startLineDisplay, anchors);
+                                if (truncation.truncatedBy === "lines") {
+                                    outputText += `\n\n[Showing lines ${startLineDisplay}-${endLineDisplay} of ${totalFileLines}. Use offset=${nextOffset} to continue.]`;
+                                }
+                                else {
+                                    outputText += `\n\n[Showing lines ${startLineDisplay}-${endLineDisplay} of ${totalFileLines} (${formatSize(DEFAULT_MAX_BYTES)} limit). Use offset=${nextOffset} to continue.]`;
+                                }
+                                details = { truncation };
+                            }
+                            else if (userLimitedLines !== undefined && startLine + userLimitedLines < allLines.length) {
+                                // User-specified limit stopped early, but the file still has more content.
+                                const remaining = allLines.length - (startLine + userLimitedLines);
+                                const nextOffset = startLine + userLimitedLines + 1;
+                                outputText = `${formatTextReadOutput(truncation.content, startLineDisplay, anchors)}\n\n[${remaining} more lines in file. Use offset=${nextOffset} to continue.]`;
+                            }
+                            else {
+                                // No truncation and no remaining user-limited content.
+                                outputText = formatTextReadOutput(truncation.content, startLineDisplay, anchors);
+                            }
+                            content = [{ type: "text", text: outputText }];
+                        }
+                        if (aborted)
+                            return;
+                        signal?.removeEventListener("abort", onAbort);
+                        resolve({ content, details });
+                    }
+                    catch (error) {
+                        signal?.removeEventListener("abort", onAbort);
+                        if (!aborted)
+                            reject(error);
+                    }
+                })();
+            });
+        },
+        renderCall(args, theme, context) {
+            const text = context.lastComponent ?? new Text("", 0, 0);
+            text.setText(formatReadCall(args, theme));
+            return text;
+        },
+        renderResult(result, options, theme, context) {
+            const text = context.lastComponent ?? new Text("", 0, 0);
+            text.setText(formatReadResult(context.args, result, options, theme, context.showImages));
+            return text;
+        },
+    };
+}
+export function createReadTool(cwd, options) {
+    return wrapToolDefinition(createReadToolDefinition(cwd, options));
+}

package/dist/core/tools/render-utils.d.ts

@@ -0,0 +1,20 @@
+import type { ImageContent, TextContent } from "@earendil-works/pi-ai/compat";
+export declare function shortenPath(path: unknown): string;
+export declare function str(value: unknown): string | null;
+export declare function replaceTabs(text: string): string;
+export declare function normalizeDisplayText(text: string): string;
+export declare function getTextOutput(result: {
+    content: Array<{
+        type: string;
+        text?: string;
+        data?: string;
+        mimeType?: string;
+    }>;
+} | undefined, showImages: boolean): string;
+export type ToolRenderResultLike<TDetails> = {
+    content: (TextContent | ImageContent)[];
+    details: TDetails;
+};
+export declare function invalidArgText(theme: {
+    fg: (name: any, text: string) => string;
+}): string;

package/dist/core/tools/render-utils.js

@@ -0,0 +1,48 @@
+import * as os from "node:os";
+import { getCapabilities, getImageDimensions, imageFallback } from "@earendil-works/pi-tui";
+import stripAnsi from "strip-ansi";
+import { sanitizeBinaryOutput } from "../../utils/shell.js";
+export function shortenPath(path) {
+    if (typeof path !== "string")
+        return "";
+    const home = os.homedir();
+    if (path.startsWith(home)) {
+        return `~${path.slice(home.length)}`;
+    }
+    return path;
+}
+export function str(value) {
+    if (typeof value === "string")
+        return value;
+    if (value == null)
+        return "";
+    return null;
+}
+export function replaceTabs(text) {
+    return text.replace(/\t/g, "   ");
+}
+export function normalizeDisplayText(text) {
+    return text.replace(/\r/g, "");
+}
+export function getTextOutput(result, showImages) {
+    if (!result)
+        return "";
+    const textBlocks = result.content.filter((c) => c.type === "text");
+    const imageBlocks = result.content.filter((c) => c.type === "image");
+    let output = textBlocks.map((c) => sanitizeBinaryOutput(stripAnsi(c.text || "")).replace(/\r/g, "")).join("\n");
+    const caps = getCapabilities();
+    if (imageBlocks.length > 0 && (!caps.images || !showImages)) {
+        const imageIndicators = imageBlocks
+            .map((img) => {
+            const mimeType = img.mimeType ?? "image/unknown";
+            const dims = img.data && img.mimeType ? (getImageDimensions(img.data, img.mimeType) ?? undefined) : undefined;
+            return imageFallback(mimeType, dims);
+        })
+            .join("\n");
+        output = output ? `${output}\n${imageIndicators}` : imageIndicators;
+    }
+    return output;
+}
+export function invalidArgText(theme) {
+    return theme.fg("error", "[invalid arg]");
+}

package/dist/core/tools/session-split.d.ts

@@ -0,0 +1,9 @@
+import { type Static } from "@sinclair/typebox";
+import { type ToolDefinition } from "../extensions/types.js";
+declare const sessionSplitSchema: import("@sinclair/typebox").TObject<{
+    reason: import("@sinclair/typebox").TOptional<import("@sinclair/typebox").TString>;
+    name: import("@sinclair/typebox").TOptional<import("@sinclair/typebox").TString>;
+}>;
+export type SessionSplitToolInput = Static<typeof sessionSplitSchema>;
+export declare function createSessionSplitToolDefinition(): ToolDefinition<typeof sessionSplitSchema>;
+export {};

package/dist/core/tools/session-split.js

@@ -0,0 +1,77 @@
+import { Type } from "@sinclair/typebox";
+import { Text } from "@earendil-works/pi-tui";
+import { defineTool } from "../extensions/types.js";
+const sessionSplitSchema = Type.Object({
+    reason: Type.Optional(Type.String({
+        description: "Short reason for the split (e.g. 'topic_change', 'new_task', 'context_reset'). Helps the user understand why the session was divided.",
+    })),
+    name: Type.Optional(Type.String({
+        description: "Optional name for the new session (e.g. 'Fix authentication bug'). If omitted, a default name is used.",
+    })),
+});
+export function createSessionSplitToolDefinition() {
+    return defineTool({
+        name: "session_split",
+        label: "session_split",
+        description: "Split the current conversation session, starting a new session from this point forward. " +
+            "Previous messages will no longer be included in the LLM context for subsequent turns. " +
+            "Call this when the user's intent has clearly shifted to a new, unrelated topic and the previous " +
+            "conversation context is no longer needed. Do NOT call this for follow-up questions or clarifications " +
+            "about the current topic. You can call this alongside other tools in the same turn. " +
+            "IMPORTANT: Call this AT MOST ONCE per turn. After the split, your context will be refreshed " +
+            "to the new session — proceed directly with the user's requested task.",
+        promptSnippet: "Split the conversation session when the user's topic has clearly changed",
+        promptGuidelines: [
+            "Only call session_split when the user's request is clearly unrelated to the current conversation topic.",
+            "Do not call session_split for follow-up questions, clarifications, or refinements of the current task.",
+            "Call session_split at most once per turn. After the split, your context is refreshed — proceed with the task.",
+        ],
+        parameters: sessionSplitSchema,
+        renderShell: "self",
+        async execute(_toolCallId, params, _signal, _onUpdate, ctx) {
+            const result = ctx?.sessionManager?.splitSession?.(params.reason ?? "intent_shift", params.name);
+            if (!result) {
+                return {
+                    content: [
+                        {
+                            type: "text",
+                            text: "Session split is not available in this runtime configuration.",
+                        },
+                    ],
+                };
+            }
+            if (result.already_split) {
+                return {
+                    content: [
+                        {
+                            type: "text",
+                            text: "The session was already split for this turn. Do NOT call session_split again — " +
+                                "proceed directly with the user's requested task.",
+                        },
+                    ],
+                };
+            }
+            return {
+                content: [
+                    {
+                        type: "text",
+                        text: `Session split successfully. New session: ${result.session_id}. ` +
+                            "Your context has been refreshed to the new session. " +
+                            "Proceed with the user's task now — do NOT call session_split again.",
+                    },
+                ],
+            };
+        },
+        renderCall(args, _theme) {
+            const reason = args?.reason ?? "intent_shift";
+            const name = args?.name ? ` → ${args.name}` : "";
+            return new Text(`session_split (${reason})${name}`, 0, 0);
+        },
+        renderResult(result, _options, _theme) {
+            const text = result.content
+                .map((c) => ("text" in c ? c.text : ""))
+                .join("\n");
+            return new Text(`\n${text}`, 0, 0);
+        },
+    });
+}

package/dist/core/tools/tool-definition-wrapper.d.ts

@@ -0,0 +1,13 @@
+import type { AgentTool } from "../agent/types.js";
+import type { ExtensionContext, ToolDefinition } from "../extensions/types.js";
+/** Wrap a ToolDefinition into an AgentTool for the core runtime. */
+export declare function wrapToolDefinition<TDetails = unknown>(definition: ToolDefinition<any, TDetails>, ctxFactory?: () => ExtensionContext): AgentTool<any, TDetails>;
+/** Wrap multiple ToolDefinitions into AgentTools for the core runtime. */
+export declare function wrapToolDefinitions(definitions: ToolDefinition<any, any>[], ctxFactory?: () => ExtensionContext): AgentTool<any>[];
+/**
+ * Synthesize a minimal ToolDefinition from an AgentTool.
+ *
+ * This keeps the internal registry definition-first even when a caller
+ * provides plain AgentTool overrides that do not include prompt metadata or renderers.
+ */
+export declare function createToolDefinitionFromAgentTool(tool: AgentTool<any>): ToolDefinition<any, unknown>;

package/dist/core/tools/tool-definition-wrapper.js

@@ -0,0 +1,32 @@
+/** Wrap a ToolDefinition into an AgentTool for the core runtime. */
+export function wrapToolDefinition(definition, ctxFactory) {
+    return {
+        name: definition.name,
+        description: definition.description,
+        parameters: definition.parameters,
+        prepareArguments: definition.prepareArguments,
+        executionMode: definition.executionMode,
+        execute: (toolCallId, params, signal, onUpdate) => definition.execute(toolCallId, params, signal, onUpdate, ctxFactory?.()),
+    };
+}
+/** Wrap multiple ToolDefinitions into AgentTools for the core runtime. */
+export function wrapToolDefinitions(definitions, ctxFactory) {
+    return definitions.map((definition) => wrapToolDefinition(definition, ctxFactory));
+}
+/**
+ * Synthesize a minimal ToolDefinition from an AgentTool.
+ *
+ * This keeps the internal registry definition-first even when a caller
+ * provides plain AgentTool overrides that do not include prompt metadata or renderers.
+ */
+export function createToolDefinitionFromAgentTool(tool) {
+    return {
+        name: tool.name,
+        label: tool.label, // label is inherited from Tool interface
+        description: tool.description,
+        parameters: tool.parameters,
+        prepareArguments: tool.prepareArguments,
+        executionMode: tool.executionMode,
+        execute: async (toolCallId, params, signal, onUpdate) => tool.execute(toolCallId, params, signal, onUpdate),
+    };
+}

package/dist/core/tools/truncate.d.ts

@@ -0,0 +1,69 @@
+/**
+ * Shared truncation utilities for tool outputs.
+ *
+ * Truncation is based on two independent limits - whichever is hit first wins:
+ * - Line limit (default: 2000 lines)
+ * - Byte limit (default: 50KB)
+ *
+ * Never returns partial lines (except bash tail truncation edge case).
+ */
+export declare const DEFAULT_MAX_LINES = 2000;
+export declare const DEFAULT_MAX_BYTES: number;
+export declare const GREP_MAX_LINE_LENGTH = 500;
+export interface TruncationResult {
+    /** The truncated content */
+    content: string;
+    /** Whether truncation occurred */
+    truncated: boolean;
+    /** Which limit was hit: "lines", "bytes", or null if not truncated */
+    truncatedBy: "lines" | "bytes" | null;
+    /** Total number of lines in the original content */
+    totalLines: number;
+    /** Total number of bytes in the original content */
+    totalBytes: number;
+    /** Number of complete lines in the truncated output */
+    outputLines: number;
+    /** Number of bytes in the truncated output */
+    outputBytes: number;
+    /** Whether the last line was partially truncated (only for tail truncation edge case) */
+    lastLinePartial: boolean;
+    /** Whether the first line exceeded the byte limit (for head truncation) */
+    firstLineExceedsLimit: boolean;
+    /** The max lines limit that was applied */
+    maxLines: number;
+    /** The max bytes limit that was applied */
+    maxBytes: number;
+}
+export interface TruncationOptions {
+    /** Maximum number of lines (default: 2000) */
+    maxLines?: number;
+    /** Maximum number of bytes (default: 50KB) */
+    maxBytes?: number;
+}
+/**
+ * Format bytes as human-readable size.
+ */
+export declare function formatSize(bytes: number): string;
+/**
+ * Truncate content from the head (keep first N lines/bytes).
+ * Suitable for file reads where you want to see the beginning.
+ *
+ * Never returns partial lines. If first line exceeds byte limit,
+ * returns empty content with firstLineExceedsLimit=true.
+ */
+export declare function truncateHead(content: string, options?: TruncationOptions): TruncationResult;
+/**
+ * Truncate content from the tail (keep last N lines/bytes).
+ * Suitable for bash output where you want to see the end (errors, final results).
+ *
+ * May return partial first line if the last line of original content exceeds byte limit.
+ */
+export declare function truncateTail(content: string, options?: TruncationOptions): TruncationResult;
+/**
+ * Truncate a single line to max characters, adding [truncated] suffix.
+ * Used for grep match lines.
+ */
+export declare function truncateLine(line: string, maxChars?: number): {
+    text: string;
+    wasTruncated: boolean;
+};