Add dupe checks to store_text_reply() & store_text_labels() in PromptRepository (#1018)

* add dupe checks to store_text_reply() & store_text_labels

* remove test export file

* add user_id to protocol.ConversationMessage

* add show_on_leaderboard ot protocol.FrontEndUser
This commit is contained in:
Andreas Köpf
2023-01-30 20:53:59 +01:00
committed by GitHub
parent b8a62e5f4f
commit a5bc9bf492
8 changed files with 70 additions and 22 deletions
@@ -40,6 +40,7 @@ class OasstErrorCode(IntEnum):
TASK_MESSAGE_TOO_LONG = 1008
TASK_MESSAGE_DUPLICATED = 1009
TASK_MESSAGE_TEXT_EMPTY = 1010
TASK_MESSAGE_DUPLICATE_REPLY = 1011
# 2000-3000: prompt_repository
INVALID_FRONTEND_MESSAGE_ID = 2000
@@ -59,6 +60,8 @@ class OasstErrorCode(IntEnum):
TEXT_LABELS_WRONG_MESSAGE_ID = 2050
TEXT_LABELS_INVALID_LABEL = 2051
TEXT_LABELS_MANDATORY_LABEL_MISSING = 2052
TEXT_LABELS_NO_SELF_LABELING = 2053
TEXT_LABELS_DUPLICATE_TASK_REPLY = 2053
TASK_NOT_FOUND = 2100
TASK_EXPIRED = 2101
@@ -35,6 +35,7 @@ class FrontEndUser(User):
deleted: bool
notes: str
created_date: Optional[datetime] = None
show_on_leaderboard: bool
class PageResult(BaseModel):
@@ -53,6 +54,7 @@ class ConversationMessage(BaseModel):
"""Represents a message in a conversation between the user and the assistant."""
id: Optional[UUID] = None
user_id: Optional[UUID]
frontend_message_id: Optional[str] = None
text: str
lang: Optional[str] # BCP 47
@@ -80,8 +82,10 @@ class Conversation(BaseModel):
class Message(ConversationMessage):
parent_id: Optional[UUID] = None
created_date: Optional[datetime] = None
parent_id: Optional[UUID]
created_date: Optional[datetime]
review_result: Optional[bool]
review_count: Optional[int]
class MessagePage(PageResult):