@ai-sdk/google 4.0.10 → 4.0.11

package/dist/internal/index.js

@@ -274,7 +274,8 @@ function appendLegacyToolResultParts(parts, toolName, outputValue, toolCallId) {
         });
         break;
       case "file": {
-        if (contentPart.data.type === "data" && getTopLevelMediaType(contentPart.mediaType) === "image") {
+        if (contentPart.data.type === "data") {
+          const topLevelMediaType = getTopLevelMediaType(contentPart.mediaType);
           parts.push(
             {
               inlineData: {
@@ -283,7 +284,7 @@ function appendLegacyToolResultParts(parts, toolName, outputValue, toolCallId) {
               }
             },
             {
-              text: "Tool executed successfully and returned this image as a response"
+              text: `Tool executed successfully and returned this ${topLevelMediaType === "image" ? "image" : "file"} as a response`
             }
           );
         } else {
@@ -1329,6 +1330,12 @@ function mapGoogleFinishReason({
 }
 
 // src/google-language-model.ts
+var configurableSafetySettingCategories = [
+  "HARM_CATEGORY_HATE_SPEECH",
+  "HARM_CATEGORY_DANGEROUS_CONTENT",
+  "HARM_CATEGORY_HARASSMENT",
+  "HARM_CATEGORY_SEXUALLY_EXPLICIT"
+];
 var GoogleLanguageModel = class _GoogleLanguageModel {
   constructor(modelId, config) {
     this.specificationVersion = "v4";
@@ -1369,7 +1376,7 @@ var GoogleLanguageModel = class _GoogleLanguageModel {
     reasoning,
     providerOptions
   }, { isStreaming = false } = {}) {
-    var _a, _b;
+    var _a, _b, _c;
     const warnings = [];
     const providerOptionsNames = this.config.provider.includes("vertex") ? ["googleVertex", "vertex"] : ["google"];
     let googleOptions;
@@ -1451,6 +1458,11 @@ var GoogleLanguageModel = class _GoogleLanguageModel {
     });
     const thinkingConfig = (googleOptions == null ? void 0 : googleOptions.thinkingConfig) || resolvedThinking ? { ...resolvedThinking, ...googleOptions == null ? void 0 : googleOptions.thinkingConfig } : void 0;
     const streamFunctionCallArguments = isStreaming && isVertexProvider ? (_a = googleOptions == null ? void 0 : googleOptions.streamFunctionCallArguments) != null ? _a : false : void 0;
+    const safetyThreshold = googleOptions == null ? void 0 : googleOptions.threshold;
+    const safetySettings = (_b = googleOptions == null ? void 0 : googleOptions.safetySettings) != null ? _b : safetyThreshold != null ? configurableSafetySettingCategories.map((category) => ({
+      category,
+      threshold: safetyThreshold
+    })) : void 0;
     const toolConfig = googleToolConfig || streamFunctionCallArguments || (googleOptions == null ? void 0 : googleOptions.retrievalConfig) ? {
       ...googleToolConfig,
       ...streamFunctionCallArguments && {
@@ -1480,7 +1492,7 @@ var GoogleLanguageModel = class _GoogleLanguageModel {
           responseSchema: (responseFormat == null ? void 0 : responseFormat.type) === "json" && responseFormat.schema != null && // Google GenAI does not support all OpenAPI Schema features,
           // so this is needed as an escape hatch:
           // TODO convert into provider option
-          ((_b = googleOptions == null ? void 0 : googleOptions.structuredOutputs) != null ? _b : true) ? convertJSONSchemaToOpenAPISchema(responseFormat.schema) : void 0,
+          ((_c = googleOptions == null ? void 0 : googleOptions.structuredOutputs) != null ? _c : true) ? convertJSONSchemaToOpenAPISchema(responseFormat.schema) : void 0,
           ...(googleOptions == null ? void 0 : googleOptions.audioTimestamp) && {
             audioTimestamp: googleOptions.audioTimestamp
           },
@@ -1496,7 +1508,7 @@ var GoogleLanguageModel = class _GoogleLanguageModel {
         },
         contents,
         systemInstruction: isGemmaModel ? void 0 : systemInstruction,
-        safetySettings: googleOptions == null ? void 0 : googleOptions.safetySettings,
+        safetySettings,
         tools: googleTools2,
         toolConfig,
         cachedContent: googleOptions == null ? void 0 : googleOptions.cachedContent,