@pyric/cli 0.1.0-alpha.16 → 0.1.0-alpha.17

package/src/serve/sandbox-session.ts

@@ -63,6 +63,7 @@ export interface SandboxSession {
   payload(): InitPayload;
   handle(req: IncomingMessage, res: ServerResponse, url: URL): boolean | Promise<boolean>;
   reloadFirestoreRules(): Promise<RulesReloadResult>;
+  reloadDatabaseRules(): Promise<RulesReloadResult>;
   close(): Promise<void>;
 }
 
@@ -93,6 +94,8 @@ export async function createSandboxSession(
   const live = {
     rules: firestore.rules,
     rulesHash: firestore.rulesHash,
+    databaseRules: database.rules,
+    databaseRulesHash: database.rulesHash,
   };
   const events = createEventHub();
   const capture: CaptureStore | undefined = (options.capture ?? true)
@@ -145,8 +148,8 @@ export async function createSandboxSession(
   const payload = (): InitPayload => ({
     rules: live.rules,
     rulesHash: live.rulesHash,
-    databaseRules: database.rules,
-    databaseRulesHash: database.rulesHash,
+    databaseRules: live.databaseRules,
+    databaseRulesHash: live.databaseRulesHash,
     databaseUrl: database.databaseUrl,
     storageRules: storage.rules,
     storageRulesHash: storage.rulesHash,
@@ -217,6 +220,31 @@ export async function createSandboxSession(
       return { kind: 'rejected', error: error instanceof Error ? error : new Error(String(error)) };
     }
   };
+  const reloadDatabaseRules = async (): Promise<RulesReloadResult> => {
+    const isMissingDatabaseRules = database.sourcePath === null;
+    if (isMissingDatabaseRules) {
+      return { kind: 'not-configured' };
+    }
+    try {
+      const updated = await loadProjectDatabaseRules(options.projectDir, options.firebaseConfig);
+      const isMissingUpdatedRules = updated.rules === null || updated.rulesHash === null;
+      if (isMissingUpdatedRules) {
+        return { kind: 'not-configured' };
+      }
+      live.databaseRules = updated.rules;
+      live.databaseRulesHash = updated.rulesHash;
+      events.broadcast('rtdb-rules-update', { rules: updated.rules, rulesHash: updated.rulesHash });
+      return { kind: 'reloaded', rulesHash: updated.rulesHash as string, clients: events.clientCount() };
+    } catch (error) {
+      const isErrorInstance = error instanceof Error;
+      let errorResult: Error = new Error(String(error));
+      if (isErrorInstance) {
+        errorResult = error as Error;
+      }
+      return { kind: 'rejected', error: errorResult };
+    }
+  };
+
   let closePromise: Promise<void> | null = null;
   const close = (): Promise<void> => {
     closePromise ??= Promise.resolve().then(() => events.close());
@@ -228,6 +256,7 @@ export async function createSandboxSession(
     payload,
     handle: namespace,
     reloadFirestoreRules,
+    reloadDatabaseRules,
     close,
   };
 }

package/src/serve/vite-ai-config.ts

@@ -9,6 +9,8 @@ const AI_PROXY_PATH = '/__pyric/ai-proxy';
 export type PyricAiEngineConfig = Extract<NonNullable<AIOptions['engine']>, { kind: string }>;
 
 export interface PyricAiOptions {
+  /** Mode for AI Logic: 'sandbox' (local mirrors) or 'production' (pass-through to Google AI / Vertex AI). */
+  mode?: 'sandbox' | 'production';
   /** Simple OpenAI-compatible model selection through Pyric's same-origin proxy. */
   model?: string;
   /** Advanced declarative engine configuration. */
@@ -18,6 +20,7 @@ export interface PyricAiOptions {
 }
 
 export interface ResolvedViteAiConfig {
+  mode: 'sandbox' | 'production';
   engineWire: AiEngineConfigWire | undefined;
   proxyUpstream: string | undefined;
 }
@@ -26,8 +29,9 @@ export interface ResolvedViteAiConfig {
 export function viteWorkerEpochSalt(
   projectKey: string,
   engineWire: AiEngineConfigWire | undefined,
+  mode: 'sandbox' | 'production' = 'sandbox',
 ): string {
-  return JSON.stringify({ projectKey, aiEngine: engineWire ?? null });
+  return JSON.stringify({ projectKey, aiEngine: engineWire ?? null, aiMode: mode });
 }
 
 /** Load env using Vite's `envDir`-relative-to-root convention. */
@@ -45,20 +49,30 @@ export function loadViteAiEnv(
 
 /** Convert a public declarative engine into the JSON-safe worker wire shape. */
 export function engineConfigToWire(engine: PyricAiEngineConfig): AiEngineConfigWire {
-  if (engine.kind === 'openai') {
-    return {
+  const isOpenAiEngine = engine.kind === 'openai';
+  if (isOpenAiEngine) {
+    const result: Record<string, unknown> = {
       kind: 'openai',
-      baseUrl: engine.baseUrl ?? AI_PROXY_PATH,
-      ...(engine.model !== undefined ? { model: engine.model } : {}),
-      ...(engine.modelMap !== undefined ? { modelMap: engine.modelMap } : {}),
+      baseUrl: engine.baseUrl !== undefined && engine.baseUrl !== null ? engine.baseUrl : AI_PROXY_PATH,
     };
+    const hasModel = engine.model !== undefined;
+    if (hasModel) {
+      result.model = engine.model;
+    }
+    const hasModelMap = engine.modelMap !== undefined;
+    if (hasModelMap) {
+      result.modelMap = engine.modelMap;
+    }
+    return result as AiEngineConfigWire;
   }
-  return {
+  const result: Record<string, unknown> = {
     kind: 'scripted',
-    ...(engine.script !== undefined
-      ? { script: engine.script as unknown as Array<Record<string, unknown>> }
-      : {}),
   };
+  const hasScript = engine.script !== undefined;
+  if (hasScript) {
+    result.script = engine.script as unknown as Array<Record<string, unknown>>;
+  }
+  return result as AiEngineConfigWire;
 }
 
 /** Resolve the plugin's explicit-options-over-Vite-env AI convention. */
@@ -66,20 +80,73 @@ export function resolveViteAiConfig(
   options: PyricAiOptions | undefined,
   env: Record<string, string | undefined>,
 ): ResolvedViteAiConfig {
-  if (options?.model !== undefined && options.engine !== undefined) {
+  const explicitModel = options?.model;
+  const explicitEngine = options?.engine;
+  const hasModelOption = explicitModel !== undefined;
+  const hasEngineOption = explicitEngine !== undefined;
+  const hasBothModelAndEngine = hasModelOption && hasEngineOption;
+  if (hasBothModelAndEngine) {
     throw new Error('@pyric/cli/vite: Choose either ai.model or ai.engine, not both.');
   }
 
-  const model = options?.model?.trim() || env.PYRIC_AI_MODEL?.trim();
-  const engineWire = options?.engine
-    ? engineConfigToWire(options.engine)
-    : model
-      ? engineConfigToWire({ kind: 'openai', baseUrl: AI_PROXY_PATH, model })
-      : undefined;
+  let mode: 'sandbox' | 'production' = 'sandbox';
+  const explicitMode = options?.mode;
+  const hasExplicitMode = explicitMode !== undefined;
+  if (hasExplicitMode) {
+    mode = explicitMode;
+  } else {
+    const isEnvProductionMode = env.PYRIC_AI_MODE === 'production';
+    const isEnvPassthroughFlag = env.PYRIC_AI_PASSTHROUGH === '1';
+    const isProductionEnv = isEnvProductionMode || isEnvPassthroughFlag;
+    if (isProductionEnv) {
+      mode = 'production';
+    }
+  }
+
+  const isProductionMode = mode === 'production';
+  const hasAnyEngineConfig = hasModelOption || hasEngineOption;
+  const isInvalidProductionConfig = isProductionMode && hasAnyEngineConfig;
+  if (isInvalidProductionConfig) {
+    throw new Error('@pyric/cli/vite: Cannot configure ai.model or ai.engine when ai.mode is set to "production".');
+  }
+
+  let model: string | undefined = undefined;
+  const envModel = env.PYRIC_AI_MODEL;
+  if (explicitModel !== undefined) {
+    model = explicitModel.trim();
+  } else if (envModel !== undefined) {
+    model = envModel.trim();
+  }
+  const hasNonEmptyModel = model !== undefined && model !== '';
+  if (!hasNonEmptyModel) {
+    model = undefined;
+  }
+
+  let engineWire: AiEngineConfigWire | undefined = undefined;
+  if (!isProductionMode) {
+    if (explicitEngine !== undefined) {
+      engineWire = engineConfigToWire(explicitEngine);
+    } else if (model !== undefined) {
+      engineWire = engineConfigToWire({ kind: 'openai', baseUrl: AI_PROXY_PATH, model });
+    }
+  }
+
+  let proxyUpstream: string | undefined = undefined;
+  const explicitProxyUpstream = options?.proxyUpstream;
+  const envProxyUpstream = env.PYRIC_AI_PROXY_UPSTREAM;
+  if (explicitProxyUpstream !== undefined) {
+    proxyUpstream = explicitProxyUpstream.trim();
+  } else if (envProxyUpstream !== undefined) {
+    proxyUpstream = envProxyUpstream.trim();
+  }
+  const hasNonEmptyProxyUpstream = proxyUpstream !== undefined && proxyUpstream !== '';
+  if (!hasNonEmptyProxyUpstream) {
+    proxyUpstream = undefined;
+  }
 
   return {
+    mode,
     engineWire,
-    proxyUpstream:
-      options?.proxyUpstream?.trim() || env.PYRIC_AI_PROXY_UPSTREAM?.trim() || undefined,
+    proxyUpstream,
   };
 }

package/src/serve/vite-generation-rules-watch.ts

@@ -8,30 +8,82 @@ export function watchViteGenerationRules(input: {
   session: SandboxSession;
 }): (() => void) | null {
   const { server, session } = input;
-  const rulesFile = session.summary.rules.firestore.sourcePath;
-  if (!rulesFile) return null;
+  const firestoreFile = session.summary.rules.firestore.sourcePath;
+  const databaseFile = session.summary.rules.database.sourcePath;
+  const hasFirestoreFile = firestoreFile !== null;
+  const hasDatabaseFile = databaseFile !== null;
+  const hasNeitherFile = !hasFirestoreFile && !hasDatabaseFile;
+  if (hasNeitherFile) {
+    return null;
+  }
 
   let debounce: ReturnType<typeof setTimeout> | null = null;
   const onRulesChange = (file: string): void => {
-    if (path.resolve(file) !== path.resolve(rulesFile)) return;
-    if (debounce) clearTimeout(debounce);
+    const resolvedFile = path.resolve(file);
+    let isFirestoreMatch = false;
+    if (hasFirestoreFile) {
+      const firestoreResolved = path.resolve(firestoreFile);
+      isFirestoreMatch = resolvedFile === firestoreResolved;
+    }
+    let isDatabaseMatch = false;
+    if (hasDatabaseFile) {
+      const databaseResolved = path.resolve(databaseFile);
+      isDatabaseMatch = resolvedFile === databaseResolved;
+    }
+    const isNeitherMatch = !isFirestoreMatch && !isDatabaseMatch;
+    if (isNeitherMatch) {
+      return;
+    }
+    const hasDebounce = debounce !== null;
+    if (hasDebounce) {
+      clearTimeout(debounce as ReturnType<typeof setTimeout>);
+    }
     debounce = setTimeout(() => {
-      void session.reloadFirestoreRules().then((result) => {
-        if (result.kind === 'reloaded') {
-          server.config.logger.info(`  ↻ [pyric] rules reloaded (${result.rulesHash})`);
-        } else if (result.kind === 'rejected') {
-          server.config.logger.warn(
-            `  ⚠ [pyric] rules NOT reloaded (last-good stays live): ${result.error.message}`,
-          );
-        }
-      });
+      if (isFirestoreMatch) {
+        void session.reloadFirestoreRules().then((result) => {
+          const isReloaded = result.kind === 'reloaded';
+          if (isReloaded) {
+            server.config.logger.info(`  ↻ [pyric] rules reloaded (${result.rulesHash})`);
+          } else {
+            const isRejected = result.kind === 'rejected';
+            if (isRejected) {
+              server.config.logger.warn(
+                `  ⚠ [pyric] rules NOT reloaded (last-good stays live): ${result.error.message}`,
+              );
+            }
+          }
+        });
+      }
+      if (isDatabaseMatch) {
+        void session.reloadDatabaseRules().then((result) => {
+          const isReloaded = result.kind === 'reloaded';
+          if (isReloaded) {
+            server.config.logger.info(`  ↻ [pyric] rtdb rules reloaded (${result.rulesHash})`);
+          } else {
+            const isRejected = result.kind === 'rejected';
+            if (isRejected) {
+              server.config.logger.warn(
+                `  ⚠ [pyric] rtdb rules NOT reloaded (last-good stays live): ${result.error.message}`,
+              );
+            }
+          }
+        });
+      }
     }, 150);
   };
 
-  server.watcher.add(rulesFile);
+  if (hasFirestoreFile) {
+    server.watcher.add(firestoreFile);
+  }
+  if (hasDatabaseFile) {
+    server.watcher.add(databaseFile);
+  }
   server.watcher.on('change', onRulesChange);
   return () => {
-    if (debounce) clearTimeout(debounce);
+    const hasDebounce = debounce !== null;
+    if (hasDebounce) {
+      clearTimeout(debounce as ReturnType<typeof setTimeout>);
+    }
     server.watcher.off('change', onRulesChange);
   };
 }

package/src/serve/vite-module-swap.ts

@@ -45,10 +45,18 @@ export function createViteModuleContext(): ViteModuleContext {
   return { entries, cliRoot: packageRootOf(entries.init) };
 }
 
+export interface ViteModuleSwapOptions {
+  getAiMode?: () => 'sandbox' | 'production';
+}
+
 /** Own the Vite and optimizer forms of the Firebase-module swap. */
-export function createViteModuleSwap(context: ViteModuleContext): ViteModuleSwap {
+export function createViteModuleSwap(
+  context: ViteModuleContext,
+  options?: ViteModuleSwapOptions,
+): ViteModuleSwap {
   const { entries, cliRoot } = context;
   const pyricRoot = pyricPackageRoot();
+  const aiMode = () => options?.getAiMode?.() ?? 'sandbox';
 
   const isPyricImporter = (importer: string | undefined): boolean => {
     if (!importer) return false;
@@ -68,28 +76,72 @@ export function createViteModuleSwap(context: ViteModuleContext): ViteModuleSwap
     name: 'pyric-sandbox-optimizer',
     setup(build) {
       build.onResolve({ filter: FIREBASE_SPECIFIER }, (args) => {
-        const subpath = FIREBASE_SPECIFIER.exec(args.path)![1]!;
-        if (SERVED_FIREBASE_SUBPATHS.has(subpath)) {
-          return { path: entries[entryKey(subpath)]! };
+        const isShadowBridgeImporter = args.importer !== undefined && args.importer !== '' &&
+          (args.importer.includes('app-ai-passthrough') || args.importer.includes('app-bridge'));
+        const isFirebaseAppSpecifier = args.path === 'firebase/app';
+        const isBypassedBridgeImport = isShadowBridgeImporter && isFirebaseAppSpecifier;
+        if (isBypassedBridgeImport) {
+          return null;
+        }
+
+        const match = FIREBASE_SPECIFIER.exec(args.path);
+        const subpath = match !== null && match[1] !== undefined ? match[1] : '';
+        const isProductionAiMode = aiMode() === 'production';
+        if (isProductionAiMode) {
+          const isAiSubpath = subpath === 'ai';
+          if (isAiSubpath) {
+            return null;
+          }
+          const isAppSubpath = subpath === 'app';
+          const passthroughEntry = entries['app-ai-passthrough'];
+          const hasPassthroughEntry = passthroughEntry !== undefined;
+          const shouldUsePassthroughBridge = isAppSubpath && hasPassthroughEntry;
+          if (shouldUsePassthroughBridge) {
+            return { path: passthroughEntry };
+          }
+        }
+
+        const isServedSubpath = SERVED_FIREBASE_SUBPATHS.has(subpath);
+        if (isServedSubpath) {
+          const key = entryKey(subpath);
+          const entryPath = entries[key];
+          if (entryPath !== undefined) {
+            return { path: entryPath };
+          }
         }
         return null;
       });
       build.onResolve({ filter: NODE_BUILTIN_RE }, (args) => {
-        if (!isOurCode(args.importer)) return null;
-        return { path: args.path.replace(/^node:/, ''), namespace: 'pyric-node-shim' };
+        const isOwnedImporter = isOurCode(args.importer);
+        if (!isOwnedImporter) {
+          return null;
+        }
+        const shimPath = args.path.replace(/^node:/, '');
+        return { path: shimPath, namespace: 'pyric-node-shim' };
+      });
+      build.onLoad({ filter: /.*/, namespace: 'pyric-node-shim' }, (args) => {
+        const content = NODE_BUILTIN_SHIMS[args.path];
+        if (content !== undefined) {
+          return { contents: content, loader: 'js' };
+        }
+        return null;
       });
-      build.onLoad({ filter: /.*/, namespace: 'pyric-node-shim' }, (args) => ({
-        contents: NODE_BUILTIN_SHIMS[args.path]!,
-        loader: 'js',
-      }));
     },
   };
 
   return {
     config() {
+      const excludedModules = [
+        ...SDK_MODULES,
+        '@firebase/app',
+        '@firebase/component',
+        '@firebase/ai',
+        '@firebase/util',
+        '@firebase/logger',
+      ];
       return {
         optimizeDeps: {
-          exclude: [...SDK_MODULES],
+          exclude: excludedModules,
           include: ['js-md5', 'js-sha256'],
           esbuildOptions: { plugins: [optimizerMirror] },
         },
@@ -97,23 +149,71 @@ export function createViteModuleSwap(context: ViteModuleContext): ViteModuleSwap
     },
     configResolved(config) {
       const allow = config.server?.fs?.allow;
-      if (!allow) return;
+      const hasAllowList = allow !== undefined && Array.isArray(allow);
+      if (!hasAllowList) {
+        return;
+      }
       for (const dir of [pyricRoot, cliRoot]) {
-        if (!allow.includes(dir)) allow.push(dir);
+        const isAlreadyAllowed = allow.includes(dir);
+        if (!isAlreadyAllowed) {
+          allow.push(dir);
+        }
       }
     },
     resolveId(source, importer) {
-      const firebase = FIREBASE_SPECIFIER.exec(source);
-      if (firebase) {
-        const subpath = firebase[1]!;
-        return SERVED_FIREBASE_SUBPATHS.has(subpath) ? entries[entryKey(subpath)]! : null;
+      const isShadowBridgeImporter = importer !== undefined &&
+        (importer.includes('app-ai-passthrough') || importer.includes('app-bridge'));
+      const isFirebaseAppSpecifier = source === 'firebase/app';
+      const isBypassedBridgeImport = isShadowBridgeImporter && isFirebaseAppSpecifier;
+      if (isBypassedBridgeImport) {
+        return null;
       }
-      const node = NODE_BUILTIN_RE.exec(source);
-      return node && isOurCode(importer) ? NODE_SHIM_PREFIX + node[2]! : null;
+
+      const firebaseMatch = FIREBASE_SPECIFIER.exec(source);
+      const isFirebaseSpecifier = firebaseMatch !== null;
+      if (isFirebaseSpecifier) {
+        const subpath = firebaseMatch[1] !== undefined ? firebaseMatch[1] : '';
+        const isProductionAiMode = aiMode() === 'production';
+        if (isProductionAiMode) {
+          const isAiSubpath = subpath === 'ai';
+          if (isAiSubpath) {
+            return null;
+          }
+          const isAppSubpath = subpath === 'app';
+          const passthroughEntry = entries['app-ai-passthrough'];
+          const hasPassthroughEntry = passthroughEntry !== undefined;
+          const shouldUsePassthroughBridge = isAppSubpath && hasPassthroughEntry;
+          if (shouldUsePassthroughBridge) {
+            return passthroughEntry;
+          }
+        }
+        const isServedSubpath = SERVED_FIREBASE_SUBPATHS.has(subpath);
+        if (isServedSubpath) {
+          const key = entryKey(subpath);
+          const entryPath = entries[key];
+          if (entryPath !== undefined) {
+            return entryPath;
+          }
+        }
+        return null;
+      }
+
+      const nodeMatch = NODE_BUILTIN_RE.exec(source);
+      const isNodeBuiltin = nodeMatch !== null && nodeMatch[2] !== undefined;
+      const isOwnedImporter = isOurCode(importer);
+      const shouldShimNodeBuiltin = isNodeBuiltin && isOwnedImporter;
+      if (shouldShimNodeBuiltin) {
+        return NODE_SHIM_PREFIX + nodeMatch[2];
+      }
+      return null;
     },
     load(id) {
-      if (!id.startsWith(NODE_SHIM_PREFIX)) return null;
-      return shimFor(id.slice(NODE_SHIM_PREFIX.length));
+      const isNodeShim = id.startsWith(NODE_SHIM_PREFIX);
+      if (!isNodeShim) {
+        return null;
+      }
+      const specifier = id.slice(NODE_SHIM_PREFIX.length);
+      return shimFor(specifier);
     },
   };
 }

package/src/serve/vite-plugin.ts

@@ -39,6 +39,8 @@
  * to the in-page sandbox (single-tab, ephemeral).
  */
 import type { Plugin, UserConfig } from 'vite';
+import { existsSync } from 'node:fs';
+import { dirname, join, resolve } from 'node:path';
 import { createViteWorkerRuntime } from './vite-worker-runtime.js';
 import type { PyricAiOptions } from './vite-ai-config.js';
 import type { PyricRuntimeChipOption } from './runtime/chip-config.js';
@@ -157,6 +159,29 @@ function resolveFunctionsOptions(
   return {};
 }
 
+/**
+ * Resolves the Firebase project root directory from Vite's server root. In
+ * split-directory or full-stack monorepo layouts where Vite operates inside a
+ * subdirectory (e.g. `/web` or `/frontend`) while hosting and security rules
+ * configuration (`firebase.json`) resides in an ancestor workspace directory,
+ * this ascends to locate the authoritative configuration root.
+ */
+function resolveFirebaseProjectDir(defaultRoot: string, explicitRoot?: string): string {
+  const hasExplicitRoot = explicitRoot !== undefined && explicitRoot.length > 0;
+  if (hasExplicitRoot) return explicitRoot as string;
+  let current = resolve(defaultRoot);
+  while (true) {
+    const candidateConfig = join(current, 'firebase.json');
+    const configExists = existsSync(candidateConfig);
+    if (configExists) return current;
+    const parent = dirname(current);
+    const isRootOfFilesystem = parent === current;
+    if (isRootOfFilesystem) break;
+    current = parent;
+  }
+  return defaultRoot;
+}
+
 /**
  * The dev-only Vite plugin. Add to `vite.config`:
  *
@@ -165,7 +190,6 @@ function resolveFunctionsOptions(
  */
 export function pyric(options: PyricOptions = {}): Plugin {
   const moduleContext = createViteModuleContext();
-  const moduleSwap = createViteModuleSwap(moduleContext);
   const { entries, cliRoot } = moduleContext;
   const workerRuntime = createViteWorkerRuntime();
   const studioEnabled = options.ui !== false;
@@ -176,6 +200,9 @@ export function pyric(options: PyricOptions = {}): Plugin {
     workerRuntime,
   };
   const pageRuntime = createVitePageRuntime(pageRuntimeInput);
+  const moduleSwap = createViteModuleSwap(moduleContext, {
+    getAiMode: () => pageRuntime.ai().mode,
+  });
 
   // Normalize the public bridge shorthand once. The active generation owns the
   // bridge/session attachment and routes the peer through the SharedWorker.
@@ -197,10 +224,21 @@ export function pyric(options: PyricOptions = {}): Plugin {
     enforce: 'pre',
 
     config(config, env) {
-      return {
-        ...moduleSwap.config(),
-        ...pageRuntime.config(config, env),
-      } as UserConfig;
+      const moduleConfig = moduleSwap.config();
+      const pageConfig = pageRuntime.config(config, env);
+      const serverWatchConfig: UserConfig = {
+        server: {
+          watch: {
+            ignored: ['**/.pyric/**'],
+          },
+        },
+      };
+      const finalizedConfig: UserConfig = {
+        ...moduleConfig,
+        ...pageConfig,
+        ...serverWatchConfig,
+      };
+      return finalizedConfig;
     },
 
     configResolved(resolved) {
@@ -220,7 +258,7 @@ export function pyric(options: PyricOptions = {}): Plugin {
       activeGeneration = null;
       await priorGeneration?.close();
 
-      const projectDir = options.root ?? server.config.root;
+      const projectDir = resolveFirebaseProjectDir(server.config.root, options.root);
       const uiEnabled = options.ui ?? true;
       const functionsOptions = resolveFunctionsOptions(options.functions);
       const ai = pageRuntime.ai();

package/src/serve/vite-sandbox-generation.ts

@@ -134,7 +134,7 @@ export async function createViteSandboxGeneration(
     const rulesConfig = dependencies.resolveRulesConfig(cwd, options.rules, firebaseConfig);
 
     try {
-      const epochSalt = viteWorkerEpochSalt(cwd, ai.engineWire);
+      const epochSalt = viteWorkerEpochSalt(cwd, ai.engineWire, ai.mode);
       await dependencies.prepareWorker(workerRuntime, epochSalt);
     } catch (error) {
       server.config.logger.warn(