pi-observational-memory 1.0.1 → 1.0.2

package/README.md

@@ -94,12 +94,14 @@ That's it. The extension hooks into Pi's compaction lifecycle automatically. No
 
 ### Extension settings
 
-Create `~/.pi/agent/observational-memory.json` (or `.pi/observational-memory.json` per project):
+Settings live under the `observational-memory` key in Pi's `settings.json` — globally at `~/.pi/agent/settings.json`, or per-project at `.pi/settings.json`. Project values override global.
 
 ```json
 {
-  "observationThreshold": 50000,
-  "reflectionThreshold": 30000
+  "observational-memory": {
+    "observationThreshold": 50000,
+    "reflectionThreshold": 30000
+  }
 }
 ```
 
@@ -115,13 +117,15 @@ The observer and reflector don't need the same capabilities as your coding agent
 
 ```json
 {
-  "compactionModel": { "provider": "openrouter", "id": "google/gemma-4-31b-it" }
+  "observational-memory": {
+    "compactionModel": { "provider": "openrouter", "id": "google/gemma-4-31b-it" }
+  }
 }
 ```
 
 ### Pi compaction settings
 
-The extension works with Pi's built-in compaction settings in `~/.pi/agent/settings.json`:
+The extension works alongside Pi's built-in compaction settings in the same `settings.json`:
 
 ```json
 {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pi-observational-memory",
-  "version": "1.0.1",
+  "version": "1.0.2",
   "description": "Observational memory extension for pi — cache-friendly tiered compaction with observations and reflections.",
   "type": "module",
   "license": "MIT",

package/src/config.ts

@@ -13,24 +13,26 @@ export const DEFAULTS: Config = {
 	reflectionThreshold: 30_000,
 };
 
-export function loadConfig(cwd: string): Config {
-	const globalPath = join(getAgentDir(), "observational-memory.json");
-	const projectPath = join(cwd, ".pi", "observational-memory.json");
-
-	let globalConfig: Partial<Config> = {};
-	let projectConfig: Partial<Config> = {};
+const SETTINGS_KEY = "observational-memory";
 
-	if (existsSync(globalPath)) {
-		try {
-			globalConfig = JSON.parse(readFileSync(globalPath, "utf-8"));
-		} catch {}
+function readNamespacedConfig(path: string): Partial<Config> {
+	if (!existsSync(path)) return {};
+	try {
+		const raw = JSON.parse(readFileSync(path, "utf-8")) as Record<string, unknown>;
+		const nested = raw[SETTINGS_KEY];
+		return nested && typeof nested === "object" ? (nested as Partial<Config>) : {};
+	} catch {
+		return {};
 	}
+}
 
-	if (existsSync(projectPath)) {
-		try {
-			projectConfig = JSON.parse(readFileSync(projectPath, "utf-8"));
-		} catch {}
-	}
+export function loadConfig(cwd: string): Config {
+	const globalPath = join(getAgentDir(), "settings.json");
+	const projectPath = join(cwd, ".pi", "settings.json");
 
-	return { ...DEFAULTS, ...globalConfig, ...projectConfig };
+	return {
+		...DEFAULTS,
+		...readNamespacedConfig(globalPath),
+		...readNamespacedConfig(projectPath),
+	};
 }

package/src/index.ts

@@ -1,6 +1,6 @@
 import { completeSimple } from "@mariozechner/pi-ai";
 import type { ExtensionAPI } from "@mariozechner/pi-coding-agent";
-import { convertToLlm, serializeConversation } from "@mariozechner/pi-coding-agent";
+import { convertToLlm, serializeConversation, SettingsManager } from "@mariozechner/pi-coding-agent";
 import { DEFAULTS, loadConfig } from "./config.js";
 import type { Config } from "./config.js";
 import { CONTEXT_USAGE_INSTRUCTIONS, OBSERVER_SYSTEM, REFLECTOR_SYSTEM } from "./prompts.js";
@@ -199,6 +199,7 @@ export default function observationalMemory(pi: ExtensionAPI) {
 			const rawTokens = estimateRawTailTokens(entries);
 			const obsTokens = estimateTokens(state.observations);
 			const refTokens = estimateTokens(state.reflections);
+			const keepRecentTokens = SettingsManager.create(ctx.cwd).getCompactionKeepRecentTokens();
 
 			const lines = [
 				"── Observational Memory ──",
@@ -209,6 +210,7 @@ export default function observationalMemory(pi: ExtensionAPI) {
 				"── Parameters ──",
 				`Observation threshold: ${config.observationThreshold.toLocaleString()}`,
 				`Reflection threshold:  ${config.reflectionThreshold.toLocaleString()}`,
+				`Keep recent tokens:    ${keepRecentTokens.toLocaleString()} (pi compaction)`,
 			];
 
 			ctx.ui.notify(lines.join("\n"), "info");