feat(vertexai): Migrate to GenAI and add GoogleAI support

Author: dlarocque

common/api-review/vertexai.api.md

@@ -18,6 +18,18 @@ export class ArraySchema extends Schema {
     toJSON(): SchemaRequest;
 }
 
+// @public
+export type Backend = GoogleAIBackend | VertexAIBackend;
+
+// @public
+export const BackendType: {
+    readonly VERTEX_AI: "VERTEX_AI";
+    readonly GOOGLE_AI: "GOOGLE_AI";
+};
+
+// @public
+export type BackendType = (typeof BackendType)[keyof typeof BackendType];
+
 // @public
 export interface BaseParams {
     // (undocumented)
@@ -239,6 +251,60 @@ export interface FunctionResponsePart {
     text?: never;
 }
 
+// @public
+export interface GenAI {
+    app: FirebaseApp;
+    backend: Backend;
+    // @deprecated
+    location: string;
+}
+
+// @public
+export class GenAIError extends FirebaseError {
+    constructor(code: GenAIErrorCode, message: string, customErrorData?: CustomErrorData | undefined);
+    // (undocumented)
+    readonly code: GenAIErrorCode;
+    // (undocumented)
+    readonly customErrorData?: CustomErrorData | undefined;
+}
+
+// @public
+const enum GenAIErrorCode {
+    API_NOT_ENABLED = "api-not-enabled",
+    ERROR = "error",
+    FETCH_ERROR = "fetch-error",
+    INVALID_CONTENT = "invalid-content",
+    INVALID_SCHEMA = "invalid-schema",
+    NO_API_KEY = "no-api-key",
+    NO_APP_ID = "no-app-id",
+    NO_MODEL = "no-model",
+    NO_PROJECT_ID = "no-project-id",
+    PARSE_FAILED = "parse-failed",
+    REQUEST_ERROR = "request-error",
+    RESPONSE_ERROR = "response-error",
+    UNSUPPORTED = "unsupported"
+}
+
+export { GenAIErrorCode }
+
+export { GenAIErrorCode as VertexAIErrorCode }
+
+// @public
+export abstract class GenAIModel {
+    // @internal
+    protected constructor(genAI: GenAI, modelName: string);
+    // @internal (undocumented)
+    protected _apiSettings: ApiSettings;
+    readonly model: string;
+    // @internal (undocumented)
+    static normalizeModelName(modelName: string, backendType: BackendType): string;
+    }
+
+// @public
+export interface GenAIOptions {
+    backend: Backend;
+}
+
 // @public
 export interface GenerateContentCandidate {
     // (undocumented)
@@ -323,8 +389,8 @@ export interface GenerativeContentBlob {
 }
 
 // @public
-export class GenerativeModel extends VertexAIModel {
-    constructor(vertexAI: VertexAI, modelParams: ModelParams, requestOptions?: RequestOptions);
+export class GenerativeModel extends GenAIModel {
+    constructor(genAI: GenAI, modelParams: ModelParams, requestOptions?: RequestOptions);
     countTokens(request: CountTokensRequest | string | Array<string | Part>): Promise<CountTokensResponse>;
     generateContent(request: GenerateContentRequest | string | Array<string | Part>): Promise<GenerateContentResult>;
     generateContentStream(request: GenerateContentRequest | string | Array<string | Part>): Promise<GenerateContentStreamResult>;
@@ -344,14 +410,25 @@ export class GenerativeModel extends VertexAIModel {
 }
 
 // @public
-export function getGenerativeModel(vertexAI: VertexAI, modelParams: ModelParams, requestOptions?: RequestOptions): GenerativeModel;
+export function getGenAI(app?: FirebaseApp, options?: GenAIOptions): GenAI;
+
+// @public
+export function getGenerativeModel(genAI: GenAI, modelParams: ModelParams, requestOptions?: RequestOptions): GenerativeModel;
 
 // @beta
-export function getImagenModel(vertexAI: VertexAI, modelParams: ImagenModelParams, requestOptions?: RequestOptions): ImagenModel;
+export function getImagenModel(genAI: GenAI, modelParams: ImagenModelParams, requestOptions?: RequestOptions): ImagenModel;
 
 // @public
 export function getVertexAI(app?: FirebaseApp, options?: VertexAIOptions): VertexAI;
 
+// @public
+export type GoogleAIBackend = {
+    backendType: typeof BackendType.GOOGLE_AI;
+};
+
+// @public
+export function googleAIBackend(): GoogleAIBackend;
+
 // @public @deprecated (undocumented)
 export interface GroundingAttribution {
     // (undocumented)
@@ -413,7 +490,8 @@ export enum HarmSeverity {
     HARM_SEVERITY_HIGH = "HARM_SEVERITY_HIGH",
     HARM_SEVERITY_LOW = "HARM_SEVERITY_LOW",
     HARM_SEVERITY_MEDIUM = "HARM_SEVERITY_MEDIUM",
-    HARM_SEVERITY_NEGLIGIBLE = "HARM_SEVERITY_NEGLIGIBLE"
+    HARM_SEVERITY_NEGLIGIBLE = "HARM_SEVERITY_NEGLIGIBLE",
+    HARM_SEVERITY_UNSUPPORTED = "HARM_SEVERITY_UNSUPPORTED"
 }
 
 // @beta
@@ -461,8 +539,8 @@ export interface ImagenInlineImage {
 }
 
 // @beta
-export class ImagenModel extends VertexAIModel {
-    constructor(vertexAI: VertexAI, modelParams: ImagenModelParams, requestOptions?: RequestOptions | undefined);
+export class ImagenModel extends GenAIModel {
+    constructor(genAI: GenAI, modelParams: ImagenModelParams, requestOptions?: RequestOptions | undefined);
     generateImages(prompt: string): Promise<ImagenGenerationResponse<ImagenInlineImage>>;
     // @internal
     generateImagesGCS(prompt: string, gcsURI: string): Promise<ImagenGenerationResponse<ImagenGCSImage>>;
@@ -627,7 +705,6 @@ export interface SafetyRating {
 export interface SafetySetting {
     // (undocumented)
     category: HarmCategory;
-    // (undocumented)
     method?: HarmBlockMethod;
     // (undocumented)
     threshold: HarmBlockThreshold;
@@ -779,46 +856,22 @@ export interface UsageMetadata {
 }
 
 // @public
-export interface VertexAI {
-    app: FirebaseApp;
-    // (undocumented)
+export type VertexAI = GenAI;
+
+// @public
+export type VertexAIBackend = {
+    backendType: typeof BackendType.VERTEX_AI;
     location: string;
-}
+};
 
 // @public
-export class VertexAIError extends FirebaseError {
-    constructor(code: VertexAIErrorCode, message: string, customErrorData?: CustomErrorData | undefined);
-    // (undocumented)
-    readonly code: VertexAIErrorCode;
-    // (undocumented)
-    readonly customErrorData?: CustomErrorData | undefined;
-}
+export function vertexAIBackend(location?: string): VertexAIBackend;
 
 // @public
-export const enum VertexAIErrorCode {
-    API_NOT_ENABLED = "api-not-enabled",
-    ERROR = "error",
-    FETCH_ERROR = "fetch-error",
-    INVALID_CONTENT = "invalid-content",
-    INVALID_SCHEMA = "invalid-schema",
-    NO_API_KEY = "no-api-key",
-    NO_APP_ID = "no-app-id",
-    NO_MODEL = "no-model",
-    NO_PROJECT_ID = "no-project-id",
-    PARSE_FAILED = "parse-failed",
-    REQUEST_ERROR = "request-error",
-    RESPONSE_ERROR = "response-error"
-}
+export const VertexAIError: typeof GenAIError;
 
 // @public
-export abstract class VertexAIModel {
-    // @internal
-    protected constructor(vertexAI: VertexAI, modelName: string);
-    // @internal (undocumented)
-    protected _apiSettings: ApiSettings;
-    readonly model: string;
-    static normalizeModelName(modelName: string): string;
-}
+export const VertexAIModel: typeof GenAIModel;
 
 // @public
 export interface VertexAIOptions {
@@ -841,4 +894,6 @@ export interface WebAttribution {
 }
 
 
+// (No @packageDocumentation comment for this package)
+
 ```

docs-devsite/_toc.yaml

@@ -516,6 +516,14 @@ toc:
     path: /docs/reference/js/vertexai.functionresponse.md
   - title: FunctionResponsePart
     path: /docs/reference/js/vertexai.functionresponsepart.md
+  - title: GenAI
+    path: /docs/reference/js/vertexai.genai.md
+  - title: GenAIError
+    path: /docs/reference/js/vertexai.genaierror.md
+  - title: GenAIModel
+    path: /docs/reference/js/vertexai.genaimodel.md
+  - title: GenAIOptions
+    path: /docs/reference/js/vertexai.genaioptions.md
   - title: GenerateContentCandidate
     path: /docs/reference/js/vertexai.generatecontentcandidate.md
   - title: GenerateContentRequest
@@ -598,12 +606,6 @@ toc:
     path: /docs/reference/js/vertexai.toolconfig.md
   - title: UsageMetadata
     path: /docs/reference/js/vertexai.usagemetadata.md
-  - title: VertexAI
-    path: /docs/reference/js/vertexai.vertexai.md
-  - title: VertexAIError
-    path: /docs/reference/js/vertexai.vertexaierror.md
-  - title: VertexAIModel
-    path: /docs/reference/js/vertexai.vertexaimodel.md
   - title: VertexAIOptions
     path: /docs/reference/js/vertexai.vertexaioptions.md
   - title: VideoMetadata

docs-devsite/index.md

@@ -27,5 +27,5 @@ https://github.com/firebase/firebase-js-sdk
 |  [@firebase/performance](./performance.md#performance_package) | The Firebase Performance Monitoring Web SDK. This SDK does not work in a Node.js environment. |
 |  [@firebase/remote-config](./remote-config.md#remote-config_package) | The Firebase Remote Config Web SDK. This SDK does not work in a Node.js environment. |
 |  [@firebase/storage](./storage.md#storage_package) | Cloud Storage for Firebase |
-|  [@firebase/vertexai](./vertexai.md#vertexai_package) | The Vertex AI in Firebase Web SDK. |
+|  [@firebase/vertexai](./vertexai.md#vertexai_package) |  |
 

docs-devsite/vertexai.genai.md

@@ -0,0 +1,64 @@
+Project: /docs/reference/js/_project.yaml
+Book: /docs/reference/_book.yaml
+page_type: reference
+
+{% comment %}
+DO NOT EDIT THIS FILE!
+This is generated by the JS SDK team, and any local changes will be
+overwritten. Changes should be made in the source code at
+https://github.com/firebase/firebase-js-sdk
+{% endcomment %}
+
+# GenAI interface
+An instance of the Firebase GenAI SDK.
+
+Do not create this instance directly. Instead, use [getGenAI()](./vertexai.md#getgenai_65c48ee)<!-- -->.
+
+<b>Signature:</b>
+
+```typescript
+export interface GenAI 
+```
+
+## Properties
+
+|  Property | Type | Description |
+|  --- | --- | --- |
+|  [app](./vertexai.genai.md#genaiapp) | [FirebaseApp](./app.firebaseapp.md#firebaseapp_interface) | The [FirebaseApp](./app.firebaseapp.md#firebaseapp_interface) this [GenAI](./vertexai.genai.md#genai_interface) instance is associated with. |
+|  [backend](./vertexai.genai.md#genaibackend) | [Backend](./vertexai.md#backend) | A [Backend](./vertexai.md#backend) instance that specifies the backend configuration. |
+|  [location](./vertexai.genai.md#genailocation) | string | The location configured for this GenAI service instance, relevant for Vertex AI backends. |
+
+## GenAI.app
+
+The [FirebaseApp](./app.firebaseapp.md#firebaseapp_interface) this [GenAI](./vertexai.genai.md#genai_interface) instance is associated with.
+
+<b>Signature:</b>
+
+```typescript
+app: FirebaseApp;
+```
+
+## GenAI.backend
+
+A [Backend](./vertexai.md#backend) instance that specifies the backend configuration.
+
+<b>Signature:</b>
+
+```typescript
+backend: Backend;
+```
+
+## GenAI.location
+
+> Warning: This API is now obsolete.
+> 
+> use `GenAI.backend.location` instead.
+> 
+
+The location configured for this GenAI service instance, relevant for Vertex AI backends.
+
+<b>Signature:</b>
+
+```typescript
+location: string;
+```

docs-devsite/vertexai.vertexaierror.md renamed to docs-devsite/vertexai.genaierror.md

@@ -9,56 +9,56 @@ overwritten. Changes should be made in the source code at
 https://github.com/firebase/firebase-js-sdk
 {% endcomment %}
 
-# VertexAIError class
+# GenAIError class
 Error class for the Vertex AI in Firebase SDK.
 
 <b>Signature:</b>
 
 ```typescript
-export declare class VertexAIError extends FirebaseError 
+export declare class GenAIError extends FirebaseError 
 ```
 <b>Extends:</b> [FirebaseError](./util.firebaseerror.md#firebaseerror_class)
 
 ## Constructors
 
 |  Constructor | Modifiers | Description |
 |  --- | --- | --- |
-|  [(constructor)(code, message, customErrorData)](./vertexai.vertexaierror.md#vertexaierrorconstructor) |  | Constructs a new instance of the <code>VertexAIError</code> class. |
+|  [(constructor)(code, message, customErrorData)](./vertexai.genaierror.md#genaierrorconstructor) |  | Constructs a new instance of the <code>GenAIError</code> class. |
 
 ## Properties
 
 |  Property | Modifiers | Type | Description |
 |  --- | --- | --- | --- |
-|  [code](./vertexai.vertexaierror.md#vertexaierrorcode) |  | [VertexAIErrorCode](./vertexai.md#vertexaierrorcode) |  |
-|  [customErrorData](./vertexai.vertexaierror.md#vertexaierrorcustomerrordata) |  | [CustomErrorData](./vertexai.customerrordata.md#customerrordata_interface) \| undefined |  |
+|  [code](./vertexai.genaierror.md#genaierrorcode) |  | [GenAIErrorCode](./vertexai.md#genaierrorcode) |  |
+|  [customErrorData](./vertexai.genaierror.md#genaierrorcustomerrordata) |  | [CustomErrorData](./vertexai.customerrordata.md#customerrordata_interface) \| undefined |  |
 
-## VertexAIError.(constructor)
+## GenAIError.(constructor)
 
-Constructs a new instance of the `VertexAIError` class.
+Constructs a new instance of the `GenAIError` class.
 
 <b>Signature:</b>
 
 ```typescript
-constructor(code: VertexAIErrorCode, message: string, customErrorData?: CustomErrorData | undefined);
+constructor(code: GenAIErrorCode, message: string, customErrorData?: CustomErrorData | undefined);
 ```
 
 #### Parameters
 
 |  Parameter | Type | Description |
 |  --- | --- | --- |
-|  code | [VertexAIErrorCode](./vertexai.md#vertexaierrorcode) | The error code from <code>[VertexAIErrorCode](./vertexai.md#vertexaierrorcode)</code>. |
+|  code | [GenAIErrorCode](./vertexai.md#genaierrorcode) | The error code from <code>[GenAIErrorCode](./vertexai.md#genaierrorcode)</code>. |
 |  message | string | A human-readable message describing the error. |
 |  customErrorData | [CustomErrorData](./vertexai.customerrordata.md#customerrordata_interface) \| undefined | Optional error data. |
 
-## VertexAIError.code
+## GenAIError.code
 
 <b>Signature:</b>
 
 ```typescript
-readonly code: VertexAIErrorCode;
+readonly code: GenAIErrorCode;
 ```
 
-## VertexAIError.customErrorData
+## GenAIError.customErrorData
 
 <b>Signature:</b>