From 63b0c3926fe6583de876198c940190986e1248aa Mon Sep 17 00:00:00 2001 From: Russell Ballestrini Date: Sun, 29 Oct 2023 20:52:07 -0400 Subject: [PATCH] dont error out if we delete message before streaming is done. modified: app.py --- app.py | 14 ++++++++------ 1 file changed, 8 insertions(+), 6 deletions(-) diff --git a/app.py b/app.py index 5d962db..439f673 100644 --- a/app.py +++ b/app.py @@ -213,9 +213,10 @@ def chat_claude(username, room, message): # Save the entire completion to the database with app.app_context(): new_message = db.session.query(Message).filter(Message.id == msg_id).one_or_none() - new_message.content = buffer - db.session.add(new_message) - db.session.commit() + if new_message: + new_message.content = buffer + db.session.add(new_message) + db.session.commit() socketio.emit("delete_processing_message", "", room=room) @@ -285,9 +286,10 @@ def chat_gpt(username, room, message): # Save the entire completion to the database with app.app_context(): new_message = db.session.query(Message).filter(Message.id == msg_id).one_or_none() - new_message.content = buffer - db.session.add(new_message) - db.session.commit() + if new_message: + new_message.content = buffer + db.session.add(new_message) + db.session.commit() socketio.emit("delete_processing_message", "", room=room)