engineering/opencompletion.com
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

dont error out if we delete message before streaming is done.

Browse source 
modified:   app.py
This commit is contained in:
Russell Ballestrini 2023-10-29 20:52:07 -04:00
parent 2a410a977b
commit 63b0c3926f
1 changed files with 8 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

14
app.py
View file

@ -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)