Created and extended greet tool extension
Public
Created and extended greet tool extension
Agent
Pi
Model
Claude Opus 4.5 (latest)
Started
Jan 20, 11:25 PM
Messages60
Downloads0
Created greet tool extension returning 'hi'.
BuildUsed greet tool and confirmed output.
VerifyAdded custom tool call and result renderer with /reads command.
Build>rm .pi/extensions/greet.ts(no output)
i want you to create an extension .pi/extensions, called greet.ts. it should have a custom tool "greet" that just returns hi when called.
The user wants me to create a custom extension with a tool called "greet" that returns "hi". I should first read the extensions documentation to understand the proper format.
Now I have a good understanding of how to create an extension with a custom tool. Let me create a simple greet extension that just returns "hi" when called.
Done. Created a simple extension with a "greet" tool that returns "hi" when called.
Created .pi/extensions/greet.ts with a greet tool that returns "hi" when called. The tool takes no parameters and will be auto-discovered when you run pi.
ok, use the tool
The user wants me to use the greet tool that was just created.
hiThe greet tool worked and returned "hi" as expected.
Done. The greet tool returned "hi".
ok, add a custom tool call and result renderer, just think about something simple
The user wants me to add custom renderCall and renderResult methods to the greet tool. Let me think of something simple but nice - maybe show a greeting emoji or styled text.
Let me add simple custom renderers. I'll make:
Done. Added custom renderers:
Updated the extension with custom renderers:
use the tool agian
The user wants me to use the greet tool again to see the custom rendering.
hiThe tool worked and returned "hi". The custom renderers should have displayed the styled output in the TUI.
Done. You should see the custom rendering with the 👋 and ✓ styled output.
add a /reads slash command, that goes through the active branch entries, and finds all assistant messages and the tool calls in them. and list the read files read via the read tool. you can find the types in @packages/ai/src/types.ts
The user wants me to add a slash command /reads that:
Let me first look at the types in packages/ai/src/types.ts to understand the structure of messages and tool calls.
>rg -n "role.*assistant|ToolCall|AssistantMessage" /Users/badlogic/workspaces/pi-mono/packages/ai/src/types.ts | head -509:import type { AssistantMessageEventStream } from "./utils/event-stream.js";
11:export type { AssistantMessageEventStream } from "./utils/event-stream.js";
113:) => AssistantMessageEventStream;
133:export interface ToolCall {
164:export interface AssistantMessage {
165: role: "assistant";
166: content: (TextContent | ThinkingContent | ToolCall)[];
186:export type Message = UserMessage | AssistantMessage | ToolResultMessage;
202:export type AssistantMessageEvent =
203: | { type: "start"; partial: AssistantMessage }
204: | { type: "text_start"; contentIndex: number; partial: AssistantMessage }
205: | { type: "text_delta"; contentIndex: number; delta: string; partial: AssistantMessage }
206: | { type: "text_end"; contentIndex: number; content: string; partial: AssistantMessage }
207: | { type: "thinking_start"; contentIndex: number; partial: AssistantMessage }
208: | { type: "thinking_delta"; contentIndex: number; delta: string; partial: AssistantMessage }
209: | { type: "thinking_end"; contentIndex: number; content: string; partial: AssistantMessage }
210: | { type: "toolcall_start"; contentIndex: number; partial: AssistantMessage }
211: | { type: "toolcall_delta"; contentIndex: number; delta: string; partial: AssistantMessage }
212: | { type: "toolcall_end"; contentIndex: number; toolCall: ToolCall; partial: AssistantMessage }
213: | { type: "done"; reason: Extract<StopReason, "stop" | "length" | "toolUse">; message: AssistantMessage }
214: | { type: "error"; reason: Extract<StopReason, "aborted" | "error">; error: AssistantMessage };Now let me look at the ToolCall interface more closely and also check what the session entries look like.
Now I understand the structure. Let me check what the session entries look like in the coding-agent. I need to understand how to access them from an extension.
>rg -n "getBranch|getEntries|entry.type" /Users/badlogic/workspaces/pi-mono/packages/coding-agent/src --include="*.ts" | head -30rg: unrecognized flag --include
similar flags that are available: --include-zeroLet me fix that - rg uses -g for glob patterns.
>rg -n "getBranch|getEntries|entry\.type" /Users/badlogic/workspaces/pi-mono/packages/coding-agent/src -g "*.ts" | head -40/Users/badlogic/workspaces/pi-mono/packages/coding-agent/src/modes/interactive/components/footer.ts:72: for (const entry of this.session.sessionManager.getEntries()) {
/Users/badlogic/workspaces/pi-mono/packages/coding-agent/src/modes/interactive/components/footer.ts:73: if (entry.type === "message" && entry.message.role === "assistant") {
/Users/badlogic/workspaces/pi-mono/packages/coding-agent/src/modes/interactive/components/tree-selector.ts:167: if (entry.type === "message" && entry.message.role === "assistant") {
/Users/badlogic/workspaces/pi-mono/packages/coding-agent/src/modes/interactive/components/tree-selector.ts:253: if (entry.type === "message" && entry.message.role === "assistant" && !isCurrentLeaf) {
/Users/badlogic/workspaces/pi-mono/packages/coding-agent/src/modes/interactive/components/tree-selector.ts:267: entry.type === "label" ||
/Users/badlogic/workspaces/pi-mono/packages/coding-agent/src/modes/interactive/components/tree-selector.ts:268: entry.type === "custom" ||
/Users/badlogic/workspaces/pi-mono/packages/coding-agent/src/modes/interactive/components/tree-selector.ts:269: entry.type === "model_change" ||
/Users/badlogic/workspaces/pi-mono/packages/coding-agent/src/modes/interactive/components/tree-selector.ts:270: entry.type === "thinking_level_change";
/Users/badlogic/workspaces/pi-mono/packages/coding-agent/src/modes/interactive/components/tree-selector.ts:275: passesFilter = entry.type === "message" && entry.message.role === "user";
/Users/badlogic/workspaces/pi-mono/packages/coding-agent/src/modes/interactive/components/tree-selector.ts:279: passesFilter = !isSettingsEntry && !(entry.type === "message" && entry.message.role === "toolResult");
/Users/badlogic/workspaces/pi-mono/packages/coding-agent/src/modes/interactive/components/tree-selector.ts:460: switch (entry.type) {
/Users/badlogic/workspaces/pi-mono/packages/coding-agent/src/modes/interactive/components/tree-selector.ts:634: switch (entry.type) {
/Users/badlogic/workspaces/pi-mono/packages/coding-agent/src/modes/interactive/interactive-mode.ts:2112: const allEntries = this.sessionManager.getEntries();
/Users/badlogic/workspaces/pi-mono/packages/coding-agent/src/modes/interactive/interactive-mode.ts:2928: if (entry.type !== "label" && entry.type !== "custom") break;
/Users/badlogic/workspaces/pi-mono/packages/coding-agent/src/modes/interactive/interactive-mode.ts:3765: const entries = this.sessionManager.getEntries();
/Users/badlogic/workspaces/pi-mono/packages/coding-agent/src/core/session-manager.ts:188: | "getBranch"
/Users/badlogic/workspaces/pi-mono/packages/coding-agent/src/core/session-manager.ts:190: | "getEntries"
/Users/badlogic/workspaces/pi-mono/packages/coding-agent/src/core/session-manager.ts:211: if (entry.type === "session") {
/Users/badlogic/workspaces/pi-mono/packages/coding-agent/src/core/session-manager.ts:221: if (entry.type === "compaction") {
/Users/badlogic/workspaces/pi-mono/packages/coding-agent/src/core/session-manager.ts:237: if (entry.type === "session") {
/Users/badlogic/workspaces/pi-mono/packages/coding-agent/src/core/session-manager.ts:243: if (entry.type === "message") {
/Users/badlogic/workspaces/pi-mono/packages/coding-agent/src/core/session-manager.ts:350: if (entry.type === "thinking_level_change") {
/Users/badlogic/workspaces/pi-mono/packages/coding-agent/src/core/session-manager.ts:352: } else if (entry.type === "model_change") {
/Users/badlogic/workspaces/pi-mono/packages/coding-agent/src/core/session-manager.ts:354: } else if (entry.type === "message" && entry.message.role === "assistant") {
/Users/badlogic/workspaces/pi-mono/packages/coding-agent/src/core/session-manager.ts:356: } else if (entry.type === "compaction") {
/Users/badlogic/workspaces/pi-mono/packages/coding-agent/src/core/session-manager.ts:369: if (entry.type === "message") {
/Users/badlogic/workspaces/pi-mono/packages/coding-agent/src/core/session-manager.ts:371: } else if (entry.type === "custom_message") {
/Users/badlogic/workspaces/pi-mono/packages/coding-agent/src/core/session-manager.ts:375: } else if (entry.type === "branch_summary" && entry.summary) {
/Users/badlogic/workspaces/pi-mono/packages/coding-agent/src/core/session-manager.ts:528: if (entry.type === "session_info") {
/Users/badlogic/workspaces/pi-mono/packages/coding-agent/src/core/session-manager.ts:535: if (entry.type !== "message") continue;
/Users/badlogic/workspaces/pi-mono/packages/coding-agent/src/core/session-manager.ts:696: if (entry.type === "session") continue;
/Users/badlogic/workspaces/pi-mono/packages/coding-agent/src/core/session-manager.ts:699: if (entry.type === "label") {
/Users/badlogic/workspaces/pi-mono/packages/coding-agent/src/core/session-manager.ts:856: const entries = this.getEntries();
/Users/badlogic/workspaces/pi-mono/packages/coding-agent/src/core/session-manager.ts:859: if (entry.type === "session_info" && entry.name) {
/Users/badlogic/workspaces/pi-mono/packages/coding-agent/src/core/session-manager.ts:961: getBranch(fromId?: string): SessionEntry[] {
/Users/badlogic/workspaces/pi-mono/packages/coding-agent/src/core/session-manager.ts:977: return buildSessionContext(this.getEntries(), this.leafId, this.byId);
/Users/badlogic/workspaces/pi-mono/packages/coding-agent/src/core/session-manager.ts:993: getEntries(): SessionEntry[] {
/Users/badlogic/workspaces/pi-mono/packages/coding-agent/src/core/session-manager.ts:1003: const entries = this.getEntries();
/Users/badlogic/workspaces/pi-mono/packages/coding-agent/src/core/session-manager.ts:1097: const path = this.getBranch(leafId);
/Users/badlogic/workspaces/pi-mono/packages/coding-agent/src/core/session-manager.ts:1264: if (entry.type !== "session") {Good, I can see that:
ctx.sessionManager.getBranch() returns entries on the current branchtype field, and for messages it's "message" with an entry.message containing the actual messageentry.message.content which is an array of TextContent | ThinkingContent | ToolCalltype: "toolCall" and a name propertyNow let me look at how the read tool stores its arguments to know what to extract.
>rg -n "parameters.*Type\.Object" /Users/badlogic/workspaces/pi-mono/packages/coding-agent/src/core/tools/read.ts | head -5(no output)Let me look at the read tool directly to see its parameters structure.
Good, so the read tool takes a path parameter. Now I have everything I need to create the /reads command. Let me update the extension.
Now I have the info I need. Let me update the extension:
Now I'll add the /reads command that:
Done. Added the /reads command that:
ctx.sessionManager.getBranch()path argument from each tool callThe user can now use /reads to see all files that have been read in the current session.
Added the /reads command. It:
ctx.sessionManager.getBranch()Try it with /reads to see all files read in this session.