@desplega.ai/agent-swarm 1.172.0 → 2.0.0

package/dist/{pi-mono-adapter-ef9wwx9m.js → pi-mono-adapter-9n2ya240.js}

@@ -1,42 +1,45 @@
 import {
   piPromptSkillName
 } from "./cli-ge593j7e.js";
+import {
+  createSwarmHooksExtension
+} from "./cli-apx87jne.js";
 import {
   readPkgVersion
 } from "./cli-vv0qtfpx.js";
 import {
-  createSwarmHooksExtension
-} from "./cli-bf7485pm.js";
+  fetchInstalledMcpServers
+} from "./cli-tng3131n.js";
+import"./cli-tr4sxp68.js";
+import"./cli-hp1n6zq3.js";
 import {
   DefaultResourceLoader,
   ModelRuntime,
   SessionManager,
   SettingsManager,
   createAgentSession,
+  createBashToolDefinition,
   createCodemodeExtension,
   createMcpExtension,
-  createToolSearchExtension
-} from "./cli-4nz8p1p5.js";
+  createToolSearchExtension,
+  defineTool
+} from "./cli-7x0g4sn1.js";
+import"./cli-c8r3kzzb.js";
+import"./cli-r106725k.js";
+import"./cli-pxeeyc53.js";
 import"./cli-bt8k5h3s.js";
-import"./cli-c5km5znr.js";
-import"./cli-r44t9wqc.js";
 import"./cli-ev91t6xp.js";
-import"./cli-pxeeyc53.js";
 import"./cli-qybkr9n3.js";
 import"./cli-vb8g5x02.js";
 import {
   getAgentDir,
   setEmbeddedQuickJSWasmPath
 } from "./cli-hd27d2md.js";
-import {
-  fetchInstalledMcpServers
-} from "./cli-tng3131n.js";
-import"./cli-cf69s3ew.js";
-import"./cli-hp1n6zq3.js";
+import"./cli-0m7wct0d.js";
 import {
   applyReasoningEffort
-} from "./cli-tmhsxhb8.js";
-import"./cli-0m7wct0d.js";
+} from "./cli-n5vwg842.js";
+import"./cli-jta2eyzc.js";
 import"./cli-7nxes1hs.js";
 import"./cli-104zsxba.js";
 import {
@@ -45,16 +48,16 @@ import {
 import {
   McpHttpClient
 } from "./cli-epan13p1.js";
-import"./cli-xczpdp9e.js";
-import"./cli-7jykv7ef.js";
+import"./cli-se8w6wny.js";
+import"./cli-e6mpjhed.js";
+import"./cli-2x0e561b.js";
 import {
   getBuiltinModel,
   getBuiltinModels
-} from "./cli-cceskcvt.js";
+} from "./cli-txqb9s38.js";
 import {
   exports_typebox
 } from "./cli-h6vtvt5c.js";
-import"./cli-2x0e561b.js";
 import"./cli-bsakbbkm.js";
 import"./cli-zxg80zs9.js";
 import"./cli-ydt3zkwv.js";
@@ -62,14 +65,14 @@ import"./cli-3er6gcsn.js";
 import"./cli-0xqde53b.js";
 import"./cli-pqetb83f.js";
 import"./cli-61625k3x.js";
-import"./cli-btbzvq95.js";
 import"./cli-1m2e086m.js";
+import"./cli-btbzvq95.js";
 import"./cli-p7z6q2m6.js";
 import"./cli-wnb9b2g3.js";
 import {
   CORE_TOOLS,
   init_tool_config
-} from "./cli-ywm1azg9.js";
+} from "./cli-hnmhtzvh.js";
 import {
   init_templates
 } from "./cli-mvzyxwg3.js";
@@ -84,7 +87,10 @@ import {
   getOpenRouterBaseUrl,
   init_openrouter_base_url
 } from "./cli-a35hzdve.js";
-import"./cli-014vns2y.js";
+import {
+  init_runtime_model_catalog,
+  snapshotCatalogRows
+} from "./cli-stfs0k5s.js";
 import"./cli-sd0mxa3c.js";
 import"./cli-w5e1d1vv.js";
 import {
@@ -93,15 +99,39 @@ import {
 } from "./cli-ej6gbxfb.js";
 import {
   init_secret_scrubber,
+  registerEnvSecrets,
+  scrubObject,
   scrubSecrets
-} from "./cli-d161g69q.js";
+} from "./cli-p1x85d2t.js";
+import {
+  ignoredBaseUrlVar,
+  init_src,
+  openAICompatibleKeyVars,
+  openAICompatibleProvider,
+  openAICompatibleProviders,
+  providerBaseUrl,
+  providerForModel,
+  providerHarnessModels,
+  providerKeyFromEnv,
+  providerKeyVars
+} from "./cli-e3g5mbkz.js";
 import {
   __require,
   __toESM
 } from "./cli-bc05qbss.js";
 
 // src/providers/pi-mono-adapter.ts
-import { existsSync, lstatSync, symlinkSync, unlinkSync } from "node:fs";
+init_src();
+import {
+  existsSync,
+  lstatSync,
+  readFileSync,
+  symlinkSync,
+  unlinkSync,
+  writeFileSync
+} from "node:fs";
+import { mkdtemp, rm } from "node:fs/promises";
+import { tmpdir } from "node:os";
 import { join } from "node:path";
 
 // node_modules/quickjs-wasi/quickjs.wasm
@@ -170,6 +200,7 @@ function classifyAwsSdkError(text) {
 // src/providers/pi-mono-adapter.ts
 init_env_flag();
 init_openrouter_base_url();
+init_runtime_model_catalog();
 init_secret_scrubber();
 function modelToCredKeys(modelStr) {
   if (!modelStr)
@@ -188,6 +219,9 @@ function modelToCredKeys(modelStr) {
       return ["OPENAI_API_KEY"];
     if (provider === "google")
       return ["GOOGLE_API_KEY", "GEMINI_API_KEY"];
+    const direct = openAICompatibleProvider(provider);
+    if (direct)
+      return providerKeyVars(direct);
   }
   return null;
 }
@@ -282,13 +316,13 @@ async function checkPiMonoCredentials(env, opts = {}) {
       hint: `MODEL_OVERRIDE=${env.MODEL_OVERRIDE} requires one of ${keyList}; or run \`pi auth login\` to create ${authFile}.`
     };
   }
-  if (env.ANTHROPIC_API_KEY || env.OPENROUTER_API_KEY || env.OPENAI_API_KEY) {
+  if (env.ANTHROPIC_API_KEY || env.OPENROUTER_API_KEY || env.OPENAI_API_KEY || openAICompatibleKeyVars().some((k) => env[k])) {
     return { ready: true, missing: [], satisfiedBy: "env" };
   }
   return {
     ready: false,
     missing: ["ANTHROPIC_API_KEY", "OPENROUTER_API_KEY", "OPENAI_API_KEY", authFile],
-    hint: "Set one of ANTHROPIC_API_KEY / OPENROUTER_API_KEY / OPENAI_API_KEY (any one suffices), or run `pi auth login` to create ~/.pi/agent/auth.json."
+    hint: `Set one of ANTHROPIC_API_KEY / OPENROUTER_API_KEY / OPENAI_API_KEY (any one suffices; ${openAICompatibleProviders().map((p) => p.envVar).join(" or ")} serve only ${openAICompatibleProviders().map((p) => `${p.id}/*`).join(" or ")} models), or run \`pi auth login\` to create ~/.pi/agent/auth.json.`
   };
 }
 function jsonSchemaToTypeBox(schema) {
@@ -349,10 +383,11 @@ var PI_RUNTIME_API_KEYS = [
   ["ANTHROPIC_API_KEY", "anthropic"],
   ["OPENAI_API_KEY", "openai"],
   ["GEMINI_API_KEY", "google"],
-  ["GOOGLE_API_KEY", "google"]
+  ["GOOGLE_API_KEY", "google"],
+  ...openAICompatibleProviders().flatMap((p) => [...p.aliasEnvVars, p.envVar].map((name) => [name, p.id]))
 ];
-async function createPiRuntimeAuth(env = process.env) {
-  const modelRuntime = await ModelRuntime.create();
+async function createPiRuntimeAuth(env = process.env, modelsPath) {
+  const modelRuntime = await ModelRuntime.create(modelsPath ? { modelsPath, modelsStorePath: join(getAgentDir(), "models-store.json") } : {});
   for (const [envKey, provider] of PI_RUNTIME_API_KEYS) {
     const apiKey = env[envKey];
     if (apiKey) {
@@ -373,84 +408,113 @@ async function readJsonFile(path) {
     return;
   }
 }
-async function ensureOpenRouterModelsOverride(agentDir, env = process.env) {
-  const baseUrl = getOpenRouterBaseUrl(env);
-  const modelsPath = join(agentDir, "models.json");
-  const markerPath = join(agentDir, OPENROUTER_OVERRIDE_MARKER);
-  if (baseUrl === DEFAULT_OPENROUTER_BASE_URL) {
-    await revertOpenRouterModelsOverride(modelsPath, markerPath);
-    return;
-  }
-  let existing = {};
-  try {
-    const file = Bun.file(modelsPath);
-    if (await file.exists()) {
-      existing = JSON.parse(await file.text());
-    }
-  } catch (err) {
-    console.warn(`[pi-mono] ${modelsPath} is unreadable/invalid (${err instanceof Error ? err.message : err}); rewriting with the OpenRouter gateway override`);
-    existing = {};
+function customProviderModelsBlock(provider, modelId, baseUrl = provider.baseUrl) {
+  const zeroCost = { input: 0, output: 0, cacheRead: 0, cacheWrite: 0 };
+  const listed = providerHarnessModels(provider, snapshotCatalogRows(provider.id));
+  const models = listed.map((m) => ({
+    id: m.id,
+    name: m.name,
+    reasoning: m.reasoning,
+    input: [...m.input],
+    contextWindow: m.contextWindow,
+    maxTokens: m.maxOutputTokens,
+    cost: { ...zeroCost },
+    compat: { ...m.compat }
+  }));
+  if (modelId && !listed.some((m) => m.id === modelId)) {
+    models.push({ id: modelId, name: modelId, cost: { ...zeroCost } });
   }
-  const providers = typeof existing.providers === "object" && existing.providers !== null ? existing.providers : {};
-  const openrouter = typeof providers.openrouter === "object" && providers.openrouter !== null ? providers.openrouter : {};
-  const priorMarker = await readJsonFile(markerPath);
-  const currentBaseUrl = typeof openrouter.baseUrl === "string" ? openrouter.baseUrl : undefined;
-  const previous = priorMarker ? priorMarker.previous : currentBaseUrl !== baseUrl ? currentBaseUrl : undefined;
-  const next = {
-    ...existing,
-    providers: {
-      ...providers,
-      openrouter: { ...openrouter, baseUrl }
-    }
-  };
-  await Bun.write(modelsPath, `${JSON.stringify(next, null, 2)}
-`);
-  const marker = {
+  return {
     baseUrl,
-    ...previous !== undefined ? { previous } : {}
+    api: "openai-completions",
+    apiKey: `$${provider.envVar}`,
+    models
   };
-  await Bun.write(markerPath, `${JSON.stringify(marker, null, 2)}
-`);
 }
-async function revertOpenRouterModelsOverride(modelsPath, markerPath) {
-  const marker = await readJsonFile(markerPath);
-  if (!marker)
-    return;
-  const removeMarker = () => Bun.file(markerPath).delete();
-  const existing = await readJsonFile(modelsPath);
-  if (!existing) {
-    await removeMarker();
-    return;
+function isPlainObject(value) {
+  return typeof value === "object" && value !== null && !Array.isArray(value);
+}
+async function writeSessionModelsJson(agentDir, env = process.env, model) {
+  const sharedPath = join(agentDir, "models.json");
+  const registry = openAICompatibleProviders();
+  const wanted = registry.filter((p) => providerKeyFromEnv(p, env) !== undefined);
+  const gateway = getOpenRouterBaseUrl(env);
+  const marker = await readJsonFile(join(agentDir, OPENROUTER_OVERRIDE_MARKER));
+  if (wanted.length === 0 && gateway === DEFAULT_OPENROUTER_BASE_URL && !marker) {
+    return { modelsPath: undefined, dispose: async () => {} };
   }
-  const providers = typeof existing.providers === "object" && existing.providers !== null ? { ...existing.providers } : undefined;
-  const openrouter = providers && typeof providers.openrouter === "object" && providers.openrouter !== null ? { ...providers.openrouter } : undefined;
-  if (!providers || !openrouter || openrouter.baseUrl !== marker.baseUrl) {
-    await removeMarker();
-    return;
+  let shared = {};
+  const sharedFile = Bun.file(sharedPath);
+  if (await sharedFile.exists()) {
+    try {
+      const parsed = JSON.parse(await sharedFile.text());
+      if (!isPlainObject(parsed))
+        throw new Error("not a JSON object");
+      shared = parsed;
+    } catch (err) {
+      console.warn(scrubSecrets(`[pi-mono] ${sharedPath} is unreadable/invalid (${err instanceof Error ? err.message : String(err)}); this session loads its own provider blocks only`));
+    }
   }
-  if (marker.previous !== undefined) {
-    openrouter.baseUrl = marker.previous;
-  } else {
-    delete openrouter.baseUrl;
+  const providers = isPlainObject(shared.providers) ? { ...shared.providers } : {};
+  const openrouter = isPlainObject(providers.openrouter) ? { ...providers.openrouter } : {};
+  if (gateway !== DEFAULT_OPENROUTER_BASE_URL) {
+    openrouter.baseUrl = gateway;
+  } else if (marker && openrouter.baseUrl === marker.baseUrl) {
+    if (marker.previous !== undefined)
+      openrouter.baseUrl = marker.previous;
+    else
+      delete openrouter.baseUrl;
   }
-  if (Object.keys(openrouter).length === 0) {
-    delete providers.openrouter;
-  } else {
+  if (Object.keys(openrouter).length > 0)
     providers.openrouter = openrouter;
+  else
+    delete providers.openrouter;
+  const modelProvider = providerForModel(model);
+  const modelId = modelProvider && model ? model.slice(modelProvider.id.length + 1) : undefined;
+  for (const provider of wanted) {
+    const ignored = ignoredBaseUrlVar(provider, env);
+    if (ignored) {
+      console.warn(scrubSecrets(`[pi-mono] ${ignored} is ignored: ${provider.displayName} is only reached at ${provider.baseUrl}`));
+    }
+    const existing = providers[provider.id];
+    if (provider.baseUrlLocked && isPlainObject(existing)) {
+      if (existing.baseUrl !== provider.baseUrl || existing.api !== "openai-completions") {
+        console.warn(scrubSecrets(`[pi-mono] ${sharedPath} providers.${provider.id} points at ${String(existing.baseUrl)}; using ${provider.baseUrl}`));
+      }
+      providers[provider.id] = {
+        ...customProviderModelsBlock(provider, modelProvider?.id === provider.id ? modelId : undefined),
+        ...existing,
+        baseUrl: provider.baseUrl,
+        api: "openai-completions"
+      };
+      continue;
+    }
+    if (existing !== undefined) {
+      console.warn(scrubSecrets(`[pi-mono] ${sharedPath} already has a providers.${provider.id} block; using it as is`));
+      continue;
+    }
+    providers[provider.id] = customProviderModelsBlock(provider, modelProvider?.id === provider.id ? modelId : undefined, providerBaseUrl(provider, env));
   }
-  const next = { ...existing, providers };
-  if (Object.keys(providers).length === 0) {
+  const next = { ...shared, providers };
+  if (Object.keys(providers).length === 0)
     delete next.providers;
-  }
-  if (Object.keys(next).length === 0) {
-    await Bun.file(modelsPath).delete();
-  } else {
-    await Bun.write(modelsPath, `${JSON.stringify(next, null, 2)}
+  const dir = await mkdtemp(join(tmpdir(), "pi-session-models-"));
+  const modelsPath = join(dir, "models.json");
+  await Bun.write(modelsPath, `${JSON.stringify(next, null, 2)}
 `);
+  return {
+    modelsPath,
+    dispose: () => rm(dir, { recursive: true, force: true })
+  };
+}
+function resolveModel(modelStr, env = process.env, modelRuntime) {
+  const model = resolveModelInner(modelStr, env, modelRuntime);
+  if (modelStr && !model) {
+    console.warn(scrubSecrets(`[pi-mono] Model "${modelStr}" did not resolve to a builtin or models.json model; pi will fall back to its default model`));
   }
-  await removeMarker();
+  return model;
 }
-function resolveModel(modelStr, env = process.env) {
+function resolveModelInner(modelStr, env, modelRuntime) {
   if (!modelStr)
     return;
   const lower = modelStr.toLowerCase();
@@ -481,11 +545,15 @@ function resolveModel(modelStr, env = process.env) {
     const slashIdx = modelStr.indexOf("/");
     const provider = modelStr.slice(0, slashIdx);
     const modelId = modelStr.slice(slashIdx + 1);
+    let builtin;
     try {
-      return getBuiltinModel(provider, modelId);
+      builtin = getBuiltinModel(provider, modelId);
     } catch {
-      return;
+      builtin = undefined;
     }
+    if (builtin)
+      return builtin;
+    return modelRuntime?.getModel(provider, modelId);
   }
   for (const provider of ["anthropic", "openai", "google"]) {
     try {
@@ -547,6 +615,24 @@ function describeAssistantTurn(message) {
   };
 }
 var TERMINAL_STORE_PROGRESS_STATUSES = new Set(["completed", "failed"]);
+function scrubSessionFileSync(path) {
+  if (!path || !existsSync(path))
+    return;
+  const text = readFileSync(path, "utf-8");
+  const scrubbed = scrubSecrets(text);
+  if (scrubbed !== text)
+    writeFileSync(path, scrubbed);
+}
+function scrubPiSessionPersistence(sessionManager) {
+  const persist = sessionManager._persist.bind(sessionManager);
+  sessionManager._persist = (entry) => {
+    const file = sessionManager.getSessionFile();
+    const existed = file ? existsSync(file) : true;
+    persist(scrubObject(entry));
+    if (!existed)
+      scrubSessionFileSync(file);
+  };
+}
 
 class PiMonoSession {
   listeners = [];
@@ -569,8 +655,10 @@ class PiMonoSession {
   terminalStoreProgressDone = false;
   emptyTurnReprompted = false;
   abortRequested = false;
-  constructor(agentSession, config, createdSymlink, appliedReasoningEffort = null) {
+  disposeModels;
+  constructor(agentSession, config, createdSymlink, appliedReasoningEffort = null, disposeModels = async () => {}) {
     this.agentSession = agentSession;
+    this.disposeModels = disposeModels;
     this.config = config;
     this.createdSymlink = createdSymlink;
     this.appliedReasoningEffort = appliedReasoningEffort;
@@ -595,7 +683,7 @@ class PiMonoSession {
     return this.config.model;
   }
   emit(event) {
-    const scrubbed = event.type === "raw_log" || event.type === "raw_stderr" ? { ...event, content: scrubSecrets(event.content) } : event;
+    const scrubbed = scrubObject(event);
     this.logFileHandle.write(`${JSON.stringify({ ...scrubbed, timestamp: new Date().toISOString() })}
 `);
     if (this.listeners.length > 0) {
@@ -746,7 +834,7 @@ class PiMonoSession {
       const cost = this.buildCostData(stats);
       if (this.terminalError) {
         const classification = classifyAwsSdkError(this.terminalError);
-        const message = classification?.message ?? this.terminalError;
+        const message = scrubSecrets(classification?.message ?? this.terminalError);
         const category = classification?.category;
         this.emit({ type: "error", message, category });
         return {
@@ -789,7 +877,7 @@ class PiMonoSession {
         sessionId: this._sessionId,
         isError: true,
         errorCategory: awsCatchError?.category,
-        failureReason: awsCatchError?.message ?? errorMessage,
+        failureReason: scrubSecrets(awsCatchError?.message ?? errorMessage),
         appliedReasoningEffort: this.appliedReasoningEffort
       };
     } finally {
@@ -799,6 +887,12 @@ class PiMonoSession {
         cleanupAgentsMdSymlink(this.config.cwd);
       }
       this.agentSession.dispose();
+      try {
+        scrubSessionFileSync(this.agentSession.sessionManager?.getSessionFile());
+      } catch (err) {
+        console.warn(scrubSecrets(`[pi-mono] could not scrub the session file: ${String(err)}`));
+      }
+      await this.disposeModels();
     }
   }
   async repromptAfterEmptyFinalTurn() {
@@ -1039,6 +1133,24 @@ function piDefaultToolAdditions(features) {
     additions.push("+codemode");
   return additions;
 }
+var PI_SPAWN_ENV_KEYS = [
+  "PATH",
+  "PI_SESSION_ID",
+  "PI_SESSION_FILE",
+  "PI_PROVIDER",
+  "PI_MODEL",
+  "PI_REASONING_LEVEL"
+];
+function buildPiShellEnv(sessionEnv, spawnEnv) {
+  const env = { ...sessionEnv };
+  for (const key of PI_SPAWN_ENV_KEYS) {
+    if (spawnEnv[key] !== undefined)
+      env[key] = spawnEnv[key];
+    else if (key !== "PATH")
+      delete env[key];
+  }
+  return env;
+}
 async function createPiResourceLoader(opts) {
   const settingsManager = SettingsManager.create(opts.cwd, opts.agentDir, {
     projectTrusted: false
@@ -1109,10 +1221,25 @@ class PiMonoAdapter {
       console.log(`\x1B[2m[${config.role}]\x1B[0m Registering ${Object.keys(piMcpServers).length} installed MCP server(s) with pi`);
     }
     const sessionEnv = config.env ?? process.env;
-    await ensureOpenRouterModelsOverride(getAgentDir(), sessionEnv);
-    const builtinModel = resolveModel(config.model, sessionEnv);
-    const modelRuntime = await createPiRuntimeAuth(sessionEnv);
-    const model = builtinModel ? modelRuntime.getModel(builtinModel.provider, builtinModel.id) ?? builtinModel : builtinModel;
+    registerEnvSecrets(sessionEnv);
+    const sessionModels = await writeSessionModelsJson(getAgentDir(), sessionEnv, config.model);
+    try {
+      return await this.createSessionWithModels(config, sessionEnv, sessionModels, {
+        createdSymlink,
+        customTools,
+        deferTools,
+        piMcpServers
+      });
+    } catch (err) {
+      await sessionModels.dispose();
+      throw err;
+    }
+  }
+  async createSessionWithModels(config, sessionEnv, sessionModels, setup) {
+    const { createdSymlink, customTools, deferTools, piMcpServers } = setup;
+    const modelRuntime = await createPiRuntimeAuth(sessionEnv, sessionModels.modelsPath);
+    const resolvedModel = resolveModel(config.model, sessionEnv, modelRuntime);
+    const model = resolvedModel ? modelRuntime.getModel(resolvedModel.provider, resolvedModel.id) ?? resolvedModel : resolvedModel;
     const swarmExtension = createSwarmHooksExtension({
       apiUrl: config.apiUrl,
       apiKey: config.apiKey,
@@ -1144,10 +1271,18 @@ class PiMonoAdapter {
     const reasoningApplication = applyReasoningEffort("pi", config.model, config.reasoningEffort);
     const reasoningSessionOptions = reasoningApplication.kind === "pi-session" ? reasoningApplication.sessionOptions : {};
     const appliedReasoningEffort = reasoningApplication.kind === "pi-session" ? config.reasoningEffort ?? null : null;
+    const bashTool = defineTool(createBashToolDefinition(config.cwd, {
+      commandPrefix: settingsManager.getShellCommandPrefix(),
+      shellPath: settingsManager.getShellPath(),
+      spawnHook: (context) => ({ ...context, env: buildPiShellEnv(sessionEnv, context.env) })
+    }));
+    const sessionManager = SessionManager.create(config.cwd);
+    scrubPiSessionPersistence(sessionManager);
     const sessionOptions = {
       cwd: config.cwd,
+      sessionManager,
       model,
-      customTools,
+      customTools: [...customTools, bashTool],
       resourceLoader,
       settingsManager,
       modelRuntime,
@@ -1155,7 +1290,7 @@ class PiMonoAdapter {
     };
     const { session } = await createAgentSession(sessionOptions);
     await session.bindExtensions({});
-    return new PiMonoSession(session, config, createdSymlink, appliedReasoningEffort);
+    return new PiMonoSession(session, config, createdSymlink, appliedReasoningEffort, sessionModels.dispose);
   }
   async canResume(sessionId) {
     try {
@@ -1179,12 +1314,13 @@ export {
   PiMonoSession,
   SWARM_TOOL_NAMESPACE,
   boundCodemodeTool,
+  buildPiShellEnv,
   checkPiMonoCredentials,
   createBoundedCodemodeExtension,
   createPiResourceLoader,
   createPiRuntimeAuth,
   createSwarmMcpExtension,
-  ensureOpenRouterModelsOverride,
+  customProviderModelsBlock,
   extractPiAssistantText,
   isCoreSwarmTool,
   isPiCodemodeEnabled,
@@ -1195,5 +1331,7 @@ export {
   piExtensionFactories,
   resolveModel,
   runBedrockSdkProbeAndEnumerate,
-  toPiMcpServers
+  scrubPiSessionPersistence,
+  toPiMcpServers,
+  writeSessionModelsJson
 };

package/dist/{pricing-refresh-s52nsa5q.js → pricing-refresh-qb0rv0ap.js}

@@ -7,29 +7,31 @@ import {
   requestModelCatalogRefresh,
   resetModelCatalogRefreshGuardForTests,
   startPricingRefreshLoop
-} from "./cli-gp1rtyvk.js";
-import"./cli-r8smj8b8.js";
-import"./cli-tgpfnm6r.js";
-import"./cli-ywm1azg9.js";
-import"./cli-hsem16hb.js";
-import"./cli-nyhb1r6x.js";
-import"./cli-7mm2r1kp.js";
+} from "./cli-z5jjpaxw.js";
+import"./cli-r62a35ea.js";
+import"./cli-ef1hg1wv.js";
+import"./cli-hnmhtzvh.js";
+import"./cli-cgtksawx.js";
+import"./cli-s2sp957d.js";
 import"./cli-dmtrxqhg.js";
+import"./cli-7mm2r1kp.js";
+import"./cli-5hkm7whv.js";
 import"./cli-mvzyxwg3.js";
-import"./cli-tadfgd0t.js";
 import"./cli-fdvbjtbw.js";
 import"./cli-ywybs5as.js";
 import"./cli-f9w1semm.js";
 import"./cli-1a2j50m0.js";
 import"./cli-a35hzdve.js";
-import"./cli-014vns2y.js";
+import"./cli-stfs0k5s.js";
 import"./cli-sd0mxa3c.js";
 import"./cli-w5e1d1vv.js";
-import"./cli-esnpehf5.js";
+import"./cli-p0sg24zp.js";
+import"./cli-k382tshb.js";
 import"./cli-zf8jjwex.js";
 import"./cli-gd4favx9.js";
 import"./cli-ej6gbxfb.js";
-import"./cli-d161g69q.js";
+import"./cli-p1x85d2t.js";
+import"./cli-e3g5mbkz.js";
 import"./cli-bc05qbss.js";
 export {
   MODEL_CATALOG_FORCE_COOLDOWN_MS,

package/dist/{rbac-roles-aq3msh4e.js → rbac-roles-1gzdwbh1.js}

@@ -1,35 +1,37 @@
 import {
   PermissionVerbSchema,
   init_rbac
-} from "./cli-7pz3b7x5.js";
+} from "./cli-przpmdgr.js";
 import {
   getDb,
   getDbClient,
   init_db1 as init_db
-} from "./cli-tgpfnm6r.js";
-import"./cli-ywm1azg9.js";
-import"./cli-hsem16hb.js";
-import"./cli-nyhb1r6x.js";
-import"./cli-7mm2r1kp.js";
+} from "./cli-ef1hg1wv.js";
+import"./cli-hnmhtzvh.js";
+import"./cli-cgtksawx.js";
+import"./cli-s2sp957d.js";
 import"./cli-dmtrxqhg.js";
+import"./cli-7mm2r1kp.js";
+import"./cli-5hkm7whv.js";
 import"./cli-mvzyxwg3.js";
-import"./cli-tadfgd0t.js";
 import"./cli-fdvbjtbw.js";
 import"./cli-ywybs5as.js";
 import"./cli-f9w1semm.js";
 import"./cli-1a2j50m0.js";
 import"./cli-a35hzdve.js";
-import"./cli-014vns2y.js";
+import"./cli-stfs0k5s.js";
 import"./cli-sd0mxa3c.js";
 import"./cli-w5e1d1vv.js";
-import"./cli-esnpehf5.js";
+import"./cli-p0sg24zp.js";
+import"./cli-k382tshb.js";
 import"./cli-zf8jjwex.js";
 import"./cli-gd4favx9.js";
 import {
   init_env_flag,
   isEnvFlagEnabled
 } from "./cli-ej6gbxfb.js";
-import"./cli-d161g69q.js";
+import"./cli-p1x85d2t.js";
+import"./cli-e3g5mbkz.js";
 import"./cli-bc05qbss.js";
 
 // src/be/rbac-roles.ts