@xopcai/xopc 0.0.8 → 0.0.11

package/dist/src/infra/update-startup.d.ts

@@ -0,0 +1,20 @@
+import type { Config } from '../config/schema.js';
+import { type UpdateAvailable } from './update-check.js';
+/** Get the cached update-available state (populated after startup check). */
+export declare function getUpdateAvailable(): UpdateAvailable | null;
+/**
+ * Main startup update check. Called once when the gateway starts (or on demand).
+ */
+export declare function runGatewayUpdateCheck(params: {
+    config: Config;
+    onUpdateAvailableChange?: (update: UpdateAvailable | null) => void;
+    /** When true, bypass checkOnStart/auto-disabled early exit and throttle (for POST /api/update/check). */
+    force?: boolean;
+}): Promise<void>;
+/**
+ * Schedule periodic update checks. Returns a cleanup function to stop the timer.
+ */
+export declare function scheduleGatewayUpdateCheck(params: {
+    config: Config;
+    onUpdateAvailableChange?: (update: UpdateAvailable | null) => void;
+}): () => void;

package/dist/src/infra/update-startup.js

@@ -0,0 +1,246 @@
+import { PACKAGE_VERSION, init_package_version } from "../package-version.js";
+import { createLogger } from "../utils/logger/index.js";
+import { init_logger } from "../utils/logger.js";
+import { resolveStateDir } from "../config/paths-state.js";
+import { init_paths } from "../config/paths.js";
+import { normalizeUpdateChannel } from "./update-channels.js";
+import { compareSemver, detectInstallKind, resolveNpmChannelTag, resolvePackageRoot } from "./update-check.js";
+import { dirname, join } from "node:path";
+import { mkdir, readFile, writeFile } from "node:fs/promises";
+import { createHash, randomUUID } from "node:crypto";
+//#region src/infra/update-startup.ts
+init_paths();
+init_package_version();
+init_logger();
+const log = createLogger("UpdateCheck");
+const UPDATE_CHECK_FILENAME = "update-check.json";
+const CHECK_INTERVAL_MS = 1440 * 60 * 1e3;
+const ONE_HOUR_MS = 3600 * 1e3;
+let updateAvailableCache = null;
+/** Get the cached update-available state (populated after startup check). */
+function getUpdateAvailable() {
+	return updateAvailableCache;
+}
+async function readState(statePath) {
+	try {
+		const raw = await readFile(statePath, "utf-8");
+		const parsed = JSON.parse(raw);
+		return parsed && typeof parsed === "object" ? parsed : {};
+	} catch {
+		return {};
+	}
+}
+async function writeState(statePath, state) {
+	await mkdir(dirname(statePath), { recursive: true });
+	await writeFile(statePath, JSON.stringify(state, null, 2), "utf-8");
+}
+function resolveCheckIntervalMs(config) {
+	const auto = config.update?.auto;
+	if (!auto?.enabled) return CHECK_INTERVAL_MS;
+	if ((normalizeUpdateChannel(config.update?.channel) ?? "stable") === "beta") {
+		const hours = auto.betaCheckIntervalHours ?? 1;
+		return Math.max(ONE_HOUR_MS / 4, Math.floor(hours * ONE_HOUR_MS));
+	}
+	return ONE_HOUR_MS;
+}
+/**
+* Compute a deterministic delay for stable auto-update rollout,
+* based on a per-installation hash to spread updates over time.
+*/
+function resolveStableJitterMs(installId, version, tag, jitterWindowMs) {
+	if (jitterWindowMs <= 0) return 0;
+	return createHash("sha256").update(`${installId}:${version}:${tag}`).digest().readUInt32BE(0) % (Math.floor(jitterWindowMs) + 1);
+}
+/**
+* Main startup update check. Called once when the gateway starts (or on demand).
+*/
+async function runGatewayUpdateCheck(params) {
+	const { config, force } = params;
+	const autoEnabled = config.update?.auto?.enabled ?? false;
+	const shouldCheckHints = config.update?.checkOnStart !== false;
+	if (!force && !shouldCheckHints && !autoEnabled) return;
+	const statePath = join(resolveStateDir(), UPDATE_CHECK_FILENAME);
+	const state = await readState(statePath);
+	const now = Date.now();
+	const lastCheckedAt = state.lastCheckedAt ? Date.parse(state.lastCheckedAt) : null;
+	if (state.lastAvailableVersion && (shouldCheckHints || force)) {
+		const comparison = compareSemver(PACKAGE_VERSION, state.lastAvailableVersion);
+		if (comparison !== null && comparison < 0) {
+			const cached = {
+				currentVersion: PACKAGE_VERSION,
+				latestVersion: state.lastAvailableVersion,
+				channel: state.lastAvailableTag ?? "latest"
+			};
+			updateAvailableCache = cached;
+			params.onUpdateAvailableChange?.(cached);
+		}
+	}
+	const checkIntervalMs = resolveCheckIntervalMs(config);
+	const shouldBypassThrottleForVersion = state.lastCheckPackageVersion === void 0 || state.lastCheckPackageVersion !== PACKAGE_VERSION;
+	if (!force && !shouldBypassThrottleForVersion && lastCheckedAt && Number.isFinite(lastCheckedAt) && now - lastCheckedAt < checkIntervalMs) return;
+	const root = await resolvePackageRoot();
+	let installKind = "unknown";
+	if (root) {
+		installKind = await detectInstallKind(root);
+		if (installKind === "git") log.info("Update check: git checkout (hint-only; use git pull to update, no auto npm install)");
+	}
+	const channel = normalizeUpdateChannel(config.update?.channel) ?? "stable";
+	const resolved = await resolveNpmChannelTag({
+		channel,
+		timeoutMs: 2500
+	});
+	const nextState = {
+		...state,
+		lastCheckedAt: new Date(now).toISOString()
+	};
+	if (!resolved.version) {
+		nextState.lastCheckPackageVersion = PACKAGE_VERSION;
+		await writeState(statePath, nextState);
+		return;
+	}
+	const comparison = compareSemver(PACKAGE_VERSION, resolved.version);
+	if (comparison !== null && comparison < 0) {
+		const updateInfo = {
+			currentVersion: PACKAGE_VERSION,
+			latestVersion: resolved.version,
+			channel: resolved.tag
+		};
+		if (shouldCheckHints || force) {
+			updateAvailableCache = updateInfo;
+			params.onUpdateAvailableChange?.(updateInfo);
+		}
+		nextState.lastAvailableVersion = resolved.version;
+		nextState.lastAvailableTag = resolved.tag;
+		const shouldNotify = state.lastNotifiedVersion !== resolved.version || state.lastNotifiedTag !== resolved.tag;
+		if ((shouldCheckHints || force) && shouldNotify) {
+			log.info({
+				currentVersion: PACKAGE_VERSION,
+				latestVersion: resolved.version,
+				tag: resolved.tag
+			}, `Update available (${resolved.tag}): v${resolved.version} (current v${PACKAGE_VERSION}). Run: xopc update`);
+			nextState.lastNotifiedVersion = resolved.version;
+			nextState.lastNotifiedTag = resolved.tag;
+		}
+		if (autoEnabled && (channel === "stable" || channel === "beta") && installKind !== "git") await handleAutoUpdate({
+			channel,
+			version: resolved.version,
+			tag: resolved.tag,
+			state,
+			nextState,
+			now,
+			root,
+			config
+		});
+	} else {
+		delete nextState.lastAvailableVersion;
+		delete nextState.lastAvailableTag;
+		updateAvailableCache = null;
+		params.onUpdateAvailableChange?.(null);
+	}
+	nextState.lastCheckPackageVersion = PACKAGE_VERSION;
+	await writeState(statePath, nextState);
+}
+async function handleAutoUpdate(params) {
+	const { channel, version, tag, state, nextState, now, root, config } = params;
+	const auto = config.update?.auto;
+	if (!auto) return;
+	const stableDelayHours = auto.stableDelayHours ?? 6;
+	const stableJitterHours = auto.stableJitterHours ?? 12;
+	const betaCheckIntervalHours = auto.betaCheckIntervalHours ?? 1;
+	const attemptIntervalMs = channel === "beta" ? Math.max(ONE_HOUR_MS / 4, Math.floor(betaCheckIntervalHours * ONE_HOUR_MS)) : ONE_HOUR_MS;
+	const lastAttemptAt = state.autoLastAttemptAt ? Date.parse(state.autoLastAttemptAt) : null;
+	if (state.autoLastAttemptVersion === version && lastAttemptAt !== null && Number.isFinite(lastAttemptAt) && now - lastAttemptAt < attemptIntervalMs) {
+		log.info({
+			version,
+			tag
+		}, "Auto-update deferred: recent attempt exists");
+		return;
+	}
+	if (channel === "stable") {
+		if (!nextState.autoInstallId) nextState.autoInstallId = state.autoInstallId?.trim() || randomUUID();
+		if (state.autoFirstSeenVersion !== version || state.autoFirstSeenTag !== tag) {
+			nextState.autoFirstSeenVersion = version;
+			nextState.autoFirstSeenTag = tag;
+			nextState.autoFirstSeenAt = new Date(now).toISOString();
+		} else nextState.autoFirstSeenAt = state.autoFirstSeenAt;
+		const firstSeenMs = nextState.autoFirstSeenAt ? Date.parse(nextState.autoFirstSeenAt) : now;
+		const baseDelayMs = Math.max(0, stableDelayHours) * ONE_HOUR_MS;
+		const jitterWindowMs = Math.max(0, stableJitterHours) * ONE_HOUR_MS;
+		const jitterMs = resolveStableJitterMs(nextState.autoInstallId, version, tag, jitterWindowMs);
+		const applyAfterMs = firstSeenMs + baseDelayMs + jitterMs;
+		if (now < applyAfterMs) {
+			log.info({
+				version,
+				tag,
+				applyAfter: new Date(applyAfterMs).toISOString()
+			}, "Auto-update deferred: stable rollout window not yet due");
+			return;
+		}
+	}
+	nextState.autoLastAttemptVersion = version;
+	nextState.autoLastAttemptAt = new Date(now).toISOString();
+	log.info({
+		channel,
+		version,
+		tag
+	}, "Starting auto-update");
+	try {
+		const { runAutoUpdateCommand } = await import("./update-runner.js");
+		const result = await runAutoUpdateCommand({
+			channel,
+			root
+		});
+		if (result.ok) {
+			nextState.autoLastSuccessVersion = version;
+			nextState.autoLastSuccessAt = new Date(now).toISOString();
+			log.info({
+				channel,
+				version,
+				tag
+			}, "Auto-update applied successfully");
+		} else log.warn({
+			channel,
+			version,
+			tag,
+			exitCode: result.exitCode,
+			reason: result.reason
+		}, `Auto-update attempt failed: ${result.reason ?? `exit ${result.exitCode}`}`);
+	} catch (err) {
+		log.error({
+			err,
+			channel,
+			version
+		}, "Auto-update command threw");
+	}
+}
+/**
+* Schedule periodic update checks. Returns a cleanup function to stop the timer.
+*/
+function scheduleGatewayUpdateCheck(params) {
+	let stopped = false;
+	let timer = null;
+	const tick = async () => {
+		if (stopped) return;
+		try {
+			await runGatewayUpdateCheck(params);
+		} catch (err) {
+			log.warn({ err }, "Periodic update check failed");
+		}
+		if (!stopped) {
+			const intervalMs = resolveCheckIntervalMs(params.config);
+			timer = setTimeout(() => void tick(), intervalMs);
+		}
+	};
+	timer = setTimeout(() => void tick(), 5e3);
+	return () => {
+		stopped = true;
+		if (timer) {
+			clearTimeout(timer);
+			timer = null;
+		}
+	};
+}
+//#endregion
+export { getUpdateAvailable, runGatewayUpdateCheck, scheduleGatewayUpdateCheck };
+
+//# sourceMappingURL=update-startup.js.map

package/dist/src/infra/update-startup.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"update-startup.js","names":[],"sources":["../../../src/infra/update-startup.ts"],"sourcesContent":["// src/infra/update-startup.ts\n\nimport { createHash, randomUUID } from 'node:crypto';\nimport { readFile, writeFile, mkdir } from 'node:fs/promises';\nimport { join, dirname } from 'node:path';\n\nimport type { Config } from '../config/schema.js';\nimport { resolveStateDir } from '../config/paths.js';\nimport { PACKAGE_VERSION } from '../package-version.js';\nimport { createLogger } from '../utils/logger.js';\n\nimport { normalizeUpdateChannel, DEFAULT_PACKAGE_CHANNEL } from './update-channels.js';\nimport {\n  compareSemver,\n  resolveNpmChannelTag,\n  detectInstallKind,\n  resolvePackageRoot,\n  type InstallKind,\n  type UpdateAvailable,\n} from './update-check.js';\n\nconst log = createLogger('UpdateCheck');\n\n// --- State persistence ---\n\nconst UPDATE_CHECK_FILENAME = 'update-check.json';\nconst CHECK_INTERVAL_MS = 24 * 60 * 60 * 1000; // 24 hours\nconst ONE_HOUR_MS = 60 * 60 * 1000;\n\ntype UpdateCheckState = {\n  /** `package.json` version at the last successful registry check (used to bypass 24h throttle when you bump the local version). */\n  lastCheckPackageVersion?: string;\n  lastCheckedAt?: string;\n  lastAvailableVersion?: string;\n  lastAvailableTag?: string;\n  lastNotifiedVersion?: string;\n  lastNotifiedTag?: string;\n  autoInstallId?: string;\n  autoFirstSeenVersion?: string;\n  autoFirstSeenTag?: string;\n  autoFirstSeenAt?: string;\n  autoLastAttemptVersion?: string;\n  autoLastAttemptAt?: string;\n  autoLastSuccessVersion?: string;\n  autoLastSuccessAt?: string;\n};\n\n// --- In-memory cache ---\n\nlet updateAvailableCache: UpdateAvailable | null = null;\n\n/** Get the cached update-available state (populated after startup check). */\nexport function getUpdateAvailable(): UpdateAvailable | null {\n  return updateAvailableCache;\n}\n\n// --- Core logic ---\n\nasync function readState(statePath: string): Promise<UpdateCheckState> {\n  try {\n    const raw = await readFile(statePath, 'utf-8');\n    const parsed = JSON.parse(raw) as UpdateCheckState;\n    return parsed && typeof parsed === 'object' ? parsed : {};\n  } catch {\n    return {};\n  }\n}\n\nasync function writeState(statePath: string, state: UpdateCheckState): Promise<void> {\n  await mkdir(dirname(statePath), { recursive: true });\n  await writeFile(statePath, JSON.stringify(state, null, 2), 'utf-8');\n}\n\nfunction resolveCheckIntervalMs(config: Config): number {\n  const auto = config.update?.auto;\n  if (!auto?.enabled) return CHECK_INTERVAL_MS;\n\n  const channel = normalizeUpdateChannel(config.update?.channel) ?? DEFAULT_PACKAGE_CHANNEL;\n  if (channel === 'beta') {\n    const hours = auto.betaCheckIntervalHours ?? 1;\n    return Math.max(ONE_HOUR_MS / 4, Math.floor(hours * ONE_HOUR_MS));\n  }\n  return ONE_HOUR_MS;\n}\n\n/**\n * Compute a deterministic delay for stable auto-update rollout,\n * based on a per-installation hash to spread updates over time.\n */\nfunction resolveStableJitterMs(\n  installId: string,\n  version: string,\n  tag: string,\n  jitterWindowMs: number,\n): number {\n  if (jitterWindowMs <= 0) return 0;\n  const hash = createHash('sha256').update(`${installId}:${version}:${tag}`).digest();\n  const bucket = hash.readUInt32BE(0);\n  return bucket % (Math.floor(jitterWindowMs) + 1);\n}\n\n/**\n * Main startup update check. Called once when the gateway starts (or on demand).\n */\nexport async function runGatewayUpdateCheck(params: {\n  config: Config;\n  onUpdateAvailableChange?: (update: UpdateAvailable | null) => void;\n  /** When true, bypass checkOnStart/auto-disabled early exit and throttle (for POST /api/update/check). */\n  force?: boolean;\n}): Promise<void> {\n  const { config, force } = params;\n\n  const autoEnabled = config.update?.auto?.enabled ?? false;\n  const shouldCheckHints = config.update?.checkOnStart !== false;\n  if (!force && !shouldCheckHints && !autoEnabled) return;\n\n  const stateDir = resolveStateDir();\n  const statePath = join(stateDir, UPDATE_CHECK_FILENAME);\n  const state = await readState(statePath);\n  const now = Date.now();\n\n  // Hydrate from persisted state if within throttle window\n  const lastCheckedAt = state.lastCheckedAt ? Date.parse(state.lastCheckedAt) : null;\n  if (state.lastAvailableVersion && (shouldCheckHints || force)) {\n    const comparison = compareSemver(PACKAGE_VERSION, state.lastAvailableVersion);\n    if (comparison !== null && comparison < 0) {\n      const cached: UpdateAvailable = {\n        currentVersion: PACKAGE_VERSION,\n        latestVersion: state.lastAvailableVersion,\n        channel: state.lastAvailableTag ?? 'latest',\n      };\n      updateAvailableCache = cached;\n      params.onUpdateAvailableChange?.(cached);\n    }\n  }\n\n  const checkIntervalMs = resolveCheckIntervalMs(config);\n  // Re-check npm when the local package version changed (e.g. after editing package.json) even within 24h.\n  const shouldBypassThrottleForVersion =\n    state.lastCheckPackageVersion === undefined || state.lastCheckPackageVersion !== PACKAGE_VERSION;\n  if (\n    !force &&\n    !shouldBypassThrottleForVersion &&\n    lastCheckedAt &&\n    Number.isFinite(lastCheckedAt) &&\n    now - lastCheckedAt < checkIntervalMs\n  ) {\n    return; // Within throttle window\n  }\n\n  // Install kind: auto-install only for npm global installs, but we still query npm in git\n  // so the Web UI / CLI can show \"newer on registry\" and the top reminder bar.\n  const root = await resolvePackageRoot();\n  let installKind: InstallKind = 'unknown';\n  if (root) {\n    installKind = await detectInstallKind(root);\n    if (installKind === 'git') {\n      log.info('Update check: git checkout (hint-only; use git pull to update, no auto npm install)');\n    }\n  }\n\n  // Query npm registry\n  const channel = normalizeUpdateChannel(config.update?.channel) ?? DEFAULT_PACKAGE_CHANNEL;\n  const resolved = await resolveNpmChannelTag({ channel, timeoutMs: 2500 });\n\n  const nextState: UpdateCheckState = {\n    ...state,\n    lastCheckedAt: new Date(now).toISOString(),\n  };\n\n  if (!resolved.version) {\n    nextState.lastCheckPackageVersion = PACKAGE_VERSION;\n    await writeState(statePath, nextState);\n    return;\n  }\n\n  const comparison = compareSemver(PACKAGE_VERSION, resolved.version);\n  if (comparison !== null && comparison < 0) {\n    // Update available\n    const updateInfo: UpdateAvailable = {\n      currentVersion: PACKAGE_VERSION,\n      latestVersion: resolved.version,\n      channel: resolved.tag,\n    };\n\n    if (shouldCheckHints || force) {\n      updateAvailableCache = updateInfo;\n      params.onUpdateAvailableChange?.(updateInfo);\n    }\n\n    nextState.lastAvailableVersion = resolved.version;\n    nextState.lastAvailableTag = resolved.tag;\n\n    // Log notification (once per version)\n    const shouldNotify =\n      state.lastNotifiedVersion !== resolved.version || state.lastNotifiedTag !== resolved.tag;\n    if ((shouldCheckHints || force) && shouldNotify) {\n      log.info(\n        { currentVersion: PACKAGE_VERSION, latestVersion: resolved.version, tag: resolved.tag },\n        `Update available (${resolved.tag}): v${resolved.version} (current v${PACKAGE_VERSION}). Run: xopc update`,\n      );\n      nextState.lastNotifiedVersion = resolved.version;\n      nextState.lastNotifiedTag = resolved.tag;\n    }\n\n    // Auto-update logic (never from a git worktree)\n    if (\n      autoEnabled &&\n      (channel === 'stable' || channel === 'beta') &&\n      installKind !== 'git'\n    ) {\n      await handleAutoUpdate({\n        channel,\n        version: resolved.version,\n        tag: resolved.tag,\n        state,\n        nextState,\n        now,\n        root,\n        config,\n      });\n    }\n  } else {\n    // Current version is up to date or newer\n    delete nextState.lastAvailableVersion;\n    delete nextState.lastAvailableTag;\n    updateAvailableCache = null;\n    params.onUpdateAvailableChange?.(null);\n  }\n\n  nextState.lastCheckPackageVersion = PACKAGE_VERSION;\n  await writeState(statePath, nextState);\n}\n\nasync function handleAutoUpdate(params: {\n  channel: 'stable' | 'beta';\n  version: string;\n  tag: string;\n  state: UpdateCheckState;\n  nextState: UpdateCheckState;\n  now: number;\n  root: string | null;\n  config: Config;\n}): Promise<void> {\n  const { channel, version, tag, state, nextState, now, root, config } = params;\n  const auto = config.update?.auto;\n  if (!auto) return;\n\n  const stableDelayHours = auto.stableDelayHours ?? 6;\n  const stableJitterHours = auto.stableJitterHours ?? 12;\n  const betaCheckIntervalHours = auto.betaCheckIntervalHours ?? 1;\n\n  // Rate limit: don't re-attempt same version within interval\n  const attemptIntervalMs =\n    channel === 'beta'\n      ? Math.max(ONE_HOUR_MS / 4, Math.floor(betaCheckIntervalHours * ONE_HOUR_MS))\n      : ONE_HOUR_MS;\n  const lastAttemptAt = state.autoLastAttemptAt ? Date.parse(state.autoLastAttemptAt) : null;\n  const recentAttempt =\n    state.autoLastAttemptVersion === version &&\n    lastAttemptAt !== null &&\n    Number.isFinite(lastAttemptAt) &&\n    now - lastAttemptAt < attemptIntervalMs;\n\n  if (recentAttempt) {\n    log.info({ version, tag }, 'Auto-update deferred: recent attempt exists');\n    return;\n  }\n\n  // Stable rollout delay + jitter\n  if (channel === 'stable') {\n    if (!nextState.autoInstallId) {\n      nextState.autoInstallId = state.autoInstallId?.trim() || randomUUID();\n    }\n    // Track first-seen time for this version\n    if (state.autoFirstSeenVersion !== version || state.autoFirstSeenTag !== tag) {\n      nextState.autoFirstSeenVersion = version;\n      nextState.autoFirstSeenTag = tag;\n      nextState.autoFirstSeenAt = new Date(now).toISOString();\n    } else {\n      nextState.autoFirstSeenAt = state.autoFirstSeenAt;\n    }\n\n    const firstSeenMs = nextState.autoFirstSeenAt ? Date.parse(nextState.autoFirstSeenAt) : now;\n    const baseDelayMs = Math.max(0, stableDelayHours) * ONE_HOUR_MS;\n    const jitterWindowMs = Math.max(0, stableJitterHours) * ONE_HOUR_MS;\n    const jitterMs = resolveStableJitterMs(nextState.autoInstallId, version, tag, jitterWindowMs);\n    const applyAfterMs = firstSeenMs + baseDelayMs + jitterMs;\n\n    if (now < applyAfterMs) {\n      log.info(\n        { version, tag, applyAfter: new Date(applyAfterMs).toISOString() },\n        'Auto-update deferred: stable rollout window not yet due',\n      );\n      return;\n    }\n  }\n\n  // Execute auto-update\n  nextState.autoLastAttemptVersion = version;\n  nextState.autoLastAttemptAt = new Date(now).toISOString();\n\n  log.info({ channel, version, tag }, 'Starting auto-update');\n\n  try {\n    const { runAutoUpdateCommand } = await import('./update-runner.js');\n    const result = await runAutoUpdateCommand({ channel, root });\n    if (result.ok) {\n      nextState.autoLastSuccessVersion = version;\n      nextState.autoLastSuccessAt = new Date(now).toISOString();\n      log.info({ channel, version, tag }, 'Auto-update applied successfully');\n    } else {\n      log.warn(\n        { channel, version, tag, exitCode: result.exitCode, reason: result.reason },\n        `Auto-update attempt failed: ${result.reason ?? `exit ${result.exitCode}`}`,\n      );\n    }\n  } catch (err) {\n    log.error({ err, channel, version }, 'Auto-update command threw');\n  }\n}\n\n/**\n * Schedule periodic update checks. Returns a cleanup function to stop the timer.\n */\nexport function scheduleGatewayUpdateCheck(params: {\n  config: Config;\n  onUpdateAvailableChange?: (update: UpdateAvailable | null) => void;\n}): () => void {\n  let stopped = false;\n  let timer: ReturnType<typeof setTimeout> | null = null;\n\n  const tick = async () => {\n    if (stopped) return;\n    try {\n      await runGatewayUpdateCheck(params);\n    } catch (err) {\n      log.warn({ err }, 'Periodic update check failed');\n    }\n    if (!stopped) {\n      const intervalMs = resolveCheckIntervalMs(params.config);\n      timer = setTimeout(() => void tick(), intervalMs);\n    }\n  };\n\n  // Initial check after a short delay (don't block startup)\n  timer = setTimeout(() => void tick(), 5000);\n\n  return () => {\n    stopped = true;\n    if (timer) {\n      clearTimeout(timer);\n      timer = null;\n    }\n  };\n}\n"],"mappings":";;;;;;;;;;;YAOqD;sBACG;aACN;AAYlD,MAAM,MAAM,aAAa,cAAc;AAIvC,MAAM,wBAAwB;AAC9B,MAAM,oBAAoB,OAAU,KAAK;AACzC,MAAM,cAAc,OAAU;AAsB9B,IAAI,uBAA+C;;AAGnD,SAAgB,qBAA6C;AAC3D,QAAO;;AAKT,eAAe,UAAU,WAA8C;AACrE,KAAI;EACF,MAAM,MAAM,MAAM,SAAS,WAAW,QAAQ;EAC9C,MAAM,SAAS,KAAK,MAAM,IAAI;AAC9B,SAAO,UAAU,OAAO,WAAW,WAAW,SAAS,EAAE;SACnD;AACN,SAAO,EAAE;;;AAIb,eAAe,WAAW,WAAmB,OAAwC;AACnF,OAAM,MAAM,QAAQ,UAAU,EAAE,EAAE,WAAW,MAAM,CAAC;AACpD,OAAM,UAAU,WAAW,KAAK,UAAU,OAAO,MAAM,EAAE,EAAE,QAAQ;;AAGrE,SAAS,uBAAuB,QAAwB;CACtD,MAAM,OAAO,OAAO,QAAQ;AAC5B,KAAI,CAAC,MAAM,QAAS,QAAO;AAG3B,MADgB,uBAAuB,OAAO,QAAQ,QAAQ,IAAA,cAC9C,QAAQ;EACtB,MAAM,QAAQ,KAAK,0BAA0B;AAC7C,SAAO,KAAK,IAAI,cAAc,GAAG,KAAK,MAAM,QAAQ,YAAY,CAAC;;AAEnE,QAAO;;;;;;AAOT,SAAS,sBACP,WACA,SACA,KACA,gBACQ;AACR,KAAI,kBAAkB,EAAG,QAAO;AAGhC,QAFa,WAAW,SAAS,CAAC,OAAO,GAAG,UAAU,GAAG,QAAQ,GAAG,MAAM,CAAC,QAAQ,CAC/D,aAAa,EAAE,IAClB,KAAK,MAAM,eAAe,GAAG;;;;;AAMhD,eAAsB,sBAAsB,QAK1B;CAChB,MAAM,EAAE,QAAQ,UAAU;CAE1B,MAAM,cAAc,OAAO,QAAQ,MAAM,WAAW;CACpD,MAAM,mBAAmB,OAAO,QAAQ,iBAAiB;AACzD,KAAI,CAAC,SAAS,CAAC,oBAAoB,CAAC,YAAa;CAGjD,MAAM,YAAY,KADD,iBAAiB,EACD,sBAAsB;CACvD,MAAM,QAAQ,MAAM,UAAU,UAAU;CACxC,MAAM,MAAM,KAAK,KAAK;CAGtB,MAAM,gBAAgB,MAAM,gBAAgB,KAAK,MAAM,MAAM,cAAc,GAAG;AAC9E,KAAI,MAAM,yBAAyB,oBAAoB,QAAQ;EAC7D,MAAM,aAAa,cAAc,iBAAiB,MAAM,qBAAqB;AAC7E,MAAI,eAAe,QAAQ,aAAa,GAAG;GACzC,MAAM,SAA0B;IAC9B,gBAAgB;IAChB,eAAe,MAAM;IACrB,SAAS,MAAM,oBAAoB;IACpC;AACD,0BAAuB;AACvB,UAAO,0BAA0B,OAAO;;;CAI5C,MAAM,kBAAkB,uBAAuB,OAAO;CAEtD,MAAM,iCACJ,MAAM,4BAA4B,KAAA,KAAa,MAAM,4BAA4B;AACnF,KACE,CAAC,SACD,CAAC,kCACD,iBACA,OAAO,SAAS,cAAc,IAC9B,MAAM,gBAAgB,gBAEtB;CAKF,MAAM,OAAO,MAAM,oBAAoB;CACvC,IAAI,cAA2B;AAC/B,KAAI,MAAM;AACR,gBAAc,MAAM,kBAAkB,KAAK;AAC3C,MAAI,gBAAgB,MAClB,KAAI,KAAK,sFAAsF;;CAKnG,MAAM,UAAU,uBAAuB,OAAO,QAAQ,QAAQ,IAAA;CAC9D,MAAM,WAAW,MAAM,qBAAqB;EAAE;EAAS,WAAW;EAAM,CAAC;CAEzE,MAAM,YAA8B;EAClC,GAAG;EACH,eAAe,IAAI,KAAK,IAAI,CAAC,aAAa;EAC3C;AAED,KAAI,CAAC,SAAS,SAAS;AACrB,YAAU,0BAA0B;AACpC,QAAM,WAAW,WAAW,UAAU;AACtC;;CAGF,MAAM,aAAa,cAAc,iBAAiB,SAAS,QAAQ;AACnE,KAAI,eAAe,QAAQ,aAAa,GAAG;EAEzC,MAAM,aAA8B;GAClC,gBAAgB;GAChB,eAAe,SAAS;GACxB,SAAS,SAAS;GACnB;AAED,MAAI,oBAAoB,OAAO;AAC7B,0BAAuB;AACvB,UAAO,0BAA0B,WAAW;;AAG9C,YAAU,uBAAuB,SAAS;AAC1C,YAAU,mBAAmB,SAAS;EAGtC,MAAM,eACJ,MAAM,wBAAwB,SAAS,WAAW,MAAM,oBAAoB,SAAS;AACvF,OAAK,oBAAoB,UAAU,cAAc;AAC/C,OAAI,KACF;IAAE,gBAAgB;IAAiB,eAAe,SAAS;IAAS,KAAK,SAAS;IAAK,EACvF,qBAAqB,SAAS,IAAI,MAAM,SAAS,QAAQ,aAAa,gBAAgB,qBACvF;AACD,aAAU,sBAAsB,SAAS;AACzC,aAAU,kBAAkB,SAAS;;AAIvC,MACE,gBACC,YAAY,YAAY,YAAY,WACrC,gBAAgB,MAEhB,OAAM,iBAAiB;GACrB;GACA,SAAS,SAAS;GAClB,KAAK,SAAS;GACd;GACA;GACA;GACA;GACA;GACD,CAAC;QAEC;AAEL,SAAO,UAAU;AACjB,SAAO,UAAU;AACjB,yBAAuB;AACvB,SAAO,0BAA0B,KAAK;;AAGxC,WAAU,0BAA0B;AACpC,OAAM,WAAW,WAAW,UAAU;;AAGxC,eAAe,iBAAiB,QASd;CAChB,MAAM,EAAE,SAAS,SAAS,KAAK,OAAO,WAAW,KAAK,MAAM,WAAW;CACvE,MAAM,OAAO,OAAO,QAAQ;AAC5B,KAAI,CAAC,KAAM;CAEX,MAAM,mBAAmB,KAAK,oBAAoB;CAClD,MAAM,oBAAoB,KAAK,qBAAqB;CACpD,MAAM,yBAAyB,KAAK,0BAA0B;CAG9D,MAAM,oBACJ,YAAY,SACR,KAAK,IAAI,cAAc,GAAG,KAAK,MAAM,yBAAyB,YAAY,CAAC,GAC3E;CACN,MAAM,gBAAgB,MAAM,oBAAoB,KAAK,MAAM,MAAM,kBAAkB,GAAG;AAOtF,KALE,MAAM,2BAA2B,WACjC,kBAAkB,QAClB,OAAO,SAAS,cAAc,IAC9B,MAAM,gBAAgB,mBAEL;AACjB,MAAI,KAAK;GAAE;GAAS;GAAK,EAAE,8CAA8C;AACzE;;AAIF,KAAI,YAAY,UAAU;AACxB,MAAI,CAAC,UAAU,cACb,WAAU,gBAAgB,MAAM,eAAe,MAAM,IAAI,YAAY;AAGvE,MAAI,MAAM,yBAAyB,WAAW,MAAM,qBAAqB,KAAK;AAC5E,aAAU,uBAAuB;AACjC,aAAU,mBAAmB;AAC7B,aAAU,kBAAkB,IAAI,KAAK,IAAI,CAAC,aAAa;QAEvD,WAAU,kBAAkB,MAAM;EAGpC,MAAM,cAAc,UAAU,kBAAkB,KAAK,MAAM,UAAU,gBAAgB,GAAG;EACxF,MAAM,cAAc,KAAK,IAAI,GAAG,iBAAiB,GAAG;EACpD,MAAM,iBAAiB,KAAK,IAAI,GAAG,kBAAkB,GAAG;EACxD,MAAM,WAAW,sBAAsB,UAAU,eAAe,SAAS,KAAK,eAAe;EAC7F,MAAM,eAAe,cAAc,cAAc;AAEjD,MAAI,MAAM,cAAc;AACtB,OAAI,KACF;IAAE;IAAS;IAAK,YAAY,IAAI,KAAK,aAAa,CAAC,aAAa;IAAE,EAClE,0DACD;AACD;;;AAKJ,WAAU,yBAAyB;AACnC,WAAU,oBAAoB,IAAI,KAAK,IAAI,CAAC,aAAa;AAEzD,KAAI,KAAK;EAAE;EAAS;EAAS;EAAK,EAAE,uBAAuB;AAE3D,KAAI;EACF,MAAM,EAAE,yBAAyB,MAAM,OAAO;EAC9C,MAAM,SAAS,MAAM,qBAAqB;GAAE;GAAS;GAAM,CAAC;AAC5D,MAAI,OAAO,IAAI;AACb,aAAU,yBAAyB;AACnC,aAAU,oBAAoB,IAAI,KAAK,IAAI,CAAC,aAAa;AACzD,OAAI,KAAK;IAAE;IAAS;IAAS;IAAK,EAAE,mCAAmC;QAEvE,KAAI,KACF;GAAE;GAAS;GAAS;GAAK,UAAU,OAAO;GAAU,QAAQ,OAAO;GAAQ,EAC3E,+BAA+B,OAAO,UAAU,QAAQ,OAAO,aAChE;UAEI,KAAK;AACZ,MAAI,MAAM;GAAE;GAAK;GAAS;GAAS,EAAE,4BAA4B;;;;;;AAOrE,SAAgB,2BAA2B,QAG5B;CACb,IAAI,UAAU;CACd,IAAI,QAA8C;CAElD,MAAM,OAAO,YAAY;AACvB,MAAI,QAAS;AACb,MAAI;AACF,SAAM,sBAAsB,OAAO;WAC5B,KAAK;AACZ,OAAI,KAAK,EAAE,KAAK,EAAE,+BAA+B;;AAEnD,MAAI,CAAC,SAAS;GACZ,MAAM,aAAa,uBAAuB,OAAO,OAAO;AACxD,WAAQ,iBAAiB,KAAK,MAAM,EAAE,WAAW;;;AAKrD,SAAQ,iBAAiB,KAAK,MAAM,EAAE,IAAK;AAE3C,cAAa;AACX,YAAU;AACV,MAAI,OAAO;AACT,gBAAa,MAAM;AACnB,WAAQ"}

package/dist/src/providers/extension-stream-bridge.d.ts

@@ -0,0 +1,3 @@
+import type { StreamFn } from '@mariozechner/pi-agent-core';
+/** Wraps {@link streamSimple} so extension-registered models use the plugin `createStream()` path. */
+export declare function createExtensionAwareStreamFn(): StreamFn;

package/dist/src/providers/extension-stream-bridge.js

@@ -0,0 +1,239 @@
+import { createLogger } from "../utils/logger/index.js";
+import { init_logger } from "../utils/logger.js";
+import { getProviderRegistry, init_plugin_registry } from "./plugin-registry.js";
+import { init_providers } from "./index.js";
+import { createAssistantMessageEventStream, streamSimple } from "@mariozechner/pi-ai";
+//#region src/providers/extension-stream-bridge.ts
+init_plugin_registry();
+init_providers();
+init_logger();
+const log = createLogger("ExtensionStreamBridge");
+function createPartialMessage(model) {
+	return {
+		role: "assistant",
+		content: [],
+		api: model.api,
+		provider: model.provider,
+		model: model.id,
+		usage: {
+			input: 0,
+			output: 0,
+			cacheRead: 0,
+			cacheWrite: 0,
+			totalTokens: 0,
+			cost: {
+				input: 0,
+				output: 0,
+				cacheRead: 0,
+				cacheWrite: 0,
+				total: 0
+			}
+		},
+		stopReason: "stop",
+		timestamp: Date.now()
+	};
+}
+function parseToolArguments(raw) {
+	if (!raw?.trim()) return {};
+	try {
+		return JSON.parse(raw);
+	} catch {
+		return {};
+	}
+}
+/** Wraps {@link streamSimple} so extension-registered models use the plugin `createStream()` path. */
+function createExtensionAwareStreamFn() {
+	return ((model, context, options) => {
+		if (model.baseUrl !== "extension://provider-plugin") return streamSimple(model, context, options);
+		const plugin = getProviderRegistry().get(model.provider);
+		if (!plugin) return streamSimple(model, context, options);
+		log.info({
+			prefix: "ExtensionStreamBridge",
+			msg: "Streaming via extension provider"
+		});
+		const stream = createAssistantMessageEventStream();
+		const params = {
+			model: model.id,
+			messages: context.messages,
+			temperature: options?.temperature,
+			maxTokens: options?.maxTokens,
+			signal: options?.signal
+		};
+		(async () => {
+			const partial = createPartialMessage(model);
+			let textContentIndex = -1;
+			let hasStarted = false;
+			try {
+				for await (const chunk of plugin.createStream(params)) {
+					if (options?.signal?.aborted) {
+						partial.stopReason = "aborted";
+						partial.errorMessage = "Request aborted";
+						stream.push({
+							type: "error",
+							reason: "aborted",
+							error: partial
+						});
+						return;
+					}
+					switch (chunk.type) {
+						case "text": {
+							if (!hasStarted) {
+								hasStarted = true;
+								stream.push({
+									type: "start",
+									partial
+								});
+							}
+							if (textContentIndex === -1) {
+								textContentIndex = partial.content.length;
+								partial.content.push({
+									type: "text",
+									text: ""
+								});
+								stream.push({
+									type: "text_start",
+									contentIndex: textContentIndex,
+									partial
+								});
+							}
+							const textContent = partial.content[textContentIndex];
+							if (textContent?.type === "text" && chunk.text) {
+								textContent.text += chunk.text;
+								stream.push({
+									type: "text_delta",
+									contentIndex: textContentIndex,
+									delta: chunk.text,
+									partial
+								});
+							}
+							break;
+						}
+						case "tool_call":
+							if (!hasStarted) {
+								hasStarted = true;
+								stream.push({
+									type: "start",
+									partial
+								});
+							}
+							if (textContentIndex !== -1) {
+								const tc = partial.content[textContentIndex];
+								if (tc?.type === "text") stream.push({
+									type: "text_end",
+									contentIndex: textContentIndex,
+									content: tc.text,
+									partial
+								});
+								textContentIndex = -1;
+							}
+							if (chunk.toolCall) {
+								const idx = partial.content.length;
+								const toolCall = {
+									type: "toolCall",
+									id: chunk.toolCall.id,
+									name: chunk.toolCall.name,
+									arguments: parseToolArguments(chunk.toolCall.arguments)
+								};
+								partial.content.push(toolCall);
+								stream.push({
+									type: "toolcall_start",
+									contentIndex: idx,
+									partial
+								});
+								stream.push({
+									type: "toolcall_delta",
+									contentIndex: idx,
+									delta: chunk.toolCall.arguments || "{}",
+									partial
+								});
+								stream.push({
+									type: "toolcall_end",
+									contentIndex: idx,
+									toolCall,
+									partial
+								});
+							}
+							break;
+						case "usage":
+							if (chunk.usage) {
+								const inputCost = chunk.usage.input / 1e6 * (model.cost?.input ?? 0);
+								const outputCost = chunk.usage.output / 1e6 * (model.cost?.output ?? 0);
+								partial.usage = {
+									input: chunk.usage.input,
+									output: chunk.usage.output,
+									cacheRead: 0,
+									cacheWrite: 0,
+									totalTokens: chunk.usage.total ?? chunk.usage.input + chunk.usage.output,
+									cost: {
+										input: inputCost,
+										output: outputCost,
+										cacheRead: 0,
+										cacheWrite: 0,
+										total: inputCost + outputCost
+									}
+								};
+							}
+							break;
+						case "error":
+							if (!hasStarted) {
+								hasStarted = true;
+								stream.push({
+									type: "start",
+									partial
+								});
+							}
+							partial.stopReason = "error";
+							partial.errorMessage = chunk.error ?? "Unknown extension provider error";
+							stream.push({
+								type: "error",
+								reason: "error",
+								error: partial
+							});
+							return;
+						case "done": break;
+					}
+				}
+				if (textContentIndex !== -1) {
+					const tc = partial.content[textContentIndex];
+					if (tc?.type === "text") stream.push({
+						type: "text_end",
+						contentIndex: textContentIndex,
+						content: tc.text,
+						partial
+					});
+				}
+				if (!hasStarted) {
+					hasStarted = true;
+					stream.push({
+						type: "start",
+						partial
+					});
+				}
+				const stopReason = partial.content.some((c) => c.type === "toolCall") ? "toolUse" : "stop";
+				partial.stopReason = stopReason;
+				stream.push({
+					type: "done",
+					reason: stopReason,
+					message: partial
+				});
+			} catch (err) {
+				if (!hasStarted) stream.push({
+					type: "start",
+					partial
+				});
+				partial.stopReason = "error";
+				partial.errorMessage = err instanceof Error ? err.message : String(err);
+				stream.push({
+					type: "error",
+					reason: "error",
+					error: partial
+				});
+			}
+		})();
+		return stream;
+	});
+}
+//#endregion
+export { createExtensionAwareStreamFn };
+
+//# sourceMappingURL=extension-stream-bridge.js.map

package/dist/src/providers/extension-stream-bridge.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"extension-stream-bridge.js","names":[],"sources":["../../../src/providers/extension-stream-bridge.ts"],"sourcesContent":["import {\n\tstreamSimple,\n\tcreateAssistantMessageEventStream,\n\ttype AssistantMessage,\n\ttype Model,\n\ttype Api,\n\ttype Context,\n\ttype SimpleStreamOptions,\n} from '@mariozechner/pi-ai';\nimport type { StreamFn } from '@mariozechner/pi-agent-core';\nimport type { ProviderStreamParams } from '../extensions/types/providers.js';\nimport { getProviderRegistry } from './plugin-registry.js';\nimport { EXTENSION_PROVIDER_BASE_URL } from './index.js';\nimport { createLogger } from '../utils/logger.js';\n\nconst log = createLogger('ExtensionStreamBridge');\n\nfunction createPartialMessage(model: Model<Api>): AssistantMessage {\n\treturn {\n\t\trole: 'assistant',\n\t\tcontent: [],\n\t\tapi: model.api,\n\t\tprovider: model.provider,\n\t\tmodel: model.id,\n\t\tusage: {\n\t\t\tinput: 0,\n\t\t\toutput: 0,\n\t\t\tcacheRead: 0,\n\t\t\tcacheWrite: 0,\n\t\t\ttotalTokens: 0,\n\t\t\tcost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0, total: 0 },\n\t\t},\n\t\tstopReason: 'stop',\n\t\ttimestamp: Date.now(),\n\t};\n}\n\nfunction parseToolArguments(raw: string | undefined): Record<string, unknown> {\n\tif (!raw?.trim()) return {};\n\ttry {\n\t\treturn JSON.parse(raw) as Record<string, unknown>;\n\t} catch {\n\t\treturn {};\n\t}\n}\n\n/** Wraps {@link streamSimple} so extension-registered models use the plugin `createStream()` path. */\nexport function createExtensionAwareStreamFn(): StreamFn {\n\treturn ((model: Model<Api>, context: Context, options?: SimpleStreamOptions) => {\n\t\tif (model.baseUrl !== EXTENSION_PROVIDER_BASE_URL) {\n\t\t\treturn streamSimple(model, context, options);\n\t\t}\n\n\t\tconst plugin = getProviderRegistry().get(model.provider);\n\t\tif (!plugin) {\n\t\t\treturn streamSimple(model, context, options);\n\t\t}\n\n\t\tlog.info({ prefix: 'ExtensionStreamBridge', msg: 'Streaming via extension provider' });\n\n\t\tconst stream = createAssistantMessageEventStream();\n\n\t\tconst params: ProviderStreamParams = {\n\t\t\tmodel: model.id,\n\t\t\tmessages: context.messages as unknown as ProviderStreamParams['messages'],\n\t\t\ttemperature: options?.temperature,\n\t\t\tmaxTokens: options?.maxTokens,\n\t\t\tsignal: options?.signal,\n\t\t};\n\n\t\tvoid (async () => {\n\t\t\tconst partial = createPartialMessage(model);\n\t\t\tlet textContentIndex = -1;\n\t\t\tlet hasStarted = false;\n\n\t\t\ttry {\n\t\t\t\tfor await (const chunk of plugin.createStream(params)) {\n\t\t\t\t\tif (options?.signal?.aborted) {\n\t\t\t\t\t\tpartial.stopReason = 'aborted';\n\t\t\t\t\t\tpartial.errorMessage = 'Request aborted';\n\t\t\t\t\t\tstream.push({ type: 'error', reason: 'aborted', error: partial });\n\t\t\t\t\t\treturn;\n\t\t\t\t\t}\n\n\t\t\t\t\tswitch (chunk.type) {\n\t\t\t\t\t\tcase 'text': {\n\t\t\t\t\t\t\tif (!hasStarted) {\n\t\t\t\t\t\t\t\thasStarted = true;\n\t\t\t\t\t\t\t\tstream.push({ type: 'start', partial });\n\t\t\t\t\t\t\t}\n\t\t\t\t\t\t\tif (textContentIndex === -1) {\n\t\t\t\t\t\t\t\ttextContentIndex = partial.content.length;\n\t\t\t\t\t\t\t\tpartial.content.push({ type: 'text', text: '' });\n\t\t\t\t\t\t\t\tstream.push({ type: 'text_start', contentIndex: textContentIndex, partial });\n\t\t\t\t\t\t\t}\n\t\t\t\t\t\t\tconst textContent = partial.content[textContentIndex];\n\t\t\t\t\t\t\tif (textContent?.type === 'text' && chunk.text) {\n\t\t\t\t\t\t\t\ttextContent.text += chunk.text;\n\t\t\t\t\t\t\t\tstream.push({\n\t\t\t\t\t\t\t\t\ttype: 'text_delta',\n\t\t\t\t\t\t\t\t\tcontentIndex: textContentIndex,\n\t\t\t\t\t\t\t\t\tdelta: chunk.text,\n\t\t\t\t\t\t\t\t\tpartial,\n\t\t\t\t\t\t\t\t});\n\t\t\t\t\t\t\t}\n\t\t\t\t\t\t\tbreak;\n\t\t\t\t\t\t}\n\t\t\t\t\t\tcase 'tool_call': {\n\t\t\t\t\t\t\tif (!hasStarted) {\n\t\t\t\t\t\t\t\thasStarted = true;\n\t\t\t\t\t\t\t\tstream.push({ type: 'start', partial });\n\t\t\t\t\t\t\t}\n\t\t\t\t\t\t\tif (textContentIndex !== -1) {\n\t\t\t\t\t\t\t\tconst tc = partial.content[textContentIndex];\n\t\t\t\t\t\t\t\tif (tc?.type === 'text') {\n\t\t\t\t\t\t\t\t\tstream.push({\n\t\t\t\t\t\t\t\t\t\ttype: 'text_end',\n\t\t\t\t\t\t\t\t\t\tcontentIndex: textContentIndex,\n\t\t\t\t\t\t\t\t\t\tcontent: tc.text,\n\t\t\t\t\t\t\t\t\t\tpartial,\n\t\t\t\t\t\t\t\t\t});\n\t\t\t\t\t\t\t\t}\n\t\t\t\t\t\t\t\ttextContentIndex = -1;\n\t\t\t\t\t\t\t}\n\t\t\t\t\t\t\tif (chunk.toolCall) {\n\t\t\t\t\t\t\t\tconst idx = partial.content.length;\n\t\t\t\t\t\t\t\tconst toolCall = {\n\t\t\t\t\t\t\t\t\ttype: 'toolCall' as const,\n\t\t\t\t\t\t\t\t\tid: chunk.toolCall.id,\n\t\t\t\t\t\t\t\t\tname: chunk.toolCall.name,\n\t\t\t\t\t\t\t\t\targuments: parseToolArguments(chunk.toolCall.arguments),\n\t\t\t\t\t\t\t\t};\n\t\t\t\t\t\t\t\tpartial.content.push(toolCall);\n\t\t\t\t\t\t\t\tstream.push({ type: 'toolcall_start', contentIndex: idx, partial });\n\t\t\t\t\t\t\t\tstream.push({\n\t\t\t\t\t\t\t\t\ttype: 'toolcall_delta',\n\t\t\t\t\t\t\t\t\tcontentIndex: idx,\n\t\t\t\t\t\t\t\t\tdelta: chunk.toolCall.arguments || '{}',\n\t\t\t\t\t\t\t\t\tpartial,\n\t\t\t\t\t\t\t\t});\n\t\t\t\t\t\t\t\tstream.push({ type: 'toolcall_end', contentIndex: idx, toolCall, partial });\n\t\t\t\t\t\t\t}\n\t\t\t\t\t\t\tbreak;\n\t\t\t\t\t\t}\n\t\t\t\t\t\tcase 'usage': {\n\t\t\t\t\t\t\tif (chunk.usage) {\n\t\t\t\t\t\t\t\tconst inputCost = (chunk.usage.input / 1_000_000) * (model.cost?.input ?? 0);\n\t\t\t\t\t\t\t\tconst outputCost = (chunk.usage.output / 1_000_000) * (model.cost?.output ?? 0);\n\t\t\t\t\t\t\t\tpartial.usage = {\n\t\t\t\t\t\t\t\t\tinput: chunk.usage.input,\n\t\t\t\t\t\t\t\t\toutput: chunk.usage.output,\n\t\t\t\t\t\t\t\t\tcacheRead: 0,\n\t\t\t\t\t\t\t\t\tcacheWrite: 0,\n\t\t\t\t\t\t\t\t\ttotalTokens: chunk.usage.total ?? chunk.usage.input + chunk.usage.output,\n\t\t\t\t\t\t\t\t\tcost: {\n\t\t\t\t\t\t\t\t\t\tinput: inputCost,\n\t\t\t\t\t\t\t\t\t\toutput: outputCost,\n\t\t\t\t\t\t\t\t\t\tcacheRead: 0,\n\t\t\t\t\t\t\t\t\t\tcacheWrite: 0,\n\t\t\t\t\t\t\t\t\t\ttotal: inputCost + outputCost,\n\t\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t\t};\n\t\t\t\t\t\t\t}\n\t\t\t\t\t\t\tbreak;\n\t\t\t\t\t\t}\n\t\t\t\t\t\tcase 'error': {\n\t\t\t\t\t\t\tif (!hasStarted) {\n\t\t\t\t\t\t\t\thasStarted = true;\n\t\t\t\t\t\t\t\tstream.push({ type: 'start', partial });\n\t\t\t\t\t\t\t}\n\t\t\t\t\t\t\tpartial.stopReason = 'error';\n\t\t\t\t\t\t\tpartial.errorMessage = chunk.error ?? 'Unknown extension provider error';\n\t\t\t\t\t\t\tstream.push({ type: 'error', reason: 'error', error: partial });\n\t\t\t\t\t\t\treturn;\n\t\t\t\t\t\t}\n\t\t\t\t\t\tcase 'done':\n\t\t\t\t\t\t\tbreak;\n\t\t\t\t\t}\n\t\t\t\t}\n\n\t\t\t\tif (textContentIndex !== -1) {\n\t\t\t\t\tconst tc = partial.content[textContentIndex];\n\t\t\t\t\tif (tc?.type === 'text') {\n\t\t\t\t\t\tstream.push({\n\t\t\t\t\t\t\ttype: 'text_end',\n\t\t\t\t\t\t\tcontentIndex: textContentIndex,\n\t\t\t\t\t\t\tcontent: tc.text,\n\t\t\t\t\t\t\tpartial,\n\t\t\t\t\t\t});\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t\tif (!hasStarted) {\n\t\t\t\t\thasStarted = true;\n\t\t\t\t\tstream.push({ type: 'start', partial });\n\t\t\t\t}\n\n\t\t\t\tconst hasToolCalls = partial.content.some(c => c.type === 'toolCall');\n\t\t\t\tconst stopReason = hasToolCalls ? 'toolUse' : 'stop';\n\t\t\t\tpartial.stopReason = stopReason;\n\t\t\t\tstream.push({ type: 'done', reason: stopReason, message: partial });\n\t\t\t} catch (err) {\n\t\t\t\tif (!hasStarted) stream.push({ type: 'start', partial });\n\t\t\t\tpartial.stopReason = 'error';\n\t\t\t\tpartial.errorMessage = err instanceof Error ? err.message : String(err);\n\t\t\t\tstream.push({ type: 'error', reason: 'error', error: partial });\n\t\t\t}\n\t\t})();\n\n\t\treturn stream;\n\t}) as StreamFn;\n}\n"],"mappings":";;;;;;sBAW2D;gBACF;aACP;AAElD,MAAM,MAAM,aAAa,wBAAwB;AAEjD,SAAS,qBAAqB,OAAqC;AAClE,QAAO;EACN,MAAM;EACN,SAAS,EAAE;EACX,KAAK,MAAM;EACX,UAAU,MAAM;EAChB,OAAO,MAAM;EACb,OAAO;GACN,OAAO;GACP,QAAQ;GACR,WAAW;GACX,YAAY;GACZ,aAAa;GACb,MAAM;IAAE,OAAO;IAAG,QAAQ;IAAG,WAAW;IAAG,YAAY;IAAG,OAAO;IAAG;GACpE;EACD,YAAY;EACZ,WAAW,KAAK,KAAK;EACrB;;AAGF,SAAS,mBAAmB,KAAkD;AAC7E,KAAI,CAAC,KAAK,MAAM,CAAE,QAAO,EAAE;AAC3B,KAAI;AACH,SAAO,KAAK,MAAM,IAAI;SACf;AACP,SAAO,EAAE;;;;AAKX,SAAgB,+BAAyC;AACxD,UAAS,OAAmB,SAAkB,YAAkC;AAC/E,MAAI,MAAM,YAAA,8BACT,QAAO,aAAa,OAAO,SAAS,QAAQ;EAG7C,MAAM,SAAS,qBAAqB,CAAC,IAAI,MAAM,SAAS;AACxD,MAAI,CAAC,OACJ,QAAO,aAAa,OAAO,SAAS,QAAQ;AAG7C,MAAI,KAAK;GAAE,QAAQ;GAAyB,KAAK;GAAoC,CAAC;EAEtF,MAAM,SAAS,mCAAmC;EAElD,MAAM,SAA+B;GACpC,OAAO,MAAM;GACb,UAAU,QAAQ;GAClB,aAAa,SAAS;GACtB,WAAW,SAAS;GACpB,QAAQ,SAAS;GACjB;AAED,GAAM,YAAY;GACjB,MAAM,UAAU,qBAAqB,MAAM;GAC3C,IAAI,mBAAmB;GACvB,IAAI,aAAa;AAEjB,OAAI;AACH,eAAW,MAAM,SAAS,OAAO,aAAa,OAAO,EAAE;AACtD,SAAI,SAAS,QAAQ,SAAS;AAC7B,cAAQ,aAAa;AACrB,cAAQ,eAAe;AACvB,aAAO,KAAK;OAAE,MAAM;OAAS,QAAQ;OAAW,OAAO;OAAS,CAAC;AACjE;;AAGD,aAAQ,MAAM,MAAd;MACC,KAAK,QAAQ;AACZ,WAAI,CAAC,YAAY;AAChB,qBAAa;AACb,eAAO,KAAK;SAAE,MAAM;SAAS;SAAS,CAAC;;AAExC,WAAI,qBAAqB,IAAI;AAC5B,2BAAmB,QAAQ,QAAQ;AACnC,gBAAQ,QAAQ,KAAK;SAAE,MAAM;SAAQ,MAAM;SAAI,CAAC;AAChD,eAAO,KAAK;SAAE,MAAM;SAAc,cAAc;SAAkB;SAAS,CAAC;;OAE7E,MAAM,cAAc,QAAQ,QAAQ;AACpC,WAAI,aAAa,SAAS,UAAU,MAAM,MAAM;AAC/C,oBAAY,QAAQ,MAAM;AAC1B,eAAO,KAAK;SACX,MAAM;SACN,cAAc;SACd,OAAO,MAAM;SACb;SACA,CAAC;;AAEH;;MAED,KAAK;AACJ,WAAI,CAAC,YAAY;AAChB,qBAAa;AACb,eAAO,KAAK;SAAE,MAAM;SAAS;SAAS,CAAC;;AAExC,WAAI,qBAAqB,IAAI;QAC5B,MAAM,KAAK,QAAQ,QAAQ;AAC3B,YAAI,IAAI,SAAS,OAChB,QAAO,KAAK;SACX,MAAM;SACN,cAAc;SACd,SAAS,GAAG;SACZ;SACA,CAAC;AAEH,2BAAmB;;AAEpB,WAAI,MAAM,UAAU;QACnB,MAAM,MAAM,QAAQ,QAAQ;QAC5B,MAAM,WAAW;SAChB,MAAM;SACN,IAAI,MAAM,SAAS;SACnB,MAAM,MAAM,SAAS;SACrB,WAAW,mBAAmB,MAAM,SAAS,UAAU;SACvD;AACD,gBAAQ,QAAQ,KAAK,SAAS;AAC9B,eAAO,KAAK;SAAE,MAAM;SAAkB,cAAc;SAAK;SAAS,CAAC;AACnE,eAAO,KAAK;SACX,MAAM;SACN,cAAc;SACd,OAAO,MAAM,SAAS,aAAa;SACnC;SACA,CAAC;AACF,eAAO,KAAK;SAAE,MAAM;SAAgB,cAAc;SAAK;SAAU;SAAS,CAAC;;AAE5E;MAED,KAAK;AACJ,WAAI,MAAM,OAAO;QAChB,MAAM,YAAa,MAAM,MAAM,QAAQ,OAAc,MAAM,MAAM,SAAS;QAC1E,MAAM,aAAc,MAAM,MAAM,SAAS,OAAc,MAAM,MAAM,UAAU;AAC7E,gBAAQ,QAAQ;SACf,OAAO,MAAM,MAAM;SACnB,QAAQ,MAAM,MAAM;SACpB,WAAW;SACX,YAAY;SACZ,aAAa,MAAM,MAAM,SAAS,MAAM,MAAM,QAAQ,MAAM,MAAM;SAClE,MAAM;UACL,OAAO;UACP,QAAQ;UACR,WAAW;UACX,YAAY;UACZ,OAAO,YAAY;UACnB;SACD;;AAEF;MAED,KAAK;AACJ,WAAI,CAAC,YAAY;AAChB,qBAAa;AACb,eAAO,KAAK;SAAE,MAAM;SAAS;SAAS,CAAC;;AAExC,eAAQ,aAAa;AACrB,eAAQ,eAAe,MAAM,SAAS;AACtC,cAAO,KAAK;QAAE,MAAM;QAAS,QAAQ;QAAS,OAAO;QAAS,CAAC;AAC/D;MAED,KAAK,OACJ;;;AAIH,QAAI,qBAAqB,IAAI;KAC5B,MAAM,KAAK,QAAQ,QAAQ;AAC3B,SAAI,IAAI,SAAS,OAChB,QAAO,KAAK;MACX,MAAM;MACN,cAAc;MACd,SAAS,GAAG;MACZ;MACA,CAAC;;AAGJ,QAAI,CAAC,YAAY;AAChB,kBAAa;AACb,YAAO,KAAK;MAAE,MAAM;MAAS;MAAS,CAAC;;IAIxC,MAAM,aADe,QAAQ,QAAQ,MAAK,MAAK,EAAE,SAAS,WAAW,GACnC,YAAY;AAC9C,YAAQ,aAAa;AACrB,WAAO,KAAK;KAAE,MAAM;KAAQ,QAAQ;KAAY,SAAS;KAAS,CAAC;YAC3D,KAAK;AACb,QAAI,CAAC,WAAY,QAAO,KAAK;KAAE,MAAM;KAAS;KAAS,CAAC;AACxD,YAAQ,aAAa;AACrB,YAAQ,eAAe,eAAe,QAAQ,IAAI,UAAU,OAAO,IAAI;AACvE,WAAO,KAAK;KAAE,MAAM;KAAS,QAAQ;KAAS,OAAO;KAAS,CAAC;;MAE7D;AAEJ,SAAO"}

package/dist/src/providers/index.d.ts

@@ -3,7 +3,12 @@
  */
 import { type Model, type Api } from '@mariozechner/pi-ai';
 import type { Config } from '../config/schema.js';
+import type { ProviderModelDefinition } from '../extensions/types/providers.js';
 export { getApiKeyFromEnv, PROVIDER_ENV_MAP } from './env-keys.js';
+/** Sentinel base URL: model is served by an extension {@link ProviderPluginRegistry} provider. */
+export declare const EXTENSION_PROVIDER_BASE_URL = "extension://provider-plugin";
+/** Map a plugin registry model to the pi-ai {@link Model} shape. */
+export declare function pluginModelToModel(providerId: string, definition: ProviderModelDefinition): Model<Api>;
 /**
  * Get API key synchronously: checks registry (models.json) first, then environment variables.
  * Use this for Agent's getApiKey callback which must be synchronous.
@@ -26,13 +31,13 @@ export declare function getApiKey(provider: string): Promise<string | undefined>
 export declare function isProviderConfiguredSync(provider: string): boolean;
 export declare function isProviderConfigured(provider: string): Promise<boolean>;
 /** Where runtime {@link getApiKey} resolves the key from (no secret values). */
-export type ProviderActiveKeySource = 'none' | 'agent' | 'gateway' | 'oauth' | 'env' | 'models_json';
+export type ProviderActiveKeySource = 'none' | 'agent' | 'gateway' | 'oauth' | 'env' | 'models_json' | 'extension';
 export declare function getProviderActiveKeySource(provider: string): Promise<ProviderActiveKeySource>;
 export declare function getConfiguredProviders(): Promise<string[]>;
 export declare function getAllModels(): readonly Model<Api>[];
 export declare function getAvailableModels(): Promise<readonly Model<Api>[]>;
 export type { Model, Api } from '@mariozechner/pi-ai';
-export type ProviderCategory = 'common' | 'specialty' | 'oauth' | 'enterprise';
+export type ProviderCategory = 'common' | 'specialty' | 'oauth' | 'enterprise' | 'extension';
 export interface ProviderMeta {
     name: string;
     category: ProviderCategory;