Skip to content

Commit e4a4084

Browse files
authored
fix: caching error of images with Amazon Nova (#1077)
1 parent 349a47e commit e4a4084

File tree

2 files changed

+4
-8
lines changed

2 files changed

+4
-8
lines changed

packages/cdk/lambda/utils/models.ts

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -343,10 +343,9 @@ const createConverseCommandInput = (
343343
// Add the string of user role and assistant role other than the system role to the conversation
344344
messages = messages.filter((message) => message.role !== 'system');
345345
const conversation = messages.map((message) => {
346-
const contentBlocks: ContentBlock[] = [
347-
{ text: message.content } as ContentBlock.TextMember,
348-
];
346+
const contentBlocks: ContentBlock[] = [];
349347

348+
// Put images, videos, and documents before the task, instruction, and user query
350349
if (message.extraData) {
351350
message.extraData.forEach((extra) => {
352351
if (extra.type === 'image' && extra.source.type === 'base64') {
@@ -394,6 +393,7 @@ const createConverseCommandInput = (
394393
});
395394
}
396395

396+
contentBlocks.push({ text: message.content });
397397
return {
398398
role:
399399
message.role === 'user'

packages/cdk/lambda/utils/promptCache.ts

Lines changed: 1 addition & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -32,11 +32,7 @@ export const applyAutoCacheToMessages = (
3232
const isToolsSupported = cacheFields.includes('tools');
3333
const cachableIndices = messages
3434
.map((message, index) => ({ message, index }))
35-
.filter(
36-
({ message }) =>
37-
message.role === 'user' &&
38-
!message.content?.some((block) => block.document || block.video)
39-
)
35+
.filter(({ message }) => message.role === 'user')
4036
.filter(
4137
({ message }) =>
4238
isToolsSupported ||

0 commit comments

Comments
 (0)