@firebase/ai 2.4.0-canary.b7e18d0ff → 2.4.0-canary.c8263c471

package/dist/src/requests/hybrid-helpers.d.ts

@@ -14,7 +14,11 @@
  * See the License for the specific language governing permissions and
  * limitations under the License.
  */
-import { GenerateContentRequest, ChromeAdapter } from '../types';
+import { GenerateContentRequest, ChromeAdapter, InferenceSource } from '../types';
+interface CallResult<Response> {
+    response: Response;
+    inferenceSource: InferenceSource;
+}
 /**
  * Dispatches a request to the appropriate backend (on-device or in-cloud)
  * based on the inference mode.
@@ -25,4 +29,5 @@ import { GenerateContentRequest, ChromeAdapter } from '../types';
  * @param inCloudCall - The function to call for in-cloud inference.
  * @returns The response from the backend.
  */
-export declare function callCloudOrDevice<Response>(request: GenerateContentRequest, chromeAdapter: ChromeAdapter | undefined, onDeviceCall: () => Promise<Response>, inCloudCall: () => Promise<Response>): Promise<Response>;
+export declare function callCloudOrDevice<Response>(request: GenerateContentRequest, chromeAdapter: ChromeAdapter | undefined, onDeviceCall: () => Promise<Response>, inCloudCall: () => Promise<Response>): Promise<CallResult<Response>>;
+export {};

package/dist/src/requests/response-helpers.d.ts

@@ -14,12 +14,12 @@
  * See the License for the specific language governing permissions and
  * limitations under the License.
  */
-import { EnhancedGenerateContentResponse, FunctionCall, GenerateContentResponse, ImagenGCSImage, ImagenInlineImage, InlineDataPart, Part } from '../types';
+import { EnhancedGenerateContentResponse, FunctionCall, GenerateContentResponse, ImagenGCSImage, ImagenInlineImage, InlineDataPart, Part, InferenceSource } from '../types';
 /**
  * Creates an EnhancedGenerateContentResponse object that has helper functions and
  * other modifications that improve usability.
  */
-export declare function createEnhancedContentResponse(response: GenerateContentResponse): EnhancedGenerateContentResponse;
+export declare function createEnhancedContentResponse(response: GenerateContentResponse, inferenceSource?: InferenceSource): EnhancedGenerateContentResponse;
 /**
  * Adds convenience helper methods to a response object, including stream
  * chunks (as long as each chunk is a complete GenerateContentResponse JSON).

package/dist/src/requests/stream-reader.d.ts

@@ -16,6 +16,7 @@
  */
 import { GenerateContentResponse, GenerateContentStreamResult } from '../types';
 import { ApiSettings } from '../types/internal';
+import { InferenceSource } from '../public-types';
 /**
  * Process a response.body stream from the backend and return an
  * iterator that provides one complete GenerateContentResponse at a time
@@ -24,7 +25,7 @@ import { ApiSettings } from '../types/internal';
  *
  * @param response - Response from a fetch call
  */
-export declare function processStream(response: Response, apiSettings: ApiSettings): GenerateContentStreamResult;
+export declare function processStream(response: Response, apiSettings: ApiSettings, inferenceSource?: InferenceSource): GenerateContentStreamResult;
 /**
  * Reads a raw stream from the fetch response and join incomplete
  * chunks, returning a new stream that provides a single complete

package/dist/src/types/enums.d.ts

@@ -349,6 +349,21 @@ export declare const InferenceMode: {
  * @beta
  */
 export type InferenceMode = (typeof InferenceMode)[keyof typeof InferenceMode];
+/**
+ * Indicates whether inference happened on-device or in-cloud.
+ *
+ * @beta
+ */
+export declare const InferenceSource: {
+    readonly ON_DEVICE: "on_device";
+    readonly IN_CLOUD: "in_cloud";
+};
+/**
+ * Indicates whether inference happened on-device or in-cloud.
+ *
+ * @beta
+ */
+export type InferenceSource = (typeof InferenceSource)[keyof typeof InferenceSource];
 /**
  * Represents the result of the code execution.
  *

package/dist/src/types/responses.d.ts

@@ -15,7 +15,7 @@
  * limitations under the License.
  */
 import { Content, FunctionCall, InlineDataPart } from './content';
-import { BlockReason, FinishReason, HarmCategory, HarmProbability, HarmSeverity, Modality } from './enums';
+import { BlockReason, FinishReason, HarmCategory, HarmProbability, HarmSeverity, InferenceSource, Modality } from './enums';
 /**
  * Result object returned from {@link GenerativeModel.generateContent} call.
  *
@@ -76,6 +76,12 @@ export interface EnhancedGenerateContentResponse extends GenerateContentResponse
      * set to `true`.
      */
     thoughtSummary: () => string | undefined;
+    /**
+     * Indicates whether inference happened on-device or in-cloud.
+     *
+     * @beta
+     */
+    inferenceSource?: InferenceSource;
 }
 /**
  * Individual response from {@link GenerativeModel.generateContent} and

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@firebase/ai",
-  "version": "2.4.0-canary.b7e18d0ff",
+  "version": "2.4.0-canary.c8263c471",
   "description": "The Firebase AI SDK",
   "author": "Firebase <firebase-support@google.com> (https://firebase.google.com/)",
   "engines": {
@@ -48,19 +48,19 @@
     "trusted-type-check": "tsec -p tsconfig.json --noEmit"
   },
   "peerDependencies": {
-    "@firebase/app": "0.14.4-canary.b7e18d0ff",
-    "@firebase/app-types": "0.9.3-canary.b7e18d0ff"
+    "@firebase/app": "0.14.4-canary.c8263c471",
+    "@firebase/app-types": "0.9.3-canary.c8263c471"
   },
   "dependencies": {
-    "@firebase/app-check-interop-types": "0.3.3-canary.b7e18d0ff",
-    "@firebase/component": "0.7.0-canary.b7e18d0ff",
-    "@firebase/logger": "0.5.0-canary.b7e18d0ff",
-    "@firebase/util": "1.13.0-canary.b7e18d0ff",
+    "@firebase/app-check-interop-types": "0.3.3-canary.c8263c471",
+    "@firebase/component": "0.7.0-canary.c8263c471",
+    "@firebase/logger": "0.5.0-canary.c8263c471",
+    "@firebase/util": "1.13.0-canary.c8263c471",
     "tslib": "^2.1.0"
   },
   "license": "Apache-2.0",
   "devDependencies": {
-    "@firebase/app": "0.14.4-canary.b7e18d0ff",
+    "@firebase/app": "0.14.4-canary.c8263c471",
     "@rollup/plugin-json": "6.1.0",
     "rollup": "2.79.2",
     "rollup-plugin-replace": "2.2.0",