fix e2e to match new package name

Author: hsubox76

e2e/sample-apps/modular.js

@@ -58,7 +58,7 @@ import {
   onValue,
   off
 } from 'firebase/database';
-import { getGenerativeModel, getVertexAI, VertexAI } from 'firebase/vertexai';
+import { getGenerativeModel, getAI } from 'firebase/ai';
 import { getDataConnect, DataConnect } from 'firebase/data-connect';
 
 /**
@@ -307,15 +307,15 @@ function callPerformance(app) {
 }
 
 /**
- * VertexAI smoke test.
+ * AI smoke test.
  * Just make sure some functions can be called without obvious errors.
  */
-async function callVertexAI(app) {
-  console.log('[VERTEXAI] start');
-  const vertexAI = getVertexAI(app);
-  const model = getGenerativeModel(vertexAI, { model: 'gemini-1.5-flash' });
+async function callAI(app) {
+  console.log('[AI] start');
+  const ai = getAI(app);
+  const model = getGenerativeModel(ai, { model: 'gemini-1.5-flash' });
   const result = await model.countTokens('abcdefg');
-  console.log(`[VERTEXAI] counted tokens: ${result.totalTokens}`);
+  console.log(`[AI] counted tokens: ${result.totalTokens}`);
 }
 
 /**
@@ -350,7 +350,7 @@ async function main() {
   callAnalytics(app);
   callPerformance(app);
   await callFunctions(app);
-  await callVertexAI(app);
+  await callAI(app);
   callDataConnect(app);
   await authLogout(app);
   console.log('DONE');

e2e/tests/modular.test.ts

@@ -86,7 +86,7 @@ import {
   StorageReference,
   deleteObject
 } from 'firebase/storage';
-import { getGenerativeModel, getVertexAI, VertexAI } from 'firebase/vertexai';
+import { getGenerativeModel, getAI, AI } from 'firebase/vertexai';
 import { getDataConnect, DataConnect } from 'firebase/data-connect';
 import { config, testAccount } from '../firebase-config';
 import 'jest';
@@ -307,13 +307,13 @@ describe('MODULAR', () => {
     });
   });
 
-  describe('VERTEXAI', () => {
-    let vertexAI: VertexAI;
+  describe.only('AI', () => {
+    let ai: AI;
     it('getVertexAI()', () => {
-      vertexAI = getVertexAI(app);
+      ai = getAI(app);
     });
     it('getGenerativeModel() and countTokens()', async () => {
-      const model = getGenerativeModel(vertexAI, { model: 'gemini-1.5-flash' });
+      const model = getGenerativeModel(ai, { model: 'gemini-1.5-flash' });
       expect(model.model).toMatch(/gemini-1.5-flash$/);
       const result = await model.countTokens('abcdefg');
       expect(result.totalTokens).toBeTruthy;