fix: regenerate image gen
This commit is contained in:
@@ -934,6 +934,16 @@ class _ChatPageState extends ConsumerState<ChatPage> {
|
||||
}
|
||||
|
||||
try {
|
||||
// If assistant message has generated images and it's the last message,
|
||||
// use image-only regenerate flow instead of text SSE regeneration
|
||||
if (message.role == 'assistant' &&
|
||||
(message.files?.any((f) => f['type'] == 'image') == true) &&
|
||||
messageIndex == messages.length - 1) {
|
||||
final regenerateImages = ref.read(regenerateLastMessageProvider);
|
||||
await regenerateImages();
|
||||
return;
|
||||
}
|
||||
|
||||
// Remove the assistant message we want to regenerate
|
||||
ref.read(chatMessagesProvider.notifier).removeLastMessage();
|
||||
|
||||
|
||||
Reference in New Issue
Block a user