@skydiveai/pi-extensions 0.1.0-beta.6 → 0.1.0-beta.61

package/dist/index.mjs

@@ -1949,11 +1949,33 @@ const subscribers = {
 	subagent: /* @__PURE__ */ new Set()
 };
 let pollerStarted = false;
+let firstPollSettled = false;
+let resolveFirstPoll = null;
+const firstPollPromise = new Promise((resolve) => {
+	resolveFirstPoll = resolve;
+});
+function markFirstPollSettled() {
+	if (firstPollSettled) return;
+	firstPollSettled = true;
+	resolveFirstPoll?.();
+}
 /** Last-polled value of a flag, or `null` if not yet resolved. */
 function getPolledFlag(name) {
 	return name === "contextManagement" ? contextManagement : subagent;
 }
 /**
+* Await the first poll already kicked by `startFeatureFlagPoller` (never a new
+* GET). Resolves when that poll settles, immediately if it already has, or
+* immediately when there's no flag source to poll. Callers on the hot path
+* should race this against their own short timeout so a slow/failed flag
+* service cannot delay first-token; a timeout just means the caller reads the
+* still-cold cache and falls back to its default, exactly as before.
+*/
+function awaitFirstFlagPoll() {
+	if (firstPollSettled || !hasFlagSource()) return Promise.resolve();
+	return firstPollPromise;
+}
+/**
 * Subscribe to changes of a flag. The callback fires only on a *transition*
 * (skipped while the value is unchanged), so a subscriber registered before the
 * first poll still learns the initial value. Returns an unsubscribe fn.
@@ -1977,10 +1999,14 @@ function apply(name, next) {
 	}
 }
 async function pollOnce() {
-	const flags = await fetchHarnessFlags();
-	if (!flags) return;
-	apply("contextManagement", flags.contextManagement ?? null);
-	apply("subagent", flags.subagent ?? null);
+	try {
+		const flags = await fetchHarnessFlags();
+		if (!flags) return;
+		apply("contextManagement", flags.contextManagement ?? null);
+		apply("subagent", flags.subagent ?? null);
+	} catch (err) {
+		log$9.debug({ err }, "feature-flag poll threw");
+	}
 }
 /**
 * Start the shared background poll (idempotent). No-op when there's no
@@ -1991,7 +2017,7 @@ async function pollOnce() {
 function startFeatureFlagPoller() {
 	if (pollerStarted || !hasFlagSource()) return;
 	pollerStarted = true;
-	pollOnce();
+	pollOnce().finally(markFirstPollSettled);
 	setInterval(() => void pollOnce(), FLAG_POLL_INTERVAL_MS).unref?.();
 }
 //#endregion
@@ -2755,6 +2781,7 @@ const soulExtension = (pi) => {
 //#endregion
 //#region src/extensions/subagent/index.ts
 const log$2 = logger.child({ module: "subagent-ext" });
+const COLD_START_FLAG_WAIT_MS = 750;
 const MAX_TASKS = 8;
 const TaskItem = Type.Object({
 	task: Type.String({ description: "The task to delegate to a subagent run." }),
@@ -2840,11 +2867,19 @@ function createSubagentExtension({ channelContext }) {
 	return (pi) => {
 		const messageId = extractMessageId(channelContext);
 		startFeatureFlagPoller();
+		let registered = false;
+		const registerOnce = () => {
+			if (registered) return;
+			registered = true;
+			pi.registerTool(buildTool(messageId));
+			log$2.info({ event: "subagent_enabled" }, "subagent tool registered");
+		};
+		onFlagChange("subagent", (enabled) => {
+			if (enabled) registerOnce();
+		});
 		pi.on("session_start", async () => {
-			if (getPolledFlag("subagent") === true) {
-				pi.registerTool(buildTool(messageId));
-				log$2.info({ event: "subagent_enabled" }, "subagent tool registered");
-			}
+			if (getPolledFlag("subagent") === null) await Promise.race([awaitFirstFlagPoll(), new Promise((resolve) => setTimeout(resolve, COLD_START_FLAG_WAIT_MS).unref?.())]);
+			if (getPolledFlag("subagent") === true) registerOnce();
 		});
 	};
 }

package/package.json

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