@agimon-ai/po-cache 0.0.0-alpha.120 → 0.0.0-alpha.121

package/dist/extensions/server.cjs

@@ -7,8 +7,8 @@ const require_promptCacheTelemetry = require("../promptCacheTelemetry-DeN4LdSo.c
 let node_crypto = require("node:crypto");
 let _agimon_ai_po_core_serverPlugin = require("@agimon-ai/po-core/serverPlugin");
 let _agimon_ai_po_core_extensionFile = require("@agimon-ai/po-core/extensionFile");
+let _agimon_ai_po_config = require("@agimon-ai/po-config");
 let _agimon_ai_po_core_platform = require("@agimon-ai/po-core/platform");
-let _agimon_ai_po_config_harnessStore = require("@agimon-ai/po-config/harnessStore");
 let _agimon_ai_po_minor_mode = require("@agimon-ai/po-minor-mode");
 let node_path = require("node:path");
 let _earendil_works_pi_coding_agent = require("@earendil-works/pi-coding-agent");
@@ -152,9 +152,9 @@ function isErrorMessage(message) {
 	const record = typeof message === "object" && message !== null ? message : void 0;
 	return record?.stopReason === "error" || record?.stopReason === "aborted";
 }
-function createCacheRuntime(dependencies, environment) {
-	const parentNamespaceEnvironment = new OwnedEnvironmentValue(require_environment.PO_PROMPT_CACHE_PARENT_NAMESPACE_ENV);
-	const rootSessionEnvironment = new OwnedEnvironmentValue(require_environment.PO_PROMPT_CACHE_ROOT_SESSION_ENV);
+function createCacheRuntime(dependencies, environment, readHarness, namespaceEnvironment) {
+	const parentNamespaceEnvironment = new OwnedEnvironmentValue(require_environment.PO_PROMPT_CACHE_PARENT_NAMESPACE_ENV, namespaceEnvironment);
+	const rootSessionEnvironment = new OwnedEnvironmentValue(require_environment.PO_PROMPT_CACHE_ROOT_SESSION_ENV, namespaceEnvironment);
 	const projectedParentNamespace = stringValue(environment[require_environment.PO_PROMPT_CACHE_PARENT_NAMESPACE_ENV]);
 	const childProjection = stringValue(environment[require_environment.PO_PROMPT_CACHE_CHILD_PROJECTION_ENV]);
 	const isProjectedChild = Boolean(projectedParentNamespace && childProjection);
@@ -164,7 +164,7 @@ function createCacheRuntime(dependencies, environment) {
 	const refreshParentNamespace = () => {
 		if (isProjectedChild) return parentNamespace;
 		if (!rootSessionId) return void 0;
-		const harness = (0, _agimon_ai_po_config_harnessStore.getSessionHarnessState)(rootSessionId);
+		const harness = readHarness();
 		parentNamespace = require_promptCacheTelemetry.createParentPromptCacheNamespace({
 			rootSessionId,
 			compositionFingerprint: harness.compositionFingerprint,
@@ -214,6 +214,12 @@ function createCacheRuntime(dependencies, environment) {
 	};
 }
 //#endregion
+//#region src/services/environment/sync.ts
+/** Node's optimizer integrates with the process-level Pi provider environment. */
+function cacheEnvironment(_environment) {
+	return process.env;
+}
+//#endregion
 //#region src/services/agentDirectory/index.ts
 /** Pi's agent directory, where the optimizer keeps its state; the edge build swaps in `index.edge.ts`. */
 function defaultAgentDirectory() {
@@ -3225,6 +3231,15 @@ function createOptimizerSupport(STATE_DIR, environment, fs) {
 	};
 }
 //#endregion
+//#region src/services/nativeOptimizer/sync.ts
+/** Process metadata used by Node's persisted optimizer shards. */
+function cacheProcessIdentity() {
+	return {
+		pid: process.pid,
+		ppid: process.ppid
+	};
+}
+//#endregion
 //#region src/services/nativeOptimizer/index.ts
 const STATUS_KEY = "pi-cache-stats";
 const PERSIST_DEBOUNCE_MS = 2e3;
@@ -3289,8 +3304,7 @@ function createNativeOptimizer(initial, fs, stateDir = defaultAgentDirectory())
 			instanceId,
 			sessionHash,
 			process: {
-				pid: process.pid,
-				ppid: process.ppid,
+				...cacheProcessIdentity(),
 				instanceStartedAt
 			},
 			lifecycle: {
@@ -3522,7 +3536,7 @@ function createNativeOptimizer(initial, fs, stateDir = defaultAgentDirectory())
 				handle: (_event, context) => serialized(async () => {
 					if (started || closed) return;
 					started = true;
-					releaseRetention = acquireEnvironmentValue("PI_CACHE_RETENTION", "long");
+					releaseRetention = acquireEnvironmentValue("PI_CACHE_RETENTION", "long", cacheEnvironment(initial.environment));
 					await support.refreshFileConfig();
 					currentGlobalEpoch = await support.readGlobalStatsEpoch();
 					await support.removeLegacyStatsFiles();
@@ -3593,10 +3607,14 @@ function createNativeOptimizer(initial, fs, stateDir = defaultAgentDirectory())
 var root_server_default = (0, _agimon_ai_po_core_extensionFile.defineRoot)(({ agent, context: root }) => {
 	if (!agent) throw new Error("Cache statistics require the native session host.");
 	const telemetry = new require_promptCacheTelemetry.PromptCacheTelemetry();
+	let config;
 	const runtime = createCacheRuntime({
 		telemetry,
 		now: Date.now
-	}, agent.context.environment);
+	}, agent.context.environment, () => {
+		if (!config) throw new Error("Cache requires the session configuration.");
+		return config.getSnapshot().harness;
+	}, cacheEnvironment(agent.context.environment));
 	const optimizer = createNativeOptimizer(agent.context, (0, _agimon_ai_po_core_platform.requirePoFs)(root));
 	return {
 		value: { hooks: optimizer.hooks.map((hook) => {
@@ -3616,9 +3634,20 @@ var root_server_default = (0, _agimon_ai_po_core_extensionFile.defineRoot)(({ ag
 			};
 			return hook;
 		}) },
-		services: [(context) => {
-			new require_promptCacheTelemetry.PromptCacheTelemetryService(context, telemetry);
-		}, runtime.plugin],
+		services: [
+			(context) => {
+				context.inject([_agimon_ai_po_config.PO_CONFIG_SERVICE], (scoped) => {
+					config = scoped.get(_agimon_ai_po_config.PO_CONFIG_SERVICE);
+					return () => {
+						config = void 0;
+					};
+				});
+			},
+			(context) => {
+				new require_promptCacheTelemetry.PromptCacheTelemetryService(context, telemetry);
+			},
+			runtime.plugin
+		],
 		async onDispose() {
 			try {
 				await optimizer.dispose();
@@ -3627,7 +3656,7 @@ var root_server_default = (0, _agimon_ai_po_core_extensionFile.defineRoot)(({ ag
 			}
 		}
 	};
-});
+}, { inject: [(0, _agimon_ai_po_core_extensionFile.defineServiceContract)(_agimon_ai_po_config.PO_CONFIG_SERVICE, { scope: "session" })] });
 //#endregion
 //#region src/extensions/workspaces/sessions/(backend)/hook/agent-settled.server.ts
 var agent_settled_server_default = (0, _agimon_ai_po_core_extensionFile.defineHook)((context) => context.root.hooks.find((hook) => hook.event === "agent_settled"));