getcompanion-ai/co-mono
1
0
Fork 0
mirror of https://github.com/getcompanion-ai/co-mono.git synced 2026-04-18 19:03:41 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix: sanitize bedrock tool call ids (#781)

Browse source
This commit is contained in:
Pablo Tovar 2026-01-16 17:51:48 +01:00 • committed by GitHub
parent c45ea27de5
commit ba8059a502
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 8 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

11
packages/ai/src/providers/amazon-bedrock.ts
View file

@ -317,6 +317,11 @@ function buildSystemPrompt(
return blocks; return blocks;
} }
function sanitizeToolCallId(id: string): string {
const sanitized = id.replace(/[^a-zA-Z0-9_-]/g, "_");
return sanitized.length > 64 ? sanitized.slice(0, 64) : sanitized;
}
function convertMessages(context: Context, model: Model<"bedrock-converse-stream">): Message[] { function convertMessages(context: Context, model: Model<"bedrock-converse-stream">): Message[] {
const result: Message[] = []; const result: Message[] = [];
const transformedMessages = transformMessages(context.messages, model); const transformedMessages = transformMessages(context.messages, model);
@ -359,7 +364,7 @@ function convertMessages(context: Context, model: Model<"bedrock-converse-stream
break; break;
case "toolCall": case "toolCall":
contentBlocks.push({ contentBlocks.push({
toolUse: { toolUseId: c.id, name: c.name, input: c.arguments }, toolUse: { toolUseId: sanitizeToolCallId(c.id), name: c.name, input: c.arguments },
}); });
break; break;
case "thinking": case "thinking":
@ -404,7 +409,7 @@ function convertMessages(context: Context, model: Model<"bedrock-converse-stream
// Add current tool result with all content blocks combined // Add current tool result with all content blocks combined
toolResults.push({ toolResults.push({
toolResult: { toolResult: {
toolUseId: m.toolCallId, toolUseId: sanitizeToolCallId(m.toolCallId),
content: m.content.map((c) => content: m.content.map((c) =>
c.type === "image" c.type === "image"
? { image: createImageBlock(c.mimeType, c.data) } ? { image: createImageBlock(c.mimeType, c.data) }
@ -420,7 +425,7 @@ function convertMessages(context: Context, model: Model<"bedrock-converse-stream
const nextMsg = transformedMessages[j] as ToolResultMessage; const nextMsg = transformedMessages[j] as ToolResultMessage;
toolResults.push({ toolResults.push({
toolResult: { toolResult: {
toolUseId: nextMsg.toolCallId, toolUseId: sanitizeToolCallId(nextMsg.toolCallId),
content: nextMsg.content.map((c) => content: nextMsg.content.map((c) =>
c.type === "image" c.type === "image"
? { image: createImageBlock(c.mimeType, c.data) } ? { image: createImageBlock(c.mimeType, c.data) }