@ai-sdk/amazon-bedrock 4.0.114 → 4.0.115

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
package/CHANGELOG.md CHANGED
@@ -1,5 +1,11 @@
1
1
  # @ai-sdk/amazon-bedrock
2
2
 
3
+ ## 4.0.115
4
+
5
+ ### Patch Changes
6
+
7
+ - c97ede5: fix(provider/amazon-bedrock): extract Cohere embedding token usage from response header
8
+
3
9
  ## 4.0.114
4
10
 
5
11
  ### Patch Changes
@@ -35,7 +35,7 @@ var import_provider_utils = require("@ai-sdk/provider-utils");
35
35
  var import_aws4fetch = require("aws4fetch");
36
36
 
37
37
  // src/version.ts
38
- var VERSION = true ? "4.0.114" : "0.0.0-test";
38
+ var VERSION = true ? "4.0.115" : "0.0.0-test";
39
39
 
40
40
  // src/bedrock-sigv4-fetch.ts
41
41
  function createSigV4FetchFunction(getCredentials, fetch, service = "bedrock") {
@@ -24,7 +24,7 @@ import {
24
24
  import { AwsV4Signer } from "aws4fetch";
25
25
 
26
26
  // src/version.ts
27
- var VERSION = true ? "4.0.114" : "0.0.0-test";
27
+ var VERSION = true ? "4.0.115" : "0.0.0-test";
28
28
 
29
29
  // src/bedrock-sigv4-fetch.ts
30
30
  function createSigV4FetchFunction(getCredentials, fetch, service = "bedrock") {
package/dist/index.js CHANGED
@@ -1734,7 +1734,7 @@ var BedrockEmbeddingModel = class {
1734
1734
  normalize: bedrockOptions.normalize
1735
1735
  };
1736
1736
  const url = this.getUrl(this.modelId);
1737
- const { value: response } = await (0, import_provider_utils5.postJsonToApi)({
1737
+ const { value: response, responseHeaders } = await (0, import_provider_utils5.postJsonToApi)({
1738
1738
  url,
1739
1739
  headers: await (0, import_provider_utils5.resolve)(
1740
1740
  (0, import_provider_utils5.combineHeaders)(await (0, import_provider_utils5.resolve)(this.config.headers), headers)
@@ -1763,7 +1763,10 @@ var BedrockEmbeddingModel = class {
1763
1763
  } else {
1764
1764
  embedding = response.embeddings.float[0];
1765
1765
  }
1766
- const tokens = "inputTextTokenCount" in response ? response.inputTextTokenCount : "inputTokenCount" in response ? (_f = response.inputTokenCount) != null ? _f : 0 : NaN;
1766
+ const headerTokenCount = Number(
1767
+ responseHeaders == null ? void 0 : responseHeaders["x-amzn-bedrock-input-token-count"]
1768
+ );
1769
+ const tokens = "inputTextTokenCount" in response ? response.inputTextTokenCount : "inputTokenCount" in response ? (_f = response.inputTokenCount) != null ? _f : 0 : headerTokenCount;
1767
1770
  return {
1768
1771
  embeddings: [embedding],
1769
1772
  usage: { tokens },
@@ -2011,7 +2014,7 @@ var import_provider_utils7 = require("@ai-sdk/provider-utils");
2011
2014
  var import_aws4fetch = require("aws4fetch");
2012
2015
 
2013
2016
  // src/version.ts
2014
- var VERSION = true ? "4.0.114" : "0.0.0-test";
2017
+ var VERSION = true ? "4.0.115" : "0.0.0-test";
2015
2018
 
2016
2019
  // src/bedrock-sigv4-fetch.ts
2017
2020
  function createSigV4FetchFunction(getCredentials, fetch, service = "bedrock") {