@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.
- package/dist/ai-public.d.ts +162 -32
- package/dist/ai.d.ts +168 -32
- package/dist/esm/index.esm.js +618 -508
- package/dist/esm/index.esm.js.map +1 -1
- package/dist/esm/src/factory-browser.d.ts +19 -0
- package/dist/esm/src/requests/hybrid-helpers.d.ts +28 -0
- package/dist/esm/src/types/chrome-adapter.d.ts +2 -2
- package/dist/esm/src/types/content.d.ts +81 -2
- package/dist/esm/src/types/enums.d.ts +53 -4
- package/dist/esm/src/types/language-model.d.ts +10 -20
- package/dist/esm/src/types/requests.d.ts +15 -7
- package/dist/index.cjs.js +619 -507
- package/dist/index.cjs.js.map +1 -1
- package/dist/index.node.cjs.js +116 -21
- package/dist/index.node.cjs.js.map +1 -1
- package/dist/index.node.mjs +115 -22
- package/dist/index.node.mjs.map +1 -1
- package/dist/src/factory-browser.d.ts +19 -0
- package/dist/src/requests/hybrid-helpers.d.ts +28 -0
- package/dist/src/types/chrome-adapter.d.ts +2 -2
- package/dist/src/types/content.d.ts +81 -2
- package/dist/src/types/enums.d.ts +53 -4
- package/dist/src/types/language-model.d.ts +10 -20
- package/dist/src/types/requests.d.ts +15 -7
- package/package.json +10 -9
|
@@ -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
|
-
*
|
|
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
|
-
* @
|
|
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
|
-
*
|
|
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
|
-
*
|
|
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
|
-
* @
|
|
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
|
-
* @
|
|
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
|
-
* @
|
|
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
|
-
* @
|
|
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
|
-
* @
|
|
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
|
-
* @
|
|
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
|
-
* @
|
|
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
|
-
* @
|
|
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
|
-
* @
|
|
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
|
-
* @
|
|
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
|
-
* @
|
|
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
|
-
* @
|
|
338
|
+
* @beta
|
|
331
339
|
*/
|
|
332
340
|
export interface HybridParams {
|
|
333
341
|
/**
|