@8monkey/pi-context-history 0.1.0 → 0.2.0

package/README.md

@@ -1,6 +1,6 @@
 # pi-context-history
 
-A [Pi](https://github.com/earendil-works/pi-coding-agent) extension that manages conversation context end-to-end. It trims old history, strips stale tool chatter from earlier turns, keeps a rolling session summary, and folds that summary back into the system prompt — so long-running conversations stay lean, focused, and continuous.
+A [Pi](https://github.com/earendil-works/pi-coding-agent) extension that manages conversation context end-to-end. It trims old history, strips stale tool chatter from earlier turns, keeps a rolling session summary, folds that summary back into the system prompt, and lets you append messages to the history by hand — so long-running conversations stay lean, focused, and continuous.
 
 Each feature is independent and can be turned off individually, so you can run the whole pipeline or just the parts you want.
 
@@ -12,6 +12,7 @@ Each feature is independent and can be turned off individually, so you can run t
 | **Strip tool history** | `context` | Removes tool calls and results from prior turns, keeping the current turn's tool interactions intact. |
 | **Generate summary** | `session_start` | Regenerates a stale rolling summary in the background when a session resumes; also available on demand via `/summarize-session`. |
 | **Inject summary** | `before_agent_start` | Folds the rolling summary into the system prompt so the model continues with prior context. |
+| **Append message** | command | Adds a user or assistant message to the end of the history via `/add-user-message` and `/add-assistant-message`. |
 
 The summary features are a producer/consumer pair: **Generate summary** writes `~/.pi/agent/summary.md`, and **Inject summary** reads it.
 
@@ -27,13 +28,16 @@ pi install npm:@8monkey/pi-context-history
 - **Strip tool history.** Tool calls and results from earlier turns are removed; the current turn is left untouched. At the start of each agent loop the extension marks where the current turn begins (scanning back past the latest tool exchange) and holds that mark steady across the loop, so a running turn is never stripped mid-flight. Before the mark, tool results are dropped and assistant messages lose their tool-call blocks (if nothing else remains, the message goes too).
 - **Generate summary.** When you resume a session whose summary has gone stale — older than `PI_SUMMARY_STALENESS_DAYS`, with a first message older than the window too — the extension rebuilds `~/.pi/agent/summary.md` by shelling out to `pi -p` (extensions, context files, and skills disabled). New and empty sessions are skipped. To override the built-in prompt, drop your own at `prompts/session-summary.md` (project `.pi/` wins over `~/.pi/`); it must contain the `{conversation_history}` placeholder.
 - **Inject summary.** Before each agent run, when `~/.pi/agent/summary.md` exists, its contents are wrapped in a `<summary date="…">` block (the date is the file's modified time) followed by an `<additional_context>` note telling the model to maintain continuity and match the existing language and tone.
+- **Append message.** `/add-user-message <text>` and `/add-assistant-message <text>` write a message with the matching role to the end of the session history. The entry is persisted immediately but enters the live context on the next rebuild (resume or branch), since Pi has no public way to inject into a running turn. Appending an assistant message requires a model to be selected, as the message records its identity.
 
 Zero runtime dependencies. Pi loads the TypeScript directly, so there's no build step. Runs under Node or Bun.
 
-## Command
+## Commands
 
 | Command | Description |
 |---|---|
+| `/add-assistant-message <text>` | Append an assistant message to the end of the history. Persisted immediately; enters context on the next rebuild. Available only when the **Append message** feature is enabled. |
+| `/add-user-message <text>` | Append a user message to the end of the history. Persisted immediately; enters context on the next rebuild. Available only when the **Append message** feature is enabled. |
 | `/summarize-session` | Regenerate the current session's summary on demand, ignoring the staleness window. Writes `~/.pi/agent/summary.md` and reports success, failure, or an empty session. Available only when the **Generate summary** feature is enabled. |
 
 ## Configuration
@@ -50,6 +54,7 @@ Every feature is **on by default** and switched off by setting its flag to `fals
 | `PI_STRIP_TOOL_HISTORY` | on | Strip tool history |
 | `PI_GENERATE_SUMMARY` | on | Generate summary |
 | `PI_INJECT_SUMMARY` | on | Inject summary |
+| `PI_APPEND_MESSAGE` | on | Append message |
 
 ### Settings
 

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "@8monkey/pi-context-history",
-	"version": "0.1.0",
+	"version": "0.2.0",
 	"description": "Pi extension that manages conversation context: trims old history, strips stale tool calls, and keeps a rolling summary fed back into the system prompt. Each feature toggles independently.",
 	"license": "MIT",
 	"type": "module",

package/src/index.ts

@@ -1,7 +1,7 @@
 import { writeFileSync } from "node:fs";
 import { homedir } from "node:os";
 import { join } from "node:path";
-import type { ExtensionAPI, ExtensionCommandContext } from "@earendil-works/pi-coding-agent";
+import type { ExtensionAPI, SessionManager } from "@earendil-works/pi-coding-agent";
 import { buildContextPrompt } from "./build-prompt.ts";
 import { featureEnabled, HISTORY_DAYS, SUMMARY_STALENESS_DAYS } from "./config.ts";
 import { filterByAge } from "./filter.ts";
@@ -17,6 +17,14 @@ import {
 
 const HISTORY_MS = HISTORY_DAYS * 86_400_000;
 const SUMMARY_PATH = join(homedir(), ".pi", "agent", "summary.md");
+const ZERO_USAGE = {
+	input: 0,
+	output: 0,
+	cacheRead: 0,
+	cacheWrite: 0,
+	totalTokens: 0,
+	cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0, total: 0 },
+};
 
 function registerTrimHistory(pi: ExtensionAPI) {
 	pi.on("context", (event) => {
@@ -76,7 +84,7 @@ function registerGenerateSummary(pi: ExtensionAPI) {
 
 	pi.registerCommand("summarize-session", {
 		description: "Regenerate the current session's summary now (ignores staleness)",
-		handler: async (_args, ctx: ExtensionCommandContext) => {
+		handler: async (_args, ctx) => {
 			const entries = ctx.sessionManager.getEntries() as TranscriptEntry[];
 			let result: string | "empty";
 			try {
@@ -106,9 +114,56 @@ function registerInjectSummary(pi: ExtensionAPI) {
 	});
 }
 
+function registerAppendMessage(pi: ExtensionAPI) {
+	pi.registerCommand("add-user-message", {
+		description: "Append a user message to the end of the conversation history",
+		handler: async (args, ctx) => {
+			const text = args.trim();
+			if (!text) {
+				if (ctx.hasUI) ctx.ui.notify("Usage: /add-user-message <text>", "warning");
+				return;
+			}
+			const session = ctx.sessionManager as SessionManager;
+			session.appendMessage({ role: "user", content: [{ type: "text", text }], timestamp: Date.now() });
+			if (ctx.hasUI) ctx.ui.notify("Appended a user message; it applies on the next session rebuild.", "info");
+		},
+	});
+
+	pi.registerCommand("add-assistant-message", {
+		description: "Append an assistant message to the end of the conversation history",
+		handler: async (args, ctx) => {
+			const text = args.trim();
+			if (!text) {
+				if (ctx.hasUI) ctx.ui.notify("Usage: /add-assistant-message <text>", "warning");
+				return;
+			}
+			if (!ctx.model) {
+				if (ctx.hasUI) ctx.ui.notify("Cannot append an assistant message: no model selected.", "error");
+				return;
+			}
+			const { api, provider, id } = ctx.model;
+			const session = ctx.sessionManager as SessionManager;
+			session.appendMessage({
+				role: "assistant",
+				content: [{ type: "text", text }],
+				api,
+				provider,
+				model: id,
+				usage: ZERO_USAGE,
+				stopReason: "stop",
+				timestamp: Date.now(),
+			});
+			if (ctx.hasUI) {
+				ctx.ui.notify("Appended an assistant message; it applies on the next session rebuild.", "info");
+			}
+		},
+	});
+}
+
 export default function (pi: ExtensionAPI) {
 	if (featureEnabled("PI_TRIM_HISTORY")) registerTrimHistory(pi);
 	if (featureEnabled("PI_STRIP_TOOL_HISTORY")) registerStripToolHistory(pi);
 	if (featureEnabled("PI_GENERATE_SUMMARY")) registerGenerateSummary(pi);
 	if (featureEnabled("PI_INJECT_SUMMARY")) registerInjectSummary(pi);
+	if (featureEnabled("PI_APPEND_MESSAGE")) registerAppendMessage(pi);
 }