@tstdl/base 0.93.167 → 0.93.170

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.
@@ -1,7 +1,6 @@
1
1
  import { vertexAI, type GeminiConfig } from '@genkit-ai/google-genai';
2
2
  import { type Genkit, type GenkitOptions } from 'genkit';
3
- import { type GeminiModelReference } from './multi-region.plugin.js';
4
- import type { VertexAiMultiLocationOptions } from './types.js';
3
+ import type { GeminiModelReference, VertexAiMultiLocationOptions } from './types.js';
5
4
  /**
6
5
  * Options for configuring the AI Module.
7
6
  */
@@ -1,419 +1,6 @@
1
1
  import type { CircuitBreakerProvider } from '../../circuit-breaker/provider.js';
2
2
  import type { Logger } from '../../logger/logger.js';
3
- import type { VertexAiMultiLocationOptions } from './types.js';
4
- export type GeminiModelReference = typeof geminiModelReference;
5
- declare const geminiModelReference: import("genkit").ModelReference<import("zod").ZodObject<{
6
- version: import("zod").ZodOptional<import("zod").ZodString>;
7
- maxOutputTokens: import("zod").ZodOptional<import("zod").ZodNumber>;
8
- topK: import("zod").ZodOptional<import("zod").ZodNumber>;
9
- stopSequences: import("zod").ZodOptional<import("zod").ZodArray<import("zod").ZodString, "many">>;
10
- } & {
11
- apiKey: import("zod").ZodOptional<import("zod").ZodString>;
12
- labels: import("zod").ZodOptional<import("zod").ZodRecord<import("zod").ZodString, import("zod").ZodString>>;
13
- temperature: import("zod").ZodOptional<import("zod").ZodNumber>;
14
- topP: import("zod").ZodOptional<import("zod").ZodNumber>;
15
- location: import("zod").ZodOptional<import("zod").ZodString>;
16
- safetySettings: import("zod").ZodOptional<import("zod").ZodArray<import("zod").ZodObject<{
17
- category: import("zod").ZodEnum<["HARM_CATEGORY_UNSPECIFIED", "HARM_CATEGORY_HATE_SPEECH", "HARM_CATEGORY_DANGEROUS_CONTENT", "HARM_CATEGORY_HARASSMENT", "HARM_CATEGORY_SEXUALLY_EXPLICIT"]>;
18
- threshold: import("zod").ZodEnum<["BLOCK_LOW_AND_ABOVE", "BLOCK_MEDIUM_AND_ABOVE", "BLOCK_ONLY_HIGH", "BLOCK_NONE"]>;
19
- }, "passthrough", import("zod").ZodTypeAny, import("zod").objectOutputType<{
20
- category: import("zod").ZodEnum<["HARM_CATEGORY_UNSPECIFIED", "HARM_CATEGORY_HATE_SPEECH", "HARM_CATEGORY_DANGEROUS_CONTENT", "HARM_CATEGORY_HARASSMENT", "HARM_CATEGORY_SEXUALLY_EXPLICIT"]>;
21
- threshold: import("zod").ZodEnum<["BLOCK_LOW_AND_ABOVE", "BLOCK_MEDIUM_AND_ABOVE", "BLOCK_ONLY_HIGH", "BLOCK_NONE"]>;
22
- }, import("zod").ZodTypeAny, "passthrough">, import("zod").objectInputType<{
23
- category: import("zod").ZodEnum<["HARM_CATEGORY_UNSPECIFIED", "HARM_CATEGORY_HATE_SPEECH", "HARM_CATEGORY_DANGEROUS_CONTENT", "HARM_CATEGORY_HARASSMENT", "HARM_CATEGORY_SEXUALLY_EXPLICIT"]>;
24
- threshold: import("zod").ZodEnum<["BLOCK_LOW_AND_ABOVE", "BLOCK_MEDIUM_AND_ABOVE", "BLOCK_ONLY_HIGH", "BLOCK_NONE"]>;
25
- }, import("zod").ZodTypeAny, "passthrough">>, "many">>;
26
- vertexRetrieval: import("zod").ZodOptional<import("zod").ZodObject<{
27
- datastore: import("zod").ZodObject<{
28
- projectId: import("zod").ZodOptional<import("zod").ZodString>;
29
- location: import("zod").ZodOptional<import("zod").ZodString>;
30
- dataStoreId: import("zod").ZodString;
31
- }, "passthrough", import("zod").ZodTypeAny, import("zod").objectOutputType<{
32
- projectId: import("zod").ZodOptional<import("zod").ZodString>;
33
- location: import("zod").ZodOptional<import("zod").ZodString>;
34
- dataStoreId: import("zod").ZodString;
35
- }, import("zod").ZodTypeAny, "passthrough">, import("zod").objectInputType<{
36
- projectId: import("zod").ZodOptional<import("zod").ZodString>;
37
- location: import("zod").ZodOptional<import("zod").ZodString>;
38
- dataStoreId: import("zod").ZodString;
39
- }, import("zod").ZodTypeAny, "passthrough">>;
40
- disableAttribution: import("zod").ZodOptional<import("zod").ZodBoolean>;
41
- }, "passthrough", import("zod").ZodTypeAny, import("zod").objectOutputType<{
42
- datastore: import("zod").ZodObject<{
43
- projectId: import("zod").ZodOptional<import("zod").ZodString>;
44
- location: import("zod").ZodOptional<import("zod").ZodString>;
45
- dataStoreId: import("zod").ZodString;
46
- }, "passthrough", import("zod").ZodTypeAny, import("zod").objectOutputType<{
47
- projectId: import("zod").ZodOptional<import("zod").ZodString>;
48
- location: import("zod").ZodOptional<import("zod").ZodString>;
49
- dataStoreId: import("zod").ZodString;
50
- }, import("zod").ZodTypeAny, "passthrough">, import("zod").objectInputType<{
51
- projectId: import("zod").ZodOptional<import("zod").ZodString>;
52
- location: import("zod").ZodOptional<import("zod").ZodString>;
53
- dataStoreId: import("zod").ZodString;
54
- }, import("zod").ZodTypeAny, "passthrough">>;
55
- disableAttribution: import("zod").ZodOptional<import("zod").ZodBoolean>;
56
- }, import("zod").ZodTypeAny, "passthrough">, import("zod").objectInputType<{
57
- datastore: import("zod").ZodObject<{
58
- projectId: import("zod").ZodOptional<import("zod").ZodString>;
59
- location: import("zod").ZodOptional<import("zod").ZodString>;
60
- dataStoreId: import("zod").ZodString;
61
- }, "passthrough", import("zod").ZodTypeAny, import("zod").objectOutputType<{
62
- projectId: import("zod").ZodOptional<import("zod").ZodString>;
63
- location: import("zod").ZodOptional<import("zod").ZodString>;
64
- dataStoreId: import("zod").ZodString;
65
- }, import("zod").ZodTypeAny, "passthrough">, import("zod").objectInputType<{
66
- projectId: import("zod").ZodOptional<import("zod").ZodString>;
67
- location: import("zod").ZodOptional<import("zod").ZodString>;
68
- dataStoreId: import("zod").ZodString;
69
- }, import("zod").ZodTypeAny, "passthrough">>;
70
- disableAttribution: import("zod").ZodOptional<import("zod").ZodBoolean>;
71
- }, import("zod").ZodTypeAny, "passthrough">>>;
72
- googleSearchRetrieval: import("zod").ZodOptional<import("zod").ZodObject<{
73
- disableAttribution: import("zod").ZodOptional<import("zod").ZodBoolean>;
74
- }, "passthrough", import("zod").ZodTypeAny, import("zod").objectOutputType<{
75
- disableAttribution: import("zod").ZodOptional<import("zod").ZodBoolean>;
76
- }, import("zod").ZodTypeAny, "passthrough">, import("zod").objectInputType<{
77
- disableAttribution: import("zod").ZodOptional<import("zod").ZodBoolean>;
78
- }, import("zod").ZodTypeAny, "passthrough">>>;
79
- functionCallingConfig: import("zod").ZodOptional<import("zod").ZodObject<{
80
- mode: import("zod").ZodOptional<import("zod").ZodEnum<["MODE_UNSPECIFIED", "AUTO", "ANY", "NONE"]>>;
81
- allowedFunctionNames: import("zod").ZodOptional<import("zod").ZodArray<import("zod").ZodString, "many">>;
82
- streamFunctionCallArguments: import("zod").ZodOptional<import("zod").ZodBoolean>;
83
- }, "passthrough", import("zod").ZodTypeAny, import("zod").objectOutputType<{
84
- mode: import("zod").ZodOptional<import("zod").ZodEnum<["MODE_UNSPECIFIED", "AUTO", "ANY", "NONE"]>>;
85
- allowedFunctionNames: import("zod").ZodOptional<import("zod").ZodArray<import("zod").ZodString, "many">>;
86
- streamFunctionCallArguments: import("zod").ZodOptional<import("zod").ZodBoolean>;
87
- }, import("zod").ZodTypeAny, "passthrough">, import("zod").objectInputType<{
88
- mode: import("zod").ZodOptional<import("zod").ZodEnum<["MODE_UNSPECIFIED", "AUTO", "ANY", "NONE"]>>;
89
- allowedFunctionNames: import("zod").ZodOptional<import("zod").ZodArray<import("zod").ZodString, "many">>;
90
- streamFunctionCallArguments: import("zod").ZodOptional<import("zod").ZodBoolean>;
91
- }, import("zod").ZodTypeAny, "passthrough">>>;
92
- retrievalConfig: import("zod").ZodOptional<import("zod").ZodObject<{
93
- latLng: import("zod").ZodOptional<import("zod").ZodObject<{
94
- latitude: import("zod").ZodOptional<import("zod").ZodNumber>;
95
- longitude: import("zod").ZodOptional<import("zod").ZodNumber>;
96
- }, "strip", import("zod").ZodTypeAny, {
97
- latitude?: number | undefined;
98
- longitude?: number | undefined;
99
- }, {
100
- latitude?: number | undefined;
101
- longitude?: number | undefined;
102
- }>>;
103
- languageCode: import("zod").ZodOptional<import("zod").ZodString>;
104
- }, "passthrough", import("zod").ZodTypeAny, import("zod").objectOutputType<{
105
- latLng: import("zod").ZodOptional<import("zod").ZodObject<{
106
- latitude: import("zod").ZodOptional<import("zod").ZodNumber>;
107
- longitude: import("zod").ZodOptional<import("zod").ZodNumber>;
108
- }, "strip", import("zod").ZodTypeAny, {
109
- latitude?: number | undefined;
110
- longitude?: number | undefined;
111
- }, {
112
- latitude?: number | undefined;
113
- longitude?: number | undefined;
114
- }>>;
115
- languageCode: import("zod").ZodOptional<import("zod").ZodString>;
116
- }, import("zod").ZodTypeAny, "passthrough">, import("zod").objectInputType<{
117
- latLng: import("zod").ZodOptional<import("zod").ZodObject<{
118
- latitude: import("zod").ZodOptional<import("zod").ZodNumber>;
119
- longitude: import("zod").ZodOptional<import("zod").ZodNumber>;
120
- }, "strip", import("zod").ZodTypeAny, {
121
- latitude?: number | undefined;
122
- longitude?: number | undefined;
123
- }, {
124
- latitude?: number | undefined;
125
- longitude?: number | undefined;
126
- }>>;
127
- languageCode: import("zod").ZodOptional<import("zod").ZodString>;
128
- }, import("zod").ZodTypeAny, "passthrough">>>;
129
- thinkingConfig: import("zod").ZodOptional<import("zod").ZodObject<{
130
- includeThoughts: import("zod").ZodOptional<import("zod").ZodBoolean>;
131
- thinkingBudget: import("zod").ZodOptional<import("zod").ZodNumber>;
132
- thinkingLevel: import("zod").ZodOptional<import("zod").ZodEnum<["MINIMAL", "LOW", "MEDIUM", "HIGH"]>>;
133
- }, "passthrough", import("zod").ZodTypeAny, import("zod").objectOutputType<{
134
- includeThoughts: import("zod").ZodOptional<import("zod").ZodBoolean>;
135
- thinkingBudget: import("zod").ZodOptional<import("zod").ZodNumber>;
136
- thinkingLevel: import("zod").ZodOptional<import("zod").ZodEnum<["MINIMAL", "LOW", "MEDIUM", "HIGH"]>>;
137
- }, import("zod").ZodTypeAny, "passthrough">, import("zod").objectInputType<{
138
- includeThoughts: import("zod").ZodOptional<import("zod").ZodBoolean>;
139
- thinkingBudget: import("zod").ZodOptional<import("zod").ZodNumber>;
140
- thinkingLevel: import("zod").ZodOptional<import("zod").ZodEnum<["MINIMAL", "LOW", "MEDIUM", "HIGH"]>>;
141
- }, import("zod").ZodTypeAny, "passthrough">>>;
142
- }, "passthrough", import("zod").ZodTypeAny, import("zod").objectOutputType<{
143
- version: import("zod").ZodOptional<import("zod").ZodString>;
144
- maxOutputTokens: import("zod").ZodOptional<import("zod").ZodNumber>;
145
- topK: import("zod").ZodOptional<import("zod").ZodNumber>;
146
- stopSequences: import("zod").ZodOptional<import("zod").ZodArray<import("zod").ZodString, "many">>;
147
- } & {
148
- apiKey: import("zod").ZodOptional<import("zod").ZodString>;
149
- labels: import("zod").ZodOptional<import("zod").ZodRecord<import("zod").ZodString, import("zod").ZodString>>;
150
- temperature: import("zod").ZodOptional<import("zod").ZodNumber>;
151
- topP: import("zod").ZodOptional<import("zod").ZodNumber>;
152
- location: import("zod").ZodOptional<import("zod").ZodString>;
153
- safetySettings: import("zod").ZodOptional<import("zod").ZodArray<import("zod").ZodObject<{
154
- category: import("zod").ZodEnum<["HARM_CATEGORY_UNSPECIFIED", "HARM_CATEGORY_HATE_SPEECH", "HARM_CATEGORY_DANGEROUS_CONTENT", "HARM_CATEGORY_HARASSMENT", "HARM_CATEGORY_SEXUALLY_EXPLICIT"]>;
155
- threshold: import("zod").ZodEnum<["BLOCK_LOW_AND_ABOVE", "BLOCK_MEDIUM_AND_ABOVE", "BLOCK_ONLY_HIGH", "BLOCK_NONE"]>;
156
- }, "passthrough", import("zod").ZodTypeAny, import("zod").objectOutputType<{
157
- category: import("zod").ZodEnum<["HARM_CATEGORY_UNSPECIFIED", "HARM_CATEGORY_HATE_SPEECH", "HARM_CATEGORY_DANGEROUS_CONTENT", "HARM_CATEGORY_HARASSMENT", "HARM_CATEGORY_SEXUALLY_EXPLICIT"]>;
158
- threshold: import("zod").ZodEnum<["BLOCK_LOW_AND_ABOVE", "BLOCK_MEDIUM_AND_ABOVE", "BLOCK_ONLY_HIGH", "BLOCK_NONE"]>;
159
- }, import("zod").ZodTypeAny, "passthrough">, import("zod").objectInputType<{
160
- category: import("zod").ZodEnum<["HARM_CATEGORY_UNSPECIFIED", "HARM_CATEGORY_HATE_SPEECH", "HARM_CATEGORY_DANGEROUS_CONTENT", "HARM_CATEGORY_HARASSMENT", "HARM_CATEGORY_SEXUALLY_EXPLICIT"]>;
161
- threshold: import("zod").ZodEnum<["BLOCK_LOW_AND_ABOVE", "BLOCK_MEDIUM_AND_ABOVE", "BLOCK_ONLY_HIGH", "BLOCK_NONE"]>;
162
- }, import("zod").ZodTypeAny, "passthrough">>, "many">>;
163
- vertexRetrieval: import("zod").ZodOptional<import("zod").ZodObject<{
164
- datastore: import("zod").ZodObject<{
165
- projectId: import("zod").ZodOptional<import("zod").ZodString>;
166
- location: import("zod").ZodOptional<import("zod").ZodString>;
167
- dataStoreId: import("zod").ZodString;
168
- }, "passthrough", import("zod").ZodTypeAny, import("zod").objectOutputType<{
169
- projectId: import("zod").ZodOptional<import("zod").ZodString>;
170
- location: import("zod").ZodOptional<import("zod").ZodString>;
171
- dataStoreId: import("zod").ZodString;
172
- }, import("zod").ZodTypeAny, "passthrough">, import("zod").objectInputType<{
173
- projectId: import("zod").ZodOptional<import("zod").ZodString>;
174
- location: import("zod").ZodOptional<import("zod").ZodString>;
175
- dataStoreId: import("zod").ZodString;
176
- }, import("zod").ZodTypeAny, "passthrough">>;
177
- disableAttribution: import("zod").ZodOptional<import("zod").ZodBoolean>;
178
- }, "passthrough", import("zod").ZodTypeAny, import("zod").objectOutputType<{
179
- datastore: import("zod").ZodObject<{
180
- projectId: import("zod").ZodOptional<import("zod").ZodString>;
181
- location: import("zod").ZodOptional<import("zod").ZodString>;
182
- dataStoreId: import("zod").ZodString;
183
- }, "passthrough", import("zod").ZodTypeAny, import("zod").objectOutputType<{
184
- projectId: import("zod").ZodOptional<import("zod").ZodString>;
185
- location: import("zod").ZodOptional<import("zod").ZodString>;
186
- dataStoreId: import("zod").ZodString;
187
- }, import("zod").ZodTypeAny, "passthrough">, import("zod").objectInputType<{
188
- projectId: import("zod").ZodOptional<import("zod").ZodString>;
189
- location: import("zod").ZodOptional<import("zod").ZodString>;
190
- dataStoreId: import("zod").ZodString;
191
- }, import("zod").ZodTypeAny, "passthrough">>;
192
- disableAttribution: import("zod").ZodOptional<import("zod").ZodBoolean>;
193
- }, import("zod").ZodTypeAny, "passthrough">, import("zod").objectInputType<{
194
- datastore: import("zod").ZodObject<{
195
- projectId: import("zod").ZodOptional<import("zod").ZodString>;
196
- location: import("zod").ZodOptional<import("zod").ZodString>;
197
- dataStoreId: import("zod").ZodString;
198
- }, "passthrough", import("zod").ZodTypeAny, import("zod").objectOutputType<{
199
- projectId: import("zod").ZodOptional<import("zod").ZodString>;
200
- location: import("zod").ZodOptional<import("zod").ZodString>;
201
- dataStoreId: import("zod").ZodString;
202
- }, import("zod").ZodTypeAny, "passthrough">, import("zod").objectInputType<{
203
- projectId: import("zod").ZodOptional<import("zod").ZodString>;
204
- location: import("zod").ZodOptional<import("zod").ZodString>;
205
- dataStoreId: import("zod").ZodString;
206
- }, import("zod").ZodTypeAny, "passthrough">>;
207
- disableAttribution: import("zod").ZodOptional<import("zod").ZodBoolean>;
208
- }, import("zod").ZodTypeAny, "passthrough">>>;
209
- googleSearchRetrieval: import("zod").ZodOptional<import("zod").ZodObject<{
210
- disableAttribution: import("zod").ZodOptional<import("zod").ZodBoolean>;
211
- }, "passthrough", import("zod").ZodTypeAny, import("zod").objectOutputType<{
212
- disableAttribution: import("zod").ZodOptional<import("zod").ZodBoolean>;
213
- }, import("zod").ZodTypeAny, "passthrough">, import("zod").objectInputType<{
214
- disableAttribution: import("zod").ZodOptional<import("zod").ZodBoolean>;
215
- }, import("zod").ZodTypeAny, "passthrough">>>;
216
- functionCallingConfig: import("zod").ZodOptional<import("zod").ZodObject<{
217
- mode: import("zod").ZodOptional<import("zod").ZodEnum<["MODE_UNSPECIFIED", "AUTO", "ANY", "NONE"]>>;
218
- allowedFunctionNames: import("zod").ZodOptional<import("zod").ZodArray<import("zod").ZodString, "many">>;
219
- streamFunctionCallArguments: import("zod").ZodOptional<import("zod").ZodBoolean>;
220
- }, "passthrough", import("zod").ZodTypeAny, import("zod").objectOutputType<{
221
- mode: import("zod").ZodOptional<import("zod").ZodEnum<["MODE_UNSPECIFIED", "AUTO", "ANY", "NONE"]>>;
222
- allowedFunctionNames: import("zod").ZodOptional<import("zod").ZodArray<import("zod").ZodString, "many">>;
223
- streamFunctionCallArguments: import("zod").ZodOptional<import("zod").ZodBoolean>;
224
- }, import("zod").ZodTypeAny, "passthrough">, import("zod").objectInputType<{
225
- mode: import("zod").ZodOptional<import("zod").ZodEnum<["MODE_UNSPECIFIED", "AUTO", "ANY", "NONE"]>>;
226
- allowedFunctionNames: import("zod").ZodOptional<import("zod").ZodArray<import("zod").ZodString, "many">>;
227
- streamFunctionCallArguments: import("zod").ZodOptional<import("zod").ZodBoolean>;
228
- }, import("zod").ZodTypeAny, "passthrough">>>;
229
- retrievalConfig: import("zod").ZodOptional<import("zod").ZodObject<{
230
- latLng: import("zod").ZodOptional<import("zod").ZodObject<{
231
- latitude: import("zod").ZodOptional<import("zod").ZodNumber>;
232
- longitude: import("zod").ZodOptional<import("zod").ZodNumber>;
233
- }, "strip", import("zod").ZodTypeAny, {
234
- latitude?: number | undefined;
235
- longitude?: number | undefined;
236
- }, {
237
- latitude?: number | undefined;
238
- longitude?: number | undefined;
239
- }>>;
240
- languageCode: import("zod").ZodOptional<import("zod").ZodString>;
241
- }, "passthrough", import("zod").ZodTypeAny, import("zod").objectOutputType<{
242
- latLng: import("zod").ZodOptional<import("zod").ZodObject<{
243
- latitude: import("zod").ZodOptional<import("zod").ZodNumber>;
244
- longitude: import("zod").ZodOptional<import("zod").ZodNumber>;
245
- }, "strip", import("zod").ZodTypeAny, {
246
- latitude?: number | undefined;
247
- longitude?: number | undefined;
248
- }, {
249
- latitude?: number | undefined;
250
- longitude?: number | undefined;
251
- }>>;
252
- languageCode: import("zod").ZodOptional<import("zod").ZodString>;
253
- }, import("zod").ZodTypeAny, "passthrough">, import("zod").objectInputType<{
254
- latLng: import("zod").ZodOptional<import("zod").ZodObject<{
255
- latitude: import("zod").ZodOptional<import("zod").ZodNumber>;
256
- longitude: import("zod").ZodOptional<import("zod").ZodNumber>;
257
- }, "strip", import("zod").ZodTypeAny, {
258
- latitude?: number | undefined;
259
- longitude?: number | undefined;
260
- }, {
261
- latitude?: number | undefined;
262
- longitude?: number | undefined;
263
- }>>;
264
- languageCode: import("zod").ZodOptional<import("zod").ZodString>;
265
- }, import("zod").ZodTypeAny, "passthrough">>>;
266
- thinkingConfig: import("zod").ZodOptional<import("zod").ZodObject<{
267
- includeThoughts: import("zod").ZodOptional<import("zod").ZodBoolean>;
268
- thinkingBudget: import("zod").ZodOptional<import("zod").ZodNumber>;
269
- thinkingLevel: import("zod").ZodOptional<import("zod").ZodEnum<["MINIMAL", "LOW", "MEDIUM", "HIGH"]>>;
270
- }, "passthrough", import("zod").ZodTypeAny, import("zod").objectOutputType<{
271
- includeThoughts: import("zod").ZodOptional<import("zod").ZodBoolean>;
272
- thinkingBudget: import("zod").ZodOptional<import("zod").ZodNumber>;
273
- thinkingLevel: import("zod").ZodOptional<import("zod").ZodEnum<["MINIMAL", "LOW", "MEDIUM", "HIGH"]>>;
274
- }, import("zod").ZodTypeAny, "passthrough">, import("zod").objectInputType<{
275
- includeThoughts: import("zod").ZodOptional<import("zod").ZodBoolean>;
276
- thinkingBudget: import("zod").ZodOptional<import("zod").ZodNumber>;
277
- thinkingLevel: import("zod").ZodOptional<import("zod").ZodEnum<["MINIMAL", "LOW", "MEDIUM", "HIGH"]>>;
278
- }, import("zod").ZodTypeAny, "passthrough">>>;
279
- }, import("zod").ZodTypeAny, "passthrough">, import("zod").objectInputType<{
280
- version: import("zod").ZodOptional<import("zod").ZodString>;
281
- maxOutputTokens: import("zod").ZodOptional<import("zod").ZodNumber>;
282
- topK: import("zod").ZodOptional<import("zod").ZodNumber>;
283
- stopSequences: import("zod").ZodOptional<import("zod").ZodArray<import("zod").ZodString, "many">>;
284
- } & {
285
- apiKey: import("zod").ZodOptional<import("zod").ZodString>;
286
- labels: import("zod").ZodOptional<import("zod").ZodRecord<import("zod").ZodString, import("zod").ZodString>>;
287
- temperature: import("zod").ZodOptional<import("zod").ZodNumber>;
288
- topP: import("zod").ZodOptional<import("zod").ZodNumber>;
289
- location: import("zod").ZodOptional<import("zod").ZodString>;
290
- safetySettings: import("zod").ZodOptional<import("zod").ZodArray<import("zod").ZodObject<{
291
- category: import("zod").ZodEnum<["HARM_CATEGORY_UNSPECIFIED", "HARM_CATEGORY_HATE_SPEECH", "HARM_CATEGORY_DANGEROUS_CONTENT", "HARM_CATEGORY_HARASSMENT", "HARM_CATEGORY_SEXUALLY_EXPLICIT"]>;
292
- threshold: import("zod").ZodEnum<["BLOCK_LOW_AND_ABOVE", "BLOCK_MEDIUM_AND_ABOVE", "BLOCK_ONLY_HIGH", "BLOCK_NONE"]>;
293
- }, "passthrough", import("zod").ZodTypeAny, import("zod").objectOutputType<{
294
- category: import("zod").ZodEnum<["HARM_CATEGORY_UNSPECIFIED", "HARM_CATEGORY_HATE_SPEECH", "HARM_CATEGORY_DANGEROUS_CONTENT", "HARM_CATEGORY_HARASSMENT", "HARM_CATEGORY_SEXUALLY_EXPLICIT"]>;
295
- threshold: import("zod").ZodEnum<["BLOCK_LOW_AND_ABOVE", "BLOCK_MEDIUM_AND_ABOVE", "BLOCK_ONLY_HIGH", "BLOCK_NONE"]>;
296
- }, import("zod").ZodTypeAny, "passthrough">, import("zod").objectInputType<{
297
- category: import("zod").ZodEnum<["HARM_CATEGORY_UNSPECIFIED", "HARM_CATEGORY_HATE_SPEECH", "HARM_CATEGORY_DANGEROUS_CONTENT", "HARM_CATEGORY_HARASSMENT", "HARM_CATEGORY_SEXUALLY_EXPLICIT"]>;
298
- threshold: import("zod").ZodEnum<["BLOCK_LOW_AND_ABOVE", "BLOCK_MEDIUM_AND_ABOVE", "BLOCK_ONLY_HIGH", "BLOCK_NONE"]>;
299
- }, import("zod").ZodTypeAny, "passthrough">>, "many">>;
300
- vertexRetrieval: import("zod").ZodOptional<import("zod").ZodObject<{
301
- datastore: import("zod").ZodObject<{
302
- projectId: import("zod").ZodOptional<import("zod").ZodString>;
303
- location: import("zod").ZodOptional<import("zod").ZodString>;
304
- dataStoreId: import("zod").ZodString;
305
- }, "passthrough", import("zod").ZodTypeAny, import("zod").objectOutputType<{
306
- projectId: import("zod").ZodOptional<import("zod").ZodString>;
307
- location: import("zod").ZodOptional<import("zod").ZodString>;
308
- dataStoreId: import("zod").ZodString;
309
- }, import("zod").ZodTypeAny, "passthrough">, import("zod").objectInputType<{
310
- projectId: import("zod").ZodOptional<import("zod").ZodString>;
311
- location: import("zod").ZodOptional<import("zod").ZodString>;
312
- dataStoreId: import("zod").ZodString;
313
- }, import("zod").ZodTypeAny, "passthrough">>;
314
- disableAttribution: import("zod").ZodOptional<import("zod").ZodBoolean>;
315
- }, "passthrough", import("zod").ZodTypeAny, import("zod").objectOutputType<{
316
- datastore: import("zod").ZodObject<{
317
- projectId: import("zod").ZodOptional<import("zod").ZodString>;
318
- location: import("zod").ZodOptional<import("zod").ZodString>;
319
- dataStoreId: import("zod").ZodString;
320
- }, "passthrough", import("zod").ZodTypeAny, import("zod").objectOutputType<{
321
- projectId: import("zod").ZodOptional<import("zod").ZodString>;
322
- location: import("zod").ZodOptional<import("zod").ZodString>;
323
- dataStoreId: import("zod").ZodString;
324
- }, import("zod").ZodTypeAny, "passthrough">, import("zod").objectInputType<{
325
- projectId: import("zod").ZodOptional<import("zod").ZodString>;
326
- location: import("zod").ZodOptional<import("zod").ZodString>;
327
- dataStoreId: import("zod").ZodString;
328
- }, import("zod").ZodTypeAny, "passthrough">>;
329
- disableAttribution: import("zod").ZodOptional<import("zod").ZodBoolean>;
330
- }, import("zod").ZodTypeAny, "passthrough">, import("zod").objectInputType<{
331
- datastore: import("zod").ZodObject<{
332
- projectId: import("zod").ZodOptional<import("zod").ZodString>;
333
- location: import("zod").ZodOptional<import("zod").ZodString>;
334
- dataStoreId: import("zod").ZodString;
335
- }, "passthrough", import("zod").ZodTypeAny, import("zod").objectOutputType<{
336
- projectId: import("zod").ZodOptional<import("zod").ZodString>;
337
- location: import("zod").ZodOptional<import("zod").ZodString>;
338
- dataStoreId: import("zod").ZodString;
339
- }, import("zod").ZodTypeAny, "passthrough">, import("zod").objectInputType<{
340
- projectId: import("zod").ZodOptional<import("zod").ZodString>;
341
- location: import("zod").ZodOptional<import("zod").ZodString>;
342
- dataStoreId: import("zod").ZodString;
343
- }, import("zod").ZodTypeAny, "passthrough">>;
344
- disableAttribution: import("zod").ZodOptional<import("zod").ZodBoolean>;
345
- }, import("zod").ZodTypeAny, "passthrough">>>;
346
- googleSearchRetrieval: import("zod").ZodOptional<import("zod").ZodObject<{
347
- disableAttribution: import("zod").ZodOptional<import("zod").ZodBoolean>;
348
- }, "passthrough", import("zod").ZodTypeAny, import("zod").objectOutputType<{
349
- disableAttribution: import("zod").ZodOptional<import("zod").ZodBoolean>;
350
- }, import("zod").ZodTypeAny, "passthrough">, import("zod").objectInputType<{
351
- disableAttribution: import("zod").ZodOptional<import("zod").ZodBoolean>;
352
- }, import("zod").ZodTypeAny, "passthrough">>>;
353
- functionCallingConfig: import("zod").ZodOptional<import("zod").ZodObject<{
354
- mode: import("zod").ZodOptional<import("zod").ZodEnum<["MODE_UNSPECIFIED", "AUTO", "ANY", "NONE"]>>;
355
- allowedFunctionNames: import("zod").ZodOptional<import("zod").ZodArray<import("zod").ZodString, "many">>;
356
- streamFunctionCallArguments: import("zod").ZodOptional<import("zod").ZodBoolean>;
357
- }, "passthrough", import("zod").ZodTypeAny, import("zod").objectOutputType<{
358
- mode: import("zod").ZodOptional<import("zod").ZodEnum<["MODE_UNSPECIFIED", "AUTO", "ANY", "NONE"]>>;
359
- allowedFunctionNames: import("zod").ZodOptional<import("zod").ZodArray<import("zod").ZodString, "many">>;
360
- streamFunctionCallArguments: import("zod").ZodOptional<import("zod").ZodBoolean>;
361
- }, import("zod").ZodTypeAny, "passthrough">, import("zod").objectInputType<{
362
- mode: import("zod").ZodOptional<import("zod").ZodEnum<["MODE_UNSPECIFIED", "AUTO", "ANY", "NONE"]>>;
363
- allowedFunctionNames: import("zod").ZodOptional<import("zod").ZodArray<import("zod").ZodString, "many">>;
364
- streamFunctionCallArguments: import("zod").ZodOptional<import("zod").ZodBoolean>;
365
- }, import("zod").ZodTypeAny, "passthrough">>>;
366
- retrievalConfig: import("zod").ZodOptional<import("zod").ZodObject<{
367
- latLng: import("zod").ZodOptional<import("zod").ZodObject<{
368
- latitude: import("zod").ZodOptional<import("zod").ZodNumber>;
369
- longitude: import("zod").ZodOptional<import("zod").ZodNumber>;
370
- }, "strip", import("zod").ZodTypeAny, {
371
- latitude?: number | undefined;
372
- longitude?: number | undefined;
373
- }, {
374
- latitude?: number | undefined;
375
- longitude?: number | undefined;
376
- }>>;
377
- languageCode: import("zod").ZodOptional<import("zod").ZodString>;
378
- }, "passthrough", import("zod").ZodTypeAny, import("zod").objectOutputType<{
379
- latLng: import("zod").ZodOptional<import("zod").ZodObject<{
380
- latitude: import("zod").ZodOptional<import("zod").ZodNumber>;
381
- longitude: import("zod").ZodOptional<import("zod").ZodNumber>;
382
- }, "strip", import("zod").ZodTypeAny, {
383
- latitude?: number | undefined;
384
- longitude?: number | undefined;
385
- }, {
386
- latitude?: number | undefined;
387
- longitude?: number | undefined;
388
- }>>;
389
- languageCode: import("zod").ZodOptional<import("zod").ZodString>;
390
- }, import("zod").ZodTypeAny, "passthrough">, import("zod").objectInputType<{
391
- latLng: import("zod").ZodOptional<import("zod").ZodObject<{
392
- latitude: import("zod").ZodOptional<import("zod").ZodNumber>;
393
- longitude: import("zod").ZodOptional<import("zod").ZodNumber>;
394
- }, "strip", import("zod").ZodTypeAny, {
395
- latitude?: number | undefined;
396
- longitude?: number | undefined;
397
- }, {
398
- latitude?: number | undefined;
399
- longitude?: number | undefined;
400
- }>>;
401
- languageCode: import("zod").ZodOptional<import("zod").ZodString>;
402
- }, import("zod").ZodTypeAny, "passthrough">>>;
403
- thinkingConfig: import("zod").ZodOptional<import("zod").ZodObject<{
404
- includeThoughts: import("zod").ZodOptional<import("zod").ZodBoolean>;
405
- thinkingBudget: import("zod").ZodOptional<import("zod").ZodNumber>;
406
- thinkingLevel: import("zod").ZodOptional<import("zod").ZodEnum<["MINIMAL", "LOW", "MEDIUM", "HIGH"]>>;
407
- }, "passthrough", import("zod").ZodTypeAny, import("zod").objectOutputType<{
408
- includeThoughts: import("zod").ZodOptional<import("zod").ZodBoolean>;
409
- thinkingBudget: import("zod").ZodOptional<import("zod").ZodNumber>;
410
- thinkingLevel: import("zod").ZodOptional<import("zod").ZodEnum<["MINIMAL", "LOW", "MEDIUM", "HIGH"]>>;
411
- }, import("zod").ZodTypeAny, "passthrough">, import("zod").objectInputType<{
412
- includeThoughts: import("zod").ZodOptional<import("zod").ZodBoolean>;
413
- thinkingBudget: import("zod").ZodOptional<import("zod").ZodNumber>;
414
- thinkingLevel: import("zod").ZodOptional<import("zod").ZodEnum<["MINIMAL", "LOW", "MEDIUM", "HIGH"]>>;
415
- }, import("zod").ZodTypeAny, "passthrough">>>;
416
- }, import("zod").ZodTypeAny, "passthrough">>>;
3
+ import type { GeminiModelReference, VertexAiMultiLocationOptions } from './types.js';
417
4
  export declare function vertexAiMultiLocation(options: VertexAiMultiLocationOptions & {
418
5
  circuitBreakerProvider: CircuitBreakerProvider;
419
6
  logger: Logger;
@@ -421,4 +8,3 @@ export declare function vertexAiMultiLocation(options: VertexAiMultiLocationOpti
421
8
  export declare namespace vertexAiMultiLocation {
422
9
  var model: <T extends GeminiModelReference>(baseModel: T) => T;
423
10
  }
424
- export {};
@@ -1,11 +1,9 @@
1
- import { vertexAI } from '@genkit-ai/google-genai';
2
1
  import { GenkitError, modelRef } from 'genkit';
3
2
  import { genkitPlugin } from 'genkit/plugin';
4
3
  import { distinct, shuffle } from '../../utils/array/index.js';
5
4
  import { isInstanceOf, isNullOrUndefined } from '../../utils/type-guards.js';
6
5
  import { millisecondsPerMinute, millisecondsPerSecond } from '../../utils/units.js';
7
6
  const pluginKey = 'vertexai-multi-location';
8
- const geminiModelReference = vertexAI.model('gemini-');
9
7
  const defaultTokenLimitThreshold = 131_072;
10
8
  export function vertexAiMultiLocation(options) {
11
9
  if (options.locations.length == 0) {