pi-run

This is a Pi extension for Pi extension developers. It enables running Typescript within the Node.js context of the Pi coding agent

Installation

pi install npm:@vamsitalupula/pi-run

How to use?

1. Create a new file by running /pi-run ./debug.ts in Pi user prompt
2. You should see a new file called debug.ts created in the current folder

import type {
    ExtensionAPI,
    ExtensionContext,
} from "@earendil-works/pi-coding-agent";

export default function (pi: ExtensionAPI, getCtx: () => ExtensionContext) {
    // Your code here
    // return getCtx().getSystemPrompt();
    return pi.getActiveTools();
}

3. Now run the file with /pi-run ./debug.ts
4. The data/object returned by the default export function is shown as a pi notification
5. The package @earendil-works/pi-coding-agent is available in the node environment running pi. But if you need LSP/linting support in your editor, you need to point your editor tools to the install location of the package one way or an other. The easiest is to just run pnpm i --save-dev @earendil-works/pi-coding-agent in the project or local directory

Examples

To see the runtime system prompt

export default function (pi: ExtensionAPI, getCtx: () => ExtensionContext) {
    return getCtx().getSystemPrompt();
}

To log tool results to a file

import { appendFileSync } from "node:fs";

export default function (pi: ExtensionAPI, getCtx: () => ExtensionContext) {
    pi.on("tool_result", async (event, ctx) => {
        const logEntry = JSON.stringify({
            timestamp: new Date().toISOString(),
            toolName: event.toolName,
            toolCallId: event.toolCallId,
            isError: event.isError,
            content: event.content,
        }, null, 2);

        appendFileSync("/tmp/pi-tool-results.log", logEntry + "\n");
    });
}

To get assistant message details with decreasing cache reads

    const ctx = getCtx();
    const branch = ctx.sessionManager.getBranch();

    const messages = branch
        .filter((entry) => entry.type === "message")
        .map((entry) => entry.message);

    return messages
        .slice(1)
        .filter((msg, index) => {
            if (msg.role !== "assistant" || !msg.usage) return false;

            const prevMsg = messages[index];
            const prevCache =
                prevMsg?.role === "assistant"
                    ? (prevMsg.usage?.cacheRead ?? 0)
                    : 0;
            const currCache = msg.usage.cacheRead ?? 0;

            return currCache > prevCache;
        })
        .map((msg) => {
            return {
                timestamp: msg.timestamp,
                responseId: msg.responseId,
                usage: msg.usage,
            };
        });