ars/iiEsaywebUIapp
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix(streaming): prevent late events from corrupting message stream

Browse Source
This commit is contained in:
cogwheel
2026-01-13 21:33:00 +05:30
parent 0d5d2ef0a0
commit 65cd902d5c
1 changed files with 43 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

54
lib/core/services/streaming_helper.dart
View File

@@ -1214,17 +1214,39 @@ ActiveSocketStream attachUnifiedChunkedStreaming({
} else if ((type == 'chat:message:delta' || type == 'message') &&
payload != null) {
// Incremental message content over socket
final content = payload['content']?.toString() ?? '';
if (content.isNotEmpty) {
appendToLastMessage(content);
updateImagesFromCurrentContent();
// Validate message ID to prevent late events from previous turns
// from corrupting the current assistant message
if (messageId != null &&
messageId.isNotEmpty &&
messageId != assistantMessageId) {
DebugLogger.log(
'Ignoring delta for wrong message: $messageId (expected $assistantMessageId)',
scope: 'streaming/helper',
);
} else {
final content = payload['content']?.toString() ?? '';
if (content.isNotEmpty) {
appendToLastMessage(content);
updateImagesFromCurrentContent();
}
}
} else if ((type == 'chat:message' || type == 'replace') &&
payload != null) {
// Full message replacement over socket
final content = payload['content']?.toString() ?? '';
if (content.isNotEmpty) {
replaceLastMessageContent(content);
// Validate message ID to prevent late events from previous turns
// from corrupting the current assistant message
if (messageId != null &&
messageId.isNotEmpty &&
messageId != assistantMessageId) {
DebugLogger.log(
'Ignoring replace for wrong message: $messageId (expected $assistantMessageId)',
scope: 'streaming/helper',
);
} else {
final content = payload['content']?.toString() ?? '';
if (content.isNotEmpty) {
replaceLastMessageContent(content);
}
}
} else if ((type == 'chat:message:files') && payload != null) {
// Alias for files event used by web client
@@ -1347,10 +1369,20 @@ ActiveSocketStream attachUnifiedChunkedStreaming({
}
}
} else if (type == 'event:message:delta' && payload != null) {
final content = payload['content']?.toString() ?? '';
if (content.isNotEmpty) {
appendToLastMessage(content);
updateImagesFromCurrentContent();
// Validate message ID to prevent late events from previous turns
if (messageId != null &&
messageId.isNotEmpty &&
messageId != assistantMessageId) {
DebugLogger.log(
'Ignoring event delta for wrong message: $messageId (expected $assistantMessageId)',
scope: 'streaming/helper',
);
} else {
final content = payload['content']?.toString() ?? '';
if (content.isNotEmpty) {
appendToLastMessage(content);
updateImagesFromCurrentContent();
}
}
} else {
// Log unknown event types to catch any follow-up events we might be missing