@ai-sdk/amazon-bedrock 5.0.0-beta.45 → 5.0.0-beta.47
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 +21 -0
- package/dist/anthropic/index.d.ts +10 -10
- package/dist/anthropic/index.js +65 -51
- package/dist/anthropic/index.js.map +1 -1
- package/dist/index.d.ts +21 -21
- package/dist/index.js +338 -298
- package/dist/index.js.map +1 -1
- package/docs/08-amazon-bedrock.mdx +66 -66
- package/package.json +3 -3
- package/src/{bedrock-api-types.ts → amazon-bedrock-api-types.ts} +60 -55
- package/src/{bedrock-chat-options.ts → amazon-bedrock-chat-language-model-options.ts} +7 -7
- package/src/{bedrock-chat-language-model.ts → amazon-bedrock-chat-language-model.ts} +211 -190
- package/src/{bedrock-embedding-options.ts → amazon-bedrock-embedding-model-options.ts} +1 -1
- package/src/{bedrock-embedding-model.ts → amazon-bedrock-embedding-model.ts} +31 -24
- package/src/{bedrock-error.ts → amazon-bedrock-error.ts} +1 -1
- package/src/{bedrock-event-stream-decoder.ts → amazon-bedrock-event-stream-decoder.ts} +1 -1
- package/src/{bedrock-event-stream-response-handler.ts → amazon-bedrock-event-stream-response-handler.ts} +3 -3
- package/src/{bedrock-image-model.ts → amazon-bedrock-image-model.ts} +40 -36
- package/src/amazon-bedrock-image-settings.ts +9 -0
- package/src/{bedrock-prepare-tools.ts → amazon-bedrock-prepare-tools.ts} +18 -17
- package/src/{bedrock-provider.ts → amazon-bedrock-provider.ts} +40 -38
- package/src/amazon-bedrock-reasoning-metadata.ts +10 -0
- package/src/{bedrock-sigv4-fetch.ts → amazon-bedrock-sigv4-fetch.ts} +13 -7
- package/src/anthropic/amazon-bedrock-anthropic-fetch.ts +104 -0
- package/src/anthropic/{bedrock-anthropic-options.ts → amazon-bedrock-anthropic-options.ts} +1 -1
- package/src/anthropic/{bedrock-anthropic-provider.ts → amazon-bedrock-anthropic-provider.ts} +18 -16
- package/src/anthropic/index.ts +19 -7
- package/src/{convert-bedrock-usage.ts → convert-amazon-bedrock-usage.ts} +3 -3
- package/src/{convert-to-bedrock-chat-messages.ts → convert-to-amazon-bedrock-chat-messages.ts} +92 -59
- package/src/index.ts +15 -8
- package/src/{map-bedrock-finish-reason.ts → map-amazon-bedrock-finish-reason.ts} +3 -3
- package/src/reranking/{bedrock-reranking-api.ts → amazon-bedrock-reranking-api.ts} +3 -3
- package/src/reranking/{bedrock-reranking-options.ts → amazon-bedrock-reranking-model-options.ts} +1 -1
- package/src/reranking/{bedrock-reranking-model.ts → amazon-bedrock-reranking-model.ts} +29 -21
- package/src/anthropic/bedrock-anthropic-fetch.ts +0 -94
- package/src/bedrock-image-settings.ts +0 -6
- package/src/bedrock-reasoning-metadata.ts +0 -10
package/dist/index.js
CHANGED
|
@@ -1,4 +1,4 @@
|
|
|
1
|
-
// src/bedrock-provider.ts
|
|
1
|
+
// src/amazon-bedrock-provider.ts
|
|
2
2
|
import { anthropicTools as anthropicTools2 } from "@ai-sdk/anthropic/internal";
|
|
3
3
|
import {
|
|
4
4
|
generateId,
|
|
@@ -8,7 +8,7 @@ import {
|
|
|
8
8
|
withUserAgentSuffix as withUserAgentSuffix2
|
|
9
9
|
} from "@ai-sdk/provider-utils";
|
|
10
10
|
|
|
11
|
-
// src/bedrock-chat-language-model.ts
|
|
11
|
+
// src/amazon-bedrock-chat-language-model.ts
|
|
12
12
|
import {
|
|
13
13
|
combineHeaders,
|
|
14
14
|
createJsonErrorResponseHandler,
|
|
@@ -26,7 +26,7 @@ import {
|
|
|
26
26
|
import { getModelCapabilities } from "@ai-sdk/anthropic/internal";
|
|
27
27
|
import { z as z4 } from "zod/v4";
|
|
28
28
|
|
|
29
|
-
// src/bedrock-api-types.ts
|
|
29
|
+
// src/amazon-bedrock-api-types.ts
|
|
30
30
|
var BEDROCK_STOP_REASONS = [
|
|
31
31
|
"stop",
|
|
32
32
|
"stop_sequence",
|
|
@@ -57,9 +57,9 @@ var BEDROCK_DOCUMENT_MIME_TYPES = {
|
|
|
57
57
|
"text/markdown": "md"
|
|
58
58
|
};
|
|
59
59
|
|
|
60
|
-
// src/bedrock-chat-options.ts
|
|
60
|
+
// src/amazon-bedrock-chat-language-model-options.ts
|
|
61
61
|
import { z } from "zod/v4";
|
|
62
|
-
var
|
|
62
|
+
var amazonBedrockFilePartProviderOptions = z.object({
|
|
63
63
|
/**
|
|
64
64
|
* Citation configuration for this document.
|
|
65
65
|
* When enabled, this document will generate citations in the response.
|
|
@@ -71,7 +71,7 @@ var bedrockFilePartProviderOptions = z.object({
|
|
|
71
71
|
enabled: z.boolean()
|
|
72
72
|
}).optional()
|
|
73
73
|
});
|
|
74
|
-
var
|
|
74
|
+
var amazonBedrockLanguageModelChatOptions = z.object({
|
|
75
75
|
/**
|
|
76
76
|
* Additional inference parameters that the model supports,
|
|
77
77
|
* beyond the base set of inference parameters that Converse
|
|
@@ -104,14 +104,14 @@ var amazonBedrockLanguageModelOptions = z.object({
|
|
|
104
104
|
serviceTier: z.enum(["reserved", "priority", "default", "flex"]).optional()
|
|
105
105
|
});
|
|
106
106
|
|
|
107
|
-
// src/bedrock-error.ts
|
|
107
|
+
// src/amazon-bedrock-error.ts
|
|
108
108
|
import { z as z2 } from "zod/v4";
|
|
109
|
-
var
|
|
109
|
+
var AmazonBedrockErrorSchema = z2.object({
|
|
110
110
|
message: z2.string(),
|
|
111
111
|
type: z2.string().nullish()
|
|
112
112
|
});
|
|
113
113
|
|
|
114
|
-
// src/bedrock-event-stream-response-handler.ts
|
|
114
|
+
// src/amazon-bedrock-event-stream-response-handler.ts
|
|
115
115
|
import { EmptyResponseBodyError } from "@ai-sdk/provider";
|
|
116
116
|
import {
|
|
117
117
|
safeParseJSON,
|
|
@@ -119,10 +119,10 @@ import {
|
|
|
119
119
|
safeValidateTypes
|
|
120
120
|
} from "@ai-sdk/provider-utils";
|
|
121
121
|
|
|
122
|
-
// src/bedrock-event-stream-decoder.ts
|
|
122
|
+
// src/amazon-bedrock-event-stream-decoder.ts
|
|
123
123
|
import { EventStreamCodec } from "@smithy/eventstream-codec";
|
|
124
124
|
import { toUtf8, fromUtf8 } from "@smithy/util-utf8";
|
|
125
|
-
function
|
|
125
|
+
function createAmazonBedrockEventStreamDecoder(body, processEvent) {
|
|
126
126
|
const codec = new EventStreamCodec(toUtf8, fromUtf8);
|
|
127
127
|
let buffer = new Uint8Array(0);
|
|
128
128
|
const textDecoder = new TextDecoder();
|
|
@@ -160,15 +160,15 @@ function createBedrockEventStreamDecoder(body, processEvent) {
|
|
|
160
160
|
);
|
|
161
161
|
}
|
|
162
162
|
|
|
163
|
-
// src/bedrock-event-stream-response-handler.ts
|
|
164
|
-
var
|
|
163
|
+
// src/amazon-bedrock-event-stream-response-handler.ts
|
|
164
|
+
var createAmazonBedrockEventStreamResponseHandler = (chunkSchema) => async ({ response }) => {
|
|
165
165
|
const responseHeaders = extractResponseHeaders(response);
|
|
166
166
|
if (response.body == null) {
|
|
167
167
|
throw new EmptyResponseBodyError({});
|
|
168
168
|
}
|
|
169
169
|
return {
|
|
170
170
|
responseHeaders,
|
|
171
|
-
value:
|
|
171
|
+
value: createAmazonBedrockEventStreamDecoder(
|
|
172
172
|
response.body,
|
|
173
173
|
async (event, controller) => {
|
|
174
174
|
if (event.messageType === "event") {
|
|
@@ -200,7 +200,7 @@ var createBedrockEventStreamResponseHandler = (chunkSchema) => async ({ response
|
|
|
200
200
|
};
|
|
201
201
|
};
|
|
202
202
|
|
|
203
|
-
// src/bedrock-prepare-tools.ts
|
|
203
|
+
// src/amazon-bedrock-prepare-tools.ts
|
|
204
204
|
import {
|
|
205
205
|
UnsupportedFunctionalityError
|
|
206
206
|
} from "@ai-sdk/provider";
|
|
@@ -248,7 +248,7 @@ async function prepareTools({
|
|
|
248
248
|
const ProviderTools = supportedTools.filter((t) => t.type === "provider");
|
|
249
249
|
const functionTools = supportedTools.filter((t) => t.type === "function");
|
|
250
250
|
let additionalTools = void 0;
|
|
251
|
-
const
|
|
251
|
+
const amazonBedrockTools = [];
|
|
252
252
|
const usingAnthropicTools = isAnthropicModel && ProviderTools.length > 0;
|
|
253
253
|
if (usingAnthropicTools) {
|
|
254
254
|
const {
|
|
@@ -275,7 +275,7 @@ async function prepareTools({
|
|
|
275
275
|
});
|
|
276
276
|
if (toolFactory != null) {
|
|
277
277
|
const fullToolDefinition = toolFactory({});
|
|
278
|
-
|
|
278
|
+
amazonBedrockTools.push({
|
|
279
279
|
toolSpec: {
|
|
280
280
|
name: tool.name,
|
|
281
281
|
inputSchema: {
|
|
@@ -294,7 +294,7 @@ async function prepareTools({
|
|
|
294
294
|
}
|
|
295
295
|
const filteredFunctionTools = (toolChoice == null ? void 0 : toolChoice.type) === "tool" ? functionTools.filter((t) => t.name === toolChoice.toolName) : functionTools;
|
|
296
296
|
for (const tool of filteredFunctionTools) {
|
|
297
|
-
|
|
297
|
+
amazonBedrockTools.push({
|
|
298
298
|
toolSpec: {
|
|
299
299
|
name: tool.name,
|
|
300
300
|
...((_a = tool.description) == null ? void 0 : _a.trim()) !== "" ? { description: tool.description } : {},
|
|
@@ -305,22 +305,22 @@ async function prepareTools({
|
|
|
305
305
|
}
|
|
306
306
|
});
|
|
307
307
|
}
|
|
308
|
-
let
|
|
309
|
-
if (!usingAnthropicTools &&
|
|
308
|
+
let amazonBedrockToolChoice = void 0;
|
|
309
|
+
if (!usingAnthropicTools && amazonBedrockTools.length > 0 && toolChoice) {
|
|
310
310
|
const type = toolChoice.type;
|
|
311
311
|
switch (type) {
|
|
312
312
|
case "auto":
|
|
313
|
-
|
|
313
|
+
amazonBedrockToolChoice = { auto: {} };
|
|
314
314
|
break;
|
|
315
315
|
case "required":
|
|
316
|
-
|
|
316
|
+
amazonBedrockToolChoice = { any: {} };
|
|
317
317
|
break;
|
|
318
318
|
case "none":
|
|
319
|
-
|
|
320
|
-
|
|
319
|
+
amazonBedrockTools.length = 0;
|
|
320
|
+
amazonBedrockToolChoice = void 0;
|
|
321
321
|
break;
|
|
322
322
|
case "tool":
|
|
323
|
-
|
|
323
|
+
amazonBedrockToolChoice = { tool: { name: toolChoice.toolName } };
|
|
324
324
|
break;
|
|
325
325
|
default: {
|
|
326
326
|
const _exhaustiveCheck = type;
|
|
@@ -330,7 +330,7 @@ async function prepareTools({
|
|
|
330
330
|
}
|
|
331
331
|
}
|
|
332
332
|
}
|
|
333
|
-
const toolConfig =
|
|
333
|
+
const toolConfig = amazonBedrockTools.length > 0 ? { tools: amazonBedrockTools, toolChoice: amazonBedrockToolChoice } : {};
|
|
334
334
|
return {
|
|
335
335
|
toolConfig,
|
|
336
336
|
additionalTools,
|
|
@@ -339,8 +339,8 @@ async function prepareTools({
|
|
|
339
339
|
};
|
|
340
340
|
}
|
|
341
341
|
|
|
342
|
-
// src/convert-bedrock-usage.ts
|
|
343
|
-
function
|
|
342
|
+
// src/convert-amazon-bedrock-usage.ts
|
|
343
|
+
function convertAmazonBedrockUsage(usage) {
|
|
344
344
|
var _a, _b;
|
|
345
345
|
if (usage == null) {
|
|
346
346
|
return {
|
|
@@ -378,7 +378,7 @@ function convertBedrockUsage(usage) {
|
|
|
378
378
|
};
|
|
379
379
|
}
|
|
380
380
|
|
|
381
|
-
// src/convert-to-bedrock-chat-messages.ts
|
|
381
|
+
// src/convert-to-amazon-bedrock-chat-messages.ts
|
|
382
382
|
import {
|
|
383
383
|
UnsupportedFunctionalityError as UnsupportedFunctionalityError2
|
|
384
384
|
} from "@ai-sdk/provider";
|
|
@@ -391,9 +391,9 @@ import {
|
|
|
391
391
|
stripFileExtension
|
|
392
392
|
} from "@ai-sdk/provider-utils";
|
|
393
393
|
|
|
394
|
-
// src/bedrock-reasoning-metadata.ts
|
|
394
|
+
// src/amazon-bedrock-reasoning-metadata.ts
|
|
395
395
|
import { z as z3 } from "zod/v4";
|
|
396
|
-
var
|
|
396
|
+
var amazonBedrockReasoningMetadataSchema = z3.object({
|
|
397
397
|
signature: z3.string().optional(),
|
|
398
398
|
redactedData: z3.string().optional()
|
|
399
399
|
});
|
|
@@ -410,26 +410,30 @@ function normalizeToolCallId(toolCallId, isMistral) {
|
|
|
410
410
|
return alphanumericChars.slice(0, 9);
|
|
411
411
|
}
|
|
412
412
|
|
|
413
|
-
// src/convert-to-bedrock-chat-messages.ts
|
|
413
|
+
// src/convert-to-amazon-bedrock-chat-messages.ts
|
|
414
414
|
function getCachePoint(providerMetadata) {
|
|
415
|
-
var _a;
|
|
416
|
-
const cachePointConfig = (_a = providerMetadata == null ? void 0 : providerMetadata.
|
|
415
|
+
var _a, _b, _c;
|
|
416
|
+
const cachePointConfig = (_c = (_a = providerMetadata == null ? void 0 : providerMetadata.amazonBedrock) == null ? void 0 : _a.cachePoint) != null ? _c : (_b = providerMetadata == null ? void 0 : providerMetadata.bedrock) == null ? void 0 : _b.cachePoint;
|
|
417
417
|
if (!cachePointConfig) {
|
|
418
418
|
return void 0;
|
|
419
419
|
}
|
|
420
420
|
return { cachePoint: cachePointConfig };
|
|
421
421
|
}
|
|
422
422
|
async function shouldEnableCitations(providerMetadata) {
|
|
423
|
-
var _a, _b;
|
|
424
|
-
const
|
|
423
|
+
var _a, _b, _c;
|
|
424
|
+
const amazonBedrockOptions = (_a = await parseProviderOptions({
|
|
425
|
+
provider: "amazonBedrock",
|
|
426
|
+
providerOptions: providerMetadata,
|
|
427
|
+
schema: amazonBedrockFilePartProviderOptions
|
|
428
|
+
})) != null ? _a : await parseProviderOptions({
|
|
425
429
|
provider: "bedrock",
|
|
426
430
|
providerOptions: providerMetadata,
|
|
427
|
-
schema:
|
|
431
|
+
schema: amazonBedrockFilePartProviderOptions
|
|
428
432
|
});
|
|
429
|
-
return (
|
|
433
|
+
return (_c = (_b = amazonBedrockOptions == null ? void 0 : amazonBedrockOptions.citations) == null ? void 0 : _b.enabled) != null ? _c : false;
|
|
430
434
|
}
|
|
431
|
-
async function
|
|
432
|
-
var _a;
|
|
435
|
+
async function convertToAmazonBedrockChatMessages(prompt, isMistral = false) {
|
|
436
|
+
var _a, _b;
|
|
433
437
|
const blocks = groupIntoBlocks(prompt);
|
|
434
438
|
let system = [];
|
|
435
439
|
const messages = [];
|
|
@@ -456,7 +460,7 @@ async function convertToBedrockChatMessages(prompt, isMistral = false) {
|
|
|
456
460
|
break;
|
|
457
461
|
}
|
|
458
462
|
case "user": {
|
|
459
|
-
const
|
|
463
|
+
const amazonBedrockContent = [];
|
|
460
464
|
for (const message of block.messages) {
|
|
461
465
|
const { role, content, providerOptions } = message;
|
|
462
466
|
switch (role) {
|
|
@@ -465,7 +469,7 @@ async function convertToBedrockChatMessages(prompt, isMistral = false) {
|
|
|
465
469
|
const part = content[j];
|
|
466
470
|
switch (part.type) {
|
|
467
471
|
case "text": {
|
|
468
|
-
|
|
472
|
+
amazonBedrockContent.push({
|
|
469
473
|
text: part.text
|
|
470
474
|
});
|
|
471
475
|
break;
|
|
@@ -487,9 +491,9 @@ async function convertToBedrockChatMessages(prompt, isMistral = false) {
|
|
|
487
491
|
const enableCitations = await shouldEnableCitations(
|
|
488
492
|
part.providerOptions
|
|
489
493
|
);
|
|
490
|
-
|
|
494
|
+
amazonBedrockContent.push({
|
|
491
495
|
document: {
|
|
492
|
-
format:
|
|
496
|
+
format: getAmazonBedrockDocumentFormat(textMediaType),
|
|
493
497
|
name: part.filename ? stripFileExtension(part.filename) : generateDocumentName(),
|
|
494
498
|
source: {
|
|
495
499
|
bytes: convertToBase64(
|
|
@@ -506,9 +510,9 @@ async function convertToBedrockChatMessages(prompt, isMistral = false) {
|
|
|
506
510
|
case "data": {
|
|
507
511
|
const fullMediaType = resolveFullMediaType({ part });
|
|
508
512
|
if (getTopLevelMediaType(fullMediaType) === "image") {
|
|
509
|
-
|
|
513
|
+
amazonBedrockContent.push({
|
|
510
514
|
image: {
|
|
511
|
-
format:
|
|
515
|
+
format: getAmazonBedrockImageFormat(fullMediaType),
|
|
512
516
|
source: {
|
|
513
517
|
bytes: convertToBase64(part.data.data)
|
|
514
518
|
}
|
|
@@ -518,9 +522,9 @@ async function convertToBedrockChatMessages(prompt, isMistral = false) {
|
|
|
518
522
|
const enableCitations = await shouldEnableCitations(
|
|
519
523
|
part.providerOptions
|
|
520
524
|
);
|
|
521
|
-
|
|
525
|
+
amazonBedrockContent.push({
|
|
522
526
|
document: {
|
|
523
|
-
format:
|
|
527
|
+
format: getAmazonBedrockDocumentFormat(fullMediaType),
|
|
524
528
|
name: part.filename ? stripFileExtension(part.filename) : generateDocumentName(),
|
|
525
529
|
source: {
|
|
526
530
|
bytes: convertToBase64(part.data.data)
|
|
@@ -559,7 +563,7 @@ async function convertToBedrockChatMessages(prompt, isMistral = false) {
|
|
|
559
563
|
functionality: `media type: ${contentPart.mediaType}`
|
|
560
564
|
});
|
|
561
565
|
}
|
|
562
|
-
const format =
|
|
566
|
+
const format = getAmazonBedrockImageFormat(
|
|
563
567
|
contentPart.mediaType
|
|
564
568
|
);
|
|
565
569
|
return {
|
|
@@ -594,7 +598,7 @@ async function convertToBedrockChatMessages(prompt, isMistral = false) {
|
|
|
594
598
|
];
|
|
595
599
|
break;
|
|
596
600
|
}
|
|
597
|
-
|
|
601
|
+
amazonBedrockContent.push({
|
|
598
602
|
toolResult: {
|
|
599
603
|
toolUseId: normalizeToolCallId(part.toolCallId, isMistral),
|
|
600
604
|
content: toolResultContent
|
|
@@ -610,14 +614,14 @@ async function convertToBedrockChatMessages(prompt, isMistral = false) {
|
|
|
610
614
|
}
|
|
611
615
|
const cachePoint = getCachePoint(providerOptions);
|
|
612
616
|
if (cachePoint) {
|
|
613
|
-
|
|
617
|
+
amazonBedrockContent.push(cachePoint);
|
|
614
618
|
}
|
|
615
619
|
}
|
|
616
|
-
messages.push({ role: "user", content:
|
|
620
|
+
messages.push({ role: "user", content: amazonBedrockContent });
|
|
617
621
|
break;
|
|
618
622
|
}
|
|
619
623
|
case "assistant": {
|
|
620
|
-
const
|
|
624
|
+
const amazonBedrockContent = [];
|
|
621
625
|
for (let j = 0; j < block.messages.length; j++) {
|
|
622
626
|
const message = block.messages[j];
|
|
623
627
|
const isLastMessage = j === block.messages.length - 1;
|
|
@@ -633,7 +637,7 @@ async function convertToBedrockChatMessages(prompt, isMistral = false) {
|
|
|
633
637
|
if (!part.text.trim() && !hasReasoningBlocks) {
|
|
634
638
|
break;
|
|
635
639
|
}
|
|
636
|
-
|
|
640
|
+
amazonBedrockContent.push({
|
|
637
641
|
text: (
|
|
638
642
|
// trim the last text part if it's the last message in the block
|
|
639
643
|
// because Bedrock does not allow trailing whitespace
|
|
@@ -649,13 +653,17 @@ async function convertToBedrockChatMessages(prompt, isMistral = false) {
|
|
|
649
653
|
break;
|
|
650
654
|
}
|
|
651
655
|
case "reasoning": {
|
|
652
|
-
const reasoningMetadata = await parseProviderOptions({
|
|
656
|
+
const reasoningMetadata = (_b = await parseProviderOptions({
|
|
657
|
+
provider: "amazonBedrock",
|
|
658
|
+
providerOptions: part.providerOptions,
|
|
659
|
+
schema: amazonBedrockReasoningMetadataSchema
|
|
660
|
+
})) != null ? _b : await parseProviderOptions({
|
|
653
661
|
provider: "bedrock",
|
|
654
662
|
providerOptions: part.providerOptions,
|
|
655
|
-
schema:
|
|
663
|
+
schema: amazonBedrockReasoningMetadataSchema
|
|
656
664
|
});
|
|
657
665
|
if ((reasoningMetadata == null ? void 0 : reasoningMetadata.signature) != null) {
|
|
658
|
-
|
|
666
|
+
amazonBedrockContent.push({
|
|
659
667
|
reasoningContent: {
|
|
660
668
|
reasoningText: {
|
|
661
669
|
text: part.text,
|
|
@@ -664,15 +672,17 @@ async function convertToBedrockChatMessages(prompt, isMistral = false) {
|
|
|
664
672
|
}
|
|
665
673
|
});
|
|
666
674
|
} else if ((reasoningMetadata == null ? void 0 : reasoningMetadata.redactedData) != null) {
|
|
667
|
-
|
|
675
|
+
amazonBedrockContent.push({
|
|
668
676
|
reasoningContent: {
|
|
669
677
|
redactedReasoning: {
|
|
670
678
|
data: reasoningMetadata.redactedData
|
|
671
679
|
}
|
|
672
680
|
}
|
|
673
681
|
});
|
|
674
|
-
} else
|
|
675
|
-
|
|
682
|
+
} else if (part.providerOptions == null || Object.keys(part.providerOptions).every(
|
|
683
|
+
(k2) => k2 === "bedrock" || k2 === "amazonBedrock"
|
|
684
|
+
)) {
|
|
685
|
+
amazonBedrockContent.push({
|
|
676
686
|
reasoningContent: {
|
|
677
687
|
reasoningText: {
|
|
678
688
|
text: trimIfLast(
|
|
@@ -688,7 +698,7 @@ async function convertToBedrockChatMessages(prompt, isMistral = false) {
|
|
|
688
698
|
break;
|
|
689
699
|
}
|
|
690
700
|
case "tool-call": {
|
|
691
|
-
|
|
701
|
+
amazonBedrockContent.push({
|
|
692
702
|
toolUse: {
|
|
693
703
|
toolUseId: normalizeToolCallId(part.toolCallId, isMistral),
|
|
694
704
|
name: part.toolName,
|
|
@@ -701,10 +711,10 @@ async function convertToBedrockChatMessages(prompt, isMistral = false) {
|
|
|
701
711
|
}
|
|
702
712
|
const cachePoint = getCachePoint(message.providerOptions);
|
|
703
713
|
if (cachePoint) {
|
|
704
|
-
|
|
714
|
+
amazonBedrockContent.push(cachePoint);
|
|
705
715
|
}
|
|
706
716
|
}
|
|
707
|
-
messages.push({ role: "assistant", content:
|
|
717
|
+
messages.push({ role: "assistant", content: amazonBedrockContent });
|
|
708
718
|
break;
|
|
709
719
|
}
|
|
710
720
|
default: {
|
|
@@ -715,7 +725,7 @@ async function convertToBedrockChatMessages(prompt, isMistral = false) {
|
|
|
715
725
|
}
|
|
716
726
|
return { system, messages };
|
|
717
727
|
}
|
|
718
|
-
function
|
|
728
|
+
function getAmazonBedrockImageFormat(mimeType) {
|
|
719
729
|
const format = BEDROCK_IMAGE_MIME_TYPES[mimeType];
|
|
720
730
|
if (!format) {
|
|
721
731
|
throw new UnsupportedFunctionalityError2({
|
|
@@ -725,7 +735,7 @@ function getBedrockImageFormat(mimeType) {
|
|
|
725
735
|
}
|
|
726
736
|
return format;
|
|
727
737
|
}
|
|
728
|
-
function
|
|
738
|
+
function getAmazonBedrockDocumentFormat(mimeType) {
|
|
729
739
|
const format = BEDROCK_DOCUMENT_MIME_TYPES[mimeType];
|
|
730
740
|
if (!format) {
|
|
731
741
|
throw new UnsupportedFunctionalityError2({
|
|
@@ -785,8 +795,8 @@ function groupIntoBlocks(prompt) {
|
|
|
785
795
|
return blocks;
|
|
786
796
|
}
|
|
787
797
|
|
|
788
|
-
// src/map-bedrock-finish-reason.ts
|
|
789
|
-
function
|
|
798
|
+
// src/map-amazon-bedrock-finish-reason.ts
|
|
799
|
+
function mapAmazonBedrockFinishReason(finishReason, isJsonResponseFromTool) {
|
|
790
800
|
switch (finishReason) {
|
|
791
801
|
case "stop_sequence":
|
|
792
802
|
case "end_turn":
|
|
@@ -803,8 +813,8 @@ function mapBedrockFinishReason(finishReason, isJsonResponseFromTool) {
|
|
|
803
813
|
}
|
|
804
814
|
}
|
|
805
815
|
|
|
806
|
-
// src/bedrock-chat-language-model.ts
|
|
807
|
-
var
|
|
816
|
+
// src/amazon-bedrock-chat-language-model.ts
|
|
817
|
+
var AmazonBedrockChatLanguageModel = class _AmazonBedrockChatLanguageModel {
|
|
808
818
|
constructor(modelId, config) {
|
|
809
819
|
this.modelId = modelId;
|
|
810
820
|
this.config = config;
|
|
@@ -821,7 +831,7 @@ var BedrockChatLanguageModel = class _BedrockChatLanguageModel {
|
|
|
821
831
|
});
|
|
822
832
|
}
|
|
823
833
|
static [WORKFLOW_DESERIALIZE](options) {
|
|
824
|
-
return new
|
|
834
|
+
return new _AmazonBedrockChatLanguageModel(options.modelId, options.config);
|
|
825
835
|
}
|
|
826
836
|
async getArgs({
|
|
827
837
|
prompt,
|
|
@@ -839,12 +849,16 @@ var BedrockChatLanguageModel = class _BedrockChatLanguageModel {
|
|
|
839
849
|
reasoning,
|
|
840
850
|
providerOptions
|
|
841
851
|
}) {
|
|
842
|
-
var _a, _b, _c, _d, _e, _f, _g, _h, _i, _j, _k, _l, _m;
|
|
843
|
-
let
|
|
852
|
+
var _a, _b, _c, _d, _e, _f, _g, _h, _i, _j, _k, _l, _m, _n, _o, _p;
|
|
853
|
+
let amazonBedrockOptions = (_b = (_a = await parseProviderOptions2({
|
|
854
|
+
provider: "amazonBedrock",
|
|
855
|
+
providerOptions,
|
|
856
|
+
schema: amazonBedrockLanguageModelChatOptions
|
|
857
|
+
})) != null ? _a : await parseProviderOptions2({
|
|
844
858
|
provider: "bedrock",
|
|
845
859
|
providerOptions,
|
|
846
|
-
schema:
|
|
847
|
-
})) != null ?
|
|
860
|
+
schema: amazonBedrockLanguageModelChatOptions
|
|
861
|
+
})) != null ? _b : {};
|
|
848
862
|
const warnings = [];
|
|
849
863
|
if (frequencyPenalty != null) {
|
|
850
864
|
warnings.push({
|
|
@@ -888,14 +902,14 @@ var BedrockChatLanguageModel = class _BedrockChatLanguageModel {
|
|
|
888
902
|
}
|
|
889
903
|
const isAnthropicModel = this.modelId.includes("anthropic");
|
|
890
904
|
const isOpenAIModel = this.modelId.startsWith("openai.");
|
|
891
|
-
|
|
905
|
+
amazonBedrockOptions = resolveAmazonBedrockReasoningConfig({
|
|
892
906
|
reasoning,
|
|
893
|
-
|
|
907
|
+
amazonBedrockOptions,
|
|
894
908
|
warnings,
|
|
895
909
|
isAnthropicModel,
|
|
896
910
|
modelId: this.modelId
|
|
897
911
|
});
|
|
898
|
-
const isThinkingEnabled = ((
|
|
912
|
+
const isThinkingEnabled = ((_c = amazonBedrockOptions.reasoningConfig) == null ? void 0 : _c.type) === "enabled" || ((_d = amazonBedrockOptions.reasoningConfig) == null ? void 0 : _d.type) === "adaptive";
|
|
899
913
|
const { supportsStructuredOutput: modelSupportsStructuredOutput } = getModelCapabilities(this.modelId);
|
|
900
914
|
const useNativeStructuredOutput = isAnthropicModel && (modelSupportsStructuredOutput || isThinkingEnabled) && (responseFormat == null ? void 0 : responseFormat.type) === "json" && responseFormat.schema != null;
|
|
901
915
|
const jsonResponseTool = (responseFormat == null ? void 0 : responseFormat.type) === "json" && responseFormat.schema != null && !useNativeStructuredOutput ? {
|
|
@@ -911,22 +925,22 @@ var BedrockChatLanguageModel = class _BedrockChatLanguageModel {
|
|
|
911
925
|
});
|
|
912
926
|
warnings.push(...toolWarnings);
|
|
913
927
|
if (additionalTools) {
|
|
914
|
-
|
|
915
|
-
...
|
|
928
|
+
amazonBedrockOptions.additionalModelRequestFields = {
|
|
929
|
+
...amazonBedrockOptions.additionalModelRequestFields,
|
|
916
930
|
...additionalTools
|
|
917
931
|
};
|
|
918
932
|
}
|
|
919
|
-
if (betas.size > 0 ||
|
|
920
|
-
const existingBetas = (
|
|
933
|
+
if (betas.size > 0 || amazonBedrockOptions.anthropicBeta) {
|
|
934
|
+
const existingBetas = (_e = amazonBedrockOptions.anthropicBeta) != null ? _e : [];
|
|
921
935
|
const mergedBetas = betas.size > 0 ? [...existingBetas, ...Array.from(betas)] : existingBetas;
|
|
922
|
-
|
|
923
|
-
...
|
|
936
|
+
amazonBedrockOptions.additionalModelRequestFields = {
|
|
937
|
+
...amazonBedrockOptions.additionalModelRequestFields,
|
|
924
938
|
anthropic_beta: mergedBetas
|
|
925
939
|
};
|
|
926
940
|
}
|
|
927
|
-
const thinkingType = (
|
|
928
|
-
const thinkingBudget = thinkingType === "enabled" ? (
|
|
929
|
-
const thinkingDisplay = thinkingType === "adaptive" ? (
|
|
941
|
+
const thinkingType = (_f = amazonBedrockOptions.reasoningConfig) == null ? void 0 : _f.type;
|
|
942
|
+
const thinkingBudget = thinkingType === "enabled" ? (_g = amazonBedrockOptions.reasoningConfig) == null ? void 0 : _g.budgetTokens : void 0;
|
|
943
|
+
const thinkingDisplay = thinkingType === "adaptive" ? (_h = amazonBedrockOptions.reasoningConfig) == null ? void 0 : _h.display : void 0;
|
|
930
944
|
const isAnthropicThinkingEnabled = isAnthropicModel && isThinkingEnabled;
|
|
931
945
|
const inferenceConfig = {
|
|
932
946
|
...maxOutputTokens != null && { maxTokens: maxOutputTokens },
|
|
@@ -942,16 +956,16 @@ var BedrockChatLanguageModel = class _BedrockChatLanguageModel {
|
|
|
942
956
|
} else {
|
|
943
957
|
inferenceConfig.maxTokens = thinkingBudget + 4096;
|
|
944
958
|
}
|
|
945
|
-
|
|
946
|
-
...
|
|
959
|
+
amazonBedrockOptions.additionalModelRequestFields = {
|
|
960
|
+
...amazonBedrockOptions.additionalModelRequestFields,
|
|
947
961
|
thinking: {
|
|
948
962
|
type: "enabled",
|
|
949
963
|
budget_tokens: thinkingBudget
|
|
950
964
|
}
|
|
951
965
|
};
|
|
952
966
|
} else if (thinkingType === "adaptive") {
|
|
953
|
-
|
|
954
|
-
...
|
|
967
|
+
amazonBedrockOptions.additionalModelRequestFields = {
|
|
968
|
+
...amazonBedrockOptions.additionalModelRequestFields,
|
|
955
969
|
thinking: {
|
|
956
970
|
type: "adaptive",
|
|
957
971
|
...thinkingDisplay != null && { display: thinkingDisplay }
|
|
@@ -959,7 +973,7 @@ var BedrockChatLanguageModel = class _BedrockChatLanguageModel {
|
|
|
959
973
|
};
|
|
960
974
|
}
|
|
961
975
|
} else if (!isAnthropicModel) {
|
|
962
|
-
if (((
|
|
976
|
+
if (((_i = amazonBedrockOptions.reasoningConfig) == null ? void 0 : _i.budgetTokens) != null) {
|
|
963
977
|
warnings.push({
|
|
964
978
|
type: "unsupported",
|
|
965
979
|
feature: "budgetTokens",
|
|
@@ -974,24 +988,24 @@ var BedrockChatLanguageModel = class _BedrockChatLanguageModel {
|
|
|
974
988
|
});
|
|
975
989
|
}
|
|
976
990
|
}
|
|
977
|
-
const maxReasoningEffort = (
|
|
991
|
+
const maxReasoningEffort = (_j = amazonBedrockOptions.reasoningConfig) == null ? void 0 : _j.maxReasoningEffort;
|
|
978
992
|
if (maxReasoningEffort != null) {
|
|
979
993
|
if (isAnthropicModel) {
|
|
980
|
-
|
|
981
|
-
...
|
|
994
|
+
amazonBedrockOptions.additionalModelRequestFields = {
|
|
995
|
+
...amazonBedrockOptions.additionalModelRequestFields,
|
|
982
996
|
output_config: {
|
|
983
|
-
...(
|
|
997
|
+
...(_k = amazonBedrockOptions.additionalModelRequestFields) == null ? void 0 : _k.output_config,
|
|
984
998
|
effort: maxReasoningEffort
|
|
985
999
|
}
|
|
986
1000
|
};
|
|
987
1001
|
} else if (isOpenAIModel) {
|
|
988
|
-
|
|
989
|
-
...
|
|
1002
|
+
amazonBedrockOptions.additionalModelRequestFields = {
|
|
1003
|
+
...amazonBedrockOptions.additionalModelRequestFields,
|
|
990
1004
|
reasoning_effort: maxReasoningEffort
|
|
991
1005
|
};
|
|
992
1006
|
} else {
|
|
993
|
-
|
|
994
|
-
...
|
|
1007
|
+
amazonBedrockOptions.additionalModelRequestFields = {
|
|
1008
|
+
...amazonBedrockOptions.additionalModelRequestFields,
|
|
995
1009
|
reasoningConfig: {
|
|
996
1010
|
...thinkingType != null && thinkingType !== "adaptive" && { type: thinkingType },
|
|
997
1011
|
...thinkingBudget != null && { budgetTokens: thinkingBudget },
|
|
@@ -1001,10 +1015,10 @@ var BedrockChatLanguageModel = class _BedrockChatLanguageModel {
|
|
|
1001
1015
|
}
|
|
1002
1016
|
}
|
|
1003
1017
|
if (useNativeStructuredOutput) {
|
|
1004
|
-
|
|
1005
|
-
...
|
|
1018
|
+
amazonBedrockOptions.additionalModelRequestFields = {
|
|
1019
|
+
...amazonBedrockOptions.additionalModelRequestFields,
|
|
1006
1020
|
output_config: {
|
|
1007
|
-
...(
|
|
1021
|
+
...(_l = amazonBedrockOptions.additionalModelRequestFields) == null ? void 0 : _l.output_config,
|
|
1008
1022
|
format: {
|
|
1009
1023
|
type: "json_schema",
|
|
1010
1024
|
schema: responseFormat.schema
|
|
@@ -1036,7 +1050,7 @@ var BedrockChatLanguageModel = class _BedrockChatLanguageModel {
|
|
|
1036
1050
|
details: "topK is not supported when thinking is enabled"
|
|
1037
1051
|
});
|
|
1038
1052
|
}
|
|
1039
|
-
const hasAnyTools = ((
|
|
1053
|
+
const hasAnyTools = ((_n = (_m = toolConfig.tools) == null ? void 0 : _m.length) != null ? _n : 0) > 0 || additionalTools;
|
|
1040
1054
|
let filteredPrompt = prompt;
|
|
1041
1055
|
if (!hasAnyTools) {
|
|
1042
1056
|
const hasToolContent = prompt.some(
|
|
@@ -1063,7 +1077,7 @@ var BedrockChatLanguageModel = class _BedrockChatLanguageModel {
|
|
|
1063
1077
|
}
|
|
1064
1078
|
}
|
|
1065
1079
|
const isMistral = isMistralModel(this.modelId);
|
|
1066
|
-
const { system, messages } = await
|
|
1080
|
+
const { system, messages } = await convertToAmazonBedrockChatMessages(
|
|
1067
1081
|
filteredPrompt,
|
|
1068
1082
|
isMistral
|
|
1069
1083
|
);
|
|
@@ -1071,26 +1085,26 @@ var BedrockChatLanguageModel = class _BedrockChatLanguageModel {
|
|
|
1071
1085
|
reasoningConfig: _,
|
|
1072
1086
|
additionalModelRequestFields: __,
|
|
1073
1087
|
serviceTier: ___,
|
|
1074
|
-
...
|
|
1075
|
-
} = (providerOptions == null ? void 0 : providerOptions.bedrock)
|
|
1088
|
+
...filteredAmazonBedrockOptions
|
|
1089
|
+
} = (_p = (_o = providerOptions == null ? void 0 : providerOptions.amazonBedrock) != null ? _o : providerOptions == null ? void 0 : providerOptions.bedrock) != null ? _p : {};
|
|
1076
1090
|
const additionalModelResponseFieldPaths = isAnthropicModel ? ["/delta/stop_sequence"] : void 0;
|
|
1077
1091
|
return {
|
|
1078
1092
|
command: {
|
|
1079
1093
|
system,
|
|
1080
1094
|
messages,
|
|
1081
|
-
additionalModelRequestFields:
|
|
1095
|
+
additionalModelRequestFields: amazonBedrockOptions.additionalModelRequestFields,
|
|
1082
1096
|
...additionalModelResponseFieldPaths && {
|
|
1083
1097
|
additionalModelResponseFieldPaths
|
|
1084
1098
|
},
|
|
1085
1099
|
...Object.keys(inferenceConfig).length > 0 && {
|
|
1086
1100
|
inferenceConfig
|
|
1087
1101
|
},
|
|
1088
|
-
...
|
|
1102
|
+
...amazonBedrockOptions.serviceTier != null && {
|
|
1089
1103
|
serviceTier: {
|
|
1090
|
-
type:
|
|
1104
|
+
type: amazonBedrockOptions.serviceTier
|
|
1091
1105
|
}
|
|
1092
1106
|
},
|
|
1093
|
-
...
|
|
1107
|
+
...filteredAmazonBedrockOptions,
|
|
1094
1108
|
...toolConfig.tools !== void 0 && toolConfig.tools.length > 0 ? { toolConfig } : {}
|
|
1095
1109
|
},
|
|
1096
1110
|
warnings,
|
|
@@ -1119,14 +1133,14 @@ var BedrockChatLanguageModel = class _BedrockChatLanguageModel {
|
|
|
1119
1133
|
headers: await this.getHeaders({ headers: options.headers }),
|
|
1120
1134
|
body: args,
|
|
1121
1135
|
failedResponseHandler: createJsonErrorResponseHandler({
|
|
1122
|
-
errorSchema:
|
|
1136
|
+
errorSchema: AmazonBedrockErrorSchema,
|
|
1123
1137
|
errorToMessage: (error) => {
|
|
1124
1138
|
var _a2;
|
|
1125
1139
|
return `${(_a2 = error.message) != null ? _a2 : "Unknown error"}`;
|
|
1126
1140
|
}
|
|
1127
1141
|
}),
|
|
1128
1142
|
successfulResponseHandler: createJsonResponseHandler(
|
|
1129
|
-
|
|
1143
|
+
AmazonBedrockResponseSchema
|
|
1130
1144
|
),
|
|
1131
1145
|
abortSignal: options.abortSignal,
|
|
1132
1146
|
fetch: this.config.fetch
|
|
@@ -1144,21 +1158,25 @@ var BedrockChatLanguageModel = class _BedrockChatLanguageModel {
|
|
|
1144
1158
|
text: part.reasoningContent.reasoningText.text
|
|
1145
1159
|
};
|
|
1146
1160
|
if (part.reasoningContent.reasoningText.signature) {
|
|
1161
|
+
const reasoningPayload = {
|
|
1162
|
+
signature: part.reasoningContent.reasoningText.signature
|
|
1163
|
+
};
|
|
1147
1164
|
reasoning.providerMetadata = {
|
|
1148
|
-
|
|
1149
|
-
|
|
1150
|
-
}
|
|
1165
|
+
amazonBedrock: reasoningPayload,
|
|
1166
|
+
bedrock: reasoningPayload
|
|
1151
1167
|
};
|
|
1152
1168
|
}
|
|
1153
1169
|
content.push(reasoning);
|
|
1154
1170
|
} else if ("redactedReasoning" in part.reasoningContent) {
|
|
1171
|
+
const redactedPayload = {
|
|
1172
|
+
redactedData: (_a = part.reasoningContent.redactedReasoning.data) != null ? _a : ""
|
|
1173
|
+
};
|
|
1155
1174
|
content.push({
|
|
1156
1175
|
type: "reasoning",
|
|
1157
1176
|
text: "",
|
|
1158
1177
|
providerMetadata: {
|
|
1159
|
-
|
|
1160
|
-
|
|
1161
|
-
}
|
|
1178
|
+
amazonBedrock: redactedPayload,
|
|
1179
|
+
bedrock: redactedPayload
|
|
1162
1180
|
}
|
|
1163
1181
|
});
|
|
1164
1182
|
}
|
|
@@ -1184,39 +1202,41 @@ var BedrockChatLanguageModel = class _BedrockChatLanguageModel {
|
|
|
1184
1202
|
}
|
|
1185
1203
|
}
|
|
1186
1204
|
const stopSequence = (_j = (_i = (_h = response.additionalModelResponseFields) == null ? void 0 : _h.delta) == null ? void 0 : _i.stop_sequence) != null ? _j : null;
|
|
1187
|
-
const
|
|
1188
|
-
|
|
1189
|
-
|
|
1190
|
-
|
|
1191
|
-
|
|
1192
|
-
|
|
1193
|
-
|
|
1194
|
-
|
|
1195
|
-
|
|
1196
|
-
|
|
1197
|
-
usage
|
|
1198
|
-
|
|
1199
|
-
|
|
1200
|
-
|
|
1201
|
-
|
|
1202
|
-
cacheDetails: response.usage.cacheDetails
|
|
1203
|
-
}
|
|
1205
|
+
const providerMetadataPayload = response.trace || response.usage || response.performanceConfig || response.serviceTier || isJsonResponseFromTool || stopSequence ? {
|
|
1206
|
+
...response.trace && typeof response.trace === "object" ? { trace: response.trace } : {},
|
|
1207
|
+
...response.performanceConfig && {
|
|
1208
|
+
performanceConfig: response.performanceConfig
|
|
1209
|
+
},
|
|
1210
|
+
...response.serviceTier && {
|
|
1211
|
+
serviceTier: response.serviceTier
|
|
1212
|
+
},
|
|
1213
|
+
...(((_k = response.usage) == null ? void 0 : _k.cacheWriteInputTokens) != null || ((_l = response.usage) == null ? void 0 : _l.cacheDetails) != null) && {
|
|
1214
|
+
usage: {
|
|
1215
|
+
...response.usage.cacheWriteInputTokens != null && {
|
|
1216
|
+
cacheWriteInputTokens: response.usage.cacheWriteInputTokens
|
|
1217
|
+
},
|
|
1218
|
+
...response.usage.cacheDetails != null && {
|
|
1219
|
+
cacheDetails: response.usage.cacheDetails
|
|
1204
1220
|
}
|
|
1205
|
-
}
|
|
1206
|
-
|
|
1207
|
-
|
|
1208
|
-
|
|
1221
|
+
}
|
|
1222
|
+
},
|
|
1223
|
+
...isJsonResponseFromTool && { isJsonResponseFromTool: true },
|
|
1224
|
+
stopSequence
|
|
1225
|
+
} : void 0;
|
|
1226
|
+
const providerMetadata = providerMetadataPayload ? {
|
|
1227
|
+
amazonBedrock: providerMetadataPayload,
|
|
1228
|
+
bedrock: providerMetadataPayload
|
|
1209
1229
|
} : void 0;
|
|
1210
1230
|
return {
|
|
1211
1231
|
content,
|
|
1212
1232
|
finishReason: {
|
|
1213
|
-
unified:
|
|
1233
|
+
unified: mapAmazonBedrockFinishReason(
|
|
1214
1234
|
response.stopReason,
|
|
1215
1235
|
isJsonResponseFromTool
|
|
1216
1236
|
),
|
|
1217
1237
|
raw: (_m = response.stopReason) != null ? _m : void 0
|
|
1218
1238
|
},
|
|
1219
|
-
usage:
|
|
1239
|
+
usage: convertAmazonBedrockUsage(response.usage),
|
|
1220
1240
|
response: {
|
|
1221
1241
|
id: (_n = responseHeaders == null ? void 0 : responseHeaders["x-amzn-requestid"]) != null ? _n : void 0,
|
|
1222
1242
|
timestamp: (responseHeaders == null ? void 0 : responseHeaders["date"]) != null ? new Date(responseHeaders["date"]) : void 0,
|
|
@@ -1241,10 +1261,12 @@ var BedrockChatLanguageModel = class _BedrockChatLanguageModel {
|
|
|
1241
1261
|
headers: await this.getHeaders({ headers: options.headers }),
|
|
1242
1262
|
body: args,
|
|
1243
1263
|
failedResponseHandler: createJsonErrorResponseHandler({
|
|
1244
|
-
errorSchema:
|
|
1264
|
+
errorSchema: AmazonBedrockErrorSchema,
|
|
1245
1265
|
errorToMessage: (error) => `${error.type}: ${error.message}`
|
|
1246
1266
|
}),
|
|
1247
|
-
successfulResponseHandler:
|
|
1267
|
+
successfulResponseHandler: createAmazonBedrockEventStreamResponseHandler(
|
|
1268
|
+
AmazonBedrockStreamSchema
|
|
1269
|
+
),
|
|
1248
1270
|
abortSignal: options.abortSignal,
|
|
1249
1271
|
fetch: this.config.fetch
|
|
1250
1272
|
});
|
|
@@ -1272,9 +1294,9 @@ var BedrockChatLanguageModel = class _BedrockChatLanguageModel {
|
|
|
1272
1294
|
},
|
|
1273
1295
|
transform(chunk, controller) {
|
|
1274
1296
|
var _a, _b, _c, _d, _e, _f, _g, _h, _i, _j, _k, _l, _m, _n, _o, _p;
|
|
1275
|
-
function enqueueError(
|
|
1297
|
+
function enqueueError(amazonBedrockError) {
|
|
1276
1298
|
finishReason = { unified: "error", raw: void 0 };
|
|
1277
|
-
controller.enqueue({ type: "error", error:
|
|
1299
|
+
controller.enqueue({ type: "error", error: amazonBedrockError });
|
|
1278
1300
|
}
|
|
1279
1301
|
if (options.includeRawChunks) {
|
|
1280
1302
|
controller.enqueue({ type: "raw", rawValue: chunk.rawValue });
|
|
@@ -1302,7 +1324,7 @@ var BedrockChatLanguageModel = class _BedrockChatLanguageModel {
|
|
|
1302
1324
|
}
|
|
1303
1325
|
if (value.messageStop) {
|
|
1304
1326
|
finishReason = {
|
|
1305
|
-
unified:
|
|
1327
|
+
unified: mapAmazonBedrockFinishReason(
|
|
1306
1328
|
value.messageStop.stopReason,
|
|
1307
1329
|
isJsonResponseFromTool
|
|
1308
1330
|
),
|
|
@@ -1328,18 +1350,20 @@ var BedrockChatLanguageModel = class _BedrockChatLanguageModel {
|
|
|
1328
1350
|
trace: value.metadata.trace
|
|
1329
1351
|
} : void 0;
|
|
1330
1352
|
if (cacheUsage || trace || value.metadata.performanceConfig || value.metadata.serviceTier) {
|
|
1331
|
-
|
|
1332
|
-
|
|
1333
|
-
|
|
1334
|
-
|
|
1335
|
-
|
|
1336
|
-
|
|
1337
|
-
|
|
1338
|
-
|
|
1339
|
-
serviceTier: value.metadata.serviceTier
|
|
1340
|
-
}
|
|
1353
|
+
const metadataPayload = {
|
|
1354
|
+
...cacheUsage,
|
|
1355
|
+
...trace,
|
|
1356
|
+
...value.metadata.performanceConfig && {
|
|
1357
|
+
performanceConfig: value.metadata.performanceConfig
|
|
1358
|
+
},
|
|
1359
|
+
...value.metadata.serviceTier && {
|
|
1360
|
+
serviceTier: value.metadata.serviceTier
|
|
1341
1361
|
}
|
|
1342
1362
|
};
|
|
1363
|
+
providerMetadata = {
|
|
1364
|
+
amazonBedrock: metadataPayload,
|
|
1365
|
+
bedrock: metadataPayload
|
|
1366
|
+
};
|
|
1343
1367
|
}
|
|
1344
1368
|
}
|
|
1345
1369
|
if (((_i = value.contentBlockStart) == null ? void 0 : _i.contentBlockIndex) != null && !((_k = (_j = value.contentBlockStart) == null ? void 0 : _j.start) == null ? void 0 : _k.toolUse)) {
|
|
@@ -1435,16 +1459,20 @@ var BedrockChatLanguageModel = class _BedrockChatLanguageModel {
|
|
|
1435
1459
|
id: String(blockIndex)
|
|
1436
1460
|
});
|
|
1437
1461
|
}
|
|
1438
|
-
|
|
1439
|
-
|
|
1440
|
-
|
|
1441
|
-
|
|
1442
|
-
|
|
1443
|
-
|
|
1444
|
-
|
|
1462
|
+
{
|
|
1463
|
+
const signaturePayload = {
|
|
1464
|
+
signature: reasoningContent.signature
|
|
1465
|
+
};
|
|
1466
|
+
controller.enqueue({
|
|
1467
|
+
type: "reasoning-delta",
|
|
1468
|
+
id: String(blockIndex),
|
|
1469
|
+
delta: "",
|
|
1470
|
+
providerMetadata: {
|
|
1471
|
+
amazonBedrock: signaturePayload,
|
|
1472
|
+
bedrock: signaturePayload
|
|
1445
1473
|
}
|
|
1446
|
-
}
|
|
1447
|
-
}
|
|
1474
|
+
});
|
|
1475
|
+
}
|
|
1448
1476
|
} else if ("data" in reasoningContent && reasoningContent.data) {
|
|
1449
1477
|
if (contentBlocks[blockIndex] == null) {
|
|
1450
1478
|
contentBlocks[blockIndex] = { type: "reasoning" };
|
|
@@ -1453,16 +1481,20 @@ var BedrockChatLanguageModel = class _BedrockChatLanguageModel {
|
|
|
1453
1481
|
id: String(blockIndex)
|
|
1454
1482
|
});
|
|
1455
1483
|
}
|
|
1456
|
-
|
|
1457
|
-
|
|
1458
|
-
|
|
1459
|
-
|
|
1460
|
-
|
|
1461
|
-
|
|
1462
|
-
|
|
1484
|
+
{
|
|
1485
|
+
const redactedPayload = {
|
|
1486
|
+
redactedData: reasoningContent.data
|
|
1487
|
+
};
|
|
1488
|
+
controller.enqueue({
|
|
1489
|
+
type: "reasoning-delta",
|
|
1490
|
+
id: String(blockIndex),
|
|
1491
|
+
delta: "",
|
|
1492
|
+
providerMetadata: {
|
|
1493
|
+
amazonBedrock: redactedPayload,
|
|
1494
|
+
bedrock: redactedPayload
|
|
1463
1495
|
}
|
|
1464
|
-
}
|
|
1465
|
-
}
|
|
1496
|
+
});
|
|
1497
|
+
}
|
|
1466
1498
|
}
|
|
1467
1499
|
}
|
|
1468
1500
|
const contentBlockStart = value.contentBlockStart;
|
|
@@ -1507,30 +1539,25 @@ var BedrockChatLanguageModel = class _BedrockChatLanguageModel {
|
|
|
1507
1539
|
}
|
|
1508
1540
|
},
|
|
1509
1541
|
flush(controller) {
|
|
1542
|
+
var _a;
|
|
1510
1543
|
if (isJsonResponseFromTool || stopSequence != null) {
|
|
1511
|
-
|
|
1512
|
-
providerMetadata.
|
|
1513
|
-
|
|
1514
|
-
|
|
1515
|
-
|
|
1516
|
-
|
|
1517
|
-
|
|
1518
|
-
|
|
1519
|
-
|
|
1520
|
-
|
|
1521
|
-
|
|
1522
|
-
|
|
1523
|
-
isJsonResponseFromTool: true
|
|
1524
|
-
},
|
|
1525
|
-
stopSequence
|
|
1526
|
-
}
|
|
1527
|
-
};
|
|
1528
|
-
}
|
|
1544
|
+
const updatePayload = {
|
|
1545
|
+
...(_a = providerMetadata == null ? void 0 : providerMetadata.amazonBedrock) != null ? _a : providerMetadata == null ? void 0 : providerMetadata.bedrock,
|
|
1546
|
+
...isJsonResponseFromTool && {
|
|
1547
|
+
isJsonResponseFromTool: true
|
|
1548
|
+
},
|
|
1549
|
+
stopSequence
|
|
1550
|
+
};
|
|
1551
|
+
providerMetadata = {
|
|
1552
|
+
...providerMetadata,
|
|
1553
|
+
amazonBedrock: updatePayload,
|
|
1554
|
+
bedrock: updatePayload
|
|
1555
|
+
};
|
|
1529
1556
|
}
|
|
1530
1557
|
controller.enqueue({
|
|
1531
1558
|
type: "finish",
|
|
1532
1559
|
finishReason,
|
|
1533
|
-
usage:
|
|
1560
|
+
usage: convertAmazonBedrockUsage(usage),
|
|
1534
1561
|
...providerMetadata && { providerMetadata }
|
|
1535
1562
|
});
|
|
1536
1563
|
}
|
|
@@ -1545,28 +1572,28 @@ var BedrockChatLanguageModel = class _BedrockChatLanguageModel {
|
|
|
1545
1572
|
return `${this.config.baseUrl()}/model/${encodedModelId}`;
|
|
1546
1573
|
}
|
|
1547
1574
|
};
|
|
1548
|
-
var
|
|
1575
|
+
var AmazonBedrockStopReasonSchema = z4.union([
|
|
1549
1576
|
z4.enum(BEDROCK_STOP_REASONS),
|
|
1550
1577
|
z4.string()
|
|
1551
1578
|
]);
|
|
1552
|
-
var
|
|
1579
|
+
var AmazonBedrockAdditionalModelResponseFieldsSchema = z4.object({
|
|
1553
1580
|
delta: z4.object({
|
|
1554
1581
|
stop_sequence: z4.string().nullish()
|
|
1555
1582
|
}).nullish()
|
|
1556
1583
|
}).catchall(z4.unknown());
|
|
1557
|
-
var
|
|
1584
|
+
var AmazonBedrockToolUseSchema = z4.object({
|
|
1558
1585
|
toolUseId: z4.string(),
|
|
1559
1586
|
name: z4.string(),
|
|
1560
1587
|
input: z4.unknown()
|
|
1561
1588
|
});
|
|
1562
|
-
var
|
|
1589
|
+
var AmazonBedrockReasoningTextSchema = z4.object({
|
|
1563
1590
|
signature: z4.string().nullish(),
|
|
1564
1591
|
text: z4.string()
|
|
1565
1592
|
});
|
|
1566
|
-
var
|
|
1593
|
+
var AmazonBedrockRedactedReasoningSchema = z4.object({
|
|
1567
1594
|
data: z4.string()
|
|
1568
1595
|
});
|
|
1569
|
-
var
|
|
1596
|
+
var AmazonBedrockResponseSchema = z4.object({
|
|
1570
1597
|
metrics: z4.object({
|
|
1571
1598
|
latencyMs: z4.number()
|
|
1572
1599
|
}).nullish(),
|
|
@@ -1575,13 +1602,13 @@ var BedrockResponseSchema = z4.object({
|
|
|
1575
1602
|
content: z4.array(
|
|
1576
1603
|
z4.object({
|
|
1577
1604
|
text: z4.string().nullish(),
|
|
1578
|
-
toolUse:
|
|
1605
|
+
toolUse: AmazonBedrockToolUseSchema.nullish(),
|
|
1579
1606
|
reasoningContent: z4.union([
|
|
1580
1607
|
z4.object({
|
|
1581
|
-
reasoningText:
|
|
1608
|
+
reasoningText: AmazonBedrockReasoningTextSchema
|
|
1582
1609
|
}),
|
|
1583
1610
|
z4.object({
|
|
1584
|
-
redactedReasoning:
|
|
1611
|
+
redactedReasoning: AmazonBedrockRedactedReasoningSchema
|
|
1585
1612
|
})
|
|
1586
1613
|
]).nullish()
|
|
1587
1614
|
})
|
|
@@ -1589,8 +1616,8 @@ var BedrockResponseSchema = z4.object({
|
|
|
1589
1616
|
role: z4.string()
|
|
1590
1617
|
})
|
|
1591
1618
|
}),
|
|
1592
|
-
stopReason:
|
|
1593
|
-
additionalModelResponseFields:
|
|
1619
|
+
stopReason: AmazonBedrockStopReasonSchema,
|
|
1620
|
+
additionalModelResponseFields: AmazonBedrockAdditionalModelResponseFieldsSchema.nullish(),
|
|
1594
1621
|
trace: z4.unknown().nullish(),
|
|
1595
1622
|
performanceConfig: z4.object({ latency: z4.string() }).nullish(),
|
|
1596
1623
|
serviceTier: z4.object({ type: z4.string() }).nullish(),
|
|
@@ -1603,7 +1630,7 @@ var BedrockResponseSchema = z4.object({
|
|
|
1603
1630
|
cacheDetails: z4.array(z4.object({ inputTokens: z4.number(), ttl: z4.string() })).nullish()
|
|
1604
1631
|
})
|
|
1605
1632
|
});
|
|
1606
|
-
var
|
|
1633
|
+
var AmazonBedrockStreamSchema = z4.object({
|
|
1607
1634
|
contentBlockDelta: z4.object({
|
|
1608
1635
|
contentBlockIndex: z4.number(),
|
|
1609
1636
|
delta: z4.union([
|
|
@@ -1625,7 +1652,7 @@ var BedrockStreamSchema = z4.object({
|
|
|
1625
1652
|
contentBlockStart: z4.object({
|
|
1626
1653
|
contentBlockIndex: z4.number(),
|
|
1627
1654
|
start: z4.object({
|
|
1628
|
-
toolUse:
|
|
1655
|
+
toolUse: AmazonBedrockToolUseSchema.nullish()
|
|
1629
1656
|
}).nullish()
|
|
1630
1657
|
}).nullish(),
|
|
1631
1658
|
contentBlockStop: z4.object({
|
|
@@ -1633,8 +1660,8 @@ var BedrockStreamSchema = z4.object({
|
|
|
1633
1660
|
}).nullish(),
|
|
1634
1661
|
internalServerException: z4.record(z4.string(), z4.unknown()).nullish(),
|
|
1635
1662
|
messageStop: z4.object({
|
|
1636
|
-
additionalModelResponseFields:
|
|
1637
|
-
stopReason:
|
|
1663
|
+
additionalModelResponseFields: AmazonBedrockAdditionalModelResponseFieldsSchema.nullish(),
|
|
1664
|
+
stopReason: AmazonBedrockStopReasonSchema
|
|
1638
1665
|
}).nullish(),
|
|
1639
1666
|
metadata: z4.object({
|
|
1640
1667
|
trace: z4.unknown().nullish(),
|
|
@@ -1652,25 +1679,25 @@ var BedrockStreamSchema = z4.object({
|
|
|
1652
1679
|
throttlingException: z4.record(z4.string(), z4.unknown()).nullish(),
|
|
1653
1680
|
validationException: z4.record(z4.string(), z4.unknown()).nullish()
|
|
1654
1681
|
});
|
|
1655
|
-
var
|
|
1682
|
+
var amazonBedrockReasoningEffortMap = {
|
|
1656
1683
|
minimal: "low",
|
|
1657
1684
|
low: "low",
|
|
1658
1685
|
medium: "medium",
|
|
1659
1686
|
high: "high",
|
|
1660
1687
|
xhigh: "max"
|
|
1661
1688
|
};
|
|
1662
|
-
function
|
|
1689
|
+
function resolveAmazonBedrockReasoningConfig({
|
|
1663
1690
|
reasoning,
|
|
1664
|
-
|
|
1691
|
+
amazonBedrockOptions,
|
|
1665
1692
|
warnings,
|
|
1666
1693
|
isAnthropicModel,
|
|
1667
1694
|
modelId
|
|
1668
1695
|
}) {
|
|
1669
1696
|
var _a;
|
|
1670
1697
|
if (!isCustomReasoning(reasoning)) {
|
|
1671
|
-
return
|
|
1698
|
+
return amazonBedrockOptions;
|
|
1672
1699
|
}
|
|
1673
|
-
const result = { ...
|
|
1700
|
+
const result = { ...amazonBedrockOptions };
|
|
1674
1701
|
if (isAnthropicModel) {
|
|
1675
1702
|
const capabilities = getModelCapabilities(modelId);
|
|
1676
1703
|
if (reasoning === "none") {
|
|
@@ -1678,13 +1705,13 @@ function resolveBedrockReasoningConfig({
|
|
|
1678
1705
|
} else if (capabilities.supportsAdaptiveThinking) {
|
|
1679
1706
|
const effort = mapReasoningToProviderEffort({
|
|
1680
1707
|
reasoning,
|
|
1681
|
-
effortMap:
|
|
1708
|
+
effortMap: amazonBedrockReasoningEffortMap,
|
|
1682
1709
|
warnings
|
|
1683
1710
|
});
|
|
1684
1711
|
result.reasoningConfig = {
|
|
1685
1712
|
type: "adaptive",
|
|
1686
1713
|
maxReasoningEffort: effort,
|
|
1687
|
-
...
|
|
1714
|
+
...amazonBedrockOptions.reasoningConfig
|
|
1688
1715
|
};
|
|
1689
1716
|
} else {
|
|
1690
1717
|
const budgetTokens = mapReasoningToProviderBudget({
|
|
@@ -1697,19 +1724,19 @@ function resolveBedrockReasoningConfig({
|
|
|
1697
1724
|
result.reasoningConfig = {
|
|
1698
1725
|
type: "enabled",
|
|
1699
1726
|
budgetTokens,
|
|
1700
|
-
...
|
|
1727
|
+
...amazonBedrockOptions.reasoningConfig
|
|
1701
1728
|
};
|
|
1702
1729
|
}
|
|
1703
1730
|
}
|
|
1704
1731
|
} else if (reasoning !== "none") {
|
|
1705
1732
|
const effort = mapReasoningToProviderEffort({
|
|
1706
1733
|
reasoning,
|
|
1707
|
-
effortMap:
|
|
1734
|
+
effortMap: amazonBedrockReasoningEffortMap,
|
|
1708
1735
|
warnings
|
|
1709
1736
|
});
|
|
1710
1737
|
result.reasoningConfig = {
|
|
1711
1738
|
maxReasoningEffort: effort,
|
|
1712
|
-
...
|
|
1739
|
+
...amazonBedrockOptions.reasoningConfig
|
|
1713
1740
|
};
|
|
1714
1741
|
}
|
|
1715
1742
|
if (((_a = result.reasoningConfig) == null ? void 0 : _a.type) === "disabled") {
|
|
@@ -1719,7 +1746,7 @@ function resolveBedrockReasoningConfig({
|
|
|
1719
1746
|
return result;
|
|
1720
1747
|
}
|
|
1721
1748
|
|
|
1722
|
-
// src/bedrock-embedding-model.ts
|
|
1749
|
+
// src/amazon-bedrock-embedding-model.ts
|
|
1723
1750
|
import {
|
|
1724
1751
|
TooManyEmbeddingValuesForCallError
|
|
1725
1752
|
} from "@ai-sdk/provider";
|
|
@@ -1735,7 +1762,7 @@ import {
|
|
|
1735
1762
|
WORKFLOW_DESERIALIZE as WORKFLOW_DESERIALIZE2
|
|
1736
1763
|
} from "@ai-sdk/provider-utils";
|
|
1737
1764
|
|
|
1738
|
-
// src/bedrock-embedding-options.ts
|
|
1765
|
+
// src/amazon-bedrock-embedding-model-options.ts
|
|
1739
1766
|
import { z as z5 } from "zod/v4";
|
|
1740
1767
|
var amazonBedrockEmbeddingModelOptionsSchema = z5.object({
|
|
1741
1768
|
/**
|
|
@@ -1787,9 +1814,9 @@ var amazonBedrockEmbeddingModelOptionsSchema = z5.object({
|
|
|
1787
1814
|
outputDimension: z5.union([z5.literal(256), z5.literal(512), z5.literal(1024), z5.literal(1536)]).optional()
|
|
1788
1815
|
});
|
|
1789
1816
|
|
|
1790
|
-
// src/bedrock-embedding-model.ts
|
|
1817
|
+
// src/amazon-bedrock-embedding-model.ts
|
|
1791
1818
|
import { z as z6 } from "zod/v4";
|
|
1792
|
-
var
|
|
1819
|
+
var AmazonBedrockEmbeddingModel = class _AmazonBedrockEmbeddingModel {
|
|
1793
1820
|
constructor(modelId, config) {
|
|
1794
1821
|
this.modelId = modelId;
|
|
1795
1822
|
this.config = config;
|
|
@@ -1805,7 +1832,7 @@ var BedrockEmbeddingModel = class _BedrockEmbeddingModel {
|
|
|
1805
1832
|
});
|
|
1806
1833
|
}
|
|
1807
1834
|
static [WORKFLOW_DESERIALIZE2](options) {
|
|
1808
|
-
return new
|
|
1835
|
+
return new _AmazonBedrockEmbeddingModel(options.modelId, options.config);
|
|
1809
1836
|
}
|
|
1810
1837
|
getUrl(modelId) {
|
|
1811
1838
|
const encodedModelId = encodeURIComponent(modelId);
|
|
@@ -1817,7 +1844,7 @@ var BedrockEmbeddingModel = class _BedrockEmbeddingModel {
|
|
|
1817
1844
|
abortSignal,
|
|
1818
1845
|
providerOptions
|
|
1819
1846
|
}) {
|
|
1820
|
-
var _a, _b, _c, _d, _e, _f;
|
|
1847
|
+
var _a, _b, _c, _d, _e, _f, _g;
|
|
1821
1848
|
if (values.length > this.maxEmbeddingsPerCall) {
|
|
1822
1849
|
throw new TooManyEmbeddingValuesForCallError({
|
|
1823
1850
|
provider: this.provider,
|
|
@@ -1826,34 +1853,38 @@ var BedrockEmbeddingModel = class _BedrockEmbeddingModel {
|
|
|
1826
1853
|
values
|
|
1827
1854
|
});
|
|
1828
1855
|
}
|
|
1829
|
-
const
|
|
1856
|
+
const amazonBedrockOptions = (_b = (_a = await parseProviderOptions3({
|
|
1857
|
+
provider: "amazonBedrock",
|
|
1858
|
+
providerOptions,
|
|
1859
|
+
schema: amazonBedrockEmbeddingModelOptionsSchema
|
|
1860
|
+
})) != null ? _a : await parseProviderOptions3({
|
|
1830
1861
|
provider: "bedrock",
|
|
1831
1862
|
providerOptions,
|
|
1832
1863
|
schema: amazonBedrockEmbeddingModelOptionsSchema
|
|
1833
|
-
})) != null ?
|
|
1864
|
+
})) != null ? _b : {};
|
|
1834
1865
|
const isNovaModel = this.modelId.startsWith("amazon.nova-") && this.modelId.includes("embed");
|
|
1835
1866
|
const isCohereModel = this.modelId.startsWith("cohere.embed-");
|
|
1836
1867
|
const args = isNovaModel ? {
|
|
1837
1868
|
taskType: "SINGLE_EMBEDDING",
|
|
1838
1869
|
singleEmbeddingParams: {
|
|
1839
|
-
embeddingPurpose: (
|
|
1840
|
-
embeddingDimension: (
|
|
1870
|
+
embeddingPurpose: (_c = amazonBedrockOptions.embeddingPurpose) != null ? _c : "GENERIC_INDEX",
|
|
1871
|
+
embeddingDimension: (_d = amazonBedrockOptions.embeddingDimension) != null ? _d : 1024,
|
|
1841
1872
|
text: {
|
|
1842
|
-
truncationMode: (
|
|
1873
|
+
truncationMode: (_e = amazonBedrockOptions.truncate) != null ? _e : "END",
|
|
1843
1874
|
value: values[0]
|
|
1844
1875
|
}
|
|
1845
1876
|
}
|
|
1846
1877
|
} : isCohereModel ? {
|
|
1847
1878
|
// Cohere embedding models on Bedrock require `input_type`.
|
|
1848
1879
|
// Without it, the service attempts other schema branches and rejects the request.
|
|
1849
|
-
input_type: (
|
|
1880
|
+
input_type: (_f = amazonBedrockOptions.inputType) != null ? _f : "search_query",
|
|
1850
1881
|
texts: [values[0]],
|
|
1851
|
-
truncate:
|
|
1852
|
-
output_dimension:
|
|
1882
|
+
truncate: amazonBedrockOptions.truncate,
|
|
1883
|
+
output_dimension: amazonBedrockOptions.outputDimension
|
|
1853
1884
|
} : {
|
|
1854
1885
|
inputText: values[0],
|
|
1855
|
-
dimensions:
|
|
1856
|
-
normalize:
|
|
1886
|
+
dimensions: amazonBedrockOptions.dimensions,
|
|
1887
|
+
normalize: amazonBedrockOptions.normalize
|
|
1857
1888
|
};
|
|
1858
1889
|
const url = this.getUrl(this.modelId);
|
|
1859
1890
|
const { value: response } = await postJsonToApi2({
|
|
@@ -1866,11 +1897,11 @@ var BedrockEmbeddingModel = class _BedrockEmbeddingModel {
|
|
|
1866
1897
|
),
|
|
1867
1898
|
body: args,
|
|
1868
1899
|
failedResponseHandler: createJsonErrorResponseHandler2({
|
|
1869
|
-
errorSchema:
|
|
1900
|
+
errorSchema: AmazonBedrockErrorSchema,
|
|
1870
1901
|
errorToMessage: (error) => `${error.type}: ${error.message}`
|
|
1871
1902
|
}),
|
|
1872
1903
|
successfulResponseHandler: createJsonResponseHandler2(
|
|
1873
|
-
|
|
1904
|
+
AmazonBedrockEmbeddingResponseSchema
|
|
1874
1905
|
),
|
|
1875
1906
|
fetch: this.config.fetch,
|
|
1876
1907
|
abortSignal
|
|
@@ -1888,7 +1919,7 @@ var BedrockEmbeddingModel = class _BedrockEmbeddingModel {
|
|
|
1888
1919
|
} else {
|
|
1889
1920
|
embedding = response.embeddings.float[0];
|
|
1890
1921
|
}
|
|
1891
|
-
const tokens = "inputTextTokenCount" in response ? response.inputTextTokenCount : "inputTokenCount" in response ? (
|
|
1922
|
+
const tokens = "inputTextTokenCount" in response ? response.inputTextTokenCount : "inputTokenCount" in response ? (_g = response.inputTokenCount) != null ? _g : 0 : NaN;
|
|
1892
1923
|
return {
|
|
1893
1924
|
embeddings: [embedding],
|
|
1894
1925
|
usage: { tokens },
|
|
@@ -1896,7 +1927,7 @@ var BedrockEmbeddingModel = class _BedrockEmbeddingModel {
|
|
|
1896
1927
|
};
|
|
1897
1928
|
}
|
|
1898
1929
|
};
|
|
1899
|
-
var
|
|
1930
|
+
var AmazonBedrockEmbeddingResponseSchema = z6.union([
|
|
1900
1931
|
// Titan-style response
|
|
1901
1932
|
z6.object({
|
|
1902
1933
|
embedding: z6.array(z6.number()),
|
|
@@ -1924,7 +1955,7 @@ var BedrockEmbeddingResponseSchema = z6.union([
|
|
|
1924
1955
|
})
|
|
1925
1956
|
]);
|
|
1926
1957
|
|
|
1927
|
-
// src/bedrock-image-model.ts
|
|
1958
|
+
// src/amazon-bedrock-image-model.ts
|
|
1928
1959
|
import {
|
|
1929
1960
|
combineHeaders as combineHeaders3,
|
|
1930
1961
|
convertUint8ArrayToBase64,
|
|
@@ -1937,14 +1968,14 @@ import {
|
|
|
1937
1968
|
WORKFLOW_DESERIALIZE as WORKFLOW_DESERIALIZE3
|
|
1938
1969
|
} from "@ai-sdk/provider-utils";
|
|
1939
1970
|
|
|
1940
|
-
// src/bedrock-image-settings.ts
|
|
1971
|
+
// src/amazon-bedrock-image-settings.ts
|
|
1941
1972
|
var modelMaxImagesPerCall = {
|
|
1942
1973
|
"amazon.nova-canvas-v1:0": 5
|
|
1943
1974
|
};
|
|
1944
1975
|
|
|
1945
|
-
// src/bedrock-image-model.ts
|
|
1976
|
+
// src/amazon-bedrock-image-model.ts
|
|
1946
1977
|
import { z as z7 } from "zod/v4";
|
|
1947
|
-
var
|
|
1978
|
+
var AmazonBedrockImageModel = class _AmazonBedrockImageModel {
|
|
1948
1979
|
constructor(modelId, config) {
|
|
1949
1980
|
this.modelId = modelId;
|
|
1950
1981
|
this.config = config;
|
|
@@ -1958,7 +1989,7 @@ var BedrockImageModel = class _BedrockImageModel {
|
|
|
1958
1989
|
});
|
|
1959
1990
|
}
|
|
1960
1991
|
static [WORKFLOW_DESERIALIZE3](options) {
|
|
1961
|
-
return new
|
|
1992
|
+
return new _AmazonBedrockImageModel(options.modelId, options.config);
|
|
1962
1993
|
}
|
|
1963
1994
|
get maxImagesPerCall() {
|
|
1964
1995
|
var _a;
|
|
@@ -1980,35 +2011,36 @@ var BedrockImageModel = class _BedrockImageModel {
|
|
|
1980
2011
|
files,
|
|
1981
2012
|
mask
|
|
1982
2013
|
}) {
|
|
1983
|
-
var _a, _b, _c, _d, _e, _f
|
|
2014
|
+
var _a, _b, _c, _d, _e, _f;
|
|
1984
2015
|
const warnings = [];
|
|
1985
2016
|
const [width, height] = size ? size.split("x").map(Number) : [];
|
|
1986
2017
|
const hasFiles = files != null && files.length > 0;
|
|
2018
|
+
const amazonBedrockOptions = (_a = providerOptions == null ? void 0 : providerOptions.amazonBedrock) != null ? _a : providerOptions == null ? void 0 : providerOptions.bedrock;
|
|
1987
2019
|
const imageGenerationConfig = {
|
|
1988
2020
|
...width ? { width } : {},
|
|
1989
2021
|
...height ? { height } : {},
|
|
1990
2022
|
...seed ? { seed } : {},
|
|
1991
2023
|
...n ? { numberOfImages: n } : {},
|
|
1992
|
-
...(
|
|
1993
|
-
...(
|
|
2024
|
+
...(amazonBedrockOptions == null ? void 0 : amazonBedrockOptions.quality) ? { quality: amazonBedrockOptions.quality } : {},
|
|
2025
|
+
...(amazonBedrockOptions == null ? void 0 : amazonBedrockOptions.cfgScale) ? { cfgScale: amazonBedrockOptions.cfgScale } : {}
|
|
1994
2026
|
};
|
|
1995
2027
|
let args;
|
|
1996
2028
|
if (hasFiles) {
|
|
1997
2029
|
const hasMask = (mask == null ? void 0 : mask.type) != null;
|
|
1998
|
-
const hasMaskPrompt = (
|
|
1999
|
-
const taskType = (
|
|
2030
|
+
const hasMaskPrompt = (amazonBedrockOptions == null ? void 0 : amazonBedrockOptions.maskPrompt) != null;
|
|
2031
|
+
const taskType = (_b = amazonBedrockOptions == null ? void 0 : amazonBedrockOptions.taskType) != null ? _b : hasMask || hasMaskPrompt ? "INPAINTING" : "IMAGE_VARIATION";
|
|
2000
2032
|
const sourceImageBase64 = getBase64Data(files[0]);
|
|
2001
2033
|
switch (taskType) {
|
|
2002
2034
|
case "INPAINTING": {
|
|
2003
2035
|
const inPaintingParams = {
|
|
2004
2036
|
image: sourceImageBase64,
|
|
2005
2037
|
...prompt ? { text: prompt } : {},
|
|
2006
|
-
...(
|
|
2038
|
+
...(amazonBedrockOptions == null ? void 0 : amazonBedrockOptions.negativeText) ? { negativeText: amazonBedrockOptions.negativeText } : {}
|
|
2007
2039
|
};
|
|
2008
2040
|
if (hasMask) {
|
|
2009
2041
|
inPaintingParams.maskImage = getBase64Data(mask);
|
|
2010
2042
|
} else if (hasMaskPrompt) {
|
|
2011
|
-
inPaintingParams.maskPrompt =
|
|
2043
|
+
inPaintingParams.maskPrompt = amazonBedrockOptions.maskPrompt;
|
|
2012
2044
|
}
|
|
2013
2045
|
args = {
|
|
2014
2046
|
taskType: "INPAINTING",
|
|
@@ -2021,13 +2053,13 @@ var BedrockImageModel = class _BedrockImageModel {
|
|
|
2021
2053
|
const outPaintingParams = {
|
|
2022
2054
|
image: sourceImageBase64,
|
|
2023
2055
|
...prompt ? { text: prompt } : {},
|
|
2024
|
-
...(
|
|
2025
|
-
...(
|
|
2056
|
+
...(amazonBedrockOptions == null ? void 0 : amazonBedrockOptions.negativeText) ? { negativeText: amazonBedrockOptions.negativeText } : {},
|
|
2057
|
+
...(amazonBedrockOptions == null ? void 0 : amazonBedrockOptions.outPaintingMode) ? { outPaintingMode: amazonBedrockOptions.outPaintingMode } : {}
|
|
2026
2058
|
};
|
|
2027
2059
|
if (hasMask) {
|
|
2028
2060
|
outPaintingParams.maskImage = getBase64Data(mask);
|
|
2029
2061
|
} else if (hasMaskPrompt) {
|
|
2030
|
-
outPaintingParams.maskPrompt =
|
|
2062
|
+
outPaintingParams.maskPrompt = amazonBedrockOptions.maskPrompt;
|
|
2031
2063
|
}
|
|
2032
2064
|
args = {
|
|
2033
2065
|
taskType: "OUTPAINTING",
|
|
@@ -2050,9 +2082,9 @@ var BedrockImageModel = class _BedrockImageModel {
|
|
|
2050
2082
|
const imageVariationParams = {
|
|
2051
2083
|
images,
|
|
2052
2084
|
...prompt ? { text: prompt } : {},
|
|
2053
|
-
...(
|
|
2054
|
-
...(
|
|
2055
|
-
similarityStrength:
|
|
2085
|
+
...(amazonBedrockOptions == null ? void 0 : amazonBedrockOptions.negativeText) ? { negativeText: amazonBedrockOptions.negativeText } : {},
|
|
2086
|
+
...(amazonBedrockOptions == null ? void 0 : amazonBedrockOptions.similarityStrength) != null ? {
|
|
2087
|
+
similarityStrength: amazonBedrockOptions.similarityStrength
|
|
2056
2088
|
} : {}
|
|
2057
2089
|
};
|
|
2058
2090
|
args = {
|
|
@@ -2070,11 +2102,11 @@ var BedrockImageModel = class _BedrockImageModel {
|
|
|
2070
2102
|
taskType: "TEXT_IMAGE",
|
|
2071
2103
|
textToImageParams: {
|
|
2072
2104
|
text: prompt,
|
|
2073
|
-
...(
|
|
2074
|
-
negativeText:
|
|
2105
|
+
...(amazonBedrockOptions == null ? void 0 : amazonBedrockOptions.negativeText) ? {
|
|
2106
|
+
negativeText: amazonBedrockOptions.negativeText
|
|
2075
2107
|
} : {},
|
|
2076
|
-
...(
|
|
2077
|
-
style:
|
|
2108
|
+
...(amazonBedrockOptions == null ? void 0 : amazonBedrockOptions.style) ? {
|
|
2109
|
+
style: amazonBedrockOptions.style
|
|
2078
2110
|
} : {}
|
|
2079
2111
|
},
|
|
2080
2112
|
imageGenerationConfig
|
|
@@ -2087,7 +2119,7 @@ var BedrockImageModel = class _BedrockImageModel {
|
|
|
2087
2119
|
details: "This model does not support aspect ratio. Use `size` instead."
|
|
2088
2120
|
});
|
|
2089
2121
|
}
|
|
2090
|
-
const currentDate = (
|
|
2122
|
+
const currentDate = (_e = (_d = (_c = this.config._internal) == null ? void 0 : _c.currentDate) == null ? void 0 : _d.call(_c)) != null ? _e : /* @__PURE__ */ new Date();
|
|
2091
2123
|
const { value: response, responseHeaders } = await postJsonToApi3({
|
|
2092
2124
|
url: this.getUrl(this.modelId),
|
|
2093
2125
|
headers: await resolve3(
|
|
@@ -2098,17 +2130,17 @@ var BedrockImageModel = class _BedrockImageModel {
|
|
|
2098
2130
|
),
|
|
2099
2131
|
body: args,
|
|
2100
2132
|
failedResponseHandler: createJsonErrorResponseHandler3({
|
|
2101
|
-
errorSchema:
|
|
2133
|
+
errorSchema: AmazonBedrockErrorSchema,
|
|
2102
2134
|
errorToMessage: (error) => `${error.type}: ${error.message}`
|
|
2103
2135
|
}),
|
|
2104
2136
|
successfulResponseHandler: createJsonResponseHandler3(
|
|
2105
|
-
|
|
2137
|
+
amazonBedrockImageResponseSchema
|
|
2106
2138
|
),
|
|
2107
2139
|
abortSignal,
|
|
2108
2140
|
fetch: this.config.fetch
|
|
2109
2141
|
});
|
|
2110
2142
|
if (response.status === "Request Moderated") {
|
|
2111
|
-
const moderationReasons = (
|
|
2143
|
+
const moderationReasons = (_f = response.details) == null ? void 0 : _f["Moderation Reasons"];
|
|
2112
2144
|
const reasons = Array.isArray(moderationReasons) ? moderationReasons : ["Unknown"];
|
|
2113
2145
|
throw new Error(
|
|
2114
2146
|
`Amazon Bedrock request was moderated: ${reasons.join(", ")}`
|
|
@@ -2141,7 +2173,7 @@ function getBase64Data(file) {
|
|
|
2141
2173
|
}
|
|
2142
2174
|
return file.data;
|
|
2143
2175
|
}
|
|
2144
|
-
var
|
|
2176
|
+
var amazonBedrockImageResponseSchema = z7.object({
|
|
2145
2177
|
// Normal successful response
|
|
2146
2178
|
images: z7.array(z7.string()).optional(),
|
|
2147
2179
|
// Moderation response fields
|
|
@@ -2153,7 +2185,7 @@ var bedrockImageResponseSchema = z7.object({
|
|
|
2153
2185
|
preview: z7.unknown().optional()
|
|
2154
2186
|
});
|
|
2155
2187
|
|
|
2156
|
-
// src/bedrock-sigv4-fetch.ts
|
|
2188
|
+
// src/amazon-bedrock-sigv4-fetch.ts
|
|
2157
2189
|
import {
|
|
2158
2190
|
combineHeaders as combineHeaders4,
|
|
2159
2191
|
normalizeHeaders,
|
|
@@ -2163,12 +2195,13 @@ import {
|
|
|
2163
2195
|
import { AwsV4Signer } from "aws4fetch";
|
|
2164
2196
|
|
|
2165
2197
|
// src/version.ts
|
|
2166
|
-
var VERSION = true ? "5.0.0-beta.
|
|
2198
|
+
var VERSION = true ? "5.0.0-beta.47" : "0.0.0-test";
|
|
2167
2199
|
|
|
2168
|
-
// src/bedrock-sigv4-fetch.ts
|
|
2169
|
-
function createSigV4FetchFunction(getCredentials, fetch
|
|
2200
|
+
// src/amazon-bedrock-sigv4-fetch.ts
|
|
2201
|
+
function createSigV4FetchFunction(getCredentials, fetch) {
|
|
2170
2202
|
return async (input, init) => {
|
|
2171
2203
|
var _a, _b;
|
|
2204
|
+
const effectiveFetch = fetch != null ? fetch : globalThis.fetch;
|
|
2172
2205
|
const request = input instanceof Request ? input : void 0;
|
|
2173
2206
|
const originalHeaders = combineHeaders4(
|
|
2174
2207
|
normalizeHeaders(request == null ? void 0 : request.headers),
|
|
@@ -2188,7 +2221,7 @@ function createSigV4FetchFunction(getCredentials, fetch = globalThis.fetch) {
|
|
|
2188
2221
|
}
|
|
2189
2222
|
const effectiveMethod = (_b = init == null ? void 0 : init.method) != null ? _b : request == null ? void 0 : request.method;
|
|
2190
2223
|
if ((effectiveMethod == null ? void 0 : effectiveMethod.toUpperCase()) !== "POST" || !effectiveBody) {
|
|
2191
|
-
return
|
|
2224
|
+
return effectiveFetch(input, {
|
|
2192
2225
|
...init,
|
|
2193
2226
|
headers: headersWithUserAgent
|
|
2194
2227
|
});
|
|
@@ -2210,7 +2243,7 @@ function createSigV4FetchFunction(getCredentials, fetch = globalThis.fetch) {
|
|
|
2210
2243
|
const signingResult = await signer.sign();
|
|
2211
2244
|
const signedHeaders = normalizeHeaders(signingResult.headers);
|
|
2212
2245
|
const combinedHeaders = combineHeaders4(headersWithUserAgent, signedHeaders);
|
|
2213
|
-
return
|
|
2246
|
+
return effectiveFetch(input, {
|
|
2214
2247
|
...init,
|
|
2215
2248
|
body,
|
|
2216
2249
|
headers: combinedHeaders
|
|
@@ -2228,8 +2261,9 @@ function prepareBodyString(body) {
|
|
|
2228
2261
|
return JSON.stringify(body);
|
|
2229
2262
|
}
|
|
2230
2263
|
}
|
|
2231
|
-
function createApiKeyFetchFunction(apiKey, fetch
|
|
2264
|
+
function createApiKeyFetchFunction(apiKey, fetch) {
|
|
2232
2265
|
return async (input, init) => {
|
|
2266
|
+
const effectiveFetch = fetch != null ? fetch : globalThis.fetch;
|
|
2233
2267
|
const originalHeaders = normalizeHeaders(init == null ? void 0 : init.headers);
|
|
2234
2268
|
const headersWithUserAgent = withUserAgentSuffix(
|
|
2235
2269
|
originalHeaders,
|
|
@@ -2239,14 +2273,14 @@ function createApiKeyFetchFunction(apiKey, fetch = globalThis.fetch) {
|
|
|
2239
2273
|
const finalHeaders = combineHeaders4(headersWithUserAgent, {
|
|
2240
2274
|
Authorization: `Bearer ${apiKey}`
|
|
2241
2275
|
});
|
|
2242
|
-
return
|
|
2276
|
+
return effectiveFetch(input, {
|
|
2243
2277
|
...init,
|
|
2244
2278
|
headers: finalHeaders
|
|
2245
2279
|
});
|
|
2246
2280
|
};
|
|
2247
2281
|
}
|
|
2248
2282
|
|
|
2249
|
-
// src/reranking/bedrock-reranking-model.ts
|
|
2283
|
+
// src/reranking/amazon-bedrock-reranking-model.ts
|
|
2250
2284
|
import {
|
|
2251
2285
|
combineHeaders as combineHeaders5,
|
|
2252
2286
|
createJsonErrorResponseHandler as createJsonErrorResponseHandler4,
|
|
@@ -2256,10 +2290,10 @@ import {
|
|
|
2256
2290
|
resolve as resolve4
|
|
2257
2291
|
} from "@ai-sdk/provider-utils";
|
|
2258
2292
|
|
|
2259
|
-
// src/reranking/bedrock-reranking-api.ts
|
|
2293
|
+
// src/reranking/amazon-bedrock-reranking-api.ts
|
|
2260
2294
|
import { lazySchema, zodSchema } from "@ai-sdk/provider-utils";
|
|
2261
2295
|
import { z as z8 } from "zod/v4";
|
|
2262
|
-
var
|
|
2296
|
+
var amazonBedrockRerankingResponseSchema = lazySchema(
|
|
2263
2297
|
() => zodSchema(
|
|
2264
2298
|
z8.object({
|
|
2265
2299
|
results: z8.array(
|
|
@@ -2273,7 +2307,7 @@ var bedrockRerankingResponseSchema = lazySchema(
|
|
|
2273
2307
|
)
|
|
2274
2308
|
);
|
|
2275
2309
|
|
|
2276
|
-
// src/reranking/bedrock-reranking-options.ts
|
|
2310
|
+
// src/reranking/amazon-bedrock-reranking-model-options.ts
|
|
2277
2311
|
import { lazySchema as lazySchema2, zodSchema as zodSchema2 } from "@ai-sdk/provider-utils";
|
|
2278
2312
|
import { z as z9 } from "zod/v4";
|
|
2279
2313
|
var amazonBedrockRerankingModelOptionsSchema = lazySchema2(
|
|
@@ -2291,8 +2325,8 @@ var amazonBedrockRerankingModelOptionsSchema = lazySchema2(
|
|
|
2291
2325
|
)
|
|
2292
2326
|
);
|
|
2293
2327
|
|
|
2294
|
-
// src/reranking/bedrock-reranking-model.ts
|
|
2295
|
-
var
|
|
2328
|
+
// src/reranking/amazon-bedrock-reranking-model.ts
|
|
2329
|
+
var AmazonBedrockRerankingModel = class {
|
|
2296
2330
|
constructor(modelId, config) {
|
|
2297
2331
|
this.modelId = modelId;
|
|
2298
2332
|
this.config = config;
|
|
@@ -2307,7 +2341,12 @@ var BedrockRerankingModel = class {
|
|
|
2307
2341
|
abortSignal,
|
|
2308
2342
|
providerOptions
|
|
2309
2343
|
}) {
|
|
2310
|
-
|
|
2344
|
+
var _a;
|
|
2345
|
+
const amazonBedrockOptions = (_a = await parseProviderOptions4({
|
|
2346
|
+
provider: "amazonBedrock",
|
|
2347
|
+
providerOptions,
|
|
2348
|
+
schema: amazonBedrockRerankingModelOptionsSchema
|
|
2349
|
+
})) != null ? _a : await parseProviderOptions4({
|
|
2311
2350
|
provider: "bedrock",
|
|
2312
2351
|
providerOptions,
|
|
2313
2352
|
schema: amazonBedrockRerankingModelOptionsSchema
|
|
@@ -2322,7 +2361,7 @@ var BedrockRerankingModel = class {
|
|
|
2322
2361
|
combineHeaders5(await resolve4(this.config.headers), headers)
|
|
2323
2362
|
),
|
|
2324
2363
|
body: {
|
|
2325
|
-
nextToken:
|
|
2364
|
+
nextToken: amazonBedrockOptions == null ? void 0 : amazonBedrockOptions.nextToken,
|
|
2326
2365
|
queries: [
|
|
2327
2366
|
{
|
|
2328
2367
|
textQuery: { text: query },
|
|
@@ -2330,10 +2369,10 @@ var BedrockRerankingModel = class {
|
|
|
2330
2369
|
}
|
|
2331
2370
|
],
|
|
2332
2371
|
rerankingConfiguration: {
|
|
2333
|
-
|
|
2372
|
+
amazonBedrockRerankingConfiguration: {
|
|
2334
2373
|
modelConfiguration: {
|
|
2335
2374
|
modelArn: `arn:aws:bedrock:${this.config.region}::foundation-model/${this.modelId}`,
|
|
2336
|
-
additionalModelRequestFields:
|
|
2375
|
+
additionalModelRequestFields: amazonBedrockOptions == null ? void 0 : amazonBedrockOptions.additionalModelRequestFields
|
|
2337
2376
|
},
|
|
2338
2377
|
numberOfResults: topN
|
|
2339
2378
|
},
|
|
@@ -2351,11 +2390,11 @@ var BedrockRerankingModel = class {
|
|
|
2351
2390
|
}))
|
|
2352
2391
|
},
|
|
2353
2392
|
failedResponseHandler: createJsonErrorResponseHandler4({
|
|
2354
|
-
errorSchema:
|
|
2393
|
+
errorSchema: AmazonBedrockErrorSchema,
|
|
2355
2394
|
errorToMessage: (error) => `${error.type}: ${error.message}`
|
|
2356
2395
|
}),
|
|
2357
2396
|
successfulResponseHandler: createJsonResponseHandler4(
|
|
2358
|
-
|
|
2397
|
+
amazonBedrockRerankingResponseSchema
|
|
2359
2398
|
),
|
|
2360
2399
|
fetch: this.config.fetch,
|
|
2361
2400
|
abortSignal
|
|
@@ -2370,7 +2409,7 @@ var BedrockRerankingModel = class {
|
|
|
2370
2409
|
}
|
|
2371
2410
|
};
|
|
2372
2411
|
|
|
2373
|
-
// src/bedrock-provider.ts
|
|
2412
|
+
// src/amazon-bedrock-provider.ts
|
|
2374
2413
|
function createAmazonBedrock(options = {}) {
|
|
2375
2414
|
const rawApiKey = loadOptionalSetting({
|
|
2376
2415
|
settingValue: options.apiKey,
|
|
@@ -2443,7 +2482,7 @@ Original error: ${errorMessage}`
|
|
|
2443
2482
|
const baseHeaders = (_a = options.headers) != null ? _a : {};
|
|
2444
2483
|
return withUserAgentSuffix2(baseHeaders, `ai-sdk/amazon-bedrock/${VERSION}`);
|
|
2445
2484
|
};
|
|
2446
|
-
const
|
|
2485
|
+
const getAmazonBedrockRuntimeBaseUrl = () => {
|
|
2447
2486
|
var _a, _b;
|
|
2448
2487
|
return (_b = withoutTrailingSlash(
|
|
2449
2488
|
(_a = options.baseURL) != null ? _a : `https://bedrock-runtime.${loadSetting({
|
|
@@ -2454,7 +2493,7 @@ Original error: ${errorMessage}`
|
|
|
2454
2493
|
})}.amazonaws.com`
|
|
2455
2494
|
)) != null ? _b : `https://bedrock-runtime.us-east-1.amazonaws.com`;
|
|
2456
2495
|
};
|
|
2457
|
-
const
|
|
2496
|
+
const getAmazonBedrockAgentRuntimeBaseUrl = () => {
|
|
2458
2497
|
var _a, _b;
|
|
2459
2498
|
return (_b = withoutTrailingSlash(
|
|
2460
2499
|
(_a = options.baseURL) != null ? _a : `https://bedrock-agent-runtime.${loadSetting({
|
|
@@ -2465,8 +2504,8 @@ Original error: ${errorMessage}`
|
|
|
2465
2504
|
})}.amazonaws.com`
|
|
2466
2505
|
)) != null ? _b : `https://bedrock-agent-runtime.us-west-2.amazonaws.com`;
|
|
2467
2506
|
};
|
|
2468
|
-
const createChatModel = (modelId) => new
|
|
2469
|
-
baseUrl:
|
|
2507
|
+
const createChatModel = (modelId) => new AmazonBedrockChatLanguageModel(modelId, {
|
|
2508
|
+
baseUrl: getAmazonBedrockRuntimeBaseUrl,
|
|
2470
2509
|
headers: getHeaders,
|
|
2471
2510
|
fetch: fetchFunction,
|
|
2472
2511
|
generateId
|
|
@@ -2479,18 +2518,18 @@ Original error: ${errorMessage}`
|
|
|
2479
2518
|
}
|
|
2480
2519
|
return createChatModel(modelId);
|
|
2481
2520
|
};
|
|
2482
|
-
const createEmbeddingModel = (modelId) => new
|
|
2483
|
-
baseUrl:
|
|
2521
|
+
const createEmbeddingModel = (modelId) => new AmazonBedrockEmbeddingModel(modelId, {
|
|
2522
|
+
baseUrl: getAmazonBedrockRuntimeBaseUrl,
|
|
2484
2523
|
headers: getHeaders,
|
|
2485
2524
|
fetch: fetchFunction
|
|
2486
2525
|
});
|
|
2487
|
-
const createImageModel = (modelId) => new
|
|
2488
|
-
baseUrl:
|
|
2526
|
+
const createImageModel = (modelId) => new AmazonBedrockImageModel(modelId, {
|
|
2527
|
+
baseUrl: getAmazonBedrockRuntimeBaseUrl,
|
|
2489
2528
|
headers: getHeaders,
|
|
2490
2529
|
fetch: fetchFunction
|
|
2491
2530
|
});
|
|
2492
|
-
const createRerankingModel = (modelId) => new
|
|
2493
|
-
baseUrl:
|
|
2531
|
+
const createRerankingModel = (modelId) => new AmazonBedrockRerankingModel(modelId, {
|
|
2532
|
+
baseUrl: getAmazonBedrockAgentRuntimeBaseUrl,
|
|
2494
2533
|
region: loadSetting({
|
|
2495
2534
|
settingValue: options.region,
|
|
2496
2535
|
settingName: "region",
|
|
@@ -2513,10 +2552,11 @@ Original error: ${errorMessage}`
|
|
|
2513
2552
|
provider.tools = anthropicTools2;
|
|
2514
2553
|
return provider;
|
|
2515
2554
|
}
|
|
2516
|
-
var
|
|
2555
|
+
var amazonBedrock = createAmazonBedrock();
|
|
2517
2556
|
export {
|
|
2518
2557
|
VERSION,
|
|
2519
|
-
|
|
2558
|
+
amazonBedrock,
|
|
2559
|
+
amazonBedrock as bedrock,
|
|
2520
2560
|
createAmazonBedrock
|
|
2521
2561
|
};
|
|
2522
2562
|
//# sourceMappingURL=index.js.map
|