@firebase/ai 2.2.1 → 2.3.0-canary.cb3bdd812

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
@@ -0,0 +1,19 @@
1
+ /**
2
+ * @license
3
+ * Copyright 2025 Google LLC
4
+ *
5
+ * Licensed under the Apache License, Version 2.0 (the "License");
6
+ * you may not use this file except in compliance with the License.
7
+ * You may obtain a copy of the License at
8
+ *
9
+ * http://www.apache.org/licenses/LICENSE-2.0
10
+ *
11
+ * Unless required by applicable law or agreed to in writing, software
12
+ * distributed under the License is distributed on an "AS IS" BASIS,
13
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14
+ * See the License for the specific language governing permissions and
15
+ * limitations under the License.
16
+ */
17
+ import { ComponentContainer, InstanceFactoryOptions } from '@firebase/component';
18
+ import { AIService } from './service';
19
+ export declare function factory(container: ComponentContainer, { instanceIdentifier }: InstanceFactoryOptions): AIService;
@@ -0,0 +1,28 @@
1
+ /**
2
+ * @license
3
+ * Copyright 2025 Google LLC
4
+ *
5
+ * Licensed under the Apache License, Version 2.0 (the "License");
6
+ * you may not use this file except in compliance with the License.
7
+ * You may obtain a copy of the License at
8
+ *
9
+ * http://www.apache.org/licenses/LICENSE-2.0
10
+ *
11
+ * Unless required by applicable law or agreed to in writing, software
12
+ * distributed under the License is distributed on an "AS IS" BASIS,
13
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14
+ * See the License for the specific language governing permissions and
15
+ * limitations under the License.
16
+ */
17
+ import { GenerateContentRequest, ChromeAdapter } from '../types';
18
+ /**
19
+ * Dispatches a request to the appropriate backend (on-device or in-cloud)
20
+ * based on the inference mode.
21
+ *
22
+ * @param request - The request to be sent.
23
+ * @param chromeAdapter - The on-device model adapter.
24
+ * @param onDeviceCall - The function to call for on-device inference.
25
+ * @param inCloudCall - The function to call for in-cloud inference.
26
+ * @returns The response from the backend.
27
+ */
28
+ export declare function callCloudOrDevice<Response>(request: GenerateContentRequest, chromeAdapter: ChromeAdapter | undefined, onDeviceCall: () => Promise<Response>, inCloudCall: () => Promise<Response>): Promise<Response>;
@@ -16,13 +16,13 @@
16
16
  */
17
17
  import { CountTokensRequest, GenerateContentRequest } from './requests';
18
18
  /**
19
- * <b>(EXPERIMENTAL)</b> Defines an inference "backend" that uses Chrome's on-device model,
19
+ * Defines an inference "backend" that uses Chrome's on-device model,
20
20
  * and encapsulates logic for detecting when on-device inference is
21
21
  * possible.
22
22
  *
23
23
  * These methods should not be called directly by the user.
24
24
  *
25
- * @public
25
+ * @beta
26
26
  */
27
27
  export interface ChromeAdapter {
28
28
  /**
@@ -14,7 +14,7 @@
14
14
  * See the License for the specific language governing permissions and
15
15
  * limitations under the License.
16
16
  */
17
- import { Role } from './enums';
17
+ import { Language, Outcome, Role } from './enums';
18
18
  /**
19
19
  * Content type for both prompts and response candidates.
20
20
  * @public
@@ -28,7 +28,7 @@ export interface Content {
28
28
  * part types.
29
29
  * @public
30
30
  */
31
- export type Part = TextPart | InlineDataPart | FunctionCallPart | FunctionResponsePart | FileDataPart;
31
+ export type Part = TextPart | InlineDataPart | FunctionCallPart | FunctionResponsePart | FileDataPart | ExecutableCodePart | CodeExecutionResultPart;
32
32
  /**
33
33
  * Content part interface if the part represents a text string.
34
34
  * @public
@@ -43,6 +43,8 @@ export interface TextPart {
43
43
  * @internal
44
44
  */
45
45
  thoughtSignature?: string;
46
+ executableCode?: never;
47
+ codeExecutionResult?: never;
46
48
  }
47
49
  /**
48
50
  * Content part interface if the part represents an image.
@@ -62,6 +64,8 @@ export interface InlineDataPart {
62
64
  * @internal
63
65
  */
64
66
  thoughtSignature?: never;
67
+ executableCode?: never;
68
+ codeExecutionResult?: never;
65
69
  }
66
70
  /**
67
71
  * Describes the input video content.
@@ -93,6 +97,8 @@ export interface FunctionCallPart {
93
97
  * @internal
94
98
  */
95
99
  thoughtSignature?: never;
100
+ executableCode?: never;
101
+ codeExecutionResult?: never;
96
102
  }
97
103
  /**
98
104
  * Content part interface if the part represents {@link FunctionResponse}.
@@ -108,6 +114,8 @@ export interface FunctionResponsePart {
108
114
  * @internal
109
115
  */
110
116
  thoughtSignature?: never;
117
+ executableCode?: never;
118
+ codeExecutionResult?: never;
111
119
  }
112
120
  /**
113
121
  * Content part interface if the part represents {@link FileData}
@@ -124,6 +132,77 @@ export interface FileDataPart {
124
132
  * @internal
125
133
  */
126
134
  thoughtSignature?: never;
135
+ executableCode?: never;
136
+ codeExecutionResult?: never;
137
+ }
138
+ /**
139
+ * Represents the code that is executed by the model.
140
+ *
141
+ * @public
142
+ */
143
+ export interface ExecutableCodePart {
144
+ text?: never;
145
+ inlineData?: never;
146
+ functionCall?: never;
147
+ functionResponse?: never;
148
+ fileData: never;
149
+ thought?: never;
150
+ /**
151
+ * @internal
152
+ */
153
+ thoughtSignature?: never;
154
+ executableCode?: ExecutableCode;
155
+ codeExecutionResult?: never;
156
+ }
157
+ /**
158
+ * Represents the code execution result from the model.
159
+ *
160
+ * @public
161
+ */
162
+ export interface CodeExecutionResultPart {
163
+ text?: never;
164
+ inlineData?: never;
165
+ functionCall?: never;
166
+ functionResponse?: never;
167
+ fileData: never;
168
+ thought?: never;
169
+ /**
170
+ * @internal
171
+ */
172
+ thoughtSignature?: never;
173
+ executableCode?: never;
174
+ codeExecutionResult?: CodeExecutionResult;
175
+ }
176
+ /**
177
+ * An interface for executable code returned by the model.
178
+ *
179
+ * @public
180
+ */
181
+ export interface ExecutableCode {
182
+ /**
183
+ * The programming language of the code.
184
+ */
185
+ language?: Language;
186
+ /**
187
+ * The source code to be executed.
188
+ */
189
+ code?: string;
190
+ }
191
+ /**
192
+ * The results of code execution run by the model.
193
+ *
194
+ * @public
195
+ */
196
+ export interface CodeExecutionResult {
197
+ /**
198
+ * The result of the code execution.
199
+ */
200
+ outcome?: Outcome;
201
+ /**
202
+ * The output from the code execution, or an error message
203
+ * if it failed.
204
+ */
205
+ output?: string;
127
206
  }
128
207
  /**
129
208
  * A predicted {@link FunctionCall} returned from the model
@@ -317,18 +317,67 @@ export declare const ResponseModality: {
317
317
  */
318
318
  export type ResponseModality = (typeof ResponseModality)[keyof typeof ResponseModality];
319
319
  /**
320
- * <b>(EXPERIMENTAL)</b>
321
320
  * Determines whether inference happens on-device or in-cloud.
322
- * @public
321
+ *
322
+ * @remarks
323
+ * <b>PREFER_ON_DEVICE:</b> Attempt to make inference calls using an
324
+ * on-device model. If on-device inference is not available, the SDK
325
+ * will fall back to using a cloud-hosted model.
326
+ * <br/>
327
+ * <b>ONLY_ON_DEVICE:</b> Only attempt to make inference calls using an
328
+ * on-device model. The SDK will not fall back to a cloud-hosted model.
329
+ * If on-device inference is not available, inference methods will throw.
330
+ * <br/>
331
+ * <b>ONLY_IN_CLOUD:</b> Only attempt to make inference calls using a
332
+ * cloud-hosted model. The SDK will not fall back to an on-device model.
333
+ * <br/>
334
+ * <b>PREFER_IN_CLOUD:</b> Attempt to make inference calls to a
335
+ * cloud-hosted model. If not available, the SDK will fall back to an
336
+ * on-device model.
337
+ *
338
+ * @beta
323
339
  */
324
340
  export declare const InferenceMode: {
325
341
  readonly PREFER_ON_DEVICE: "prefer_on_device";
326
342
  readonly ONLY_ON_DEVICE: "only_on_device";
327
343
  readonly ONLY_IN_CLOUD: "only_in_cloud";
344
+ readonly PREFER_IN_CLOUD: "prefer_in_cloud";
328
345
  };
329
346
  /**
330
- * <b>(EXPERIMENTAL)</b>
331
347
  * Determines whether inference happens on-device or in-cloud.
332
- * @public
348
+ *
349
+ * @beta
333
350
  */
334
351
  export type InferenceMode = (typeof InferenceMode)[keyof typeof InferenceMode];
352
+ /**
353
+ * Represents the result of the code execution.
354
+ *
355
+ * @public
356
+ */
357
+ export declare const Outcome: {
358
+ UNSPECIFIED: string;
359
+ OK: string;
360
+ FAILED: string;
361
+ DEADLINE_EXCEEDED: string;
362
+ };
363
+ /**
364
+ * Represents the result of the code execution.
365
+ *
366
+ * @public
367
+ */
368
+ export type Outcome = (typeof Outcome)[keyof typeof Outcome];
369
+ /**
370
+ * The programming language of the code.
371
+ *
372
+ * @public
373
+ */
374
+ export declare const Language: {
375
+ UNSPECIFIED: string;
376
+ PYTHON: string;
377
+ };
378
+ /**
379
+ * The programming language of the code.
380
+ *
381
+ * @public
382
+ */
383
+ export type Language = (typeof Language)[keyof typeof Language];
@@ -39,9 +39,8 @@ export declare enum Availability {
39
39
  'AVAILABLE' = "available"
40
40
  }
41
41
  /**
42
- * <b>(EXPERIMENTAL)</b>
43
42
  * Configures the creation of an on-device language model session.
44
- * @public
43
+ * @beta
45
44
  */
46
45
  export interface LanguageModelCreateCoreOptions {
47
46
  topK?: number;
@@ -49,69 +48,60 @@ export interface LanguageModelCreateCoreOptions {
49
48
  expectedInputs?: LanguageModelExpected[];
50
49
  }
51
50
  /**
52
- * <b>(EXPERIMENTAL)</b>
53
51
  * Configures the creation of an on-device language model session.
54
- * @public
52
+ * @beta
55
53
  */
56
54
  export interface LanguageModelCreateOptions extends LanguageModelCreateCoreOptions {
57
55
  signal?: AbortSignal;
58
56
  initialPrompts?: LanguageModelMessage[];
59
57
  }
60
58
  /**
61
- * <b>(EXPERIMENTAL)</b>
62
59
  * Options for an on-device language model prompt.
63
- * @public
60
+ * @beta
64
61
  */
65
62
  export interface LanguageModelPromptOptions {
66
63
  responseConstraint?: object;
67
64
  }
68
65
  /**
69
- * <b>(EXPERIMENTAL)</b>
70
66
  * Options for the expected inputs for an on-device language model.
71
- * @public
67
+ * @beta
72
68
  */ export interface LanguageModelExpected {
73
69
  type: LanguageModelMessageType;
74
70
  languages?: string[];
75
71
  }
76
72
  /**
77
- * <b>(EXPERIMENTAL)</b>
78
73
  * An on-device language model prompt.
79
- * @public
74
+ * @beta
80
75
  */
81
76
  export type LanguageModelPrompt = LanguageModelMessage[];
82
77
  /**
83
- * <b>(EXPERIMENTAL)</b>
84
78
  * An on-device language model message.
85
- * @public
79
+ * @beta
86
80
  */
87
81
  export interface LanguageModelMessage {
88
82
  role: LanguageModelMessageRole;
89
83
  content: LanguageModelMessageContent[];
90
84
  }
91
85
  /**
92
- * <b>(EXPERIMENTAL)</b>
93
86
  * An on-device language model content object.
94
- * @public
87
+ * @beta
95
88
  */
96
89
  export interface LanguageModelMessageContent {
97
90
  type: LanguageModelMessageType;
98
91
  value: LanguageModelMessageContentValue;
99
92
  }
100
93
  /**
101
- * <b>(EXPERIMENTAL)</b>
102
94
  * Allowable roles for on-device language model usage.
103
- * @public
95
+ * @beta
104
96
  */
105
97
  export type LanguageModelMessageRole = 'system' | 'user' | 'assistant';
106
98
  /**
107
- * <b>(EXPERIMENTAL)</b>
108
99
  * Allowable types for on-device language model messages.
109
- * @public
100
+ * @beta
110
101
  */
111
102
  export type LanguageModelMessageType = 'text' | 'image' | 'audio';
112
103
  /**
113
- * <b>(EXPERIMENTAL)</b>
114
104
  * Content formats that can be provided as on-device message content.
115
- * @public
105
+ * @beta
116
106
  */
117
107
  export type LanguageModelMessageContentValue = ImageBitmapSource | AudioBuffer | BufferSource | string;
@@ -218,7 +218,7 @@ export interface RequestOptions {
218
218
  * Defines a tool that model can call to access external knowledge.
219
219
  * @public
220
220
  */
221
- export type Tool = FunctionDeclarationsTool | GoogleSearchTool;
221
+ export type Tool = FunctionDeclarationsTool | GoogleSearchTool | CodeExecutionTool;
222
222
  /**
223
223
  * Structured representation of a function declaration as defined by the
224
224
  * {@link https://spec.openapis.org/oas/v3.0.3 | OpenAPI 3.0 specification}.
@@ -262,8 +262,6 @@ export interface GoogleSearchTool {
262
262
  /**
263
263
  * Specifies the Google Search configuration.
264
264
  * Currently, this is an empty object, but it's reserved for future configuration options.
265
- * Specifies the Google Search configuration. Currently, this is an empty object, but it's
266
- * reserved for future configuration options.
267
265
  *
268
266
  * When using this feature, you are required to comply with the "Grounding with Google Search"
269
267
  * usage requirements for your chosen API provider: {@link https://ai.google.dev/gemini-api/terms#grounding-with-google-search | Gemini Developer API}
@@ -272,6 +270,18 @@ export interface GoogleSearchTool {
272
270
  */
273
271
  googleSearch: GoogleSearch;
274
272
  }
273
+ /**
274
+ * A tool that enables the model to use code execution.
275
+ *
276
+ * @public
277
+ */
278
+ export interface CodeExecutionTool {
279
+ /**
280
+ * Specifies the Google Search configuration.
281
+ * Currently, this is an empty object, but it's reserved for future configuration options.
282
+ */
283
+ codeExecution: {};
284
+ }
275
285
  /**
276
286
  * Specifies the Google Search configuration.
277
287
  *
@@ -315,19 +325,17 @@ export interface FunctionCallingConfig {
315
325
  allowedFunctionNames?: string[];
316
326
  }
317
327
  /**
318
- * <b>(EXPERIMENTAL)</b>
319
328
  * Encapsulates configuration for on-device inference.
320
329
  *
321
- * @public
330
+ * @beta
322
331
  */
323
332
  export interface OnDeviceParams {
324
333
  createOptions?: LanguageModelCreateOptions;
325
334
  promptOptions?: LanguageModelPromptOptions;
326
335
  }
327
336
  /**
328
- * <b>(EXPERIMENTAL)</b>
329
337
  * Configures hybrid inference.
330
- * @public
338
+ * @beta
331
339
  */
332
340
  export interface HybridParams {
333
341
  /**
package/package.json CHANGED
@@ -1,6 +1,6 @@
1
1
  {
2
2
  "name": "@firebase/ai",
3
- "version": "2.2.1",
3
+ "version": "2.3.0-canary.cb3bdd812",
4
4
  "description": "The Firebase AI SDK",
5
5
  "author": "Firebase <firebase-support@google.com> (https://firebase.google.com/)",
6
6
  "engines": {
@@ -35,7 +35,7 @@
35
35
  "dev": "rollup -c -w",
36
36
  "update-responses": "../../scripts/update_vertexai_responses.sh",
37
37
  "testsetup": "yarn update-responses && yarn ts-node ./test-utils/convert-mocks.ts",
38
- "test": "run-p --npm-path npm lint test:browser",
38
+ "test": "run-p --npm-path npm lint type-check test:browser",
39
39
  "test:ci": "yarn testsetup && node ../../scripts/run_tests_in_ci.js -s test",
40
40
  "test:skip-clone": "karma start",
41
41
  "test:browser": "yarn testsetup && karma start",
@@ -44,22 +44,23 @@
44
44
  "test:integration:node": "TS_NODE_COMPILER_OPTIONS='{\"module\":\"commonjs\"}' mocha integration/**/*.test.ts --config ../../config/mocharc.node.js",
45
45
  "api-report": "api-extractor run --local --verbose",
46
46
  "typings:public": "node ../../scripts/build/use_typings.js ./dist/ai-public.d.ts",
47
+ "type-check": "yarn tsc --noEmit",
47
48
  "trusted-type-check": "tsec -p tsconfig.json --noEmit"
48
49
  },
49
50
  "peerDependencies": {
50
- "@firebase/app": "0.x",
51
- "@firebase/app-types": "0.x"
51
+ "@firebase/app": "0.14.3-canary.cb3bdd812",
52
+ "@firebase/app-types": "0.9.3-canary.cb3bdd812"
52
53
  },
53
54
  "dependencies": {
54
- "@firebase/app-check-interop-types": "0.3.3",
55
- "@firebase/component": "0.7.0",
56
- "@firebase/logger": "0.5.0",
57
- "@firebase/util": "1.13.0",
55
+ "@firebase/app-check-interop-types": "0.3.3-canary.cb3bdd812",
56
+ "@firebase/component": "0.7.0-canary.cb3bdd812",
57
+ "@firebase/logger": "0.5.0-canary.cb3bdd812",
58
+ "@firebase/util": "1.13.0-canary.cb3bdd812",
58
59
  "tslib": "^2.1.0"
59
60
  },
60
61
  "license": "Apache-2.0",
61
62
  "devDependencies": {
62
- "@firebase/app": "0.14.2",
63
+ "@firebase/app": "0.14.3-canary.cb3bdd812",
63
64
  "@rollup/plugin-json": "6.1.0",
64
65
  "rollup": "2.79.2",
65
66
  "rollup-plugin-replace": "2.2.0",