wassname/Open-Assistant
1
0
Fork 0
mirror of https://github.com/wassname/Open-Assistant.git synced 2026-07-11 00:30:06 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix flagged_message upsert logic

Browse Source
This commit is contained in:
Andreas Köpf
2023-02-08 00:52:05 +01:00
parent 66b7ed2b9b
commit 5d2e8a245e
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
backend/oasst_backend/prompt_repository.py
+4 -4
View File
@@ -1096,11 +1096,11 @@ WHERE message.id = cc.id;
# Add to flagged_message table if the red flag emoji is applied
if emoji == protocol_schema.EmojiCode.red_flag:
flagged_message = FlaggedMessage(
message_id=message_id, processed=False, created_date=datetime.now().astimezone()
flagged_message = FlaggedMessage(message_id=message_id, processed=False, created_date=utcnow())
insert_stmt = pg.insert(FlaggedMessage).values(**flagged_message.dict())
upsert_stmt = insert_stmt.on_conflict_do_update(
constraint="flagged_message_pkey", set_=flagged_message.dict()
)
insert_stmt = pg.insert(FlaggedMessage).values(**flagged_message.__dict__)
upsert_stmt = insert_stmt.on_conflict_do_update(constraint="message_id", set_=flagged_message.__dict__)
self.db.execute(upsert_stmt)
# insert emoji record & increment count