Skip to content

Commit fff7272

Browse files
authored
Delete message evaluations on chat delete (#3424)
1 parent c25f2a8 commit fff7272

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

inference/server/oasst_inference_server/user_chat_repository.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -96,6 +96,10 @@ async def delete_chat(self, chat_id: str) -> models.DbChat:
9696
message_ids = [message.id for message in chat.messages]
9797
# delete reports associated with messages
9898
await self.session.exec(sqlmodel.delete(models.DbReport).where(models.DbReport.message_id.in_(message_ids)))
99+
# delete message evaluations associated with message
100+
await self.session.exec(
101+
sqlmodel.delete(models.DbMessageEval).where(models.DbMessageEval.selected_message_id.in_(message_ids))
102+
)
99103
# delete messages
100104
await self.session.exec(sqlmodel.delete(models.DbMessage).where(models.DbMessage.chat_id == chat_id))
101105
# delete chat

0 commit comments

Comments
 (0)