This commit is contained in:
Evan 2026-08-14 12:26:45 +00:00 committed by GitHub
commit 96a06a73c5
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -812,10 +812,11 @@ class AdvancedChatAppGenerateTaskPipeline(GraphRuntimeStateSupport):
reason=QueueMessageReplaceEvent.MessageReplaceReason.OUTPUT_MODERATION,
)
# Save message unless it has already been persisted on pause.
if not self._message_saved_on_pause:
with self._database_session() as session:
self._save_message(session=session, graph_runtime_state=resolved_state)
# Always save the final message state, even if it was already
# persisted on pause. The previous snapshot (PAUSED status, empty
# answer) must be replaced with the completed answer.
with self._database_session() as session:
self._save_message(session=session, graph_runtime_state=resolved_state)
yield self._message_end_to_stream_response()