add a case and test

Author: hsubox76

packages/ai/src/api.test.ts

@@ -57,6 +57,14 @@ describe('Top level API', () => {
       expect(ai.backend).to.be.instanceOf(VertexAIBackend);
       expect(ai.options?.appCheck?.limitedUseTokens).to.be.true;
     });
+    it('works with options: appCheck option is empty', () => {
+      const ai = getAI(getFullApp(), {
+        backend: new VertexAIBackend('us-central1'),
+        appCheck: {}
+      });
+      expect(ai.backend).to.be.instanceOf(VertexAIBackend);
+      expect(ai.options?.appCheck?.limitedUseTokens).to.be.false;
+    });
     it('works with options: backend specified only', () => {
       const ai = getAI(getFullApp(), {
         backend: new VertexAIBackend('us-central1')

packages/ai/src/api.ts

@@ -79,9 +79,17 @@ export function getAI(app: FirebaseApp = getApp(), options?: AIOptions): AI {
 
   const backend = options?.backend ?? new GoogleAIBackend();
 
-  const finalOptions = {
-    appCheck: options?.appCheck ?? { limitedUseTokens: false }
-  };
+  const finalOptions: Omit<AIOptions, 'backend'> = {};
+
+  if (options?.appCheck) {
+    if (options.appCheck.hasOwnProperty('limitedUseTokens')) {
+      finalOptions.appCheck = options.appCheck;
+    } else {
+      finalOptions.appCheck = { ...options.appCheck, limitedUseTokens: false };
+    }
+  } else {
+    finalOptions.appCheck = { limitedUseTokens: false };
+  }
 
   const identifier = encodeInstanceIdentifier(backend);
   const aiInstance = AIProvider.getImmediate({

packages/ai/test-utils/get-fake-firebase-services.ts

@@ -38,13 +38,7 @@ export function getFullApp(fakeAppParams?: {
   appId?: string;
   apiKey?: string;
 }): FirebaseApp {
-  _registerComponent(
-    new Component(
-      AI_TYPE,
-      factory,
-      ComponentType.PUBLIC
-    )
-  );
+  _registerComponent(new Component(AI_TYPE, factory, ComponentType.PUBLIC));
   _registerComponent(
     new Component(
       'app-check-internal',