Fix tests so they work in node

Author: hsubox76

packages/ai/package.json

@@ -39,7 +39,9 @@
     "test:ci": "yarn testsetup && node ../../scripts/run_tests_in_ci.js -s test",
     "test:skip-clone": "karma start",
     "test:browser": "yarn testsetup && karma start",
+    "test:node": "TS_NODE_COMPILER_OPTIONS='{\"module\":\"commonjs\"}' mocha --require ts-node/register --require src/index.node.ts 'src/**/!(chrome)*.test.ts' --config ../../config/mocharc.node.js",
     "test:integration": "karma start --integration",
+    "test:integration:node": "TS_NODE_COMPILER_OPTIONS='{\"module\":\"commonjs\"}' mocha integration/**/*.test.ts --config ../../config/mocharc.node.js",
     "api-report": "api-extractor run --local --verbose",
     "typings:public": "node ../../scripts/build/use_typings.js ./dist/ai-public.d.ts",
     "trusted-type-check": "tsec -p tsconfig.json --noEmit"

packages/ai/src/api.ts

@@ -32,8 +32,6 @@ import { AIError } from './errors';
 import { AIModel, GenerativeModel, ImagenModel } from './models';
 import { encodeInstanceIdentifier } from './helpers';
 import { GoogleAIBackend } from './backend';
-import { ChromeAdapterImpl } from './methods/chrome-adapter';
-import { LanguageModel } from './types/language-model';
 
 export { ChatSession } from './methods/chat-session';
 export * from './requests/schema-builder';
@@ -124,15 +122,17 @@ export function getGenerativeModel(
       `Must provide a model name. Example: getGenerativeModel({ model: 'my-model-name' })`
     );
   }
-  let chromeAdapter: ChromeAdapterImpl | undefined;
-  // Do not initialize a ChromeAdapter if we are not in hybrid mode.
-  if (typeof window !== 'undefined' && hybridParams.mode) {
-    chromeAdapter = new ChromeAdapterImpl(
-      window.LanguageModel as LanguageModel,
-      hybridParams.mode,
-      hybridParams.onDeviceParams
-    );
-  }
+
+  /**
+   * An AIService registered by index.node.ts will not have a
+   * chromeAdapterFactory() method.
+   */
+  const chromeAdapter = (ai as AIService).chromeAdapterFactory?.(
+    hybridParams.mode,
+    typeof window && window,
+    hybridParams.onDeviceParams
+  );
+
   return new GenerativeModel(ai, inCloudParams, requestOptions, chromeAdapter);
 }
 

packages/ai/src/index.ts

@@ -34,10 +34,12 @@ import { name, version } from '../package.json';
 import { decodeInstanceIdentifier } from './helpers';
 import { AIError } from './api';
 import { AIErrorCode } from './types';
+import { chromeAdapterFactory } from './methods/chrome-adapter';
+import { LanguageModel } from './types/language-model';
 
 declare global {
   interface Window {
-    [key: string]: unknown;
+    languageModel: LanguageModel;
   }
 }
 
@@ -58,7 +60,14 @@ export function factory(
   const app = container.getProvider('app').getImmediate();
   const auth = container.getProvider('auth-internal');
   const appCheckProvider = container.getProvider('app-check-internal');
-  return new AIService(app, backend, auth, appCheckProvider);
+
+  return new AIService(
+    app,
+    backend,
+    auth,
+    appCheckProvider,
+    chromeAdapterFactory
+  );
 }
 
 function registerAI(): void {

packages/ai/src/methods/chrome-adapter.test.ts renamed to packages/ai/src/methods/chrome-adapter-test.ts

@@ -372,3 +372,21 @@ export class ChromeAdapterImpl implements ChromeAdapter {
     } as Response;
   }
 }
+
+/**
+ * Creates a ChromeAdapterImpl on demand.
+ */
+export function chromeAdapterFactory(
+  mode: InferenceMode,
+  window?: Global | Window,
+  params?: OnDeviceParams
+): ChromeAdapterImpl | undefined {
+  // Do not initialize a ChromeAdapter if we are not in hybrid mode.
+  if (typeof window !== 'undefined' && mode) {
+    return new ChromeAdapterImpl(
+      (window as Window).languageModel as LanguageModel,
+      mode,
+      params
+    );
+  }
+}

packages/ai/src/methods/chrome-adapter.ts

@@ -16,7 +16,7 @@
  */
 
 import { FirebaseApp, _FirebaseService } from '@firebase/app';
-import { AI, AIOptions } from './public-types';
+import { AI, AIOptions, InferenceMode, OnDeviceParams } from './public-types';
 import {
   AppCheckInternalComponentName,
   FirebaseAppCheckInternal
@@ -27,6 +27,7 @@ import {
   FirebaseAuthInternalName
 } from '@firebase/auth-interop-types';
 import { Backend, VertexAIBackend } from './backend';
+import { ChromeAdapterImpl } from './methods/chrome-adapter';
 
 export class AIService implements AI, _FirebaseService {
   auth: FirebaseAuthInternal | null;
@@ -38,7 +39,12 @@ export class AIService implements AI, _FirebaseService {
     public app: FirebaseApp,
     public backend: Backend,
     authProvider?: Provider<FirebaseAuthInternalName>,
-    appCheckProvider?: Provider<AppCheckInternalComponentName>
+    appCheckProvider?: Provider<AppCheckInternalComponentName>,
+    public chromeAdapterFactory?: (
+      mode: InferenceMode,
+      window?: Global | Window,
+      params?: OnDeviceParams
+    ) => ChromeAdapterImpl | undefined
   ) {
     const appCheck = appCheckProvider?.getImmediate({ optional: true });
     const auth = authProvider?.getImmediate({ optional: true });

packages/ai/src/service.ts

@@ -19,15 +19,15 @@
 const { readdirSync, readFileSync, writeFileSync } = require('node:fs');
 const { join } = require('node:path');
 
+type BackendName = import('./types').BackendName; // Import type without triggering ES module detection
+
 const MOCK_RESPONSES_DIR_PATH = join(
   __dirname,
   'vertexai-sdk-test-data',
   'mock-responses'
 );
 const MOCK_LOOKUP_OUTPUT_PATH = join(__dirname, 'mocks-lookup.ts');
 
-type BackendName = 'vertexAI' | 'googleAI';
-
 const mockDirs: Record<BackendName, string> = {
   vertexAI: join(MOCK_RESPONSES_DIR_PATH, 'vertexai'),
   googleAI: join(MOCK_RESPONSES_DIR_PATH, 'googleai')

packages/ai/test-utils/convert-mocks.ts

@@ -15,6 +15,7 @@
  * limitations under the License.
  */
 
+import { BackendName } from './types';
 import { vertexAIMocksLookup, googleAIMocksLookup } from './mocks-lookup';
 
 const mockSetMaps: Record<BackendName, Record<string, string>> = {

packages/ai/test-utils/mock-response.ts

@@ -0,0 +1,18 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *   http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+export type BackendName = 'vertexAI' | 'googleAI';