Remove default expected input types

Author: erikeldridge

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

@@ -124,7 +124,10 @@ describe('ChromeAdapter', () => {
       );
       const adapter = new ChromeAdapter(
         languageModelProvider,
-        'prefer_on_device'
+        'prefer_on_device',
+        {
+          expectedInputs: [{ type: 'image' }]
+        }
       );
       const expectedOnDeviceParams = {
         expectedInputs: [{ type: 'image' }]
@@ -221,17 +224,14 @@ describe('ChromeAdapter', () => {
       );
       const promptOutput = 'hi';
       const promptStub = stub(languageModel, 'prompt').resolves(promptOutput);
-      const onDeviceParams = {
-        systemPrompt: 'be yourself'
-      } as LanguageModelCreateOptions;
       const expectedOnDeviceParams = {
         systemPrompt: 'be yourself',
         expectedInputs: [{ type: 'image' }]
       } as LanguageModelCreateOptions;
       const adapter = new ChromeAdapter(
         languageModelProvider,
         'prefer_on_device',
-        onDeviceParams
+        expectedOnDeviceParams
       );
       const request = {
         contents: [{ role: 'user', parts: [{ text: 'anything' }] }]
@@ -270,17 +270,14 @@ describe('ChromeAdapter', () => {
       );
       const promptOutput = 'hi';
       const promptStub = stub(languageModel, 'prompt').resolves(promptOutput);
-      const onDeviceParams = {
-        systemPrompt: 'be yourself'
-      } as LanguageModelCreateOptions;
       const expectedOnDeviceParams = {
         systemPrompt: 'be yourself',
         expectedInputs: [{ type: 'image' }]
       } as LanguageModelCreateOptions;
       const adapter = new ChromeAdapter(
         languageModelProvider,
         'prefer_on_device',
-        onDeviceParams
+        expectedOnDeviceParams
       );
       const request = {
         contents: [
@@ -379,14 +376,13 @@ describe('ChromeAdapter', () => {
           }
         })
       );
-      const onDeviceParams = {} as LanguageModelCreateOptions;
       const expectedOnDeviceParams = {
         expectedInputs: [{ type: 'image' }]
       } as LanguageModelCreateOptions;
       const adapter = new ChromeAdapter(
         languageModelProvider,
         'prefer_on_device',
-        onDeviceParams
+        expectedOnDeviceParams
       );
       const request = {
         contents: [{ role: 'user', parts: [{ text: 'anything' }] }]
@@ -423,14 +419,13 @@ describe('ChromeAdapter', () => {
           }
         })
       );
-      const onDeviceParams = {} as LanguageModelCreateOptions;
       const expectedOnDeviceParams = {
         expectedInputs: [{ type: 'image' }]
       } as LanguageModelCreateOptions;
       const adapter = new ChromeAdapter(
         languageModelProvider,
         'prefer_on_device',
-        onDeviceParams
+        expectedOnDeviceParams
       );
       const request = {
         contents: [

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

@@ -42,9 +42,7 @@ export class ChromeAdapter {
     private languageModelProvider?: LanguageModel,
     private mode?: InferenceMode,
     private onDeviceParams: LanguageModelCreateOptions = {}
-  ) {
-    this.addImageTypeAsExpectedInput();
-  }
+  ) {}
 
   /**
    * Checks if a given request can be made on-device.
@@ -237,11 +235,6 @@ export class ChromeAdapter {
     return newSession;
   }
 
-  private addImageTypeAsExpectedInput(): void {
-    // Defaults to support image inputs for convenience.
-    this.onDeviceParams.expectedInputs ??= [{ type: 'image' }];
-  }
-
   /**
    * Formats string returned by Chrome as a {@link Response} returned by Vertex.
    */