pi-live-context 0.0.1

package/CHANGELOG.md

@@ -0,0 +1 @@
+# pi-live-context

package/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2026 Braden Lamb
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

package/README.md

@@ -0,0 +1,40 @@
+# pi-live-context
+
+A [Pi](https://pi.dev) extension that reloads `AGENTS.md` every turn and sends it as a user message instead of system prompt text.
+
+## Install
+
+```bash
+# from npm
+pi install npm:pi-live-context
+
+# or from git
+pi install git:github.com/bradennss/pi-live-context
+
+# try it for a single run without installing
+pi -e npm:pi-live-context
+```
+
+## How it works
+
+Pi discovers `AGENTS.md` and `CLAUDE.md` the same way it usually does, but sends it just before your message instead of including it in the system prompt. This makes agents follow global and project instructions more reliably, with the tradeoff of consuming more input tokens over time.
+
+## Requirements
+
+- Node.js >= 20.
+- Pi 0.85 or newer.
+
+## Development
+
+```bash
+pnpm install
+pnpm run check
+```
+
+## Contributing
+
+Every change that affects the published package needs a [changeset](https://github.com/changesets/changesets):
+
+```bash
+pnpm changeset
+```

package/SECURITY.md

@@ -0,0 +1,13 @@
+# Security Policy
+
+## Scope and behavior
+
+`pi-live-context` is a Pi extension. Like all Pi extensions it runs with your full user permissions. This extension:
+
+- Reads the `AGENTS.md` and `CLAUDE.md` files Pi already discovers, through Pi's own `loadProjectContextFiles`, once per provider request.
+- Removes that content from the system prompt and sends it as a hidden message instead. The content still goes to your model provider, in the same request it would have gone in before.
+- Writes nothing to disk, makes no network calls of its own, collects no telemetry, and reads no credentials. It has no runtime dependencies beyond Pi itself.
+
+## Reporting a vulnerability
+
+Please report suspected vulnerabilities privately via a [GitHub security advisory](https://github.com/bradennss/pi-live-context/security/advisories/new). You will receive an acknowledgement within a few days.

package/index.ts

@@ -0,0 +1,59 @@
+/**
+ * pi-live-context takes the AGENTS.md files out of the system prompt and sends
+ * them as a hidden message before the newest user message, reloaded from disk
+ * before every request.
+ */
+import {
+  type ExtensionAPI,
+  type ExtensionContext,
+  getAgentDir,
+  loadProjectContextFiles,
+} from "@earendil-works/pi-coding-agent";
+import {
+  buildContextMessage,
+  insertBeforeLastUserMessage,
+} from "./src/context-message.ts";
+import { stripContextFiles } from "./src/system-prompt.ts";
+
+const LAYOUT_WARNING =
+  "pi-live-context is off for this session: this version of Pi lays out context files in a way the extension cannot remove from the system prompt.";
+
+export default function liveContext(pi: ExtensionAPI): void {
+  let relocating = false;
+  let warned = false;
+
+  function warnOnce(ctx: ExtensionContext): void {
+    if (warned) {
+      return;
+    }
+    warned = true;
+    ctx.ui.notify(LAYOUT_WARNING, "warning");
+  }
+
+  pi.on("before_agent_start", (event, ctx) => {
+    const contextFiles = event.systemPromptOptions.contextFiles ?? [];
+    const systemPrompt = stripContextFiles(event.systemPrompt, contextFiles);
+    relocating = contextFiles.length > 0 && systemPrompt !== event.systemPrompt;
+    if (contextFiles.length > 0 && !relocating) {
+      warnOnce(ctx);
+    }
+    return relocating ? { systemPrompt } : undefined;
+  });
+
+  pi.on("context", (event, ctx) => {
+    if (!relocating) {
+      return;
+    }
+    const agentDir = getAgentDir();
+    const contextFiles = loadProjectContextFiles({ cwd: ctx.cwd, agentDir });
+    if (contextFiles.length === 0) {
+      return;
+    }
+    return {
+      messages: insertBeforeLastUserMessage(
+        event.messages,
+        buildContextMessage(contextFiles, agentDir),
+      ),
+    };
+  });
+}

package/package.json

@@ -0,0 +1,74 @@
+{
+  "name": "pi-live-context",
+  "version": "0.0.1",
+  "description": "Pi extension that reloads AGENTS.md every turn and sends it as a user message instead of system prompt text.",
+  "type": "module",
+  "license": "MIT",
+  "author": "Braden Lamb <hi@braden.lol>",
+  "repository": {
+    "type": "git",
+    "url": "git+https://github.com/bradennss/pi-live-context.git"
+  },
+  "homepage": "https://github.com/bradennss/pi-live-context#readme",
+  "bugs": {
+    "url": "https://github.com/bradennss/pi-live-context/issues"
+  },
+  "keywords": [
+    "pi-package",
+    "pi",
+    "pi-extension",
+    "pi-coding-agent",
+    "agents-md",
+    "context",
+    "system-prompt"
+  ],
+  "exports": {
+    ".": {
+      "types": "./index.ts",
+      "import": "./index.ts",
+      "default": "./index.ts"
+    }
+  },
+  "files": [
+    "index.ts",
+    "src",
+    "README.md",
+    "LICENSE",
+    "SECURITY.md",
+    "CHANGELOG.md"
+  ],
+  "engines": {
+    "node": ">=20"
+  },
+  "pi": {
+    "extensions": [
+      "./index.ts"
+    ]
+  },
+  "peerDependencies": {
+    "@earendil-works/pi-coding-agent": "*"
+  },
+  "devDependencies": {
+    "@changesets/changelog-github": "^1.0.0",
+    "@changesets/cli": "^3.0.1",
+    "@earendil-works/pi-coding-agent": "^0.85.1",
+    "@eslint/js": "^10.0.1",
+    "@types/node": "^22.0.0",
+    "eslint": "^10.8.1",
+    "eslint-config-prettier": "^10.1.8",
+    "husky": "^9.1.7",
+    "prettier": "^3.9.6",
+    "typescript": "^5.6.0",
+    "typescript-eslint": "^8.67.0",
+    "vitest": "^4.1.11"
+  },
+  "scripts": {
+    "typecheck": "tsc --noEmit",
+    "lint": "eslint . --max-warnings 0",
+    "format": "prettier --write .",
+    "format:check": "prettier --check .",
+    "test": "vitest run",
+    "test:watch": "vitest",
+    "check": "pnpm run typecheck && pnpm run lint && pnpm run format:check && pnpm test"
+  }
+}

package/src/context-file.ts

@@ -0,0 +1,4 @@
+export interface ContextFile {
+  path: string;
+  content: string;
+}

package/src/context-message.ts

@@ -0,0 +1,83 @@
+import path from "node:path";
+import type { ContextEvent } from "@earendil-works/pi-coding-agent";
+import type { ContextFile } from "./context-file.ts";
+
+export const CONTEXT_MESSAGE_TYPE = "live-context";
+
+type AgentMessage = ContextEvent["messages"][number];
+type CustomMessage = Extract<AgentMessage, { role: "custom" }>;
+
+function isGlobal(file: ContextFile, agentDir: string): boolean {
+  return path.resolve(path.dirname(file.path)) === path.resolve(agentDir);
+}
+
+function renderBlock(
+  tag: string,
+  heading: string,
+  contextFiles: ContextFile[],
+): string {
+  const files = contextFiles
+    .map(
+      ({ path: filePath, content }) =>
+        `<instructions path="${filePath}">\n${content}\n</instructions>`,
+    )
+    .join("\n\n");
+  return `<${tag}>\n\n${heading}\n\n${files}\n\n</${tag}>`;
+}
+
+export function buildContextMessageContent(
+  contextFiles: ContextFile[],
+  agentDir: string,
+): string {
+  const global = contextFiles.filter((file) => isGlobal(file, agentDir));
+  const project = contextFiles.filter((file) => !isGlobal(file, agentDir));
+  const blocks: string[] = [];
+  if (global.length > 0) {
+    blocks.push(
+      renderBlock(
+        "global_context",
+        "Global instructions and guidelines:",
+        global,
+      ),
+    );
+  }
+  if (project.length > 0) {
+    blocks.push(
+      renderBlock(
+        "project_context",
+        "Project-specific instructions and guidelines:",
+        project,
+      ),
+    );
+  }
+  return blocks.join("\n\n");
+}
+
+export function buildContextMessage(
+  contextFiles: ContextFile[],
+  agentDir: string,
+  timestamp: number = Date.now(),
+): CustomMessage {
+  return {
+    role: "custom",
+    customType: CONTEXT_MESSAGE_TYPE,
+    content: buildContextMessageContent(contextFiles, agentDir),
+    display: false,
+    timestamp,
+  };
+}
+
+export function insertBeforeLastUserMessage<M extends { role: string }>(
+  messages: M[],
+  message: M,
+): M[] {
+  const result = [...messages];
+  for (let index = result.length - 1; index >= 0; index--) {
+    if (result[index].role === "user") {
+      result.splice(index, 0, message);
+      return result;
+    }
+  }
+  result.push(message);
+  return result;
+}

package/src/system-prompt.ts

@@ -0,0 +1,22 @@
+import type { ContextFile } from "./context-file.ts";
+
+/** Mirrors the `<project_context>` block `buildSystemPrompt` appends for context files. */
+function renderSystemPromptBlock(contextFiles: ContextFile[]): string {
+  let block = "\n\n<project_context>\n\n";
+  block += "Project-specific instructions and guidelines:\n\n";
+  for (const { path, content } of contextFiles) {
+    block += `<project_instructions path="${path}">\n${content}\n</project_instructions>\n\n`;
+  }
+  block += "</project_context>\n";
+  return block;
+}
+
+export function stripContextFiles(
+  systemPrompt: string,
+  contextFiles: ContextFile[],
+): string {
+  if (contextFiles.length === 0) {
+    return systemPrompt;
+  }
+  return systemPrompt.replace(renderSystemPromptBlock(contextFiles), "");
+}