Skip to content

Commit 6a1c51c

Browse files
committed
Ignore adding retry to retry messages, just resend the existing chat request
1 parent 0870a80 commit 6a1c51c

File tree

1 file changed

+12
-0
lines changed

1 file changed

+12
-0
lines changed

llms/ui/modules/chat/index.mjs

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -423,6 +423,18 @@ export function useChatPrompt(ctx) {
423423
}
424424
}
425425

426+
if (text === 'retry') {
427+
let lastMessage = messages[messages.length - 1]
428+
if (lastMessage.role === 'user' && lastMessage.content[0].text === 'retry') {
429+
messages.pop()
430+
// Also remove assistant message before it
431+
lastMessage = messages[messages.length - 1]
432+
if (lastMessage.role === 'assistant') {
433+
messages.pop()
434+
}
435+
}
436+
}
437+
426438
const request = createRequest({ model })
427439

428440
// Add Thread History

0 commit comments

Comments
 (0)