@skydiveai/pi-extensions 0.1.0-beta.2591 → 0.1.0-beta.2595

package/dist/index.mjs

@@ -4,8 +4,8 @@ import { UserBuilder, restHandler } from "@a2a-js/sdk/server/express";
 import { buildAgentCard, chainMiddleware, composeHandlers, createAgentExecutor, createProtocolHandlers, getCurrentTraceparent, logger, mountAt, requestHeaders, requestUrl, webHandlerToMiddleware } from "@skydiveai/pi-server";
 import { mkdir, open, readFile, readdir, stat, unlink, writeFile } from "node:fs/promises";
 import { basename, dirname, join, relative, resolve } from "node:path";
-import { z } from "zod";
 import { pathToFileURL } from "node:url";
+import { z } from "zod";
 import { CallToolResultSchema } from "@modelcontextprotocol/sdk/types.js";
 import { Type } from "typebox";
 import { Client } from "@modelcontextprotocol/sdk/client/index.js";
@@ -242,167 +242,6 @@ function createHealthHandler({ metadata }) {
 	};
 }
 //#endregion
-//#region src/extensions/context-management-config.ts
-/**
-* Configuration for the context-management capability (tool-output trimming
-* and client-side microcompact). See the design notes in the extension for
-* what each layer does; this module is purely the env → config surface.
-*
-* The harness is provider-agnostic and runs inside the sandbox, where its
-* only config channel is the environment (loaded by the platform env
-* middleware before any session starts — see platform-env-middleware.ts). So
-* the LaunchDarkly flag `harness-context-management-enabled` is resolved by
-* the platform when it provisions the sandbox and passed through as
-* `SKYDIVE_CONTEXT_MANAGEMENT`; the individual knobs override the defaults
-* below when present.
-*
-* Resolution is defensive: a malformed value never throws (this config is
-* read on the hot path before every LLM call), it falls back to the default
-* for that knob and logs once.
-*/
-const log$15 = logger.child({ module: "context-management-config" });
-const DEFAULT_CONTEXT_MANAGEMENT_CONFIG = {
-	enabled: false,
-	perResultMaxBytes: 16 * 1024,
-	keepRecentToolResults: 3,
-	coldCacheGapSeconds: 240,
-	warmClearTriggerTokens: 5e4,
-	clearAtLeastTokens: 1e4,
-	excludeTools: [],
-	maxModelCallsPerTurn: 80,
-	nativeAnthropicEdits: false
-};
-const boolFromEnv = (value, fallback) => {
-	if (value === void 0) return fallback;
-	const normalized = value.trim().toLowerCase();
-	if (normalized === "1" || normalized === "true") return true;
-	if (normalized === "0" || normalized === "false") return false;
-	return fallback;
-};
-const positiveInt = (fallback) => z.coerce.number().int().positive().catch(fallback);
-const nonNegativeInt = (fallback) => z.coerce.number().int().nonnegative().catch(fallback);
-const configSchema = z.object({
-	perResultMaxBytes: positiveInt(DEFAULT_CONTEXT_MANAGEMENT_CONFIG.perResultMaxBytes),
-	keepRecentToolResults: nonNegativeInt(DEFAULT_CONTEXT_MANAGEMENT_CONFIG.keepRecentToolResults),
-	coldCacheGapSeconds: positiveInt(DEFAULT_CONTEXT_MANAGEMENT_CONFIG.coldCacheGapSeconds),
-	warmClearTriggerTokens: positiveInt(DEFAULT_CONTEXT_MANAGEMENT_CONFIG.warmClearTriggerTokens),
-	clearAtLeastTokens: positiveInt(DEFAULT_CONTEXT_MANAGEMENT_CONFIG.clearAtLeastTokens),
-	maxModelCallsPerTurn: positiveInt(DEFAULT_CONTEXT_MANAGEMENT_CONFIG.maxModelCallsPerTurn)
-});
-const parseExcludeTools = (value) => {
-	if (!value) return DEFAULT_CONTEXT_MANAGEMENT_CONFIG.excludeTools;
-	return value.split(",").map((name) => name.trim()).filter((name) => name.length > 0);
-};
-/**
-* Reads the context-management config from `env` (defaults to `process.env`).
-* Never throws — invalid values fall back to defaults.
-*/
-function resolveContextManagementConfig(env = process.env) {
-	if (!boolFromEnv(env.SKYDIVE_CONTEXT_MANAGEMENT, false)) return { ...DEFAULT_CONTEXT_MANAGEMENT_CONFIG };
-	const parsed = configSchema.safeParse({
-		perResultMaxBytes: env.SKYDIVE_CTX_PER_RESULT_MAX_BYTES,
-		keepRecentToolResults: env.SKYDIVE_CTX_KEEP_RECENT,
-		coldCacheGapSeconds: env.SKYDIVE_CTX_COLD_GAP_SECONDS,
-		warmClearTriggerTokens: env.SKYDIVE_CTX_WARM_TRIGGER_TOKENS,
-		clearAtLeastTokens: env.SKYDIVE_CTX_CLEAR_AT_LEAST_TOKENS,
-		maxModelCallsPerTurn: env.SKYDIVE_CTX_MAX_MODEL_CALLS
-	});
-	if (!parsed.success) {
-		log$15.warn({
-			event: "context_management_config_invalid",
-			err: parsed.error
-		}, "falling back to default context-management config");
-		return {
-			...DEFAULT_CONTEXT_MANAGEMENT_CONFIG,
-			enabled: true
-		};
-	}
-	return {
-		enabled: true,
-		...parsed.data,
-		excludeTools: parseExcludeTools(env.SKYDIVE_CTX_EXCLUDE_TOOLS),
-		nativeAnthropicEdits: boolFromEnv(env.SKYDIVE_CTX_NATIVE_ANTHROPIC_EDITS, DEFAULT_CONTEXT_MANAGEMENT_CONFIG.nativeAnthropicEdits)
-	};
-}
-//#endregion
-//#region src/extensions/context-management-runtime.ts
-/**
-* Live, mutable view of the context-management config.
-*
-* The env-derived config (context-management-config.ts) is the boot-time
-* default. On top of it, the platform can deliver a global on/off at runtime
-* via the LaunchDarkly flag `harness-context-management-enabled` — resolved
-* server-side and polled by the harness (see the poller in
-* context-management.ts). This holder is where that override lands so a flip
-* (especially a kill-switch) reaches long-lived sandboxes without a restart.
-*
-* Only the master `enabled` toggle is overridable at runtime; the per-knob
-* tunables stay env-derived. `enabled` resolves to the flag override when the
-* platform has reported one, else the env value.
-*/
-let baseConfig = null;
-let flagOverride = null;
-function base() {
-	baseConfig ??= resolveContextManagementConfig();
-	return baseConfig;
-}
-/** The effective config, with the runtime flag override applied to `enabled`. */
-function getContextManagementConfig() {
-	const resolved = base();
-	return {
-		...resolved,
-		enabled: flagOverride ?? resolved.enabled
-	};
-}
-/**
-* Apply the platform-reported flag value. `null` clears the override (fall back
-* to the env default) — used when the phone-home result is indeterminate so a
-* transient failure never silently changes behavior.
-*/
-function setContextManagementFlagOverride(enabled) {
-	flagOverride = enabled;
-}
-/**
-* Whether a phone-home channel exists to learn the flag at runtime. When false
-* (e.g. a bare local CLI with no platform API), the env value is the only
-* source and there's nothing to poll.
-*/
-function hasFlagSource(env = process.env) {
-	return Boolean(env.SKYDIVE_API_URL ?? env.ANYONE_API_URL);
-}
-//#endregion
-//#region src/iteration-cap.ts
-function installIterationCap({ session, log }, configOverride = null) {
-	const readConfig = () => configOverride ?? getContextManagementConfig();
-	if (!readConfig().enabled && (configOverride !== null || !hasFlagSource())) return;
-	const agent = session.agent;
-	if (typeof agent.createLoopConfig !== "function") throw new Error("installIterationCap: session.agent.createLoopConfig is missing — pi-agent-core internals changed; update iteration-cap.ts.");
-	const original = agent.createLoopConfig.bind(agent);
-	agent.createLoopConfig = (options) => {
-		const loopConfig = original(options);
-		const previousStop = loopConfig.shouldStopAfterTurn;
-		let modelCalls = 0;
-		return {
-			...loopConfig,
-			shouldStopAfterTurn: async (ctx) => {
-				if (previousStop && await previousStop(ctx)) return true;
-				const config = readConfig();
-				if (!config.enabled) return false;
-				modelCalls += 1;
-				if (modelCalls >= config.maxModelCallsPerTurn) {
-					log.warn({
-						event: "iteration_cap_reached",
-						modelCalls,
-						max: config.maxModelCallsPerTurn
-					}, "reached per-turn model-call cap; stopping turn gracefully");
-					return true;
-				}
-				return false;
-			}
-		};
-	};
-}
-//#endregion
 //#region src/extensions/capability-soul-nudge.ts
 /**
 * One-line reminder appended to a tool-update continuation when the agent
@@ -420,7 +259,7 @@ function installIterationCap({ session, log }, configOverride = null) {
 const CAPABILITY_SOUL_NUDGE = "New capability gained — once the current task is done, if this changes what you can do for the user, record it in `soul.md` so it carries into future conversations rather than being rediscovered from scratch (then commit and push).";
 //#endregion
 //#region src/extensions/tool-call-summary.ts
-const log$14 = logger.child({ module: "tool-call-summary-extension" });
+const log$15 = logger.child({ module: "tool-call-summary-extension" });
 /**
 * The injected parameter name: a namespaced sentinel, so it can never collide
 * with a real tool argument and is unmistakable in transcripts and logs. The
@@ -628,7 +467,7 @@ function withSummaryStrippedBeforeValidation(tool) {
 		try {
 			stripped = withoutInjectedSummary(args);
 		} catch (err) {
-			log$14.error({
+			log$15.error({
 				err,
 				event: "tool_call_summary_prepare_strip_failed",
 				toolName: tool.name
@@ -658,7 +497,7 @@ function stripInjectedSummary(pi, toolName, input) {
 		if (toolDeclaresSummaryParam(pi, toolName)) return;
 		delete input[TOOL_CALL_SUMMARY_FIELD];
 	} catch (err) {
-		log$14.error({
+		log$15.error({
 			err,
 			event: "tool_call_summary_strip_failed",
 			toolName
@@ -674,7 +513,7 @@ function buildInjectedPayload(pi, payload) {
 	try {
 		return injectToolCallSummary(payload, getStrictToolNames(pi));
 	} catch (err) {
-		log$14.error({
+		log$15.error({
 			err,
 			event: "tool_call_summary_injection_failed"
 		}, "tool_call_summary injection failed; passing payload through unchanged");
@@ -711,7 +550,7 @@ const toolCallSummaryExtension = (pi) => {
 * or `ToolDefinition[]`. Files starting with `_` or `.` are skipped, so
 * `tools/_example.ts` documents the shape without registering.
 */
-const log$13 = logger.child({ module: "local-tools-extension" });
+const log$14 = logger.child({ module: "local-tools-extension" });
 const TOOLS_DIRNAME = "tools";
 const fileState = /* @__PURE__ */ new Map();
 let pendingLocalToolsUpdate = null;
@@ -855,7 +694,7 @@ async function reconcileAndQueue({ pi, dir, reason }) {
 		dir
 	});
 	if (reason !== "session_start" && summaryHasChanges$1(summary)) pendingLocalToolsUpdate = summary;
-	log$13.info({
+	log$14.info({
 		event: "local_tools_reconcile",
 		reason,
 		total_tools: summary.totalTools,
@@ -877,7 +716,7 @@ const localToolsExtension = (pi) => {
 				reason: "session_start"
 			});
 		} catch (err) {
-			log$13.error({
+			log$14.error({
 				err,
 				event: "local_tools_reconcile_failed"
 			}, "local tools reconcile failed");
@@ -889,7 +728,7 @@ const localToolsExtension = (pi) => {
 		try {
 			current = await listToolFiles(dir);
 		} catch (err) {
-			log$13.warn({
+			log$14.warn({
 				err,
 				event: "local_tools_listing_failed"
 			}, "tools/ listing failed");
@@ -911,7 +750,7 @@ const localToolsExtension = (pi) => {
 				reason: "auto_reload"
 			});
 		} catch (err) {
-			log$13.error({
+			log$14.error({
 				err,
 				event: "local_tools_auto_reload_failed"
 			}, "auto-reload after tools/ change failed");
@@ -1368,7 +1207,7 @@ async function loadMcpConfig(path) {
 * Clients are keyed by JSON-stringified config and reused across
 * reloads — only changed configs reconnect.
 */
-const log$12 = logger.child({ module: "mcp-extension" });
+const log$13 = logger.child({ module: "mcp-extension" });
 async function closeConnected(connected) {
 	try {
 		await connected.client.close();
@@ -1899,7 +1738,7 @@ var McpExtension = class {
 		});
 		this.lastConfigMtimeMs = await readConfigMtimeMs(configPath);
 		if (reason !== "session_start" && summaryHasChanges(summary)) this.pendingMcpUpdate = summary;
-		log$12.info({
+		log$13.info({
 			event: "mcp_reconcile",
 			reason,
 			total_tools: summary.totalTools,
@@ -1926,7 +1765,7 @@ var McpExtension = class {
 						contextWindow: ctx.model?.contextWindow
 					});
 				} catch (err) {
-					log$12.error({
+					log$13.error({
 						err,
 						event: "mcp_reconcile_failed"
 					}, "MCP reconcile failed");
@@ -1938,7 +1777,7 @@ var McpExtension = class {
 				try {
 					mtime = await readConfigMtimeMs(configPath);
 				} catch (err) {
-					log$12.warn({
+					log$13.warn({
 						err,
 						event: "mcp_mtime_check_failed"
 					}, "mtime check on mcp.config.json failed");
@@ -1953,7 +1792,7 @@ var McpExtension = class {
 						contextWindow: ctx.model?.contextWindow
 					});
 				} catch (err) {
-					log$12.error({
+					log$13.error({
 						err,
 						event: "mcp_auto_reload_failed"
 					}, "auto-reload after mcp.config.json change failed");
@@ -2284,7 +2123,7 @@ const HEARTBEAT_THROTTLE_MS = 6e4;
 const TOOL_HEARTBEAT_INTERVAL_MS = 5e3;
 const MAX_TOOL_HEARTBEATS = 1440 * 60 * 1e3 / TOOL_HEARTBEAT_INTERVAL_MS;
 const DAEMON_URL = "http://localhost:38994";
-const log$11 = logger.child({ module: "platform-ext" });
+const log$12 = logger.child({ module: "platform-ext" });
 function sandboxClient() {
 	const apiUrl = apiBaseUrl();
 	if (!apiUrl) return null;
@@ -2352,7 +2191,7 @@ async function fetchHarnessFlags() {
 	try {
 		const res = await client["feature-flags"].$get();
 		if (!res.ok) {
-			log$11.debug({
+			log$12.debug({
 				status: res.status,
 				event: "feature_flags_fetch_failed"
 			}, "feature-flags fetch failed");
@@ -2364,7 +2203,7 @@ async function fetchHarnessFlags() {
 			closingText: body.closingText ?? null
 		};
 	} catch (err) {
-		log$11.debug({
+		log$12.debug({
 			err,
 			event: "feature_flags_fetch_error"
 		}, "feature-flags request errored");
@@ -2375,7 +2214,7 @@ function postHeartbeat({ messageId }) {
 	const client = sandboxClient();
 	if (!client) return;
 	client.heartbeat.$post({ json: { messageId } }).catch((err) => {
-		log$11.debug({
+		log$12.debug({
 			err,
 			event: "heartbeat_failed"
 		}, "heartbeat failed");
@@ -2387,7 +2226,7 @@ async function resolveConversationFromApi(messageId) {
 	try {
 		const res = await client["message-conversation"].$get({ query: { messageId } });
 		if (!res.ok) {
-			log$11.warn({
+			log$12.warn({
 				status: res.status,
 				messageId,
 				event: "resolve_conversation_failed"
@@ -2396,7 +2235,7 @@ async function resolveConversationFromApi(messageId) {
 		}
 		return (await res.json()).conversationId ?? null;
 	} catch (err) {
-		log$11.warn({
+		log$12.warn({
 			err,
 			messageId,
 			event: "resolve_conversation_error"
@@ -2421,13 +2260,13 @@ async function putBackgroundTaskJournalSpec({ messageId, spec }) {
 			messageId,
 			spec
 		} });
-		if (!res.ok) log$11.warn({
+		if (!res.ok) log$12.warn({
 			status: res.status,
 			taskId: spec.id,
 			event: "bg_journal_put_failed"
 		}, "bg-task journal PUT failed");
 	} catch (err) {
-		log$11.warn({
+		log$12.warn({
 			err,
 			taskId: spec.id,
 			event: "bg_journal_put_failed"
@@ -2443,7 +2282,7 @@ async function deleteBackgroundTaskJournalSpec({ messageId, taskId }) {
 			taskId
 		} });
 	} catch (err) {
-		log$11.debug({
+		log$12.debug({
 			err,
 			taskId,
 			event: "bg_journal_delete_failed"
@@ -2458,7 +2297,7 @@ async function listBackgroundTaskJournalSpecs({ messageId }) {
 		if (!res.ok) return [];
 		return (await res.json()).specs ?? [];
 	} catch (err) {
-		log$11.debug({
+		log$12.debug({
 			err,
 			event: "bg_journal_list_failed"
 		}, "bg-task journal GET failed");
@@ -2472,7 +2311,7 @@ function postBackgroundTasksSnapshot({ messageId, tasks }) {
 		messageId,
 		tasks
 	} }).catch((err) => {
-		log$11.debug({
+		log$12.debug({
 			err,
 			event: "bg_tasks_snapshot_failed"
 		}, "bg-tasks snapshot publish failed");
@@ -2546,7 +2385,7 @@ function createToolHeartbeat({ messageId }) {
 			}
 			heartbeatCount++;
 			if (heartbeatCount > MAX_TOOL_HEARTBEATS) {
-				log$11.warn({
+				log$12.warn({
 					heartbeatCount,
 					activeToolCalls: [...activeToolCalls]
 				}, "tool heartbeat max reached, stopping");
@@ -2577,7 +2416,7 @@ function postToDaemon(path, body) {
 		headers: { "content-type": "application/json" },
 		body: JSON.stringify(body)
 	}).catch((err) => {
-		log$11.debug({
+		log$12.debug({
 			err,
 			path,
 			event: "daemon_post_failed"
@@ -2586,7 +2425,7 @@ function postToDaemon(path, body) {
 }
 function createPlatformExtensions({ sessionId, channelContext }) {
 	return (pi) => {
-		log$11.info({
+		log$12.info({
 			sessionId,
 			hasChannelContext: Boolean(channelContext)
 		}, "platform extension initialized");
@@ -2635,12 +2474,141 @@ function createPlatformExtensions({ sessionId, channelContext }) {
 			});
 		});
 		pi.on("agent_end", () => {
-			log$11.info({ sessionId }, "session ending");
+			log$12.info({ sessionId }, "session ending");
 			postToDaemon("/session/end", { sessionId });
 		});
 	};
 }
 //#endregion
+//#region src/extensions/context-management-config.ts
+/**
+* Configuration for the context-management capability (tool-output trimming
+* and client-side microcompact). See the design notes in the extension for
+* what each layer does; this module is purely the env → config surface.
+*
+* The harness is provider-agnostic and runs inside the sandbox, where its
+* only config channel is the environment (loaded by the platform env
+* middleware before any session starts — see platform-env-middleware.ts).
+* `SKYDIVE_CONTEXT_MANAGEMENT` is the env-side master toggle; the individual
+* `SKYDIVE_CTX_*` knobs override the defaults below only when that toggle is
+* set. Prod never sets it: the LaunchDarkly flag
+* `harness-context-management-enabled` reaches running sandboxes through the
+* feature-flag poller (context-management-runtime.ts), which overlays only
+* `enabled`. So in prod every knob resolves to its default regardless of what
+* the platform forwards — see the knob-gating follow-up before relying on one.
+*
+* Resolution is defensive: a malformed value never throws (this config is
+* read on the hot path before every LLM call), it falls back to the default
+* for that knob and logs once.
+*/
+const log$11 = logger.child({ module: "context-management-config" });
+const DEFAULT_CONTEXT_MANAGEMENT_CONFIG = {
+	enabled: false,
+	perResultMaxBytes: 16 * 1024,
+	keepRecentToolResults: 3,
+	coldCacheGapSeconds: 240,
+	warmClearTriggerTokens: 5e4,
+	clearAtLeastTokens: 1e4,
+	excludeTools: [],
+	nativeAnthropicEdits: false
+};
+const boolFromEnv = (value, fallback) => {
+	if (value === void 0) return fallback;
+	const normalized = value.trim().toLowerCase();
+	if (normalized === "1" || normalized === "true") return true;
+	if (normalized === "0" || normalized === "false") return false;
+	return fallback;
+};
+const positiveInt = (fallback) => z.coerce.number().int().positive().catch(fallback);
+const nonNegativeInt = (fallback) => z.coerce.number().int().nonnegative().catch(fallback);
+const configSchema = z.object({
+	perResultMaxBytes: positiveInt(DEFAULT_CONTEXT_MANAGEMENT_CONFIG.perResultMaxBytes),
+	keepRecentToolResults: nonNegativeInt(DEFAULT_CONTEXT_MANAGEMENT_CONFIG.keepRecentToolResults),
+	coldCacheGapSeconds: positiveInt(DEFAULT_CONTEXT_MANAGEMENT_CONFIG.coldCacheGapSeconds),
+	warmClearTriggerTokens: positiveInt(DEFAULT_CONTEXT_MANAGEMENT_CONFIG.warmClearTriggerTokens),
+	clearAtLeastTokens: positiveInt(DEFAULT_CONTEXT_MANAGEMENT_CONFIG.clearAtLeastTokens)
+});
+const parseExcludeTools = (value) => {
+	if (!value) return DEFAULT_CONTEXT_MANAGEMENT_CONFIG.excludeTools;
+	return value.split(",").map((name) => name.trim()).filter((name) => name.length > 0);
+};
+/**
+* Reads the context-management config from `env` (defaults to `process.env`).
+* Never throws — invalid values fall back to defaults.
+*/
+function resolveContextManagementConfig(env = process.env) {
+	if (!boolFromEnv(env.SKYDIVE_CONTEXT_MANAGEMENT, false)) return { ...DEFAULT_CONTEXT_MANAGEMENT_CONFIG };
+	const parsed = configSchema.safeParse({
+		perResultMaxBytes: env.SKYDIVE_CTX_PER_RESULT_MAX_BYTES,
+		keepRecentToolResults: env.SKYDIVE_CTX_KEEP_RECENT,
+		coldCacheGapSeconds: env.SKYDIVE_CTX_COLD_GAP_SECONDS,
+		warmClearTriggerTokens: env.SKYDIVE_CTX_WARM_TRIGGER_TOKENS,
+		clearAtLeastTokens: env.SKYDIVE_CTX_CLEAR_AT_LEAST_TOKENS
+	});
+	if (!parsed.success) {
+		log$11.warn({
+			event: "context_management_config_invalid",
+			err: parsed.error
+		}, "falling back to default context-management config");
+		return {
+			...DEFAULT_CONTEXT_MANAGEMENT_CONFIG,
+			enabled: true
+		};
+	}
+	return {
+		enabled: true,
+		...parsed.data,
+		excludeTools: parseExcludeTools(env.SKYDIVE_CTX_EXCLUDE_TOOLS),
+		nativeAnthropicEdits: boolFromEnv(env.SKYDIVE_CTX_NATIVE_ANTHROPIC_EDITS, DEFAULT_CONTEXT_MANAGEMENT_CONFIG.nativeAnthropicEdits)
+	};
+}
+//#endregion
+//#region src/extensions/context-management-runtime.ts
+/**
+* Live, mutable view of the context-management config.
+*
+* The env-derived config (context-management-config.ts) is the boot-time
+* default. On top of it, the platform can deliver a global on/off at runtime
+* via the LaunchDarkly flag `harness-context-management-enabled` — resolved
+* server-side and polled by the harness (see the poller in
+* context-management.ts). This holder is where that override lands so a flip
+* (especially a kill-switch) reaches long-lived sandboxes without a restart.
+*
+* Only the master `enabled` toggle is overridable at runtime; the per-knob
+* tunables stay env-derived. `enabled` resolves to the flag override when the
+* platform has reported one, else the env value.
+*/
+let baseConfig = null;
+let flagOverride = null;
+function base() {
+	baseConfig ??= resolveContextManagementConfig();
+	return baseConfig;
+}
+/** The effective config, with the runtime flag override applied to `enabled`. */
+function getContextManagementConfig() {
+	const resolved = base();
+	return {
+		...resolved,
+		enabled: flagOverride ?? resolved.enabled
+	};
+}
+/**
+* Apply the platform-reported flag value. `null` clears the override (fall back
+* to the env default) — used when the phone-home result is indeterminate so a
+* transient failure never silently changes behavior.
+*/
+function setContextManagementFlagOverride(enabled) {
+	flagOverride = enabled;
+}
+/**
+* Whether a phone-home channel exists to learn the flag at runtime. When false
+* (e.g. a bare local CLI with no platform API), the env value is the only
+* source and there's nothing to poll.
+*/
+function hasFlagSource(env = process.env) {
+	return Boolean(env.SKYDIVE_API_URL ?? env.ANYONE_API_URL);
+}
+//#endregion
 //#region src/extensions/feature-flags-poll.ts
 /**
 * Shared harness feature-flag poll.
@@ -2855,10 +2823,7 @@ function createHarness(options) {
 	}, "platform runtime versions"));
 	const serverOptions = {
 		...options,
-		onSessionSetup: options.onSessionSetup ?? ((args) => {
-			installToolUpdateAutoStop(args);
-			installIterationCap(args);
-		}),
+		onSessionSetup: options.onSessionSetup ?? installToolUpdateAutoStop,
 		postPrompt: options.postPrompt ?? (async (args) => {
 			await runToolUpdateLoop(args);
 			await runClosingTextLoop(args);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@skydiveai/pi-extensions",
-  "version": "0.1.0-beta.2591",
+  "version": "0.1.0-beta.2595",
   "homepage": "https://skydive.com",
   "license": "MIT",
   "author": "Create, Inc.",