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

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.3" : "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;
@@ -840,7 +844,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 +940,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 +994,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 +1168,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 +1299,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 +1315,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
@@ -1461,14 +1475,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 +1541,7 @@ function createGoogleGenerativeAI(options = {}) {
 }
 var google = createGoogleGenerativeAI();
 export {
+  VERSION,
   createGoogleGenerativeAI,
   google
 };