refactor(vertexai): initial setup to use ai package

Author: russellwheatley

packages/ai/lib/index.ts

@@ -25,9 +25,15 @@ import { GenerativeModel } from './models/generative-model';
 export { ChatSession } from './methods/chat-session';
 export * from './requests/schema-builder';
 
-export { GenerativeModel };
-
-export { AIError };
+export {
+  GenerativeModel,
+  AIError,
+  AIErrorCode,
+  ModelParams,
+  RequestOptions,
+  GoogleAIBackend,
+  VertexAIBackend,
+};
 
 /**
  * Returns the default {@link AI} instance that is associated with the provided

packages/vertexai/lib/index.ts

@@ -17,17 +17,22 @@
 
 import './polyfills';
 import { getApp, ReactNativeFirebase } from '@react-native-firebase/app';
-import { ModelParams, RequestOptions, VertexAIErrorCode } from './types';
+import {
+  getGenerativeModel as getGenerativeModelFromAI,
+  getAI,
+  VertexAIBackend,
+  GenerativeModel,
+  RequestOptions,
+  ModelParams,
+} from '@react-native-firebase/ai';
+import { VertexAIErrorCode } from './types';
 import { DEFAULT_LOCATION } from './constants';
 import { VertexAI, VertexAIOptions } from './public-types';
 import { VertexAIError } from './errors';
-import { GenerativeModel } from './models/generative-model';
 import { VertexAIService } from './service';
 export { ChatSession } from './methods/chat-session';
 export * from './requests/schema-builder';
 
-export { GenerativeModel };
-
 export { VertexAIError };
 
 /**
@@ -69,5 +74,10 @@ export function getGenerativeModel(
       `Must provide a model name. Example: getGenerativeModel({ model: 'my-model-name' })`,
     );
   }
-  return new GenerativeModel(vertexAI, modelParams, requestOptions);
+
+  const ai = getAI(vertexAI.app, {
+    backend: new VertexAIBackend(vertexAI.location),
+  });
+
+  return getGenerativeModelFromAI(ai, modelParams, requestOptions);
 }

packages/vertexai/tsconfig.json

@@ -1,6 +1,5 @@
 {
   "compilerOptions": {
-    "rootDir": ".",
     "allowUnreachableCode": false,
     "allowUnusedLabels": false,
     "esModuleInterop": true,
@@ -27,6 +26,8 @@
       "@react-native-firebase/app": ["../app/lib"],
       "@react-native-firebase/auth": ["../auth/lib"],
       "@react-native-firebase/app-check": ["../app-check/lib"],
+      "@react-native-firebase/ai": ["../ai/lib"],
     }
-  }
+  },
+  "include": ["lib/**/*", "../ai/lib/**/*"]
 }

tsconfig-jest.json

@@ -6,6 +6,7 @@
       "@react-native-firebase/app": ["packages/app/lib"],
       "@react-native-firebase/auth": ["packages/auth/lib"],
       "@react-native-firebase/app-check": ["packages/app-check/lib"],
+      "@react-native-firebase/ai": ["packages/ai/lib"],
     }
   }
 }