From 5d2e8a245ecd8c02d51ddf619ed60f7f16afe4b6 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Andreas=20K=C3=B6pf?= Date: Wed, 8 Feb 2023 00:52:05 +0100 Subject: [PATCH] fix flagged_message upsert logic --- backend/oasst_backend/prompt_repository.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/backend/oasst_backend/prompt_repository.py b/backend/oasst_backend/prompt_repository.py index 958eaeba..f2119e3b 100644 --- a/backend/oasst_backend/prompt_repository.py +++ b/backend/oasst_backend/prompt_repository.py @@ -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