@ai-sdk/google 2.1.0-beta.1 → 2.1.0-beta.11

package/dist/index.mjs

@@ -2,9 +2,13 @@
 import {
   generateId as generateId2,
   loadApiKey,
-  withoutTrailingSlash
+  withoutTrailingSlash,
+  withUserAgentSuffix
 } from "@ai-sdk/provider-utils";
 
+// src/version.ts
+var VERSION = true ? "2.1.0-beta.11" : "0.0.0-test";
+
 // src/google-generative-ai-embedding-model.ts
 import {
   TooManyEmbeddingValuesForCallError
@@ -68,7 +72,7 @@ var googleGenerativeAIEmbeddingProviderOptions = z2.object({
 // src/google-generative-ai-embedding-model.ts
 var GoogleGenerativeAIEmbeddingModel = class {
   constructor(modelId, config) {
-    this.specificationVersion = "v2";
+    this.specificationVersion = "v3";
     this.maxEmbeddingsPerCall = 2048;
     this.supportsParallelCalls = true;
     this.modelId = modelId;
@@ -277,7 +281,7 @@ import {
 } from "@ai-sdk/provider";
 import { convertToBase64 } from "@ai-sdk/provider-utils";
 function convertToGoogleGenerativeAIMessages(prompt, options) {
-  var _a;
+  var _a, _b;
   const systemInstructionParts = [];
   const contents = [];
   let systemMessagesAllowed = true;
@@ -329,12 +333,12 @@ function convertToGoogleGenerativeAIMessages(prompt, options) {
         contents.push({
           role: "model",
           parts: content.map((part) => {
-            var _a2, _b, _c, _d, _e, _f;
+            var _a2, _b2, _c, _d, _e, _f;
             switch (part.type) {
               case "text": {
                 return part.text.length === 0 ? void 0 : {
                   text: part.text,
-                  thoughtSignature: (_b = (_a2 = part.providerOptions) == null ? void 0 : _a2.google) == null ? void 0 : _b.thoughtSignature
+                  thoughtSignature: (_b2 = (_a2 = part.providerOptions) == null ? void 0 : _a2.google) == null ? void 0 : _b2.thoughtSignature
                 };
               }
               case "reasoning": {
@@ -419,7 +423,7 @@ function convertToGoogleGenerativeAIMessages(prompt, options) {
                 name: part.toolName,
                 response: {
                   name: part.toolName,
-                  content: output.value
+                  content: output.type === "execution-denied" ? (_b = output.reason) != null ? _b : "Tool execution denied." : output.value
                 }
               }
             });
@@ -513,7 +517,18 @@ var googleGenerativeAIProviderOptions = z4.object({
    *
    * https://cloud.google.com/vertex-ai/generative-ai/docs/multimodal/add-labels-to-api-calls
    */
-  labels: z4.record(z4.string(), z4.string()).optional()
+  labels: z4.record(z4.string(), z4.string()).optional(),
+  /**
+   * Optional. If specified, the media resolution specified will be used.
+   *
+   * https://ai.google.dev/api/generate-content#MediaResolution
+   */
+  mediaResolution: z4.enum([
+    "MEDIA_RESOLUTION_UNSPECIFIED",
+    "MEDIA_RESOLUTION_LOW",
+    "MEDIA_RESOLUTION_MEDIUM",
+    "MEDIA_RESOLUTION_HIGH"
+  ]).optional()
 });
 
 // src/google-prepare-tools.ts
@@ -528,7 +543,12 @@ function prepareTools({
   var _a;
   tools = (tools == null ? void 0 : tools.length) ? tools : void 0;
   const toolWarnings = [];
-  const isGemini2 = modelId.includes("gemini-2");
+  const isLatest = [
+    "gemini-flash-latest",
+    "gemini-flash-lite-latest",
+    "gemini-pro-latest"
+  ].some((id) => id === modelId);
+  const isGemini2 = modelId.includes("gemini-2") || isLatest;
   const supportsDynamicRetrieval = modelId.includes("gemini-1.5-flash") && !modelId.includes("-8b");
   if (tools == null) {
     return { tools: void 0, toolConfig: void 0, toolWarnings };
@@ -748,7 +768,7 @@ var urlContext = createProviderDefinedToolFactory2({
 // src/google-generative-ai-language-model.ts
 var GoogleGenerativeAILanguageModel = class {
   constructor(modelId, config) {
-    this.specificationVersion = "v2";
+    this.specificationVersion = "v3";
     var _a;
     this.modelId = modelId;
     this.config = config;
@@ -826,7 +846,10 @@ var GoogleGenerativeAILanguageModel = class {
           },
           // provider options:
           responseModalities: googleOptions == null ? void 0 : googleOptions.responseModalities,
-          thinkingConfig: googleOptions == null ? void 0 : googleOptions.thinkingConfig
+          thinkingConfig: googleOptions == null ? void 0 : googleOptions.thinkingConfig,
+          ...(googleOptions == null ? void 0 : googleOptions.mediaResolution) && {
+            mediaResolution: googleOptions.mediaResolution
+          }
         },
         contents,
         systemInstruction: isGemmaModel ? void 0 : systemInstruction,
@@ -840,7 +863,7 @@ var GoogleGenerativeAILanguageModel = class {
     };
   }
   async doGenerate(options) {
-    var _a, _b, _c, _d, _e, _f, _g, _h, _i, _j, _k, _l;
+    var _a, _b, _c, _d, _e, _f, _g, _h, _i, _j, _k, _l, _m;
     const { args, warnings } = await this.getArgs(options);
     const body = JSON.stringify(args);
     const mergedHeaders = combineHeaders2(
@@ -936,9 +959,10 @@ var GoogleGenerativeAILanguageModel = class {
       warnings,
       providerMetadata: {
         google: {
-          groundingMetadata: (_j = candidate.groundingMetadata) != null ? _j : null,
-          urlContextMetadata: (_k = candidate.urlContextMetadata) != null ? _k : null,
-          safetyRatings: (_l = candidate.safetyRatings) != null ? _l : null,
+          promptFeedback: (_j = response.promptFeedback) != null ? _j : null,
+          groundingMetadata: (_k = candidate.groundingMetadata) != null ? _k : null,
+          urlContextMetadata: (_l = candidate.urlContextMetadata) != null ? _l : null,
+          safetyRatings: (_m = candidate.safetyRatings) != null ? _m : null,
           usageMetadata: usageMetadata != null ? usageMetadata : null
         }
       },
@@ -989,7 +1013,7 @@ var GoogleGenerativeAILanguageModel = class {
             controller.enqueue({ type: "stream-start", warnings });
           },
           transform(chunk, controller) {
-            var _a, _b, _c, _d, _e, _f, _g, _h, _i, _j, _k;
+            var _a, _b, _c, _d, _e, _f, _g, _h, _i, _j, _k, _l;
             if (options.includeRawChunks) {
               controller.enqueue({ type: "raw", rawValue: chunk.rawValue });
             }
@@ -1163,9 +1187,10 @@ var GoogleGenerativeAILanguageModel = class {
               });
               providerMetadata = {
                 google: {
-                  groundingMetadata: (_i = candidate.groundingMetadata) != null ? _i : null,
-                  urlContextMetadata: (_j = candidate.urlContextMetadata) != null ? _j : null,
-                  safetyRatings: (_k = candidate.safetyRatings) != null ? _k : null
+                  promptFeedback: (_i = value.promptFeedback) != null ? _i : null,
+                  groundingMetadata: (_j = candidate.groundingMetadata) != null ? _j : null,
+                  urlContextMetadata: (_k = candidate.urlContextMetadata) != null ? _k : null,
+                  safetyRatings: (_l = candidate.safetyRatings) != null ? _l : null
                 }
               };
               if (usageMetadata != null) {
@@ -1293,7 +1318,11 @@ var responseSchema = z7.object({
       urlContextMetadata: urlContextMetadataSchema.nullish()
     })
   ),
-  usageMetadata: usageSchema.nullish()
+  usageMetadata: usageSchema.nullish(),
+  promptFeedback: z7.object({
+    blockReason: z7.string().nullish(),
+    safetyRatings: z7.array(safetyRatingSchema).nullish()
+  }).nullish()
 });
 var chunkSchema = z7.object({
   candidates: z7.array(
@@ -1305,7 +1334,11 @@ var chunkSchema = z7.object({
       urlContextMetadata: urlContextMetadataSchema.nullish()
     })
   ).nullish(),
-  usageMetadata: usageSchema.nullish()
+  usageMetadata: usageSchema.nullish(),
+  promptFeedback: z7.object({
+    blockReason: z7.string().nullish(),
+    safetyRatings: z7.array(safetyRatingSchema).nullish()
+  }).nullish()
 });
 
 // src/tool/code-execution.ts
@@ -1363,7 +1396,7 @@ var GoogleGenerativeAIImageModel = class {
     this.modelId = modelId;
     this.settings = settings;
     this.config = config;
-    this.specificationVersion = "v2";
+    this.specificationVersion = "v3";
   }
   get maxImagesPerCall() {
     var _a;
@@ -1461,14 +1494,17 @@ var googleImageProviderOptionsSchema = z9.object({
 function createGoogleGenerativeAI(options = {}) {
   var _a;
   const baseURL = (_a = withoutTrailingSlash(options.baseURL)) != null ? _a : "https://generativelanguage.googleapis.com/v1beta";
-  const getHeaders = () => ({
-    "x-goog-api-key": loadApiKey({
-      apiKey: options.apiKey,
-      environmentVariableName: "GOOGLE_GENERATIVE_AI_API_KEY",
-      description: "Google Generative AI"
-    }),
-    ...options.headers
-  });
+  const getHeaders = () => withUserAgentSuffix(
+    {
+      "x-goog-api-key": loadApiKey({
+        apiKey: options.apiKey,
+        environmentVariableName: "GOOGLE_GENERATIVE_AI_API_KEY",
+        description: "Google Generative AI"
+      }),
+      ...options.headers
+    },
+    `ai-sdk/google/${VERSION}`
+  );
   const createChatModel = (modelId) => {
     var _a2;
     return new GoogleGenerativeAILanguageModel(modelId, {
@@ -1524,6 +1560,7 @@ function createGoogleGenerativeAI(options = {}) {
 }
 var google = createGoogleGenerativeAI();
 export {
+  VERSION,
   createGoogleGenerativeAI,
   google
 };