pi-local-agents-only 0.1.18 → 0.1.20

package/CHANGELOG.md

@@ -2,6 +2,18 @@
 
 ## Unreleased
 
+## 0.1.20 - 2026-06-15
+
+- updated the local pi development baseline to `@earendil-works/pi-coding-agent` `0.79.4` and refreshed the npm lockfile
+- validated project-trust no-UI/global-block stripping behavior and package load under pi `0.79.4`
+
+## 0.1.19 - 2026-06-04
+
+- updated the local pi development baseline to `@earendil-works/pi-coding-agent` `0.78.1` and regenerated the npm lockfile
+- aligned prompt rewriting with pi `0.78.1` system-prompt metadata by using `before_agent_start` `systemPromptOptions` for loaded global context files
+- removed legacy Markdown context stripping now that current pi prompt context uses XML boundaries
+- refreshed CI to test supported Node.js release lines only
+
 ## 0.1.18 - 2026-05-28
 
 - updated the local pi development baseline to `@earendil-works/pi-coding-agent` `0.77.0` and regenerated the npm lockfile

package/README.md

@@ -16,7 +16,7 @@ Or install it directly from GitHub with pi:
 pi install https://github.com/fitchmultz/pi-local-agents-only
 ```
 
-Compatibility note: this package is tested against the current pi release during each package update, and pi-bundled runtime packages are declared as optional wildcard peers. That keeps installs forward-open for future pi releases: npm peer ranges should not block users from trying a newer pi, though runtime behavior is only verified against the tested baseline until a follow-up package release confirms it.
+Compatibility note: this package is currently tested against pi `0.78.1` as the suggested floor, and pi-bundled runtime packages are declared as optional wildcard peers. That keeps installs forward-open for future pi releases: npm peer ranges should not block users from trying a newer pi, though runtime behavior is only verified against the tested baseline until a follow-up package release confirms it.
 
 ## Use
 

package/extensions/local-agents-only.js

@@ -5,7 +5,7 @@
  * Responsibilities: Detect repo and worktree opt-in state, manage repo and global toggles, add a local-only guardrail, and remove matching global context blocks before model calls.
  * Scope: Works as a pi extension package. It changes only the prompt the model sees, not pi's startup header.
  * Usage: Install the package, then use `/local-agents-only on|off|status|global-on|global-off`.
- * Invariants/Assumptions: pi injects context files as XML `<project_instructions path="...">` blocks in current releases and previously used Markdown `## /absolute/path` blocks; git worktrees that share a common git dir should share local-agents-only state.
+ * Invariants/Assumptions: pi injects context files as XML `<project_instructions path="...">` blocks in current releases; git worktrees that share a common git dir should share local-agents-only state.
  */
 
 import { execFileSync } from "node:child_process";
@@ -15,10 +15,10 @@ import { dirname, join, resolve } from "node:path";
 
 /** @typedef {import("@earendil-works/pi-coding-agent").ExtensionAPI} ExtensionAPI */
 /** @typedef {import("@earendil-works/pi-coding-agent").ExtensionContext} ExtensionContext */
+/** @typedef {import("@earendil-works/pi-coding-agent").BuildSystemPromptOptions} BuildSystemPromptOptions */
 /** @typedef {{ projects: string[]; repositories: string[] }} LocalAgentsOnlyConfig */
 /** @typedef {{ start: string; projectRoot: string; repoId: string; worktreeRoots: string[] }} ProjectState */
 /** @typedef {{ enabled: boolean; source: "env" | "marker" | "global-config" | "default" }} Mode */
-/** @typedef {{ path: string; start: number; end: number }} ContextBlock */
 /** @typedef {{ prompt: string; removedPaths: string[] }} StripResult */
 
 class ConfigError extends Error {
@@ -37,13 +37,9 @@ const MARKER = join(".pi", COMMAND);
 const GLOBAL_CONTEXT_FILES = ["AGENTS.md", "CLAUDE.md"];
 const ENV_TRUE = ["1", "true", "yes", "on"];
 const ENV_FALSE = ["0", "false", "no", "off"];
-const PROJECT_CONTEXT_HEADER = "\n\n# Project Context\n\nProject-specific instructions and guidelines:\n\n";
 const PROJECT_CONTEXT_XML_START = "<project_context>";
 const PROJECT_CONTEXT_XML_END = "</project_context>";
 const PROJECT_CONTEXT_XML_PREFIX = "<project_context>\n\nProject-specific instructions and guidelines:\n\n";
-const SKILLS_HEADER = "\n\nThe following skills provide specialized instructions for specific tasks.";
-const DATE_HEADER = "\nCurrent date:";
-const CONTEXT_BLOCK_HEADER = /^## ([^\n]+(?:AGENTS|CLAUDE)\.md)\n\n/gm;
 const CONTEXT_XML_BLOCK = /<project_instructions path="([^"]+(?:AGENTS|CLAUDE)\.md)">\n[\s\S]*?<\/project_instructions>\n*/g;
 const emptyConfig = () => ({ projects: [], repositories: [] });
 
@@ -257,28 +253,17 @@ const getExistingGlobalContextPaths = (agentDir = getAgentDir()) =>
 	getGlobalContextPaths(agentDir).filter((path) => existsSync(path));
 
 /**
- * @param {string} prompt
- * @param {number} offset
- * @returns {number}
+ * @param {BuildSystemPromptOptions | undefined} options
+ * @param {string} [agentDir]
+ * @returns {string[]}
  */
-const getContextSectionEnd = (prompt, offset) => {
-	const candidates = [prompt.indexOf(SKILLS_HEADER, offset), prompt.indexOf(DATE_HEADER, offset)].filter(
-		(index) => index !== -1,
+const getLoadedGlobalContextPaths = (options, agentDir = getAgentDir()) => {
+	const globalPathKeys = new Set(getGlobalContextPaths(agentDir).map(normalizePath));
+	return uniqueSorted(
+		(options?.contextFiles ?? [])
+			.map((file) => file.path)
+			.filter((path) => globalPathKeys.has(normalizePath(path))),
 	);
-	return candidates.length > 0 ? Math.min(...candidates) : prompt.length;
-};
-
-/**
- * @param {string} contextSection
- * @returns {ContextBlock[]}
- */
-const getContextBlocks = (contextSection) => {
-	const matches = [...contextSection.matchAll(CONTEXT_BLOCK_HEADER)];
-	return matches.map((match, index) => ({
-		path: match[1],
-		start: match.index ?? 0,
-		end: index + 1 < matches.length ? (matches[index + 1].index ?? contextSection.length) : contextSection.length,
-	}));
 };
 
 /**
@@ -286,7 +271,7 @@ const getContextBlocks = (contextSection) => {
  * @param {string[]} [globalPaths]
  * @returns {StripResult}
  */
-const stripXmlGlobalContext = (prompt, globalPaths = getGlobalContextPaths()) => {
+const stripGlobalContext = (prompt, globalPaths = getGlobalContextPaths()) => {
 	const contextTagStart = prompt.lastIndexOf(PROJECT_CONTEXT_XML_START);
 	if (contextTagStart === -1) {
 		return { prompt, removedPaths: [] };
@@ -312,7 +297,7 @@ const stripXmlGlobalContext = (prompt, globalPaths = getGlobalContextPaths()) =>
 		if (globalPathKeys.has(normalizePath(path))) {
 			removedPaths.push(path);
 		} else {
-			keptBlocks.push(blockText.endsWith("\n\n") ? blockText : `${blockText}\n`);
+			keptBlocks.push(blockText.trimEnd());
 		}
 	}
 	if (removedPaths.length === 0) {
@@ -321,68 +306,14 @@ const stripXmlGlobalContext = (prompt, globalPaths = getGlobalContextPaths()) =>
 	const prefix = prompt.slice(0, contextTagStart).replace(/\n\n$/u, "\n");
 	const suffix = prompt.slice(sectionEnd);
 	if (keptBlocks.length === 0) {
-		return { prompt: `${prefix}${suffix}`, removedPaths: uniqueSorted(removedPaths) };
+		return { prompt: `${prefix}${suffix.replace(/^\n/u, "")}`, removedPaths: uniqueSorted(removedPaths) };
 	}
 	return {
-		prompt: `${prefix}${PROJECT_CONTEXT_XML_PREFIX}${keptBlocks.join("\n")}${PROJECT_CONTEXT_XML_END}${suffix}`,
+		prompt: `${prefix}${PROJECT_CONTEXT_XML_PREFIX}${keptBlocks.join("\n\n")}\n${PROJECT_CONTEXT_XML_END}${suffix}`,
 		removedPaths: uniqueSorted(removedPaths),
 	};
 };
 
-/**
- * @param {string} prompt
- * @param {string[]} [globalPaths]
- * @returns {StripResult}
- */
-const stripMarkdownGlobalContext = (prompt, globalPaths = getGlobalContextPaths()) => {
-	const sectionStart = prompt.lastIndexOf(PROJECT_CONTEXT_HEADER);
-	if (sectionStart === -1) {
-		return { prompt, removedPaths: [] };
-	}
-	const contextStart = sectionStart + PROJECT_CONTEXT_HEADER.length;
-	const sectionEnd = getContextSectionEnd(prompt, contextStart);
-	const contextSection = prompt.slice(contextStart, sectionEnd);
-	const blocks = getContextBlocks(contextSection);
-	if (blocks.length === 0) {
-		return { prompt, removedPaths: [] };
-	}
-	const globalPathKeys = new Set(globalPaths.map(normalizePath));
-	/** @type {string[]} */
-	const keptBlocks = [];
-	/** @type {string[]} */
-	const removedPaths = [];
-	for (const block of blocks) {
-		const blockText = contextSection.slice(block.start, block.end);
-		if (globalPathKeys.has(normalizePath(block.path))) {
-			removedPaths.push(block.path);
-		} else {
-			keptBlocks.push(blockText);
-		}
-	}
-	if (removedPaths.length === 0) {
-		return { prompt, removedPaths: [] };
-	}
-	const prefix = prompt.slice(0, sectionStart);
-	const suffix = prompt.slice(sectionEnd);
-	if (keptBlocks.length === 0) {
-		return { prompt: `${prefix}${suffix}`, removedPaths: uniqueSorted(removedPaths) };
-	}
-	return {
-		prompt: `${prefix}${PROJECT_CONTEXT_HEADER}${keptBlocks.join("")}${suffix}`,
-		removedPaths: uniqueSorted(removedPaths),
-	};
-};
-
-/**
- * @param {string} prompt
- * @param {string[]} [globalPaths]
- * @returns {StripResult}
- */
-const stripGlobalContext = (prompt, globalPaths = getGlobalContextPaths()) => {
-	const xmlResult = stripXmlGlobalContext(prompt, globalPaths);
-	return xmlResult.removedPaths.length > 0 ? xmlResult : stripMarkdownGlobalContext(prompt, globalPaths);
-};
-
 /**
  * @param {string} start
  * @returns {string | undefined}
@@ -484,22 +415,31 @@ const buildLocalOnlyNotice = (paths = getExistingGlobalContextPaths(getAgentDir(
 /**
  * @param {string} prompt
  * @param {string} [agentDir]
+ * @param {BuildSystemPromptOptions} [systemPromptOptions]
  * @returns {string}
  */
-const applyLocalOnlyPrompt = (prompt, agentDir = getAgentDir()) => {
-	const { prompt: stripped, removedPaths } = stripGlobalContext(prompt, getGlobalContextPaths(agentDir));
+const applyLocalOnlyPrompt = (prompt, agentDir = getAgentDir(), systemPromptOptions = undefined) => {
+	const loadedGlobalPaths = getLoadedGlobalContextPaths(systemPromptOptions, agentDir);
+	const globalPaths = loadedGlobalPaths.length > 0 ? loadedGlobalPaths : getGlobalContextPaths(agentDir);
+	const { prompt: stripped, removedPaths } = stripGlobalContext(prompt, globalPaths);
 	const notice = buildLocalOnlyNotice(
-		removedPaths.length > 0 ? removedPaths : getExistingGlobalContextPaths(agentDir),
+		removedPaths.length > 0 ? removedPaths : loadedGlobalPaths.length > 0 ? loadedGlobalPaths : getExistingGlobalContextPaths(agentDir),
 	);
 	return notice ? `${stripped}\n\n${notice}` : stripped;
 };
 
+/** @param {ExtensionContext} ctx */
+const isProjectTrusted = (ctx) => typeof ctx.isProjectTrusted === "function" ? ctx.isProjectTrusted() : true;
+
+/** @param {ExtensionContext} ctx */
+const getTrustedMode = (ctx) => getMode(ctx.cwd, undefined, undefined, { projectTrusted: isProjectTrusted(ctx) });
+
 /** @param {ExtensionContext} ctx */
 const setStatus = (ctx) => {
 	if (!ctx.hasUI) {
 		return;
 	}
-	const mode = getMode(ctx.cwd);
+	const mode = getTrustedMode(ctx);
 	ctx.ui.setStatus(COMMAND, mode.enabled ? `AGENTS: local-only (${mode.source})` : undefined);
 };
 
@@ -563,15 +503,16 @@ export function findProjectRoot(start = process.cwd()) {
  * @param {string | ProjectState} [start]
  * @param {string | undefined} [envValue]
  * @param {string} [configPath]
+ * @param {{ projectTrusted?: boolean }} [options]
  * @returns {Mode}
  */
-export function getMode(start = process.cwd(), envValue = process.env.PI_LOCAL_AGENTS_ONLY, configPath = CONFIG()) {
+export function getMode(start = process.cwd(), envValue = process.env.PI_LOCAL_AGENTS_ONLY, configPath = CONFIG(), options = {}) {
 	const state = typeof start === "string" ? getProjectState(start) : start;
 	const envToggle = getEnvToggle(envValue);
 	if (envToggle !== undefined) {
 		return { enabled: envToggle, source: "env" };
 	}
-	if (hasMarker(state)) {
+	if (options.projectTrusted !== false && hasMarker(state)) {
 		return { enabled: true, source: "marker" };
 	}
 	const { projects, repositories } = readConfig(configPath);
@@ -642,7 +583,7 @@ export default function localAgentsOnly(pi) {
 					return;
 				}
 				case "status": {
-					const mode = getMode(state);
+					const mode = getMode(state, undefined, undefined, { projectTrusted: isProjectTrusted(ctx) });
 					ctx.ui.notify(
 						`local-agents-only: ${mode.enabled ? `enabled via ${mode.source}` : "disabled"} (${state.projectRoot})`,
 						"info",
@@ -666,6 +607,8 @@ export default function localAgentsOnly(pi) {
 	});
 
 	pi.on("before_agent_start", (event, ctx) => {
-		return getMode(ctx.cwd).enabled ? { systemPrompt: applyLocalOnlyPrompt(event.systemPrompt) } : undefined;
+		return getTrustedMode(ctx).enabled
+			? { systemPrompt: applyLocalOnlyPrompt(event.systemPrompt, getAgentDir(), event.systemPromptOptions) }
+			: undefined;
 	});
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pi-local-agents-only",
-  "version": "0.1.18",
+  "version": "0.1.20",
   "description": "Pi extension that strips global AGENTS.md and CLAUDE.md from the effective prompt for selected projects.",
   "author": "Mitch Fultz (https://github.com/fitchmultz)",
   "license": "MIT",
@@ -37,7 +37,7 @@
     "prepublishOnly": "npm run check"
   },
   "devDependencies": {
-    "@earendil-works/pi-coding-agent": "^0.77.0",
+    "@earendil-works/pi-coding-agent": "0.79.4",
     "@types/node": "^25.9.1",
     "typescript": "^6.0.3"
   },