@firebase/ai 2.1.0-canary.9b63cd60e → 2.1.0-canary.cbef6c6e5

package/dist/src/index.d.ts

@@ -5,9 +5,10 @@
  */
 import { AIService } from './service';
 import { ComponentContainer, InstanceFactoryOptions } from '@firebase/component';
+import { LanguageModel } from './types/language-model';
 declare global {
     interface Window {
-        [key: string]: unknown;
+        LanguageModel: LanguageModel;
     }
 }
 export declare function factory(container: ComponentContainer, { instanceIdentifier }: InstanceFactoryOptions): AIService;

package/dist/src/methods/chrome-adapter.d.ts

@@ -23,9 +23,9 @@ import { LanguageModel } from '../types/language-model';
  * possible.
  */
 export declare class ChromeAdapterImpl implements ChromeAdapter {
-    private languageModelProvider;
-    private mode;
-    private onDeviceParams;
+    languageModelProvider: LanguageModel;
+    mode: InferenceMode;
+    onDeviceParams: OnDeviceParams;
     static SUPPORTED_MIME_TYPES: string[];
     private isDownloading;
     private downloadPromise;
@@ -118,3 +118,7 @@ export declare class ChromeAdapterImpl implements ChromeAdapter {
      */
     private static toStreamResponse;
 }
+/**
+ * Creates a ChromeAdapterImpl on demand.
+ */
+export declare function chromeAdapterFactory(mode: InferenceMode, window?: Window, params?: OnDeviceParams): ChromeAdapterImpl | undefined;

package/dist/src/service.d.ts

@@ -15,19 +15,21 @@
  * limitations under the License.
  */
 import { FirebaseApp, _FirebaseService } from '@firebase/app';
-import { AI, AIOptions } from './public-types';
+import { AI, AIOptions, InferenceMode, OnDeviceParams } from './public-types';
 import { AppCheckInternalComponentName, FirebaseAppCheckInternal } from '@firebase/app-check-interop-types';
 import { Provider } from '@firebase/component';
 import { FirebaseAuthInternal, FirebaseAuthInternalName } from '@firebase/auth-interop-types';
 import { Backend } from './backend';
+import { ChromeAdapterImpl } from './methods/chrome-adapter';
 export declare class AIService implements AI, _FirebaseService {
     app: FirebaseApp;
     backend: Backend;
+    chromeAdapterFactory?: ((mode: InferenceMode, window?: Window, params?: OnDeviceParams) => ChromeAdapterImpl | undefined) | undefined;
     auth: FirebaseAuthInternal | null;
     appCheck: FirebaseAppCheckInternal | null;
     _options?: Omit<AIOptions, 'backend'>;
     location: string;
-    constructor(app: FirebaseApp, backend: Backend, authProvider?: Provider<FirebaseAuthInternalName>, appCheckProvider?: Provider<AppCheckInternalComponentName>);
+    constructor(app: FirebaseApp, backend: Backend, authProvider?: Provider<FirebaseAuthInternalName>, appCheckProvider?: Provider<AppCheckInternalComponentName>, chromeAdapterFactory?: ((mode: InferenceMode, window?: Window, params?: OnDeviceParams) => ChromeAdapterImpl | undefined) | undefined);
     _delete(): Promise<void>;
     set options(optionsToSet: AIOptions);
     get options(): AIOptions | undefined;

package/dist/src/types/imagen/internal.d.ts

@@ -59,6 +59,14 @@ export interface ImagenResponseInternal {
          * The reason why the image was filtered.
          */
         raiFilteredReason?: string;
+        /**
+         * The safety attributes.
+         *
+         * This type is currently unused in the SDK. It is sent back because our requests set
+         * `includeSafetyAttributes`. This property is currently only used to avoid throwing an error
+         * when encountering this unsupported prediction type.
+         */
+        safetyAttributes?: unknown;
     }>;
 }
 /**
@@ -81,6 +89,7 @@ export interface ImagenResponseInternal {
  *     "personGeneration": "allow_all",
  *     "sampleCount": 2,
  *     "includeRaiReason": true,
+ *     "includeSafetyAttributes": true,
  *     "aspectRatio": "9:16"
  *   }
  * }
@@ -108,6 +117,7 @@ export interface PredictRequestBody {
         safetyFilterLevel?: string;
         personGeneration?: string;
         includeRaiReason: boolean;
+        includeSafetyAttributes: boolean;
     };
 }
 /**

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@firebase/ai",
-  "version": "2.1.0-canary.9b63cd60e",
+  "version": "2.1.0-canary.cbef6c6e5",
   "description": "The Firebase AI SDK",
   "author": "Firebase <firebase-support@google.com> (https://firebase.google.com/)",
   "engines": {
@@ -39,25 +39,27 @@
     "test:ci": "yarn testsetup && node ../../scripts/run_tests_in_ci.js -s test",
     "test:skip-clone": "karma start",
     "test:browser": "yarn testsetup && karma start",
+    "test:node": "TS_NODE_COMPILER_OPTIONS='{\"module\":\"commonjs\"}' mocha --require ts-node/register --require src/index.node.ts 'src/**/!(*-browser)*.test.ts' --config ../../config/mocharc.node.js",
     "test:integration": "karma start --integration",
+    "test:integration:node": "TS_NODE_COMPILER_OPTIONS='{\"module\":\"commonjs\"}' mocha integration/**/*.test.ts --config ../../config/mocharc.node.js",
     "api-report": "api-extractor run --local --verbose",
     "typings:public": "node ../../scripts/build/use_typings.js ./dist/ai-public.d.ts",
     "trusted-type-check": "tsec -p tsconfig.json --noEmit"
   },
   "peerDependencies": {
-    "@firebase/app": "0.14.1-canary.9b63cd60e",
-    "@firebase/app-types": "0.9.3-canary.9b63cd60e"
+    "@firebase/app": "0.14.1-canary.cbef6c6e5",
+    "@firebase/app-types": "0.9.3-canary.cbef6c6e5"
   },
   "dependencies": {
-    "@firebase/app-check-interop-types": "0.3.3-canary.9b63cd60e",
-    "@firebase/component": "0.7.0-canary.9b63cd60e",
-    "@firebase/logger": "0.5.0-canary.9b63cd60e",
-    "@firebase/util": "1.13.0-canary.9b63cd60e",
+    "@firebase/app-check-interop-types": "0.3.3-canary.cbef6c6e5",
+    "@firebase/component": "0.7.0-canary.cbef6c6e5",
+    "@firebase/logger": "0.5.0-canary.cbef6c6e5",
+    "@firebase/util": "1.13.0-canary.cbef6c6e5",
     "tslib": "^2.1.0"
   },
   "license": "Apache-2.0",
   "devDependencies": {
-    "@firebase/app": "0.14.1-canary.9b63cd60e",
+    "@firebase/app": "0.14.1-canary.cbef6c6e5",
     "@rollup/plugin-json": "6.1.0",
     "rollup": "2.79.2",
     "rollup-plugin-replace": "2.2.0",