@selesai/code 0.13.13 → 0.13.14

package/dist/core/tools/read.js

@@ -0,0 +1,327 @@
+import { basename, dirname, isAbsolute, relative, resolve as resolvePath, sep } from "node:path";
+import { Text } from "@earendil-works/pi-tui";
+import { constants } from "fs";
+import { access as fsAccess, readFile as fsReadFile } from "fs/promises";
+import { Type } from "typebox";
+import { getReadmePath } from "../../config.js";
+import { keyHint, keyText } from "../../modes/interactive/components/keybinding-hints.js";
+import { getLanguageFromPath, highlightCode } from "../../modes/interactive/theme/theme.js";
+import { processImage } from "../../utils/image-process.js";
+import { detectSupportedImageMimeTypeFromFile } from "../../utils/mime.js";
+import { formatPathRelativeToCwdOrAbsolute } from "../../utils/paths.js";
+import { getExperimentalToolSampling } from "../experimental.js";
+import { captionImageWithModel } from "../vision-caption.js";
+import { resolveReadPathAsync, resolveToCwd } from "./path-utils.js";
+import { getTextOutput, renderToolPath, replaceTabs, 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" })),
+});
+const COMPACT_RESOURCE_FILE_NAMES = new Set(["AGENTS.md", "AGENTS.MD", "CLAUDE.md", "CLAUDE.MD"]);
+const defaultReadOperations = {
+    readFile: (path) => fsReadFile(path),
+    access: (path) => fsAccess(path, constants.R_OK),
+    detectImageMimeType: detectSupportedImageMimeTypeFromFile,
+};
+function formatReadLineRange(args, theme) {
+    if (args?.offset === undefined && args?.limit === undefined)
+        return "";
+    const startLine = args.offset ?? 1;
+    const endLine = args.limit !== undefined ? startLine + args.limit - 1 : "";
+    return theme.fg("warning", `:${startLine}${endLine ? `-${endLine}` : ""}`);
+}
+function formatReadCall(args, theme, cwd) {
+    const pathDisplay = renderToolPath(str(args?.file_path ?? args?.path), theme, cwd);
+    return `${theme.fg("toolTitle", theme.bold("read"))} ${pathDisplay}${formatReadLineRange(args, theme)}`;
+}
+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.]";
+}
+/**
+ * Ask a vision model to describe an image. Returns the caption text, or null if
+ * the caption model/credentials are unavailable or the request fails.
+ */
+export async function captionImage(image, captionModelId, ctx, signal) {
+    if (!captionModelId || !ctx?.modelRegistry || !ctx.model)
+        return null;
+    const slash = captionModelId.indexOf("/");
+    if (slash <= 0 || slash === captionModelId.length - 1)
+        return null;
+    const provider = captionModelId.slice(0, slash);
+    const modelId = captionModelId.slice(slash + 1);
+    const captionModel = ctx.modelRegistry.find(provider, modelId);
+    if (!captionModel || !captionModel.input.includes("image"))
+        return null;
+    let auth;
+    try {
+        auth = await ctx.modelRegistry.getApiKeyAndHeaders(captionModel);
+    }
+    catch {
+        return null;
+    }
+    if (!auth.ok)
+        return null;
+    return captionImageWithModel(captionModel, image, {
+        apiKey: auth.apiKey,
+        headers: auth.headers,
+        signal,
+    });
+}
+function toPosixPath(filePath) {
+    return filePath.split(sep).join("/");
+}
+function getPiDocsClassification(absolutePath) {
+    const packageRoot = dirname(getReadmePath());
+    const relativePath = relative(resolvePath(packageRoot), resolvePath(absolutePath));
+    if (relativePath === "" ||
+        relativePath === ".." ||
+        relativePath.startsWith(`..${sep}`) ||
+        isAbsolute(relativePath)) {
+        return undefined;
+    }
+    const label = toPosixPath(relativePath);
+    if (label === "README.md" || label.startsWith("docs/") || label.startsWith("examples/")) {
+        return { kind: "docs", label };
+    }
+    return undefined;
+}
+function getCompactReadClassification(args, cwd) {
+    const rawPath = str(args?.file_path ?? args?.path);
+    if (!rawPath)
+        return undefined;
+    const absolutePath = resolveToCwd(rawPath, cwd);
+    const fileName = basename(absolutePath);
+    if (fileName === "SKILL.md") {
+        return { kind: "skill", label: basename(dirname(absolutePath)) || fileName };
+    }
+    const docsClassification = getPiDocsClassification(absolutePath);
+    if (docsClassification)
+        return docsClassification;
+    if (COMPACT_RESOURCE_FILE_NAMES.has(fileName)) {
+        return { kind: "resource", label: formatPathRelativeToCwdOrAbsolute(absolutePath, cwd) };
+    }
+    return undefined;
+}
+function formatCompactReadCall(classification, args, theme) {
+    const expandHint = theme.fg("dim", ` (${keyText("app.tools.expand")} to expand)`);
+    if (classification.kind === "skill") {
+        return (theme.fg("customMessageLabel", `\x1b[1m[skill]\x1b[22m `) +
+            theme.fg("customMessageText", classification.label) +
+            formatReadLineRange(args, theme) +
+            expandHint);
+    }
+    return (theme.fg("toolTitle", theme.bold(`read ${classification.kind}`)) +
+        " " +
+        theme.fg("accent", classification.label) +
+        formatReadLineRange(args, theme) +
+        expandHint);
+}
+function formatReadResult(args, result, options, theme, showImages, _cwd, isError) {
+    if (!options.expanded && !isError) {
+        return "";
+    }
+    const rawPath = str(args?.file_path ?? args?.path);
+    const output = getTextOutput(result, showImages);
+    const lang = !isError && 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")}${theme.fg("muted", ")")}`;
+    }
+    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 imageCaptionModel = options?.imageCaptionModel;
+    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, bmp). 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). 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."],
+        parameters: readSchema,
+        constrainedSampling: getExperimentalToolSampling(),
+        async execute(_toolCallId, { path, offset, limit }, signal, _onUpdate, ctx) {
+            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 {
+                        const absolutePath = await resolveReadPathAsync(path, cwd);
+                        if (aborted)
+                            return;
+                        // 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 processed = await processImage(buffer, mimeType, { autoResizeImages });
+                            if (!processed.ok) {
+                                let textNote = `Read image file [${mimeType}]\n${processed.message}`;
+                                if (nonVisionImageNote)
+                                    textNote += `\n${nonVisionImageNote}`;
+                                content = [{ type: "text", text: textNote }];
+                            }
+                            else {
+                                let textNote = `Read image file [${processed.mimeType}]`;
+                                if (processed.hints.length > 0)
+                                    textNote += `\n${processed.hints.join("\n")}`;
+                                const image = { type: "image", data: processed.data, mimeType: processed.mimeType };
+                                // Vision relay: when the active model cannot accept images, ask a
+                                // configured vision model to describe the image and use that text instead.
+                                if (nonVisionImageNote) {
+                                    const caption = await captionImage(image, imageCaptionModel, ctx, signal);
+                                    if (caption) {
+                                        content = [
+                                            {
+                                                type: "text",
+                                                text: `Read image file [${processed.mimeType}] (described by vision model)\n${caption}`,
+                                            },
+                                        ];
+                                        if (aborted)
+                                            return;
+                                        resolve({ content, details });
+                                        return;
+                                    }
+                                }
+                                if (nonVisionImageNote)
+                                    textNote += `\n${nonVisionImageNote}`;
+                                content = [
+                                    { type: "text", text: textNote },
+                                    image,
+                                ];
+                            }
+                        }
+                        else {
+                            // Read text content.
+                            const buffer = await ops.readFile(absolutePath);
+                            const textContent = buffer.toString("utf-8");
+                            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 = truncation.content;
+                                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 = `${truncation.content}\n\n[${remaining} more lines in file. Use offset=${nextOffset} to continue.]`;
+                            }
+                            else {
+                                // No truncation and no remaining user-limited content.
+                                outputText = truncation.content;
+                            }
+                            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);
+            const classification = !context.expanded ? getCompactReadClassification(args, context.cwd) : undefined;
+            text.setText(classification
+                ? formatCompactReadCall(classification, args, theme)
+                : formatReadCall(args, theme, context.cwd));
+            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, context.cwd, context.isError));
+            return text;
+        },
+    };
+}
+export function createReadTool(cwd, options) {
+    return wrapToolDefinition(createReadToolDefinition(cwd, options));
+}

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

@@ -0,0 +1,23 @@
+import type { ImageContent, TextContent } from "@earendil-works/pi-ai";
+import type { Theme } from "../../modes/interactive/theme/theme.ts";
+export declare function shortenPath(path: unknown): string;
+export declare function linkPath(styledText: string, rawPath: string, cwd: string): 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: Theme): string;
+export declare function renderToolPath(rawPath: string | null, theme: Theme, cwd: string, options?: {
+    emptyFallback?: string;
+}): string;

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

@@ -0,0 +1,64 @@
+import * as os from "node:os";
+import { pathToFileURL } from "node:url";
+import { getCapabilities, getImageDimensions, hyperlink, imageFallback } from "@earendil-works/pi-tui";
+import { stripAnsi } from "../../utils/ansi.js";
+import { resolvePath } from "../../utils/paths.js";
+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 linkPath(styledText, rawPath, cwd) {
+    if (!getCapabilities().hyperlinks)
+        return styledText;
+    const absolutePath = resolvePath(rawPath, cwd);
+    return hyperlink(styledText, pathToFileURL(absolutePath).href);
+}
+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]");
+}
+export function renderToolPath(rawPath, theme, cwd, options) {
+    if (rawPath === null)
+        return invalidArgText(theme);
+    const value = rawPath || options?.emptyFallback;
+    if (!value)
+        return theme.fg("toolOutput", "...");
+    return linkPath(theme.fg("accent", shortenPath(value)), value, cwd);
+}

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

@@ -0,0 +1,13 @@
+import type { AgentTool } from "@earendil-works/pi-agent-core";
+import type { ExtensionContext, ToolDefinition } from "../extensions/types.ts";
+/** 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 AgentSession's 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,35 @@
+/** Wrap a ToolDefinition into an AgentTool for the core runtime. */
+export function wrapToolDefinition(definition, ctxFactory) {
+    return {
+        name: definition.name,
+        label: definition.label,
+        description: definition.description,
+        parameters: definition.parameters,
+        constrainedSampling: definition.constrainedSampling,
+        prepareArguments: definition.prepareArguments,
+        executionMode: definition.executionMode,
+        execute: (toolCallId, params, signal, onUpdate, ctx) => definition.execute(toolCallId, params, signal, onUpdate, ctx ?? 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 AgentSession's 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,
+        description: tool.description,
+        parameters: tool.parameters,
+        constrainedSampling: tool.constrainedSampling,
+        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;
+};