Inject LanguageModel provider

Author: erikeldridge

packages/vertexai/src/api.ts

@@ -31,6 +31,7 @@ import {
 import { VertexAIError } from './errors';
 import { VertexAIModel, GenerativeModel, ImagenModel } from './models';
 import { ChromeAdapter } from './methods/chrome-adapter';
+import { LanguageModel } from './types/language-model';
 
 export { ChatSession } from './methods/chat-session';
 export * from './requests/schema-builder';
@@ -96,6 +97,7 @@ export function getGenerativeModel(
     vertexAI,
     inCloudParams,
     new ChromeAdapter(
+      window.LanguageModel as LanguageModel,
       hybridParams.mode,
       hybridParams.onDeviceParams
     ),

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

@@ -16,14 +16,18 @@
  */
 
 import { GenerateContentRequest, InferenceMode } from '../types';
-import { LanguageModelCreateOptions } from '../types/language-model';
+import {
+  LanguageModel,
+  LanguageModelCreateOptions
+} from '../types/language-model';
 
 /**
  * Defines an inference "backend" that uses Chrome's on-device model,
  * and encapsulates logic for detecting when on-device is possible.
  */
 export class ChromeAdapter {
   constructor(
+    private languageModelProvider?: LanguageModel,
     private mode?: InferenceMode,
     private onDeviceParams?: LanguageModelCreateOptions
   ) {}

packages/vertexai/src/methods/generate-content.test.ts

@@ -312,6 +312,7 @@ describe('generateContent()', () => {
     const chromeAdapter = new ChromeAdapter();
     const isAvailableStub = stub(chromeAdapter, 'isAvailable').resolves(true);
     const mockResponse = getMockResponse(
+      'vertexAI',
       'unary-success-basic-reply-short.json'
     );
     const generateContentStub = stub(