@ai-sdk/google-vertex 5.0.0-beta.61 → 5.0.0-beta.63

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.
Files changed (45) hide show
  1. package/CHANGELOG.md +33 -0
  2. package/dist/anthropic/edge/index.d.ts +14 -9
  3. package/dist/anthropic/edge/index.js +15 -13
  4. package/dist/anthropic/edge/index.js.map +1 -1
  5. package/dist/anthropic/index.d.ts +14 -9
  6. package/dist/anthropic/index.js +14 -12
  7. package/dist/anthropic/index.js.map +1 -1
  8. package/dist/edge/index.d.ts +3 -3
  9. package/dist/edge/index.js +197 -157
  10. package/dist/edge/index.js.map +1 -1
  11. package/dist/index.d.ts +7 -7
  12. package/dist/index.js +199 -159
  13. package/dist/index.js.map +1 -1
  14. package/dist/maas/edge/index.d.ts +3 -3
  15. package/dist/maas/edge/index.js +12 -8
  16. package/dist/maas/edge/index.js.map +1 -1
  17. package/dist/maas/index.d.ts +3 -3
  18. package/dist/maas/index.js +11 -7
  19. package/dist/maas/index.js.map +1 -1
  20. package/docs/16-google-vertex.mdx +77 -77
  21. package/package.json +7 -7
  22. package/src/anthropic/edge/google-vertex-anthropic-provider-edge.ts +16 -11
  23. package/src/anthropic/edge/index.ts +6 -2
  24. package/src/anthropic/google-vertex-anthropic-provider-node.ts +16 -11
  25. package/src/anthropic/google-vertex-anthropic-provider.ts +10 -10
  26. package/src/anthropic/index.ts +6 -2
  27. package/src/edge/google-vertex-provider-edge.ts +9 -10
  28. package/src/edge/index.ts +8 -1
  29. package/src/google-vertex-auth-google-auth-library.ts +1 -2
  30. package/src/google-vertex-config.ts +1 -1
  31. package/src/google-vertex-embedding-model.ts +13 -5
  32. package/src/google-vertex-image-model-options.ts +74 -0
  33. package/src/google-vertex-image-model.ts +66 -113
  34. package/src/google-vertex-provider-base.ts +245 -0
  35. package/src/google-vertex-provider.ts +35 -233
  36. package/src/google-vertex-video-model-options.ts +49 -0
  37. package/src/google-vertex-video-model.ts +32 -68
  38. package/src/index.ts +12 -5
  39. package/src/maas/edge/google-vertex-maas-provider-edge.ts +7 -8
  40. package/src/maas/edge/index.ts +6 -2
  41. package/src/maas/google-vertex-maas-provider-node.ts +7 -8
  42. package/src/maas/google-vertex-maas-provider.ts +7 -5
  43. package/src/maas/index.ts +6 -2
  44. package/src/google-vertex-provider-node.ts +0 -48
  45. /package/src/{google-vertex-embedding-options.ts → google-vertex-embedding-model-options.ts} +0 -0
@@ -8,40 +8,20 @@ import {
8
8
  convertUint8ArrayToBase64,
9
9
  createJsonResponseHandler,
10
10
  delay,
11
- type FetchFunction,
12
- lazySchema,
13
11
  parseProviderOptions,
14
12
  postJsonToApi,
15
- type Resolvable,
16
13
  resolve,
17
- zodSchema,
14
+ type FetchFunction,
15
+ type Resolvable,
18
16
  } from '@ai-sdk/provider-utils';
19
17
  import { z } from 'zod/v4';
20
18
  import { googleVertexFailedResponseHandler } from './google-vertex-error';
19
+ import {
20
+ googleVertexVideoModelOptionsSchema,
21
+ type GoogleVertexVideoModelOptions,
22
+ } from './google-vertex-video-model-options';
21
23
  import type { GoogleVertexVideoModelId } from './google-vertex-video-settings';
22
24
 
23
- export type GoogleVertexVideoModelOptions = {
24
- // Polling configuration
25
- pollIntervalMs?: number | null;
26
- pollTimeoutMs?: number | null;
27
-
28
- // Video generation options
29
- personGeneration?: 'dont_allow' | 'allow_adult' | 'allow_all' | null;
30
- negativePrompt?: string | null;
31
- generateAudio?: boolean | null;
32
-
33
- // Output configuration
34
- gcsOutputDirectory?: string | null;
35
-
36
- // Reference images (for style/asset reference)
37
- referenceImages?: Array<{
38
- bytesBase64Encoded?: string;
39
- gcsUri?: string;
40
- }> | null;
41
-
42
- [key: string]: unknown; // For passthrough
43
- };
44
-
45
25
  interface GoogleVertexVideoModelConfig {
46
26
  provider: string;
47
27
  baseURL: string;
@@ -76,11 +56,16 @@ export class GoogleVertexVideoModel implements Experimental_VideoModelV4 {
76
56
  const currentDate = this.config._internal?.currentDate?.() ?? new Date();
77
57
  const warnings: SharedV4Warning[] = [];
78
58
 
79
- const vertexOptions = (await parseProviderOptions({
80
- provider: 'vertex',
59
+ const googleVertexOptions = ((await parseProviderOptions({
60
+ provider: 'googleVertex',
81
61
  providerOptions: options.providerOptions,
82
62
  schema: googleVertexVideoModelOptionsSchema,
83
- })) as GoogleVertexVideoModelOptions | undefined;
63
+ })) ??
64
+ (await parseProviderOptions({
65
+ provider: 'vertex',
66
+ providerOptions: options.providerOptions,
67
+ schema: googleVertexVideoModelOptionsSchema,
68
+ }))) as GoogleVertexVideoModelOptions | undefined;
84
69
 
85
70
  const instances: Array<Record<string, unknown>> = [{}];
86
71
  const instance = instances[0];
@@ -110,8 +95,8 @@ export class GoogleVertexVideoModel implements Experimental_VideoModelV4 {
110
95
  }
111
96
  }
112
97
 
113
- if (vertexOptions?.referenceImages != null) {
114
- instance.referenceImages = vertexOptions.referenceImages;
98
+ if (googleVertexOptions?.referenceImages != null) {
99
+ instance.referenceImages = googleVertexOptions.referenceImages;
115
100
  }
116
101
 
117
102
  const parameters: Record<string, unknown> = {
@@ -140,8 +125,8 @@ export class GoogleVertexVideoModel implements Experimental_VideoModelV4 {
140
125
  parameters.seed = options.seed;
141
126
  }
142
127
 
143
- if (vertexOptions != null) {
144
- const opts = vertexOptions;
128
+ if (googleVertexOptions != null) {
129
+ const opts = googleVertexOptions;
145
130
 
146
131
  if (
147
132
  opts.personGeneration !== undefined &&
@@ -190,7 +175,7 @@ export class GoogleVertexVideoModel implements Experimental_VideoModelV4 {
190
175
  parameters,
191
176
  },
192
177
  successfulResponseHandler: createJsonResponseHandler(
193
- vertexOperationSchema,
178
+ googleVertexOperationSchema,
194
179
  ),
195
180
  failedResponseHandler: googleVertexFailedResponseHandler,
196
181
  abortSignal: options.abortSignal,
@@ -205,8 +190,8 @@ export class GoogleVertexVideoModel implements Experimental_VideoModelV4 {
205
190
  });
206
191
  }
207
192
 
208
- const pollIntervalMs = vertexOptions?.pollIntervalMs ?? 10000; // 10 seconds
209
- const pollTimeoutMs = vertexOptions?.pollTimeoutMs ?? 600000; // 10 minutes
193
+ const pollIntervalMs = googleVertexOptions?.pollIntervalMs ?? 10000; // 10 seconds
194
+ const pollTimeoutMs = googleVertexOptions?.pollTimeoutMs ?? 600000; // 10 minutes
210
195
 
211
196
  const startTime = Date.now();
212
197
  let finalOperation = operation;
@@ -240,7 +225,7 @@ export class GoogleVertexVideoModel implements Experimental_VideoModelV4 {
240
225
  operationName,
241
226
  },
242
227
  successfulResponseHandler: createJsonResponseHandler(
243
- vertexOperationSchema,
228
+ googleVertexOperationSchema,
244
229
  ),
245
230
  failedResponseHandler: googleVertexFailedResponseHandler,
246
231
  abortSignal: options.abortSignal,
@@ -314,16 +299,20 @@ export class GoogleVertexVideoModel implements Experimental_VideoModelV4 {
314
299
  modelId: this.modelId,
315
300
  headers: responseHeaders,
316
301
  },
317
- providerMetadata: {
318
- 'google-vertex': {
319
- videos: videoMetadata,
320
- },
321
- },
302
+ providerMetadata: (() => {
303
+ const payload = { videos: videoMetadata };
304
+ return {
305
+ googleVertex: payload,
306
+ // Legacy keys preserved for backward compatibility.
307
+ 'google-vertex': payload,
308
+ vertex: payload,
309
+ };
310
+ })(),
322
311
  };
323
312
  }
324
313
  }
325
314
 
326
- const vertexOperationSchema = z.object({
315
+ const googleVertexOperationSchema = z.object({
327
316
  name: z.string().nullish(),
328
317
  done: z.boolean().nullish(),
329
318
  error: z
@@ -348,28 +337,3 @@ const vertexOperationSchema = z.object({
348
337
  })
349
338
  .nullish(),
350
339
  });
351
-
352
- const googleVertexVideoModelOptionsSchema = lazySchema(() =>
353
- zodSchema(
354
- z
355
- .object({
356
- pollIntervalMs: z.number().positive().nullish(),
357
- pollTimeoutMs: z.number().positive().nullish(),
358
- personGeneration: z
359
- .enum(['dont_allow', 'allow_adult', 'allow_all'])
360
- .nullish(),
361
- negativePrompt: z.string().nullish(),
362
- generateAudio: z.boolean().nullish(),
363
- gcsOutputDirectory: z.string().nullish(),
364
- referenceImages: z
365
- .array(
366
- z.object({
367
- bytesBase64Encoded: z.string().nullish(),
368
- gcsUri: z.string().nullish(),
369
- }),
370
- )
371
- .nullish(),
372
- })
373
- .passthrough(),
374
- ),
375
- );
package/src/index.ts CHANGED
@@ -1,18 +1,25 @@
1
- export type { GoogleVertexEmbeddingModelOptions } from './google-vertex-embedding-options';
1
+ export type { GoogleVertexEmbeddingModelOptions } from './google-vertex-embedding-model-options';
2
2
  export type {
3
3
  GoogleVertexImageModelOptions,
4
4
  /** @deprecated Use `GoogleVertexImageModelOptions` instead. */
5
5
  GoogleVertexImageModelOptions as GoogleVertexImageProviderOptions,
6
- } from './google-vertex-image-model';
6
+ } from './google-vertex-image-model-options';
7
7
  export type {
8
8
  GoogleVertexVideoModelOptions,
9
9
  /** @deprecated Use `GoogleVertexVideoModelOptions` instead. */
10
10
  GoogleVertexVideoModelOptions as GoogleVertexVideoProviderOptions,
11
- } from './google-vertex-video-model';
11
+ } from './google-vertex-video-model-options';
12
12
  export type { GoogleVertexVideoModelId } from './google-vertex-video-settings';
13
- export { createVertex, vertex } from './google-vertex-provider-node';
13
+ export {
14
+ createGoogleVertex,
15
+ /** @deprecated Use `createGoogleVertex` instead. */
16
+ createGoogleVertex as createVertex,
17
+ googleVertex,
18
+ /** @deprecated Use `googleVertex` instead. */
19
+ googleVertex as vertex,
20
+ } from './google-vertex-provider';
14
21
  export type {
15
22
  GoogleVertexProvider,
16
23
  GoogleVertexProviderSettings,
17
- } from './google-vertex-provider-node';
24
+ } from './google-vertex-provider';
18
25
  export { VERSION } from './version';
@@ -1,14 +1,13 @@
1
- import { FetchFunction, resolve } from '@ai-sdk/provider-utils';
1
+ import { resolve, type FetchFunction } from '@ai-sdk/provider-utils';
2
2
  import {
3
3
  generateAuthToken,
4
- GoogleCredentials,
4
+ type GoogleCredentials,
5
5
  } from '../../edge/google-vertex-auth-edge';
6
6
  import {
7
- createVertexMaas as createVertexMaasOriginal,
8
- GoogleVertexMaasProvider,
9
- GoogleVertexMaasProviderSettings as GoogleVertexMaasProviderSettingsOriginal,
7
+ createGoogleVertexMaas as createVertexMaasOriginal,
8
+ type GoogleVertexMaasProvider,
9
+ type GoogleVertexMaasProviderSettings as GoogleVertexMaasProviderSettingsOriginal,
10
10
  } from '../google-vertex-maas-provider';
11
-
12
11
  export type { GoogleVertexMaasProvider };
13
12
 
14
13
  export interface GoogleVertexMaasProviderSettings extends GoogleVertexMaasProviderSettingsOriginal {
@@ -27,7 +26,7 @@ export interface GoogleVertexMaasProviderSettings extends GoogleVertexMaasProvid
27
26
  *
28
27
  * @see https://cloud.google.com/vertex-ai/generative-ai/docs/maas/use-open-models
29
28
  */
30
- export function createVertexMaas(
29
+ export function createGoogleVertexMaas(
31
30
  options: GoogleVertexMaasProviderSettings = {},
32
31
  ): GoogleVertexMaasProvider {
33
32
  // Create a custom fetch wrapper that adds auth headers
@@ -62,4 +61,4 @@ export function createVertexMaas(
62
61
  /**
63
62
  * Default Google Vertex AI MaaS provider instance for Edge runtimes.
64
63
  */
65
- export const vertexMaas = createVertexMaas();
64
+ export const googleVertexMaas = createGoogleVertexMaas();
@@ -1,6 +1,10 @@
1
1
  export {
2
- createVertexMaas,
3
- vertexMaas,
2
+ createGoogleVertexMaas,
3
+ /** @deprecated Use `createGoogleVertexMaas` instead. */
4
+ createGoogleVertexMaas as createVertexMaas,
5
+ googleVertexMaas,
6
+ /** @deprecated Use `googleVertexMaas` instead. */
7
+ googleVertexMaas as vertexMaas,
4
8
  } from './google-vertex-maas-provider-edge';
5
9
  export type {
6
10
  GoogleVertexMaasProvider,
@@ -1,12 +1,11 @@
1
- import { FetchFunction, resolve } from '@ai-sdk/provider-utils';
2
- import { GoogleAuthOptions } from 'google-auth-library';
1
+ import { resolve, type FetchFunction } from '@ai-sdk/provider-utils';
2
+ import type { GoogleAuthOptions } from 'google-auth-library';
3
3
  import { generateAuthToken } from '../google-vertex-auth-google-auth-library';
4
4
  import {
5
- createVertexMaas as createVertexMaasOriginal,
6
- GoogleVertexMaasProvider,
7
- GoogleVertexMaasProviderSettings as GoogleVertexMaasProviderSettingsOriginal,
5
+ createGoogleVertexMaas as createVertexMaasOriginal,
6
+ type GoogleVertexMaasProvider,
7
+ type GoogleVertexMaasProviderSettings as GoogleVertexMaasProviderSettingsOriginal,
8
8
  } from './google-vertex-maas-provider';
9
-
10
9
  export type { GoogleVertexMaasProvider };
11
10
 
12
11
  export interface GoogleVertexMaasProviderSettings extends GoogleVertexMaasProviderSettingsOriginal {
@@ -26,7 +25,7 @@ export interface GoogleVertexMaasProviderSettings extends GoogleVertexMaasProvid
26
25
  *
27
26
  * @see https://cloud.google.com/vertex-ai/generative-ai/docs/maas/use-open-models
28
27
  */
29
- export function createVertexMaas(
28
+ export function createGoogleVertexMaas(
30
29
  options: GoogleVertexMaasProviderSettings = {},
31
30
  ): GoogleVertexMaasProvider {
32
31
  // Create a custom fetch wrapper that adds auth headers
@@ -61,4 +60,4 @@ export function createVertexMaas(
61
60
  /**
62
61
  * Default Google Vertex AI MaaS provider instance for Node.js.
63
62
  */
64
- export const vertexMaas = createVertexMaas();
63
+ export const googleVertexMaas = createGoogleVertexMaas();
@@ -1,11 +1,13 @@
1
- import { createOpenAICompatible } from '@ai-sdk/openai-compatible';
2
- import type { OpenAICompatibleProvider } from '@ai-sdk/openai-compatible';
3
1
  import {
4
- FetchFunction,
2
+ createOpenAICompatible,
3
+ type OpenAICompatibleProvider,
4
+ } from '@ai-sdk/openai-compatible';
5
+ import {
5
6
  loadOptionalSetting,
6
7
  loadSetting,
7
- Resolvable,
8
8
  withoutTrailingSlash,
9
+ type FetchFunction,
10
+ type Resolvable,
9
11
  } from '@ai-sdk/provider-utils';
10
12
  import type { GoogleVertexMaasModelId } from './google-vertex-maas-options';
11
13
 
@@ -55,7 +57,7 @@ export interface GoogleVertexMaasProviderSettings {
55
57
  *
56
58
  * @see https://cloud.google.com/vertex-ai/generative-ai/docs/maas/use-open-models
57
59
  */
58
- export function createVertexMaas(
60
+ export function createGoogleVertexMaas(
59
61
  options: GoogleVertexMaasProviderSettings = {},
60
62
  ): GoogleVertexMaasProvider {
61
63
  // Lazy-load settings to support loading from environment variables at runtime
package/src/maas/index.ts CHANGED
@@ -1,6 +1,10 @@
1
1
  export {
2
- createVertexMaas,
3
- vertexMaas,
2
+ createGoogleVertexMaas,
3
+ /** @deprecated Use `createGoogleVertexMaas` instead. */
4
+ createGoogleVertexMaas as createVertexMaas,
5
+ googleVertexMaas,
6
+ /** @deprecated Use `googleVertexMaas` instead. */
7
+ googleVertexMaas as vertexMaas,
4
8
  } from './google-vertex-maas-provider-node';
5
9
  export type {
6
10
  GoogleVertexMaasProvider,
@@ -1,48 +0,0 @@
1
- import { loadOptionalSetting, resolve } from '@ai-sdk/provider-utils';
2
- import { GoogleAuthOptions } from 'google-auth-library';
3
- import { generateAuthToken } from './google-vertex-auth-google-auth-library';
4
- import {
5
- createVertex as createVertexOriginal,
6
- GoogleVertexProvider,
7
- GoogleVertexProviderSettings as GoogleVertexProviderSettingsOriginal,
8
- } from './google-vertex-provider';
9
-
10
- export interface GoogleVertexProviderSettings extends GoogleVertexProviderSettingsOriginal {
11
- /**
12
- * Optional. The Authentication options provided by google-auth-library.
13
- * Complete list of authentication options is documented in the
14
- * GoogleAuthOptions interface:
15
- * https://github.com/googleapis/google-auth-library-nodejs/blob/main/src/auth/googleauth.ts.
16
- */
17
- googleAuthOptions?: GoogleAuthOptions;
18
- }
19
-
20
- export type { GoogleVertexProvider };
21
-
22
- export function createVertex(
23
- options: GoogleVertexProviderSettings = {},
24
- ): GoogleVertexProvider {
25
- const apiKey = loadOptionalSetting({
26
- settingValue: options.apiKey,
27
- environmentVariableName: 'GOOGLE_VERTEX_API_KEY',
28
- });
29
-
30
- if (apiKey) {
31
- return createVertexOriginal(options);
32
- }
33
-
34
- return createVertexOriginal({
35
- ...options,
36
- headers: async () => ({
37
- Authorization: `Bearer ${await generateAuthToken(
38
- options.googleAuthOptions,
39
- )}`,
40
- ...(await resolve(options.headers)),
41
- }),
42
- });
43
- }
44
-
45
- /**
46
- * Default Google Vertex AI provider instance.
47
- */
48
- export const vertex = createVertex();