Fix e2e to use new AI package

Author: hsubox76

e2e/package.json

@@ -17,7 +17,7 @@
   "author": "",
   "license": "ISC",
   "dependencies": {
-    "firebase": "11.3.0"
+    "firebase": "^11.8.0-20250512211235"
   },
   "devDependencies": {
     "@babel/core": "7.26.8",

e2e/sample-apps/modular.js

@@ -58,7 +58,7 @@ import {
   onValue,
   off
 } from 'firebase/database';
-import { getGenerativeModel, getAI } from 'firebase/ai';
+import { getGenerativeModel, getAI, VertexAIBackend } from 'firebase/ai';
 import { getDataConnect, DataConnect } from 'firebase/data-connect';
 
 /**
@@ -312,7 +312,7 @@ function callPerformance(app) {
  */
 async function callAI(app) {
   console.log('[AI] start');
-  const ai = getAI(app);
+  const ai = getAI(app, { backend: new VertexAIBackend() });
   const model = getGenerativeModel(ai, { model: 'gemini-1.5-flash' });
   const result = await model.countTokens('abcdefg');
   console.log(`[AI] counted tokens: ${result.totalTokens}`);

e2e/tests/modular.test.ts

@@ -86,7 +86,12 @@ import {
   StorageReference,
   deleteObject
 } from 'firebase/storage';
-import { getGenerativeModel, getAI, AI } from 'firebase/vertexai';
+import {
+  getGenerativeModel,
+  getAI,
+  AI,
+  VertexAIBackend
+} from 'firebase/vertexai';
 import { getDataConnect, DataConnect } from 'firebase/data-connect';
 import { config, testAccount } from '../firebase-config';
 import 'jest';
@@ -307,10 +312,10 @@ describe('MODULAR', () => {
     });
   });
 
-  describe.only('AI', () => {
+  describe('AI', () => {
     let ai: AI;
     it('getVertexAI()', () => {
-      ai = getAI(app);
+      ai = getAI(app, { backend: new VertexAIBackend() });
     });
     it('getGenerativeModel() and countTokens()', async () => {
       const model = getGenerativeModel(ai, { model: 'gemini-1.5-flash' });