@ai-sdk/amazon-bedrock 4.0.165 → 4.0.166

package/dist/index.mjs

@@ -806,7 +806,7 @@ async function convertToBedrockChatMessages(prompt, isMistral = false) {
           }
           pushCachePoint(bedrockContent, message.providerOptions);
         }
-        if (bedrockContent.length > 0) {
+        if (bedrockContent.some((block2) => !("cachePoint" in block2))) {
           messages.push({ role: "assistant", content: bedrockContent });
         }
         break;
@@ -945,7 +945,7 @@ var BedrockChatLanguageModel = class {
     toolChoice,
     providerOptions
   }) {
-    var _a, _b, _c, _d, _e, _f, _g, _h, _i, _j, _k, _l, _m, _n, _o, _p;
+    var _a, _b, _c, _d, _e, _f, _g, _h, _i, _j, _k, _l, _m, _n, _o, _p, _q;
     const bedrockOptions = (_a = await parseProviderOptions2({
       provider: "bedrock",
       providerOptions,
@@ -997,11 +997,11 @@ var BedrockChatLanguageModel = class {
         details: "Only text and json response formats are supported."
       });
     }
-    const isAnthropicModel = this.modelId.includes("anthropic");
-    const openAIModelId = (_b = /^(?:[^.]+\.)?(openai\..+)$/.exec(this.modelId)) == null ? void 0 : _b[1];
+    const isAnthropicModel = this.modelId.includes("anthropic") || this.modelId.includes(":application-inference-profile/") && ((_b = bedrockOptions.reasoningConfig) == null ? void 0 : _b.budgetTokens) != null;
+    const openAIModelId = (_c = /^(?:[^.]+\.)?(openai\..+)$/.exec(this.modelId)) == null ? void 0 : _c[1];
     const isOpenAIModel = openAIModelId != null;
-    const isOpenAIGptOssModel = (_c = openAIModelId == null ? void 0 : openAIModelId.startsWith("openai.gpt-oss-")) != null ? _c : false;
-    const isThinkingEnabled = ((_d = bedrockOptions.reasoningConfig) == null ? void 0 : _d.type) === "enabled" || ((_e = bedrockOptions.reasoningConfig) == null ? void 0 : _e.type) === "adaptive";
+    const isOpenAIGptOssModel = (_d = openAIModelId == null ? void 0 : openAIModelId.startsWith("openai.gpt-oss-")) != null ? _d : false;
+    const isThinkingEnabled = ((_e = bedrockOptions.reasoningConfig) == null ? void 0 : _e.type) === "enabled" || ((_f = bedrockOptions.reasoningConfig) == null ? void 0 : _f.type) === "adaptive";
     const { supportsStructuredOutput: modelSupportsStructuredOutput } = getModelCapabilities(this.modelId);
     const useNativeStructuredOutput = isAnthropicModel && supportsNativeStructuredOutput(this.modelId) && (modelSupportsStructuredOutput || isThinkingEnabled) && (responseFormat == null ? void 0 : responseFormat.type) === "json" && responseFormat.schema != null;
     const useJsonInstructionForStructuredOutput = isAnthropicModel && !supportsStrictTools(this.modelId) && (responseFormat == null ? void 0 : responseFormat.type) === "json" && responseFormat.schema != null && tools != null && tools.length > 0;
@@ -1025,16 +1025,16 @@ var BedrockChatLanguageModel = class {
       };
     }
     if (betas.size > 0 || bedrockOptions.anthropicBeta) {
-      const existingBetas = (_f = bedrockOptions.anthropicBeta) != null ? _f : [];
+      const existingBetas = (_g = bedrockOptions.anthropicBeta) != null ? _g : [];
       const mergedBetas = betas.size > 0 ? [...existingBetas, ...Array.from(betas)] : existingBetas;
       bedrockOptions.additionalModelRequestFields = {
         ...bedrockOptions.additionalModelRequestFields,
         anthropic_beta: mergedBetas
       };
     }
-    const thinkingType = (_g = bedrockOptions.reasoningConfig) == null ? void 0 : _g.type;
-    const thinkingBudget = thinkingType === "enabled" ? (_h = bedrockOptions.reasoningConfig) == null ? void 0 : _h.budgetTokens : void 0;
-    const thinkingDisplay = thinkingType === "adaptive" ? (_i = bedrockOptions.reasoningConfig) == null ? void 0 : _i.display : void 0;
+    const thinkingType = (_h = bedrockOptions.reasoningConfig) == null ? void 0 : _h.type;
+    const thinkingBudget = thinkingType === "enabled" ? (_i = bedrockOptions.reasoningConfig) == null ? void 0 : _i.budgetTokens : void 0;
+    const thinkingDisplay = thinkingType === "adaptive" ? (_j = bedrockOptions.reasoningConfig) == null ? void 0 : _j.display : void 0;
     const isAnthropicThinkingEnabled = isAnthropicModel && isThinkingEnabled;
     const inferenceConfig = {
       ...maxOutputTokens != null && { maxTokens: maxOutputTokens },
@@ -1067,7 +1067,7 @@ var BedrockChatLanguageModel = class {
         };
       }
     } else if (!isAnthropicModel) {
-      if (((_j = bedrockOptions.reasoningConfig) == null ? void 0 : _j.budgetTokens) != null) {
+      if (((_k = bedrockOptions.reasoningConfig) == null ? void 0 : _k.budgetTokens) != null) {
         warnings.push({
           type: "unsupported",
           feature: "budgetTokens",
@@ -1082,13 +1082,13 @@ var BedrockChatLanguageModel = class {
         });
       }
     }
-    const maxReasoningEffort = (_k = bedrockOptions.reasoningConfig) == null ? void 0 : _k.maxReasoningEffort;
+    const maxReasoningEffort = (_l = bedrockOptions.reasoningConfig) == null ? void 0 : _l.maxReasoningEffort;
     if (maxReasoningEffort != null) {
       if (isAnthropicModel) {
         bedrockOptions.additionalModelRequestFields = {
           ...bedrockOptions.additionalModelRequestFields,
           output_config: {
-            ...(_l = bedrockOptions.additionalModelRequestFields) == null ? void 0 : _l.output_config,
+            ...(_m = bedrockOptions.additionalModelRequestFields) == null ? void 0 : _m.output_config,
             effort: maxReasoningEffort
           }
         };
@@ -1099,7 +1099,7 @@ var BedrockChatLanguageModel = class {
         } : {
           ...bedrockOptions.additionalModelRequestFields,
           reasoning: {
-            ...(_m = bedrockOptions.additionalModelRequestFields) == null ? void 0 : _m.reasoning,
+            ...(_n = bedrockOptions.additionalModelRequestFields) == null ? void 0 : _n.reasoning,
             effort: maxReasoningEffort
           }
         };
@@ -1118,7 +1118,7 @@ var BedrockChatLanguageModel = class {
       bedrockOptions.additionalModelRequestFields = {
         ...bedrockOptions.additionalModelRequestFields,
         output_config: {
-          ...(_n = bedrockOptions.additionalModelRequestFields) == null ? void 0 : _n.output_config,
+          ...(_o = bedrockOptions.additionalModelRequestFields) == null ? void 0 : _o.output_config,
           format: {
             type: "json_schema",
             schema: sanitizeJsonSchema(responseFormat.schema)
@@ -1150,7 +1150,7 @@ var BedrockChatLanguageModel = class {
         details: "topK is not supported when thinking is enabled"
       });
     }
-    const hasAnyTools = ((_p = (_o = toolConfig.tools) == null ? void 0 : _o.length) != null ? _p : 0) > 0 || additionalTools;
+    const hasAnyTools = ((_q = (_p = toolConfig.tools) == null ? void 0 : _p.length) != null ? _q : 0) > 0 || additionalTools;
     let filteredPrompt = prompt;
     if (!hasAnyTools) {
       const hasToolContent = prompt.some(
@@ -1833,6 +1833,9 @@ var BedrockStreamSchema = z4.object({
     contentBlockIndex: z4.number(),
     delta: z4.union([
       z4.object({ text: z4.string() }),
+      z4.object({
+        citation: z4.record(z4.string(), z4.unknown())
+      }),
       z4.object({ toolUse: z4.object({ input: z4.string() }) }),
       z4.object({
         reasoningContent: z4.object({ text: z4.string() })
@@ -1962,7 +1965,11 @@ var BedrockEmbeddingModel = class {
     this.supportsParallelCalls = true;
   }
   get maxEmbeddingsPerCall() {
-    return isCohereEmbeddingModel(this.modelId) ? 96 : 1;
+    return this.modelFamily === "cohere" ? 96 : 1;
+  }
+  get modelFamily() {
+    var _a;
+    return (_a = this.config.modelFamily) != null ? _a : detectEmbeddingModelFamily(this.modelId);
   }
   getUrl(modelId) {
     const encodedModelId = encodeURIComponent(modelId);
@@ -1988,9 +1995,8 @@ var BedrockEmbeddingModel = class {
       providerOptions,
       schema: amazonBedrockEmbeddingModelOptionsSchema
     })) != null ? _a : {};
-    const isNovaModel = isNovaEmbeddingModel(this.modelId);
-    const isCohereModel = isCohereEmbeddingModel(this.modelId);
-    const args = isNovaModel ? {
+    const modelFamily = this.modelFamily;
+    const args = modelFamily === "nova" ? {
       taskType: "SINGLE_EMBEDDING",
       singleEmbeddingParams: {
         embeddingPurpose: (_b = bedrockOptions.embeddingPurpose) != null ? _b : "GENERIC_INDEX",
@@ -2000,7 +2006,7 @@ var BedrockEmbeddingModel = class {
           value: values[0]
         }
       }
-    } : isCohereModel ? {
+    } : modelFamily === "cohere" ? {
       // Cohere embedding models on Bedrock require `input_type`.
       // Without it, the service attempts other schema branches and rejects the request.
       input_type: (_e = bedrockOptions.inputType) != null ? _e : "search_query",
@@ -2057,7 +2063,10 @@ function isCohereEmbeddingModel(modelId) {
   return modelId.includes("cohere.embed-");
 }
 function isNovaEmbeddingModel(modelId) {
-  return modelId.startsWith("amazon.nova-") && modelId.includes("embed");
+  return modelId.includes("amazon.nova-") && modelId.includes("embed");
+}
+function detectEmbeddingModelFamily(modelId) {
+  return isNovaEmbeddingModel(modelId) ? "nova" : isCohereEmbeddingModel(modelId) ? "cohere" : "titan";
 }
 var BedrockEmbeddingResponseSchema = z6.union([
   // Titan-style response
@@ -2311,7 +2320,7 @@ import {
 import { AwsV4Signer } from "aws4fetch";
 
 // src/version.ts
-var VERSION = true ? "4.0.165" : "0.0.0-test";
+var VERSION = true ? "4.0.166" : "0.0.0-test";
 
 // src/bedrock-sigv4-fetch.ts
 function createSigV4FetchFunction(getCredentials, fetch, service = "bedrock") {
@@ -2629,10 +2638,11 @@ Original error: ${errorMessage}`
     }
     return createChatModel(modelId);
   };
-  const createEmbeddingModel = (modelId) => new BedrockEmbeddingModel(modelId, {
+  const createEmbeddingModel = (modelId, settings = {}) => new BedrockEmbeddingModel(modelId, {
     baseUrl: getBedrockRuntimeBaseUrl,
     headers: getHeaders,
-    fetch: fetchFunction
+    fetch: fetchFunction,
+    modelFamily: settings.modelFamily
   });
   const createImageModel = (modelId) => new BedrockImageModel(modelId, {
     baseUrl: getBedrockRuntimeBaseUrl,