fix(ai): allow auth and app check to be passed in

Author: russellwheatley

packages/ai/lib/index.ts

@@ -66,6 +66,8 @@ export function getAI(
     app,
     backend: options.backend,
     location: (options.backend as VertexAIBackend)?.location || '',
+    appCheck: options.appCheck || null,
+    auth: options.auth || null,
   } as AI;
 }
 

packages/ai/lib/models/ai-model.ts

@@ -66,6 +66,7 @@ export abstract class AIModel {
         location: ai.location,
         backend: ai.backend,
       };
+
       if ((ai as AIService).appCheck) {
         this._apiSettings.getAppCheckToken = () => (ai as AIService).appCheck!.getToken();
       }

packages/ai/lib/public-types.ts

@@ -33,6 +33,8 @@ export interface AIOptions {
    * The backend configuration to use for the AI service instance.
    */
   backend: Backend;
+  appCheck?: FirebaseAppCheckTypes.Module | null;
+  auth?: FirebaseAuthTypes.Module | null;
 }
 
 /**