@ai-sdk/amazon-bedrock 3.1.0-beta.1 → 3.1.0-beta.3

package/dist/index.mjs

@@ -3,8 +3,14 @@ import {
   generateId,
   loadOptionalSetting,
   loadSetting,
-  withoutTrailingSlash
+  withoutTrailingSlash,
+  withUserAgentSuffix as withUserAgentSuffix2
 } from "@ai-sdk/provider-utils";
+
+// src/version.ts
+var VERSION = true ? "3.1.0-beta.3" : "0.0.0-test";
+
+// src/bedrock-provider.ts
 import { anthropicTools as anthropicTools2 } from "@ai-sdk/anthropic/internal";
 
 // src/bedrock-chat-language-model.ts
@@ -1308,7 +1314,7 @@ var BedrockEmbeddingModel = class {
   constructor(modelId, config) {
     this.modelId = modelId;
     this.config = config;
-    this.specificationVersion = "v2";
+    this.specificationVersion = "v3";
     this.provider = "amazon-bedrock";
     this.maxEmbeddingsPerCall = 1;
     this.supportsParallelCalls = true;
@@ -1390,7 +1396,7 @@ var BedrockImageModel = class {
   constructor(modelId, config) {
     this.modelId = modelId;
     this.config = config;
-    this.specificationVersion = "v2";
+    this.specificationVersion = "v3";
     this.provider = "amazon-bedrock";
   }
   get maxImagesPerCall() {
@@ -1502,11 +1508,6 @@ import {
   getRuntimeEnvironmentUserAgent
 } from "@ai-sdk/provider-utils";
 import { AwsV4Signer } from "aws4fetch";
-
-// src/version.ts
-var VERSION = true ? "3.1.0-beta.1" : "0.0.0-test";
-
-// src/bedrock-sigv4-fetch.ts
 function createSigV4FetchFunction(getCredentials, fetch = globalThis.fetch) {
   return async (input, init) => {
     var _a;
@@ -1654,15 +1655,17 @@ Original error: ${errorMessage}`
       })}.amazonaws.com`
     )) != null ? _b : `https://bedrock-runtime.us-east-1.amazonaws.com`;
   };
-  const createChatModel = (modelId) => {
+  const getHeaders = () => {
     var _a;
-    return new BedrockChatLanguageModel(modelId, {
-      baseUrl: getBaseUrl,
-      headers: (_a = options.headers) != null ? _a : {},
-      fetch: fetchFunction,
-      generateId
-    });
+    const baseHeaders = (_a = options.headers) != null ? _a : {};
+    return withUserAgentSuffix2(baseHeaders, `ai-sdk/amazon-bedrock/${VERSION}`);
   };
+  const createChatModel = (modelId) => new BedrockChatLanguageModel(modelId, {
+    baseUrl: getBaseUrl,
+    headers: getHeaders,
+    fetch: fetchFunction,
+    generateId
+  });
   const provider = function(modelId) {
     if (new.target) {
       throw new Error(
@@ -1671,22 +1674,16 @@ Original error: ${errorMessage}`
     }
     return createChatModel(modelId);
   };
-  const createEmbeddingModel = (modelId) => {
-    var _a;
-    return new BedrockEmbeddingModel(modelId, {
-      baseUrl: getBaseUrl,
-      headers: (_a = options.headers) != null ? _a : {},
-      fetch: fetchFunction
-    });
-  };
-  const createImageModel = (modelId) => {
-    var _a;
-    return new BedrockImageModel(modelId, {
-      baseUrl: getBaseUrl,
-      headers: (_a = options.headers) != null ? _a : {},
-      fetch: fetchFunction
-    });
-  };
+  const createEmbeddingModel = (modelId) => new BedrockEmbeddingModel(modelId, {
+    baseUrl: getBaseUrl,
+    headers: getHeaders,
+    fetch: fetchFunction
+  });
+  const createImageModel = (modelId) => new BedrockImageModel(modelId, {
+    baseUrl: getBaseUrl,
+    headers: getHeaders,
+    fetch: fetchFunction
+  });
   provider.languageModel = createChatModel;
   provider.embedding = createEmbeddingModel;
   provider.textEmbedding = createEmbeddingModel;