pi-repl-py 0.3.0 → 0.4.0

package/index.ts

@@ -4,6 +4,7 @@ import { basename, join } from "node:path";
 import { homedir } from "node:os";
 import type { ExtensionAPI } from "@mariozechner/pi-coding-agent";
 import { Type } from "typebox";
+import { withSkillsBlock } from "./src/extension/skill-hook.js";
 import { EngineManager } from "./src/engine/index.js";
 import { ExecuteCellComponent, type ExecuteDetails, type ExecuteRenderState } from "./src/extension/render.js";
 import { EngineLifecycle } from "./src/extension/session-engine.js";
@@ -109,6 +110,16 @@ export default function (pi: ExtensionAPI) {
 		return { content: event.content, details: stashed.details, isError: true };
 	});
 
+	// --- pi gates skills on the read tool (absent in repl); re-emit them via withSkillsBlock. ---
+	pi.on("before_agent_start", (event) => {
+		if (!active()) return;
+		const systemPrompt = withSkillsBlock(
+			event.systemPrompt,
+			event.systemPromptOptions?.skills ?? [],
+		);
+		return systemPrompt === undefined ? undefined : { systemPrompt };
+	});
+
 	pi.registerTool<typeof executeSchema, ExecuteDetails, Partial<ExecuteRenderState>>({
 		name: "execute",
 		label: "execute",

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "pi-repl-py",
-	"version": "0.3.0",
+	"version": "0.4.0",
 	"type": "module",
 	"description": "A pi extension with a single tool: execute, running a TypeScript host with a persistent Python (ipykernel) evaluator and a user-configurable toolbox of functions.",
 	"keywords": [

package/src/extension/prompt.ts

@@ -6,14 +6,17 @@
 // more signal; the machine reads every line every turn.
 
 export const executeToolDescription =
-	"You have one tool: a real `ipython` kernel that stays alive across cells and turns. " +
-	"This persistent Python workspace is your only surface — it does the work of bash, read, write, edit, " +
-	"search, and file handling, and everything you define (variables, imports, helpers loaded from " +
-	"`~/.pi/agent/pi-repl/helpers/`) survives for reuse in later cells. A cell returns its final expression; " +
-	"printed output is captured separately.";
+	"Execute Python cells in a persistent ipython kernel that stays alive across cells and turns. " +
+	"It replaces the default read, bash, edit, write, and search tools — file work, shell commands, and " +
+	"searches all run as Python. Everything you define (variables, imports, and helpers preloaded into the " +
+	"workspace namespace) survives for reuse in later cells. A cell returns its final expression; printed " +
+	"output is captured separately. Oversized output is truncated: 1,000,000 characters per cell, 4,096 per " +
+	"line. Reads are expensive — every printed value enters the context, so hold artifacts in variables, " +
+	"parse before printing, and print only the small bounded slice the next decision needs. Keep cells lean; " +
+	"full-file dumps and raw result lists bloat the conversation.";
 
 export const executePromptSnippet =
-	"The persistent Python workspace is your only tool: keep artifacts in variables across cells for reuse, use the loaded helpers, prefer surgical reads/edits over full-file dumps and rewrites, and parse before you print so context stays lean.";
+	"Execute Python cells in a persistent ipython kernel (replaces read, bash, edit, write, and search; state survives across cells and turns)";
 
 // --- the workspace doctrine riding the execute tool ---
 export function buildPromptGuidelines(preloaded: string[]): string[] {
@@ -45,7 +48,7 @@ export function buildPromptGuidelines(preloaded: string[]): string[] {
 		...(preloaded.length
 			? [
 					"## Helpers",
-					"These helpers are given to you by the user to use directly (loaded from `~/.pi/agent/pi-repl/helpers/`). Descriptions appear below.",
+					"These helpers are already defined in the workspace namespace. Use them by name as you would any other loaded function, class, or variable. Their code already executed at kernel boot. Descriptions appear below.",
 					"",
 					...preloaded,
 					"",

package/src/extension/skill-hook.ts

@@ -0,0 +1,22 @@
+// --- skills cannot reach the prompt in --repl: pi gates <available_skills> on the read tool
+// (hasRead), which repl doesn't have. Re-emit them with pi's own formatter in pi's slot. ---
+import { formatSkillsForPrompt, type Skill } from "@mariozechner/pi-coding-agent";
+
+const CWD_MARKER = "\nCurrent working directory:"; // skills sit just before this, pi's last line
+const READ_LINE = "Use the read tool to load a skill's file when the task matches its description."; // canon line
+const EXECUTE_LINE = "Load a skill's SKILL.md file contents via execute (read the file with Python)."; // repl has no read
+
+/** The prompt with the skills block in pi's slot; undefined if nothing should change. */
+export function withSkillsBlock(
+	prompt: string,
+	skills: Skill[],
+	alreadyPresent = prompt.includes("<available_skills>"),
+): string | undefined {
+	if (skills.length === 0) return undefined;
+	let extra = formatSkillsForPrompt(skills);
+	if (!extra) return undefined;
+	if (alreadyPresent) return undefined;
+	extra = extra.replace(READ_LINE, EXECUTE_LINE);
+	const idx = prompt.indexOf(CWD_MARKER);
+	return idx === -1 ? prompt + extra : prompt.slice(0, idx) + extra + prompt.slice(idx);
+}