revert change to import from @firebase/ai

Author: dlarocque

packages/ai/integration/chat.test.ts

@@ -23,7 +23,7 @@ import {
   HarmCategory,
   SafetySetting,
   getGenerativeModel
-} from '@firebase/ai';
+} from '../src';
 import { testConfigs, TOKEN_COUNT_DELTA } from './constants';
 
 describe('Chat Session', () => {

packages/ai/integration/constants.ts

@@ -23,7 +23,7 @@ import {
   GoogleAIBackend,
   VertexAIBackend,
   getAI
-} from '@firebase/ai';
+} from '../src';
 import { FIREBASE_CONFIG } from './firebase-config';
 
 const app = initializeApp(FIREBASE_CONFIG);

packages/ai/integration/count-tokens.test.ts

@@ -30,7 +30,7 @@ import {
   InlineDataPart,
   FileDataPart,
   BackendType
-} from '@firebase/ai';
+} from '../src';
 import {
   AUDIO_MIME_TYPE,
   IMAGE_MIME_TYPE,

packages/ai/integration/generate-content.test.ts

@@ -24,7 +24,7 @@ import {
   Modality,
   SafetySetting,
   getGenerativeModel
-} from '@firebase/ai';
+} from '../src';
 import { testConfigs, TOKEN_COUNT_DELTA } from './constants';
 
 describe('Generate Content', () => {