@opencode-ai/ai 0.0.0-bootstrap.0 → 0.0.0-next-15617

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.
Files changed (151) hide show
  1. package/README.md +176 -1
  2. package/dist/cache-policy.d.ts +2 -0
  3. package/dist/cache-policy.js +108 -0
  4. package/dist/index.d.ts +14 -0
  5. package/dist/index.js +9 -0
  6. package/dist/llm.d.ts +223 -0
  7. package/dist/llm.js +81 -0
  8. package/dist/protocols/anthropic-messages.d.ts +503 -0
  9. package/dist/protocols/anthropic-messages.js +745 -0
  10. package/dist/protocols/bedrock-converse.d.ts +476 -0
  11. package/dist/protocols/bedrock-converse.js +526 -0
  12. package/dist/protocols/bedrock-event-stream.d.ts +9 -0
  13. package/dist/protocols/bedrock-event-stream.js +64 -0
  14. package/dist/protocols/gemini.d.ts +220 -0
  15. package/dist/protocols/gemini.js +420 -0
  16. package/dist/protocols/google-vertex-anthropic.d.ts +494 -0
  17. package/dist/protocols/google-vertex-anthropic.js +33 -0
  18. package/dist/protocols/google-vertex-gemini.d.ts +59 -0
  19. package/dist/protocols/google-vertex-gemini.js +18 -0
  20. package/dist/protocols/index.d.ts +9 -0
  21. package/dist/protocols/index.js +9 -0
  22. package/dist/protocols/openai-chat.d.ts +391 -0
  23. package/dist/protocols/openai-chat.js +423 -0
  24. package/dist/protocols/openai-compatible-chat.d.ts +73 -0
  25. package/dist/protocols/openai-compatible-chat.js +20 -0
  26. package/dist/protocols/openai-compatible-responses.d.ts +85 -0
  27. package/dist/protocols/openai-compatible-responses.js +18 -0
  28. package/dist/protocols/openai-responses.d.ts +613 -0
  29. package/dist/protocols/openai-responses.js +828 -0
  30. package/dist/protocols/shared.d.ts +192 -0
  31. package/dist/protocols/shared.js +236 -0
  32. package/dist/protocols/utils/bedrock-auth.d.ts +18 -0
  33. package/dist/protocols/utils/bedrock-auth.js +41 -0
  34. package/dist/protocols/utils/bedrock-cache.d.ts +15 -0
  35. package/dist/protocols/utils/bedrock-cache.js +29 -0
  36. package/dist/protocols/utils/bedrock-media.d.ts +49 -0
  37. package/dist/protocols/utils/bedrock-media.js +68 -0
  38. package/dist/protocols/utils/cache.d.ts +6 -0
  39. package/dist/protocols/utils/cache.js +8 -0
  40. package/dist/protocols/utils/gemini-tool-schema.d.ts +2 -0
  41. package/dist/protocols/utils/gemini-tool-schema.js +83 -0
  42. package/dist/protocols/utils/lifecycle.d.ts +19 -0
  43. package/dist/protocols/utils/lifecycle.js +65 -0
  44. package/dist/protocols/utils/openai-options.d.ts +22 -0
  45. package/dist/protocols/utils/openai-options.js +64 -0
  46. package/dist/protocols/utils/tool-schema.d.ts +7 -0
  47. package/dist/protocols/utils/tool-schema.js +79 -0
  48. package/dist/protocols/utils/tool-stream.d.ts +155 -0
  49. package/dist/protocols/utils/tool-stream.js +136 -0
  50. package/dist/protocols.d.ts +1 -0
  51. package/dist/protocols.js +1 -0
  52. package/dist/provider-error.d.ts +13 -0
  53. package/dist/provider-error.js +122 -0
  54. package/dist/provider-package.d.ts +13 -0
  55. package/dist/provider-package.js +1 -0
  56. package/dist/provider.d.ts +23 -0
  57. package/dist/provider.js +2 -0
  58. package/dist/providers/amazon-bedrock.d.ts +150 -0
  59. package/dist/providers/amazon-bedrock.js +41 -0
  60. package/dist/providers/anthropic-compatible.d.ts +183 -0
  61. package/dist/providers/anthropic-compatible.js +44 -0
  62. package/dist/providers/anthropic.d.ts +181 -0
  63. package/dist/providers/anthropic.js +39 -0
  64. package/dist/providers/azure/chat.d.ts +2 -0
  65. package/dist/providers/azure/chat.js +1 -0
  66. package/dist/providers/azure/responses.d.ts +2 -0
  67. package/dist/providers/azure/responses.js +1 -0
  68. package/dist/providers/azure.d.ts +183 -0
  69. package/dist/providers/azure.js +89 -0
  70. package/dist/providers/cloudflare.d.ts +232 -0
  71. package/dist/providers/cloudflare.js +87 -0
  72. package/dist/providers/github-copilot.d.ts +167 -0
  73. package/dist/providers/github-copilot.js +47 -0
  74. package/dist/providers/google-vertex/anthropic.d.ts +2 -0
  75. package/dist/providers/google-vertex/anthropic.js +1 -0
  76. package/dist/providers/google-vertex-anthropic.d.ts +181 -0
  77. package/dist/providers/google-vertex-anthropic.js +46 -0
  78. package/dist/providers/google-vertex-shared.d.ts +22 -0
  79. package/dist/providers/google-vertex-shared.js +61 -0
  80. package/dist/providers/google-vertex.d.ts +93 -0
  81. package/dist/providers/google-vertex.js +49 -0
  82. package/dist/providers/google.d.ts +85 -0
  83. package/dist/providers/google.js +33 -0
  84. package/dist/providers/index.d.ts +15 -0
  85. package/dist/providers/index.js +15 -0
  86. package/dist/providers/openai/chat.d.ts +2 -0
  87. package/dist/providers/openai/chat.js +1 -0
  88. package/dist/providers/openai/responses.d.ts +2 -0
  89. package/dist/providers/openai/responses.js +1 -0
  90. package/dist/providers/openai-compatible/responses.d.ts +1 -0
  91. package/dist/providers/openai-compatible/responses.js +1 -0
  92. package/dist/providers/openai-compatible-profile.d.ts +43 -0
  93. package/dist/providers/openai-compatible-profile.js +12 -0
  94. package/dist/providers/openai-compatible-responses.d.ts +107 -0
  95. package/dist/providers/openai-compatible-responses.js +33 -0
  96. package/dist/providers/openai-compatible.d.ts +130 -0
  97. package/dist/providers/openai-compatible.js +55 -0
  98. package/dist/providers/openai-options.d.ts +30 -0
  99. package/dist/providers/openai-options.js +42 -0
  100. package/dist/providers/openai.d.ts +264 -0
  101. package/dist/providers/openai.js +62 -0
  102. package/dist/providers/openrouter.d.ts +321 -0
  103. package/dist/providers/openrouter.js +65 -0
  104. package/dist/providers/xai.d.ts +169 -0
  105. package/dist/providers/xai.js +43 -0
  106. package/dist/providers.d.ts +1 -0
  107. package/dist/providers.js +1 -0
  108. package/dist/route/auth-options.d.ts +34 -0
  109. package/dist/route/auth-options.js +14 -0
  110. package/dist/route/auth.d.ts +49 -0
  111. package/dist/route/auth.js +89 -0
  112. package/dist/route/client.d.ts +330 -0
  113. package/dist/route/client.js +211 -0
  114. package/dist/route/endpoint.d.ts +28 -0
  115. package/dist/route/endpoint.js +21 -0
  116. package/dist/route/executor.d.ts +12 -0
  117. package/dist/route/executor.js +229 -0
  118. package/dist/route/framing.d.ts +23 -0
  119. package/dist/route/framing.js +4 -0
  120. package/dist/route/index.d.ts +16 -0
  121. package/dist/route/index.js +9 -0
  122. package/dist/route/protocol.d.ts +76 -0
  123. package/dist/route/protocol.js +17 -0
  124. package/dist/route/transport/http.d.ts +33 -0
  125. package/dist/route/transport/http.js +98 -0
  126. package/dist/route/transport/index.d.ts +27 -0
  127. package/dist/route/transport/index.js +4 -0
  128. package/dist/route/transport/websocket.d.ts +56 -0
  129. package/dist/route/transport/websocket.js +181 -0
  130. package/dist/route.d.ts +1 -0
  131. package/dist/route.js +1 -0
  132. package/dist/schema/errors.d.ts +153 -0
  133. package/dist/schema/errors.js +150 -0
  134. package/dist/schema/events.d.ts +4292 -0
  135. package/dist/schema/events.js +482 -0
  136. package/dist/schema/ids.d.ts +30 -0
  137. package/dist/schema/ids.js +18 -0
  138. package/dist/schema/index.d.ts +5 -0
  139. package/dist/schema/index.js +5 -0
  140. package/dist/schema/messages.d.ts +385 -0
  141. package/dist/schema/messages.js +246 -0
  142. package/dist/schema/options.d.ts +143 -0
  143. package/dist/schema/options.js +195 -0
  144. package/dist/tool-runtime.d.ts +15 -0
  145. package/dist/tool-runtime.js +35 -0
  146. package/dist/tool.d.ts +134 -0
  147. package/dist/tool.js +66 -0
  148. package/dist/utils/record.d.ts +2 -0
  149. package/dist/utils/record.js +2 -0
  150. package/package.json +36 -9
  151. package/index.js +0 -1
@@ -0,0 +1,526 @@
1
+ import { Effect, Schema } from "effect";
2
+ import { Route } from "../route/client";
3
+ import { Endpoint } from "../route/endpoint";
4
+ import { Protocol } from "../route/protocol";
5
+ import { LLMError, LLMEvent, Usage, } from "../schema";
6
+ import { BedrockEventStream } from "./bedrock-event-stream";
7
+ import { classifyProviderFailure } from "../provider-error";
8
+ import { JsonObject, optionalArray, ProviderShared } from "./shared";
9
+ import { BedrockAuth } from "./utils/bedrock-auth";
10
+ import { BedrockCache } from "./utils/bedrock-cache";
11
+ import { BedrockMedia } from "./utils/bedrock-media";
12
+ import { Lifecycle } from "./utils/lifecycle";
13
+ import { ToolSchemaProjection } from "./utils/tool-schema";
14
+ import { ToolStream } from "./utils/tool-stream";
15
+ const ADAPTER = "bedrock-converse";
16
+ // =============================================================================
17
+ // Request Body Schema
18
+ // =============================================================================
19
+ const BedrockTextBlock = Schema.Struct({
20
+ text: Schema.String,
21
+ });
22
+ const BedrockToolUseBlock = Schema.Struct({
23
+ toolUse: Schema.Struct({
24
+ toolUseId: Schema.String,
25
+ name: Schema.String,
26
+ input: Schema.Unknown,
27
+ }),
28
+ });
29
+ const BedrockToolResultContentItem = Schema.Union([
30
+ Schema.Struct({ text: Schema.String }),
31
+ Schema.Struct({ json: Schema.Unknown }),
32
+ BedrockMedia.ImageBlock,
33
+ ]);
34
+ const BedrockToolResultBlock = Schema.Struct({
35
+ toolResult: Schema.Struct({
36
+ toolUseId: Schema.String,
37
+ content: Schema.Array(BedrockToolResultContentItem),
38
+ status: Schema.optional(Schema.Literals(["success", "error"])),
39
+ }),
40
+ });
41
+ const BedrockReasoningBlock = Schema.Struct({
42
+ reasoningContent: Schema.Struct({
43
+ reasoningText: Schema.optional(Schema.Struct({
44
+ text: Schema.String,
45
+ signature: Schema.optional(Schema.String),
46
+ })),
47
+ }),
48
+ });
49
+ const BedrockUserBlock = Schema.Union([
50
+ BedrockTextBlock,
51
+ BedrockMedia.ImageBlock,
52
+ BedrockMedia.DocumentBlock,
53
+ BedrockToolResultBlock,
54
+ BedrockCache.CachePointBlock,
55
+ ]);
56
+ const BedrockAssistantBlock = Schema.Union([
57
+ BedrockTextBlock,
58
+ BedrockReasoningBlock,
59
+ BedrockToolUseBlock,
60
+ BedrockCache.CachePointBlock,
61
+ ]);
62
+ const BedrockMessage = Schema.Union([
63
+ Schema.Struct({ role: Schema.Literal("user"), content: Schema.Array(BedrockUserBlock) }),
64
+ Schema.Struct({ role: Schema.Literal("assistant"), content: Schema.Array(BedrockAssistantBlock) }),
65
+ ]).pipe(Schema.toTaggedUnion("role"));
66
+ const BedrockSystemBlock = Schema.Union([BedrockTextBlock, BedrockCache.CachePointBlock]);
67
+ const BedrockToolSpec = Schema.Struct({
68
+ toolSpec: Schema.Struct({
69
+ name: Schema.String,
70
+ description: Schema.String,
71
+ inputSchema: Schema.Struct({
72
+ json: JsonObject,
73
+ }),
74
+ }),
75
+ });
76
+ const BedrockTool = Schema.Union([BedrockToolSpec, BedrockCache.CachePointBlock]);
77
+ const BedrockToolChoice = Schema.Union([
78
+ Schema.Struct({ auto: Schema.Struct({}) }),
79
+ Schema.Struct({ any: Schema.Struct({}) }),
80
+ Schema.Struct({ tool: Schema.Struct({ name: Schema.String }) }),
81
+ ]);
82
+ const BedrockBodyFields = {
83
+ modelId: Schema.String,
84
+ messages: Schema.Array(BedrockMessage),
85
+ system: optionalArray(BedrockSystemBlock),
86
+ inferenceConfig: Schema.optional(Schema.Struct({
87
+ maxTokens: Schema.optional(Schema.Number),
88
+ temperature: Schema.optional(Schema.Number),
89
+ topP: Schema.optional(Schema.Number),
90
+ stopSequences: optionalArray(Schema.String),
91
+ })),
92
+ toolConfig: Schema.optional(Schema.Struct({
93
+ tools: Schema.Array(BedrockTool),
94
+ toolChoice: Schema.optional(BedrockToolChoice),
95
+ })),
96
+ additionalModelRequestFields: Schema.optional(JsonObject),
97
+ };
98
+ const BedrockConverseBody = Schema.Struct(BedrockBodyFields);
99
+ const BedrockUsageSchema = Schema.Struct({
100
+ inputTokens: Schema.optional(Schema.Number),
101
+ outputTokens: Schema.optional(Schema.Number),
102
+ totalTokens: Schema.optional(Schema.Number),
103
+ cacheReadInputTokens: Schema.optional(Schema.Number),
104
+ cacheWriteInputTokens: Schema.optional(Schema.Number),
105
+ });
106
+ // Streaming event shape — the AWS event stream wraps each JSON payload by its
107
+ // `:event-type` header (e.g. `messageStart`, `contentBlockDelta`). We
108
+ // reconstruct that wrapping in `decodeFrames` below so the event schema can
109
+ // stay a plain discriminated record.
110
+ const BedrockEvent = Schema.Struct({
111
+ messageStart: Schema.optional(Schema.Struct({ role: Schema.String })),
112
+ contentBlockStart: Schema.optional(Schema.Struct({
113
+ contentBlockIndex: Schema.Number,
114
+ start: Schema.optional(Schema.Struct({
115
+ toolUse: Schema.optional(Schema.Struct({ toolUseId: Schema.String, name: Schema.String })),
116
+ })),
117
+ })),
118
+ contentBlockDelta: Schema.optional(Schema.Struct({
119
+ contentBlockIndex: Schema.Number,
120
+ delta: Schema.optional(Schema.Struct({
121
+ text: Schema.optional(Schema.String),
122
+ toolUse: Schema.optional(Schema.Struct({ input: Schema.String })),
123
+ reasoningContent: Schema.optional(Schema.Struct({
124
+ text: Schema.optional(Schema.String),
125
+ signature: Schema.optional(Schema.String),
126
+ })),
127
+ })),
128
+ })),
129
+ contentBlockStop: Schema.optional(Schema.Struct({ contentBlockIndex: Schema.Number })),
130
+ messageStop: Schema.optional(Schema.Struct({
131
+ stopReason: Schema.String,
132
+ additionalModelResponseFields: Schema.optional(Schema.Unknown),
133
+ })),
134
+ metadata: Schema.optional(Schema.Struct({
135
+ usage: Schema.optional(BedrockUsageSchema),
136
+ metrics: Schema.optional(Schema.Unknown),
137
+ })),
138
+ internalServerException: Schema.optional(Schema.Struct({ message: Schema.String })),
139
+ modelStreamErrorException: Schema.optional(Schema.Struct({ message: Schema.String })),
140
+ validationException: Schema.optional(Schema.Struct({ message: Schema.String })),
141
+ throttlingException: Schema.optional(Schema.Struct({ message: Schema.String })),
142
+ serviceUnavailableException: Schema.optional(Schema.Struct({ message: Schema.String })),
143
+ });
144
+ // =============================================================================
145
+ // Request Lowering
146
+ // =============================================================================
147
+ const lowerToolSpec = (tool, inputSchema) => ({
148
+ toolSpec: {
149
+ name: tool.name,
150
+ description: tool.description,
151
+ inputSchema: { json: inputSchema },
152
+ },
153
+ });
154
+ const lowerTools = (compatibility, breakpoints, tools) => {
155
+ const result = [];
156
+ for (const tool of tools) {
157
+ result.push(lowerToolSpec(tool, ToolSchemaProjection.modelCompatibility(tool.inputSchema, compatibility)));
158
+ const cachePoint = BedrockCache.block(breakpoints, tool.cache);
159
+ if (cachePoint)
160
+ result.push(cachePoint);
161
+ }
162
+ return result;
163
+ };
164
+ const textWithCache = (breakpoints, text, cache) => {
165
+ const cachePoint = BedrockCache.block(breakpoints, cache);
166
+ return cachePoint ? [{ text }, cachePoint] : [{ text }];
167
+ };
168
+ const lowerToolChoice = (toolChoice) => ProviderShared.matchToolChoice("Bedrock Converse", toolChoice, {
169
+ auto: () => ({ auto: {} }),
170
+ none: () => undefined,
171
+ required: () => ({ any: {} }),
172
+ tool: (name) => ({ tool: { name } }),
173
+ });
174
+ const bedrockMetadata = (metadata) => ({ bedrock: metadata });
175
+ const reasoningSignature = (part) => {
176
+ const bedrock = part.providerMetadata?.bedrock;
177
+ return (part.encrypted ??
178
+ (ProviderShared.isRecord(bedrock) && typeof bedrock.signature === "string" ? bedrock.signature : undefined));
179
+ };
180
+ const lowerToolCall = (part) => ({
181
+ toolUse: {
182
+ toolUseId: part.id,
183
+ name: part.name,
184
+ input: part.input,
185
+ },
186
+ });
187
+ const lowerToolResultContent = Effect.fn("BedrockConverse.lowerToolResultContent")(function* (part) {
188
+ if (part.result.type === "text" || part.result.type === "error")
189
+ return [{ text: ProviderShared.toolResultText(part) }];
190
+ if (part.result.type === "json")
191
+ return [{ json: part.result.value }];
192
+ const content = [];
193
+ for (const item of part.result.value) {
194
+ if (item.type === "text") {
195
+ content.push({ text: item.text });
196
+ continue;
197
+ }
198
+ const media = yield* BedrockMedia.lower({
199
+ type: "media",
200
+ mediaType: item.mime,
201
+ data: item.uri,
202
+ filename: item.name,
203
+ });
204
+ if (!("image" in media))
205
+ return yield* ProviderShared.invalidRequest("Bedrock Converse only supports image media in tool results");
206
+ content.push(media);
207
+ }
208
+ return content;
209
+ });
210
+ const lowerToolResult = Effect.fn("BedrockConverse.lowerToolResult")(function* (part) {
211
+ return {
212
+ toolResult: {
213
+ toolUseId: part.id,
214
+ content: yield* lowerToolResultContent(part),
215
+ status: part.result.type === "error" ? "error" : "success",
216
+ },
217
+ };
218
+ });
219
+ const lowerMessages = Effect.fn("BedrockConverse.lowerMessages")(function* (request, breakpoints) {
220
+ const messages = [];
221
+ for (const message of request.messages) {
222
+ if (message.role === "system") {
223
+ const part = yield* ProviderShared.wrappedSystemUpdate("Bedrock Converse", message);
224
+ const content = textWithCache(breakpoints, part.text, part.cache);
225
+ const previous = messages.at(-1);
226
+ if (previous?.role === "user")
227
+ messages[messages.length - 1] = { role: "user", content: [...previous.content, ...content] };
228
+ else
229
+ messages.push({ role: "user", content });
230
+ continue;
231
+ }
232
+ if (message.role === "user") {
233
+ const content = [];
234
+ for (const part of message.content) {
235
+ if (!ProviderShared.supportsContent(part, ["text", "media"]))
236
+ return yield* ProviderShared.unsupportedContent("Bedrock Converse", "user", ["text", "media"]);
237
+ if (part.type === "text") {
238
+ content.push(...textWithCache(breakpoints, part.text, part.cache));
239
+ continue;
240
+ }
241
+ if (part.type === "media") {
242
+ content.push(yield* BedrockMedia.lower(part));
243
+ continue;
244
+ }
245
+ }
246
+ messages.push({ role: "user", content });
247
+ continue;
248
+ }
249
+ if (message.role === "assistant") {
250
+ const content = [];
251
+ for (const part of message.content) {
252
+ if (!ProviderShared.supportsContent(part, ["text", "reasoning", "tool-call"]))
253
+ return yield* ProviderShared.unsupportedContent("Bedrock Converse", "assistant", [
254
+ "text",
255
+ "reasoning",
256
+ "tool-call",
257
+ ]);
258
+ if (part.type === "text") {
259
+ content.push(...textWithCache(breakpoints, part.text, part.cache));
260
+ continue;
261
+ }
262
+ if (part.type === "reasoning") {
263
+ content.push({
264
+ reasoningContent: {
265
+ reasoningText: { text: part.text, signature: reasoningSignature(part) },
266
+ },
267
+ });
268
+ continue;
269
+ }
270
+ if (part.type === "tool-call") {
271
+ content.push(lowerToolCall(part));
272
+ continue;
273
+ }
274
+ }
275
+ messages.push({ role: "assistant", content });
276
+ continue;
277
+ }
278
+ const content = [];
279
+ for (const part of message.content) {
280
+ if (!ProviderShared.supportsContent(part, ["tool-result"]))
281
+ return yield* ProviderShared.unsupportedContent("Bedrock Converse", "tool", ["tool-result"]);
282
+ content.push(yield* lowerToolResult(part));
283
+ const cachePoint = BedrockCache.block(breakpoints, part.cache);
284
+ if (cachePoint)
285
+ content.push(cachePoint);
286
+ }
287
+ messages.push({ role: "user", content });
288
+ }
289
+ return messages;
290
+ });
291
+ // System prompts share the cache-point convention: emit the text block, then
292
+ // optionally a positional `cachePoint` marker.
293
+ const lowerSystem = (breakpoints, system) => system.flatMap((part) => textWithCache(breakpoints, part.text, part.cache));
294
+ const fromRequest = Effect.fn("BedrockConverse.fromRequest")(function* (request) {
295
+ const toolChoice = request.toolChoice ? yield* lowerToolChoice(request.toolChoice) : undefined;
296
+ const generation = request.generation;
297
+ // Bedrock-Claude shares Anthropic's 4-breakpoint cap. Spend the budget in
298
+ // tools → system → messages order to favour the highest-impact prefixes.
299
+ const breakpoints = BedrockCache.breakpoints();
300
+ const toolConfig = request.tools.length > 0 && request.toolChoice?.type !== "none"
301
+ ? { tools: lowerTools(request.model.compatibility?.toolSchema, breakpoints, request.tools), toolChoice }
302
+ : undefined;
303
+ const system = request.system.length === 0 ? undefined : lowerSystem(breakpoints, request.system);
304
+ const messages = yield* lowerMessages(request, breakpoints);
305
+ if (breakpoints.dropped > 0) {
306
+ yield* Effect.logWarning(`Bedrock Converse: dropped ${breakpoints.dropped} cache breakpoint(s); the API allows at most ${BedrockCache.BEDROCK_BREAKPOINT_CAP} per request.`);
307
+ }
308
+ return {
309
+ modelId: request.model.id,
310
+ messages,
311
+ system,
312
+ inferenceConfig: generation?.maxTokens === undefined &&
313
+ generation?.temperature === undefined &&
314
+ generation?.topP === undefined &&
315
+ (generation?.stop === undefined || generation.stop.length === 0)
316
+ ? undefined
317
+ : {
318
+ maxTokens: generation?.maxTokens,
319
+ temperature: generation?.temperature,
320
+ topP: generation?.topP,
321
+ stopSequences: generation?.stop,
322
+ },
323
+ toolConfig,
324
+ // Converse's base inferenceConfig has no topK; Anthropic/Nova accept it
325
+ // as a model-specific field, so it goes through additionalModelRequestFields.
326
+ additionalModelRequestFields: generation?.topK === undefined ? undefined : { top_k: generation.topK },
327
+ };
328
+ });
329
+ // =============================================================================
330
+ // Stream Parsing
331
+ // =============================================================================
332
+ const mapFinishReason = (reason) => {
333
+ if (reason === "end_turn" || reason === "stop_sequence")
334
+ return "stop";
335
+ if (reason === "max_tokens")
336
+ return "length";
337
+ if (reason === "tool_use")
338
+ return "tool-calls";
339
+ if (reason === "content_filtered" || reason === "guardrail_intervened")
340
+ return "content-filter";
341
+ return "unknown";
342
+ };
343
+ // AWS Bedrock Converse reports `inputTokens` (inclusive total) with
344
+ // `cacheReadInputTokens` and `cacheWriteInputTokens` as subsets. Pass
345
+ // the total through and derive the non-cached breakdown. Bedrock does
346
+ // not break reasoning out of `outputTokens` for any current model.
347
+ const mapUsage = (usage) => {
348
+ if (!usage)
349
+ return undefined;
350
+ const cacheTotal = (usage.cacheReadInputTokens ?? 0) + (usage.cacheWriteInputTokens ?? 0);
351
+ const nonCached = ProviderShared.subtractTokens(usage.inputTokens, cacheTotal);
352
+ return new Usage({
353
+ inputTokens: usage.inputTokens,
354
+ outputTokens: usage.outputTokens,
355
+ nonCachedInputTokens: nonCached,
356
+ cacheReadInputTokens: usage.cacheReadInputTokens,
357
+ cacheWriteInputTokens: usage.cacheWriteInputTokens,
358
+ totalTokens: ProviderShared.totalTokens(usage.inputTokens, usage.outputTokens, usage.totalTokens),
359
+ providerMetadata: { bedrock: usage },
360
+ });
361
+ };
362
+ const step = (state, event) => Effect.gen(function* () {
363
+ if (event.contentBlockStart?.start?.toolUse) {
364
+ const index = event.contentBlockStart.contentBlockIndex;
365
+ const events = [];
366
+ const lifecycle = Lifecycle.stepStart(state.lifecycle, events);
367
+ return [
368
+ {
369
+ ...state,
370
+ lifecycle,
371
+ tools: ToolStream.start(state.tools, index, {
372
+ id: event.contentBlockStart.start.toolUse.toolUseId,
373
+ name: event.contentBlockStart.start.toolUse.name,
374
+ }),
375
+ },
376
+ [
377
+ ...events,
378
+ LLMEvent.toolInputStart({
379
+ id: event.contentBlockStart.start.toolUse.toolUseId,
380
+ name: event.contentBlockStart.start.toolUse.name,
381
+ }),
382
+ ],
383
+ ];
384
+ }
385
+ if (event.contentBlockDelta?.delta?.text) {
386
+ const events = [];
387
+ return [
388
+ {
389
+ ...state,
390
+ lifecycle: Lifecycle.textDelta(state.lifecycle, events, `text-${event.contentBlockDelta.contentBlockIndex}`, event.contentBlockDelta.delta.text),
391
+ },
392
+ events,
393
+ ];
394
+ }
395
+ if (event.contentBlockDelta?.delta?.reasoningContent) {
396
+ const index = event.contentBlockDelta.contentBlockIndex;
397
+ const reasoning = event.contentBlockDelta.delta.reasoningContent;
398
+ const events = [];
399
+ return [
400
+ {
401
+ ...state,
402
+ lifecycle: reasoning.text
403
+ ? Lifecycle.reasoningDelta(state.lifecycle, events, `reasoning-${index}`, reasoning.text)
404
+ : state.lifecycle,
405
+ reasoningSignatures: reasoning.signature
406
+ ? { ...state.reasoningSignatures, [index]: reasoning.signature }
407
+ : state.reasoningSignatures,
408
+ },
409
+ events,
410
+ ];
411
+ }
412
+ if (event.contentBlockDelta?.delta?.toolUse) {
413
+ const index = event.contentBlockDelta.contentBlockIndex;
414
+ const result = ToolStream.appendExisting(ADAPTER, state.tools, index, event.contentBlockDelta.delta.toolUse.input, "Bedrock Converse tool delta is missing its tool call");
415
+ if (ToolStream.isError(result))
416
+ return yield* result;
417
+ const events = [];
418
+ const lifecycle = result.events.length ? Lifecycle.stepStart(state.lifecycle, events) : state.lifecycle;
419
+ events.push(...result.events);
420
+ return [{ ...state, lifecycle, tools: result.tools }, events];
421
+ }
422
+ if (event.contentBlockStop) {
423
+ const index = event.contentBlockStop.contentBlockIndex;
424
+ const result = yield* ToolStream.finish(ADAPTER, state.tools, index);
425
+ const events = [];
426
+ const resultEvents = result.events ?? [];
427
+ const lifecycle = resultEvents.length
428
+ ? Lifecycle.stepStart(state.lifecycle, events)
429
+ : Lifecycle.reasoningEnd(Lifecycle.textEnd(state.lifecycle, events, `text-${index}`), events, `reasoning-${index}`, state.reasoningSignatures[index]
430
+ ? bedrockMetadata({ signature: state.reasoningSignatures[index] })
431
+ : undefined);
432
+ events.push(...resultEvents);
433
+ return [
434
+ {
435
+ ...state,
436
+ hasToolCalls: resultEvents.some(LLMEvent.is.toolCall) ? true : state.hasToolCalls,
437
+ lifecycle,
438
+ tools: result.tools,
439
+ reasoningSignatures: Object.fromEntries(Object.entries(state.reasoningSignatures).filter(([key]) => key !== String(index))),
440
+ },
441
+ events,
442
+ ];
443
+ }
444
+ if (event.messageStop) {
445
+ return [
446
+ {
447
+ ...state,
448
+ pendingFinish: { reason: mapFinishReason(event.messageStop.stopReason), usage: state.pendingFinish?.usage },
449
+ },
450
+ [],
451
+ ];
452
+ }
453
+ if (event.metadata) {
454
+ const usage = mapUsage(event.metadata.usage);
455
+ return [{ ...state, pendingFinish: { reason: state.pendingFinish?.reason ?? "stop", usage } }, []];
456
+ }
457
+ const exception = [
458
+ ["internalServerException", event.internalServerException],
459
+ ["modelStreamErrorException", event.modelStreamErrorException],
460
+ ["serviceUnavailableException", event.serviceUnavailableException],
461
+ ["throttlingException", event.throttlingException],
462
+ ["validationException", event.validationException],
463
+ ].find((entry) => entry[1] !== undefined);
464
+ if (exception) {
465
+ return yield* new LLMError({
466
+ module: ADAPTER,
467
+ method: "stream",
468
+ reason: classifyProviderFailure({
469
+ message: exception[1]?.message ?? "Bedrock Converse stream error",
470
+ code: exception[0],
471
+ }),
472
+ });
473
+ }
474
+ return [state, []];
475
+ });
476
+ const framing = BedrockEventStream.framing(ADAPTER);
477
+ const onHalt = (state) => state.pendingFinish
478
+ ? (() => {
479
+ const events = [];
480
+ Lifecycle.finish(state.lifecycle, events, {
481
+ reason: state.pendingFinish.reason === "stop" && state.hasToolCalls ? "tool-calls" : state.pendingFinish.reason,
482
+ usage: state.pendingFinish.usage,
483
+ });
484
+ return events;
485
+ })()
486
+ : [];
487
+ // =============================================================================
488
+ // Protocol And Bedrock Route
489
+ // =============================================================================
490
+ /**
491
+ * The Bedrock Converse protocol — request body construction, body schema, and
492
+ * the streaming-event state machine.
493
+ */
494
+ export const protocol = Protocol.make({
495
+ id: ADAPTER,
496
+ body: {
497
+ schema: BedrockConverseBody,
498
+ from: fromRequest,
499
+ },
500
+ stream: {
501
+ event: BedrockEvent,
502
+ initial: () => ({
503
+ tools: ToolStream.empty(),
504
+ pendingFinish: undefined,
505
+ hasToolCalls: false,
506
+ lifecycle: Lifecycle.initial(),
507
+ reasoningSignatures: {},
508
+ }),
509
+ step,
510
+ onHalt,
511
+ },
512
+ });
513
+ export const route = Route.make({
514
+ id: ADAPTER,
515
+ provider: "bedrock",
516
+ providerMetadataKey: "bedrock",
517
+ protocol,
518
+ // Bedrock's URL embeds the region in the route endpoint host and the
519
+ // validated modelId in the path. We read the validated body so the URL
520
+ // matches the body that gets signed.
521
+ endpoint: Endpoint.path(({ body }) => `/model/${encodeURIComponent(body.modelId)}/converse-stream`),
522
+ auth: BedrockAuth.auth,
523
+ framing,
524
+ });
525
+ export const sigV4Auth = BedrockAuth.sigV4;
526
+ export * as BedrockConverse from "./bedrock-converse";
@@ -0,0 +1,9 @@
1
+ import { Framing } from "../route/framing";
2
+ /**
3
+ * AWS event-stream framing for Bedrock Converse. Each frame is decoded by
4
+ * `@smithy/eventstream-codec` (length + header + payload + CRC) and rewrapped
5
+ * under its `:event-type` header so the chunk schema can match the JSON
6
+ * payload directly.
7
+ */
8
+ export declare const framing: (route: string) => Framing.Definition<object>;
9
+ export * as BedrockEventStream from "./bedrock-event-stream";
@@ -0,0 +1,64 @@
1
+ import { EventStreamCodec } from "@smithy/eventstream-codec";
2
+ import { fromUtf8, toUtf8 } from "@smithy/util-utf8";
3
+ import { Effect, Stream } from "effect";
4
+ import { Framing } from "../route/framing";
5
+ import { ProviderShared } from "./shared";
6
+ // Bedrock streams responses using the AWS event stream binary protocol — each
7
+ // frame is `[length:4][headers-length:4][prelude-crc:4][headers][payload][crc:4]`.
8
+ // We use `@smithy/eventstream-codec` to validate framing and CRCs, then
9
+ // reconstruct the JSON wrapping by `:event-type` so the chunk schema can match.
10
+ const eventCodec = new EventStreamCodec(toUtf8, fromUtf8);
11
+ const utf8 = new TextDecoder();
12
+ const initialFrameBuffer = { buffer: new Uint8Array(0), offset: 0 };
13
+ const appendChunk = (state, chunk) => {
14
+ const remaining = state.buffer.length - state.offset;
15
+ // Compact: drop the consumed prefix and append the new chunk in one alloc.
16
+ // This bounds buffer growth to at most one network chunk past the live
17
+ // window, regardless of stream length.
18
+ const next = new Uint8Array(remaining + chunk.length);
19
+ next.set(state.buffer.subarray(state.offset), 0);
20
+ next.set(chunk, remaining);
21
+ return { buffer: next, offset: 0 };
22
+ };
23
+ const consumeFrames = (route) => (state, chunk) => Effect.gen(function* () {
24
+ let cursor = appendChunk(state, chunk);
25
+ const out = [];
26
+ while (cursor.buffer.length - cursor.offset >= 4) {
27
+ const view = cursor.buffer.subarray(cursor.offset);
28
+ const totalLength = new DataView(view.buffer, view.byteOffset, view.byteLength).getUint32(0, false);
29
+ if (view.length < totalLength)
30
+ break;
31
+ const decoded = yield* Effect.try({
32
+ try: () => eventCodec.decode(view.subarray(0, totalLength)),
33
+ catch: (error) => ProviderShared.eventError(route, `Failed to decode Bedrock Converse event-stream frame: ${error instanceof Error ? error.message : String(error)}`),
34
+ });
35
+ cursor = { buffer: cursor.buffer, offset: cursor.offset + totalLength };
36
+ if (decoded.headers[":message-type"]?.value !== "event")
37
+ continue;
38
+ const eventType = decoded.headers[":event-type"]?.value;
39
+ if (typeof eventType !== "string")
40
+ continue;
41
+ const payload = utf8.decode(decoded.body);
42
+ if (!payload)
43
+ continue;
44
+ // The AWS event stream pads short payloads with a `p` field. Drop it
45
+ // before handing the object to the chunk schema. JSON decode goes
46
+ // through the shared Schema-driven codec to satisfy the package rule
47
+ // against ad-hoc `JSON.parse` calls.
48
+ const parsed = (yield* ProviderShared.parseJson(route, payload, "Failed to parse Bedrock Converse event-stream payload"));
49
+ delete parsed.p;
50
+ out.push({ [eventType]: parsed });
51
+ }
52
+ return [cursor, out];
53
+ });
54
+ /**
55
+ * AWS event-stream framing for Bedrock Converse. Each frame is decoded by
56
+ * `@smithy/eventstream-codec` (length + header + payload + CRC) and rewrapped
57
+ * under its `:event-type` header so the chunk schema can match the JSON
58
+ * payload directly.
59
+ */
60
+ export const framing = (route) => ({
61
+ id: "aws-event-stream",
62
+ frame: (bytes) => bytes.pipe(Stream.mapAccumEffect(() => initialFrameBuffer, consumeFrames(route))),
63
+ });
64
+ export * as BedrockEventStream from "./bedrock-event-stream";