pi-blackhole 0.4.2 → 0.4.3

package/example-config.json

@@ -96,7 +96,7 @@
   "observationsPoolTargetTokens": 10000,
   "reflectorInputMaxTokens": 80000,
   "dropperInputMaxTokens": 80000,
-  "dropperPressureThreshold": 0.70,
+  "dropperPressureThreshold": 0.7,
   "observerChunkMaxTokens": 40000,
   "observerPreambleMaxTokens": 0,
   "agentMaxTurns": 16,

package/index.ts

@@ -16,69 +16,43 @@ import { registerConsolidationTrigger } from "./src/om/consolidation.js";
 import { registerCompactionTrigger } from "./src/om/compaction-trigger.js";
 import { registerRecallTool } from "./src/tools/recall";
 import { Runtime } from "./src/om/runtime.js";
+import { captureRegisteredProviderStreams } from "./src/om/provider-stream.js";
 
 export default (pi: ExtensionAPI) => {
-	// ── Bridge: capture custom provider stream functions for jiti-loaded agents ──
-	// pi-blackhole's consolidation agents are loaded via jiti with moduleCache: false,
-	// which creates a separate pi-ai instance whose apiProviderRegistry lacks custom
-	// providers (e.g., claude-bridge registered by other extensions). This bridge stores
-	// streamSimple functions in a Symbol.for() global so agents can access them without
-	// going through pi-ai's registry.
-	//
-	// There are two mechanisms:
-	// 1. Wrap pi.registerProvider to capture streamSimple at registration time.
-	//    This handles providers registered AFTER our factory runs.
-	// 2. On agent_start, scan modelRegistry.registeredProviders for any providers
-	//    that registered BEFORE our factory ran (different extension load order).
-	const PROVIDER_STREAMS_KEY = Symbol.for("pi-blackhole:provider-streams");
-	const providerStreams: Map<string, Function> = (globalThis as any)[PROVIDER_STREAMS_KEY] ??= new Map();
-
-	const origRegisterProvider = pi.registerProvider.bind(pi);
-	pi.registerProvider = ((name: string, config: any) => {
-		if (config && config.streamSimple && config.api) {
-			providerStreams.set(config.api, config.streamSimple);
-		}
-		origRegisterProvider(name, config);
-	}) as typeof pi.registerProvider;
-
-	// Fallback: on agent_start, capture providers that registered before our wrapper
-	// (handles the case where pi-blackhole loads after another provider extension).
-	// Uses a dedicated flag instead of checking providerStreams.size so the scan
-	// always runs once even if the wrapper already captured some providers.
-	let hasScannedFallback = false;
-	pi.on("agent_start", (_event: unknown, ctx: any) => {
-		if (hasScannedFallback) return;
-		hasScannedFallback = true
-		// modelRegistry.registeredProviders is declared private in TypeScript but is a
-		// regular JS class field at runtime. We access it via bracket notation for
-		// future-proofing against potential #private migration.
-		const registry = (ctx as any)?.modelRegistry;
-		const registered: Map<string, any> | undefined = registry?.["registeredProviders"];
-		if (registered && typeof registered.forEach === "function") {
-			registered.forEach((config: any, _name: string) => {
-				if (config && config.streamSimple && config.api && !providerStreams.has(config.api)) {
-					providerStreams.set(config.api, config.streamSimple);
-				}
-			});
-		}
-	});
-
-	scaffoldSettings();
-
-	const omRuntime = new Runtime();
-
-	// Observational memory: background consolidation pipeline
-	registerConsolidationTrigger(pi, omRuntime);      // agent_start + turn_end → observer/reflector/dropper
-	registerCompactionTrigger(pi, omRuntime);         // agent_end → auto-compaction
-
-	// Pi-vcc: compaction + om injection
-	registerBeforeCompactHook(pi, omRuntime);          // session_before_compact → pi-vcc + om content
-
-	// Commands
-	registerPiVccCommand(pi, omRuntime);               // /pi-vcc (needs runtime for noAutoCompact flush)
-	registerMemoryCommand(pi, omRuntime);              // /blackhole-memory [status|view|full]
-	registerVccRecallCommand(pi);                        // /blackhole-recall <query>
-
-	// Tools
-	registerRecallTool(pi);                            // unified recall (#N + [12char])
+  // ── Bridge: capture custom provider stream functions for jiti-loaded agents ──
+  // pi-blackhole's consolidation agents are loaded via jiti with moduleCache: false,
+  // which creates a separate pi-ai instance whose apiProviderRegistry lacks custom
+  // providers (e.g., claude-bridge registered by other extensions). This bridge stores
+  // streamSimple functions in a Symbol.for() global so agents can access them without
+  // going through pi-ai's registry.
+  //
+  // Capture custom provider streams from Pi's model registry before each run.
+  // This works regardless of extension load order and includes providers added
+  // after startup.
+  const PROVIDER_STREAMS_KEY = Symbol.for("pi-blackhole:provider-streams");
+  const providerStreams: Map<string, Function> = ((globalThis as any)[
+    PROVIDER_STREAMS_KEY
+  ] ??= new Map());
+  pi.on("agent_start", (_event: unknown, ctx: any) => {
+    captureRegisteredProviderStreams(ctx.modelRegistry, providerStreams);
+  });
+
+  scaffoldSettings();
+
+  const omRuntime = new Runtime();
+
+  // Observational memory: background consolidation pipeline
+  registerConsolidationTrigger(pi, omRuntime); // agent_start + turn_end → observer/reflector/dropper
+  registerCompactionTrigger(pi, omRuntime); // agent_end → auto-compaction
+
+  // Pi-vcc: compaction + om injection
+  registerBeforeCompactHook(pi, omRuntime); // session_before_compact → pi-vcc + om content
+
+  // Commands
+  registerPiVccCommand(pi, omRuntime); // /pi-vcc (needs runtime for noAutoCompact flush)
+  registerMemoryCommand(pi, omRuntime); // /blackhole-memory [status|view|full]
+  registerVccRecallCommand(pi); // /blackhole-recall <query>
+
+  // Tools
+  registerRecallTool(pi); // unified recall (#N + [12char])
 };

package/package.json

@@ -1,22 +1,32 @@
 {
   "name": "pi-blackhole",
-  "version": "0.4.2",
+  "version": "0.4.3",
   "packageManager": "pnpm@11.2.2",
   "description": "Unified compaction + observational memory extension for Pi — compresses conversation context while preserving durable observations and reflections",
   "license": "MIT",
-  "main": "index.ts",
+  "type": "module",
+  "main": "./dist/index.js",
   "files": [
+    "dist/",
     "*.ts",
     "src/**/*.ts",
     "!src/**/*.test.ts",
     "!tests/",
+    "tsup.config.ts",
     "LICENSE",
     "README.md",
     "example-config.json"
   ],
   "scripts": {
-    "check": "tsc --noEmit",
-    "lint": "eslint src/"
+    "build": "tsup",
+    "typecheck": "tsc --noEmit",
+    "lint": "eslint .",
+    "format": "prettier --write .",
+    "format:check": "prettier --check .",
+    "test": "vitest run",
+    "test:watch": "vitest",
+    "check": "pnpm typecheck && pnpm lint",
+    "prepare": "node scripts/prepare.mjs"
   },
   "keywords": [
     "pi-package",
@@ -47,19 +57,21 @@
   },
   "pi": {
     "extensions": [
-      "./index.ts"
+      "./dist/index.js"
     ]
   },
   "devDependencies": {
-    "@earendil-works/pi-agent-core": "0.81.1",
-    "@earendil-works/pi-ai": "0.81.1",
-    "@earendil-works/pi-coding-agent": "0.81.1",
-    "@earendil-works/pi-tui": "0.81.1",
+    "@earendil-works/pi-agent-core": "0.83.0",
+    "@earendil-works/pi-ai": "0.83.0",
+    "@earendil-works/pi-coding-agent": "0.83.0",
+    "@earendil-works/pi-tui": "0.83.0",
     "@typescript-eslint/eslint-plugin": "8.60.0",
     "@typescript-eslint/parser": "8.60.0",
     "eslint": "9.39.4",
     "husky": "9.1.7",
+    "tsup": "^8.5.1",
     "lint-staged": "15.5.2",
+    "prettier": "3.9.6",
     "typebox": "1.1.38",
     "typescript": "5.9.3",
     "vitest": "4.1.9"