@ai-sdk/amazon-bedrock 3.0.21 → 3.0.23

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.0.23" : "0.0.0-test";
+
+// src/bedrock-provider.ts
 import { anthropicTools as anthropicTools2 } from "@ai-sdk/anthropic/internal";
 
 // src/bedrock-chat-language-model.ts
@@ -1498,23 +1504,32 @@ function convertHeadersToRecord(headers) {
 // src/bedrock-sigv4-fetch.ts
 import {
   combineHeaders as combineHeaders4,
-  removeUndefinedEntries
+  withUserAgentSuffix,
+  getRuntimeEnvironmentUserAgent
 } from "@ai-sdk/provider-utils";
 import { AwsV4Signer } from "aws4fetch";
 function createSigV4FetchFunction(getCredentials, fetch = globalThis.fetch) {
   return async (input, init) => {
     var _a;
+    const originalHeaders = extractHeaders(init == null ? void 0 : init.headers);
+    const headersWithUserAgent = withUserAgentSuffix(
+      originalHeaders,
+      `ai-sdk/amazon-bedrock/${VERSION}`,
+      getRuntimeEnvironmentUserAgent()
+    );
     if (((_a = init == null ? void 0 : init.method) == null ? void 0 : _a.toUpperCase()) !== "POST" || !(init == null ? void 0 : init.body)) {
-      return fetch(input, init);
+      return fetch(input, {
+        ...init,
+        headers: headersWithUserAgent
+      });
     }
     const url = typeof input === "string" ? input : input instanceof URL ? input.href : input.url;
-    const originalHeaders = extractHeaders(init.headers);
     const body = prepareBodyString(init.body);
     const credentials = await getCredentials();
     const signer = new AwsV4Signer({
       url,
       method: "POST",
-      headers: Object.entries(removeUndefinedEntries(originalHeaders)),
+      headers: Object.entries(headersWithUserAgent),
       body,
       region: credentials.region,
       accessKeyId: credentials.accessKeyId,
@@ -1524,12 +1539,11 @@ function createSigV4FetchFunction(getCredentials, fetch = globalThis.fetch) {
     });
     const signingResult = await signer.sign();
     const signedHeaders = convertHeadersToRecord(signingResult.headers);
+    const combinedHeaders = combineHeaders4(headersWithUserAgent, signedHeaders);
     return fetch(input, {
       ...init,
       body,
-      headers: removeUndefinedEntries(
-        combineHeaders4(originalHeaders, signedHeaders)
-      )
+      headers: combinedHeaders
     });
   };
 }
@@ -1547,13 +1561,17 @@ function prepareBodyString(body) {
 function createApiKeyFetchFunction(apiKey, fetch = globalThis.fetch) {
   return async (input, init) => {
     const originalHeaders = extractHeaders(init == null ? void 0 : init.headers);
+    const headersWithUserAgent = withUserAgentSuffix(
+      originalHeaders,
+      `ai-sdk/amazon-bedrock/${VERSION}`,
+      getRuntimeEnvironmentUserAgent()
+    );
+    const finalHeaders = combineHeaders4(headersWithUserAgent, {
+      Authorization: `Bearer ${apiKey}`
+    });
     return fetch(input, {
       ...init,
-      headers: removeUndefinedEntries(
-        combineHeaders4(originalHeaders, {
-          Authorization: `Bearer ${apiKey}`
-        })
-      )
+      headers: finalHeaders
     });
   };
 }
@@ -1637,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(
@@ -1654,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;
@@ -1681,6 +1695,7 @@ Original error: ${errorMessage}`
 }
 var bedrock = createAmazonBedrock();
 export {
+  VERSION,
   bedrock,
   createAmazonBedrock
 };