@ai-sdk/google 4.0.0-beta.21 → 4.0.0-beta.23

package/dist/index.mjs

@@ -7,7 +7,7 @@ import {
 } from "@ai-sdk/provider-utils";
 
 // src/version.ts
-var VERSION = true ? "4.0.0-beta.21" : "0.0.0-test";
+var VERSION = true ? "4.0.0-beta.23" : "0.0.0-test";
 
 // src/google-generative-ai-embedding-model.ts
 import {
@@ -408,7 +408,11 @@ function isEmptyObjectSchema(jsonSchema) {
 import {
   UnsupportedFunctionalityError
 } from "@ai-sdk/provider";
-import { convertToBase64 } from "@ai-sdk/provider-utils";
+import {
+  convertToBase64,
+  isProviderReference,
+  resolveProviderReference
+} from "@ai-sdk/provider-utils";
 var dataUrlRegex = /^data:([^;,]+);base64,(.+)$/s;
 function parseBase64DataUrl(value) {
   const match = dataUrlRegex.exec(value);
@@ -543,19 +547,36 @@ function convertToGoogleGenerativeAIMessages(prompt, options) {
             }
             case "file": {
               const mediaType = part.mediaType === "image/*" ? "image/jpeg" : part.mediaType;
-              parts.push(
-                part.data instanceof URL ? {
+              if (part.data instanceof URL) {
+                parts.push({
                   fileData: {
                     mimeType: mediaType,
                     fileUri: part.data.toString()
                   }
-                } : {
+                });
+              } else if (isProviderReference(part.data)) {
+                if (providerOptionsName === "vertex") {
+                  throw new UnsupportedFunctionalityError({
+                    functionality: "file parts with provider references"
+                  });
+                }
+                parts.push({
+                  fileData: {
+                    mimeType: mediaType,
+                    fileUri: resolveProviderReference({
+                      reference: part.data,
+                      provider: "google"
+                    })
+                  }
+                });
+              } else {
+                parts.push({
                   inlineData: {
                     mimeType: mediaType,
                     data: convertToBase64(part.data)
                   }
-                }
-              );
+                });
+              }
               break;
             }
           }
@@ -606,6 +627,24 @@ function convertToGoogleGenerativeAIMessages(prompt, options) {
                     functionality: "File data URLs in assistant messages are not supported"
                   });
                 }
+                if (isProviderReference(part.data)) {
+                  if (providerOptionsName === "vertex") {
+                    throw new UnsupportedFunctionalityError({
+                      functionality: "file parts with provider references"
+                    });
+                  }
+                  return {
+                    fileData: {
+                      mimeType: part.mediaType,
+                      fileUri: resolveProviderReference({
+                        reference: part.data,
+                        provider: "google"
+                      })
+                    },
+                    ...(providerOpts == null ? void 0 : providerOpts.thought) === true ? { thought: true } : {},
+                    thoughtSignature
+                  };
+                }
                 return {
                   inlineData: {
                     mimeType: part.mediaType,
@@ -2540,23 +2579,186 @@ var googleImageModelOptionsSchema = lazySchema11(
   )
 );
 
-// src/google-generative-ai-video-model.ts
+// src/google-generative-ai-files.ts
 import {
   AISDKError
 } from "@ai-sdk/provider";
 import {
   combineHeaders as combineHeaders4,
-  convertUint8ArrayToBase64,
   createJsonResponseHandler as createJsonResponseHandler4,
   delay,
-  getFromApi,
   lazySchema as lazySchema12,
   parseProviderOptions as parseProviderOptions4,
-  postJsonToApi as postJsonToApi4,
-  resolve as resolve4,
-  zodSchema as zodSchema12
+  zodSchema as zodSchema12,
+  getFromApi
 } from "@ai-sdk/provider-utils";
 import { z as z14 } from "zod/v4";
+var GoogleGenerativeAIFiles = class {
+  constructor(config) {
+    this.config = config;
+    this.specificationVersion = "v4";
+  }
+  get provider() {
+    return this.config.provider;
+  }
+  async uploadFile(options) {
+    var _a, _b, _c, _d;
+    const googleOptions = await parseProviderOptions4({
+      provider: "google",
+      providerOptions: options.providerOptions,
+      schema: googleFilesUploadOptionsSchema
+    });
+    const resolvedHeaders = this.config.headers();
+    const fetchFn = (_a = this.config.fetch) != null ? _a : globalThis.fetch;
+    const warnings = [];
+    if (options.filename != null) {
+      warnings.push({ type: "unsupported", feature: "filename" });
+    }
+    const data = options.data;
+    const fileBytes = data instanceof Uint8Array ? data : Uint8Array.from(atob(data), (c) => c.charCodeAt(0));
+    const mediaType = options.mediaType;
+    const displayName = googleOptions == null ? void 0 : googleOptions.displayName;
+    const baseOrigin = this.config.baseURL.replace(/\/v1beta$/, "");
+    const initResponse = await fetchFn(`${baseOrigin}/upload/v1beta/files`, {
+      method: "POST",
+      headers: {
+        ...resolvedHeaders,
+        "X-Goog-Upload-Protocol": "resumable",
+        "X-Goog-Upload-Command": "start",
+        "X-Goog-Upload-Header-Content-Length": String(fileBytes.length),
+        "X-Goog-Upload-Header-Content-Type": mediaType,
+        "Content-Type": "application/json"
+      },
+      body: JSON.stringify({
+        file: {
+          ...displayName != null ? { display_name: displayName } : {}
+        }
+      })
+    });
+    if (!initResponse.ok) {
+      const errorBody = await initResponse.text();
+      throw new AISDKError({
+        name: "GOOGLE_FILES_UPLOAD_ERROR",
+        message: `Failed to initiate resumable upload: ${initResponse.status} ${errorBody}`
+      });
+    }
+    const uploadUrl = initResponse.headers.get("x-goog-upload-url");
+    if (!uploadUrl) {
+      throw new AISDKError({
+        name: "GOOGLE_FILES_UPLOAD_ERROR",
+        message: "No upload URL returned from initiation request"
+      });
+    }
+    const uploadResponse = await fetchFn(uploadUrl, {
+      method: "POST",
+      headers: {
+        "Content-Length": String(fileBytes.length),
+        "X-Goog-Upload-Offset": "0",
+        "X-Goog-Upload-Command": "upload, finalize"
+      },
+      body: fileBytes
+    });
+    if (!uploadResponse.ok) {
+      const errorBody = await uploadResponse.text();
+      throw new AISDKError({
+        name: "GOOGLE_FILES_UPLOAD_ERROR",
+        message: `Failed to upload file data: ${uploadResponse.status} ${errorBody}`
+      });
+    }
+    const uploadResult = await uploadResponse.json();
+    let file = uploadResult.file;
+    const pollIntervalMs = (_b = googleOptions == null ? void 0 : googleOptions.pollIntervalMs) != null ? _b : 2e3;
+    const pollTimeoutMs = (_c = googleOptions == null ? void 0 : googleOptions.pollTimeoutMs) != null ? _c : 3e5;
+    const startTime = Date.now();
+    while (file.state === "PROCESSING") {
+      if (Date.now() - startTime > pollTimeoutMs) {
+        throw new AISDKError({
+          name: "GOOGLE_FILES_UPLOAD_TIMEOUT",
+          message: `File processing timed out after ${pollTimeoutMs}ms`
+        });
+      }
+      await delay(pollIntervalMs);
+      const { value: fileStatus } = await getFromApi({
+        url: `${this.config.baseURL}/${file.name}`,
+        headers: combineHeaders4(resolvedHeaders),
+        successfulResponseHandler: createJsonResponseHandler4(
+          googleFileResponseSchema
+        ),
+        failedResponseHandler: googleFailedResponseHandler,
+        fetch: this.config.fetch
+      });
+      file = fileStatus;
+    }
+    if (file.state === "FAILED") {
+      throw new AISDKError({
+        name: "GOOGLE_FILES_UPLOAD_FAILED",
+        message: `File processing failed for ${file.name}`
+      });
+    }
+    return {
+      warnings,
+      providerReference: { google: file.uri },
+      mediaType: (_d = file.mimeType) != null ? _d : options.mediaType,
+      providerMetadata: {
+        google: {
+          name: file.name,
+          displayName: file.displayName,
+          mimeType: file.mimeType,
+          sizeBytes: file.sizeBytes,
+          state: file.state,
+          uri: file.uri,
+          ...file.createTime != null ? { createTime: file.createTime } : {},
+          ...file.updateTime != null ? { updateTime: file.updateTime } : {},
+          ...file.expirationTime != null ? { expirationTime: file.expirationTime } : {},
+          ...file.sha256Hash != null ? { sha256Hash: file.sha256Hash } : {}
+        }
+      }
+    };
+  }
+};
+var googleFileResponseSchema = lazySchema12(
+  () => zodSchema12(
+    z14.object({
+      name: z14.string(),
+      displayName: z14.string().nullish(),
+      mimeType: z14.string(),
+      sizeBytes: z14.string().nullish(),
+      createTime: z14.string().nullish(),
+      updateTime: z14.string().nullish(),
+      expirationTime: z14.string().nullish(),
+      sha256Hash: z14.string().nullish(),
+      uri: z14.string(),
+      state: z14.string()
+    })
+  )
+);
+var googleFilesUploadOptionsSchema = lazySchema12(
+  () => zodSchema12(
+    z14.object({
+      displayName: z14.string().nullish(),
+      pollIntervalMs: z14.number().positive().nullish(),
+      pollTimeoutMs: z14.number().positive().nullish()
+    }).passthrough()
+  )
+);
+
+// src/google-generative-ai-video-model.ts
+import {
+  AISDKError as AISDKError2
+} from "@ai-sdk/provider";
+import {
+  combineHeaders as combineHeaders5,
+  convertUint8ArrayToBase64 as convertUint8ArrayToBase642,
+  createJsonResponseHandler as createJsonResponseHandler5,
+  delay as delay2,
+  getFromApi as getFromApi2,
+  lazySchema as lazySchema13,
+  parseProviderOptions as parseProviderOptions5,
+  postJsonToApi as postJsonToApi5,
+  resolve as resolve4,
+  zodSchema as zodSchema13
+} from "@ai-sdk/provider-utils";
+import { z as z15 } from "zod/v4";
 var GoogleGenerativeAIVideoModel = class {
   constructor(modelId, config) {
     this.modelId = modelId;
@@ -2573,7 +2775,7 @@ var GoogleGenerativeAIVideoModel = class {
     var _a, _b, _c, _d, _e, _f, _g, _h;
     const currentDate = (_c = (_b = (_a = this.config._internal) == null ? void 0 : _a.currentDate) == null ? void 0 : _b.call(_a)) != null ? _c : /* @__PURE__ */ new Date();
     const warnings = [];
-    const googleOptions = await parseProviderOptions4({
+    const googleOptions = await parseProviderOptions5({
       provider: "google",
       providerOptions: options.providerOptions,
       schema: googleVideoModelOptionsSchema
@@ -2591,7 +2793,7 @@ var GoogleGenerativeAIVideoModel = class {
           details: "Google Generative AI video models require base64-encoded images. URL will be ignored."
         });
       } else {
-        const base64Data = typeof options.image.data === "string" ? options.image.data : convertUint8ArrayToBase64(options.image.data);
+        const base64Data = typeof options.image.data === "string" ? options.image.data : convertUint8ArrayToBase642(options.image.data);
         instance.image = {
           inlineData: {
             mimeType: options.image.mediaType || "image/png",
@@ -2657,9 +2859,9 @@ var GoogleGenerativeAIVideoModel = class {
         }
       }
     }
-    const { value: operation } = await postJsonToApi4({
+    const { value: operation } = await postJsonToApi5({
       url: `${this.config.baseURL}/models/${this.modelId}:predictLongRunning`,
-      headers: combineHeaders4(
+      headers: combineHeaders5(
         await resolve4(this.config.headers),
         options.headers
       ),
@@ -2667,7 +2869,7 @@ var GoogleGenerativeAIVideoModel = class {
         instances,
         parameters
       },
-      successfulResponseHandler: createJsonResponseHandler4(
+      successfulResponseHandler: createJsonResponseHandler5(
         googleOperationSchema
       ),
       failedResponseHandler: googleFailedResponseHandler,
@@ -2676,7 +2878,7 @@ var GoogleGenerativeAIVideoModel = class {
     });
     const operationName = operation.name;
     if (!operationName) {
-      throw new AISDKError({
+      throw new AISDKError2({
         name: "GOOGLE_VIDEO_GENERATION_ERROR",
         message: "No operation name returned from API"
       });
@@ -2688,25 +2890,25 @@ var GoogleGenerativeAIVideoModel = class {
     let responseHeaders;
     while (!finalOperation.done) {
       if (Date.now() - startTime > pollTimeoutMs) {
-        throw new AISDKError({
+        throw new AISDKError2({
           name: "GOOGLE_VIDEO_GENERATION_TIMEOUT",
           message: `Video generation timed out after ${pollTimeoutMs}ms`
         });
       }
-      await delay(pollIntervalMs);
+      await delay2(pollIntervalMs);
       if ((_f = options.abortSignal) == null ? void 0 : _f.aborted) {
-        throw new AISDKError({
+        throw new AISDKError2({
           name: "GOOGLE_VIDEO_GENERATION_ABORTED",
           message: "Video generation request was aborted"
         });
       }
-      const { value: statusOperation, responseHeaders: pollHeaders } = await getFromApi({
+      const { value: statusOperation, responseHeaders: pollHeaders } = await getFromApi2({
         url: `${this.config.baseURL}/${operationName}`,
-        headers: combineHeaders4(
+        headers: combineHeaders5(
           await resolve4(this.config.headers),
           options.headers
         ),
-        successfulResponseHandler: createJsonResponseHandler4(
+        successfulResponseHandler: createJsonResponseHandler5(
           googleOperationSchema
         ),
         failedResponseHandler: googleFailedResponseHandler,
@@ -2717,14 +2919,14 @@ var GoogleGenerativeAIVideoModel = class {
       responseHeaders = pollHeaders;
     }
     if (finalOperation.error) {
-      throw new AISDKError({
+      throw new AISDKError2({
         name: "GOOGLE_VIDEO_GENERATION_FAILED",
         message: `Video generation failed: ${finalOperation.error.message}`
       });
     }
     const response = finalOperation.response;
     if (!((_g = response == null ? void 0 : response.generateVideoResponse) == null ? void 0 : _g.generatedSamples) || response.generateVideoResponse.generatedSamples.length === 0) {
-      throw new AISDKError({
+      throw new AISDKError2({
         name: "GOOGLE_VIDEO_GENERATION_ERROR",
         message: `No videos in response. Response: ${JSON.stringify(finalOperation)}`
       });
@@ -2747,7 +2949,7 @@ var GoogleGenerativeAIVideoModel = class {
       }
     }
     if (videos.length === 0) {
-      throw new AISDKError({
+      throw new AISDKError2({
         name: "GOOGLE_VIDEO_GENERATION_ERROR",
         message: "No valid videos in response"
       });
@@ -2768,37 +2970,37 @@ var GoogleGenerativeAIVideoModel = class {
     };
   }
 };
-var googleOperationSchema = z14.object({
-  name: z14.string().nullish(),
-  done: z14.boolean().nullish(),
-  error: z14.object({
-    code: z14.number().nullish(),
-    message: z14.string(),
-    status: z14.string().nullish()
+var googleOperationSchema = z15.object({
+  name: z15.string().nullish(),
+  done: z15.boolean().nullish(),
+  error: z15.object({
+    code: z15.number().nullish(),
+    message: z15.string(),
+    status: z15.string().nullish()
   }).nullish(),
-  response: z14.object({
-    generateVideoResponse: z14.object({
-      generatedSamples: z14.array(
-        z14.object({
-          video: z14.object({
-            uri: z14.string().nullish()
+  response: z15.object({
+    generateVideoResponse: z15.object({
+      generatedSamples: z15.array(
+        z15.object({
+          video: z15.object({
+            uri: z15.string().nullish()
           }).nullish()
         })
       ).nullish()
     }).nullish()
   }).nullish()
 });
-var googleVideoModelOptionsSchema = lazySchema12(
-  () => zodSchema12(
-    z14.object({
-      pollIntervalMs: z14.number().positive().nullish(),
-      pollTimeoutMs: z14.number().positive().nullish(),
-      personGeneration: z14.enum(["dont_allow", "allow_adult", "allow_all"]).nullish(),
-      negativePrompt: z14.string().nullish(),
-      referenceImages: z14.array(
-        z14.object({
-          bytesBase64Encoded: z14.string().nullish(),
-          gcsUri: z14.string().nullish()
+var googleVideoModelOptionsSchema = lazySchema13(
+  () => zodSchema13(
+    z15.object({
+      pollIntervalMs: z15.number().positive().nullish(),
+      pollTimeoutMs: z15.number().positive().nullish(),
+      personGeneration: z15.enum(["dont_allow", "allow_adult", "allow_all"]).nullish(),
+      negativePrompt: z15.string().nullish(),
+      referenceImages: z15.array(
+        z15.object({
+          bytesBase64Encoded: z15.string().nullish(),
+          gcsUri: z15.string().nullish()
         })
       ).nullish()
     }).passthrough()
@@ -2855,6 +3057,12 @@ function createGoogleGenerativeAI(options = {}) {
     headers: getHeaders,
     fetch: options.fetch
   });
+  const createFiles = () => new GoogleGenerativeAIFiles({
+    provider: providerName,
+    baseURL,
+    headers: getHeaders,
+    fetch: options.fetch
+  });
   const createVideoModel = (modelId) => {
     var _a2;
     return new GoogleGenerativeAIVideoModel(modelId, {
@@ -2885,6 +3093,7 @@ function createGoogleGenerativeAI(options = {}) {
   provider.imageModel = createImageModel;
   provider.video = createVideoModel;
   provider.videoModel = createVideoModel;
+  provider.files = createFiles;
   provider.tools = googleTools;
   return provider;
 }