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

package/dist/internal/index.mjs

@@ -174,7 +174,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);
@@ -309,19 +313,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;
             }
           }
@@ -372,6 +393,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,