mirror of
https://github.com/wassname/Open-Assistant.git
synced 2026-08-11 11:13:12 +08:00
Add user emoji augmentation for message queries (#937)
* add disposition to text labeling tasks
* add emoji stats to ConversationMessage
* add user emoji augmentation for message queries
* add auth_method,username to message queries (query emoji status)
* add auth_method+username for single message
* fix param name typo
* only join rows when message.emojis != JSON.NULL
* formatting
* make sure emojis and user_emojis default to {}, []
* remove init_user(), use fresh empty default collections
This commit is contained in:
@@ -57,6 +57,8 @@ class ConversationMessage(BaseModel):
|
||||
text: str
|
||||
lang: Optional[str] # BCP 47
|
||||
is_assistant: bool
|
||||
emojis: Optional[dict[str, int]] = None
|
||||
user_emojis: Optional[list[str]] = None
|
||||
|
||||
|
||||
class Conversation(BaseModel):
|
||||
@@ -80,7 +82,6 @@ class Conversation(BaseModel):
|
||||
class Message(ConversationMessage):
|
||||
parent_id: Optional[UUID] = None
|
||||
created_date: Optional[datetime] = None
|
||||
emojis: Optional[dict] = None
|
||||
|
||||
|
||||
class MessagePage(PageResult):
|
||||
@@ -223,27 +224,34 @@ class LabelTaskMode(str, enum.Enum):
|
||||
full = "full"
|
||||
|
||||
|
||||
class LabelInitialPromptTask(Task):
|
||||
"""A task to label an initial prompt."""
|
||||
class LabelTaskDisposition(str, enum.Enum):
|
||||
"""Reason why the task was issued."""
|
||||
|
||||
type: Literal["label_initial_prompt"] = "label_initial_prompt"
|
||||
quality = "quality"
|
||||
spam = "spam"
|
||||
|
||||
|
||||
class AbstractLabelTask(Task):
|
||||
message_id: UUID
|
||||
prompt: str
|
||||
valid_labels: list[str]
|
||||
mandatory_labels: Optional[list[str]]
|
||||
mode: Optional[LabelTaskMode]
|
||||
disposition: Optional[LabelTaskDisposition]
|
||||
|
||||
|
||||
class LabelConversationReplyTask(Task):
|
||||
class LabelInitialPromptTask(AbstractLabelTask):
|
||||
"""A task to label an initial prompt."""
|
||||
|
||||
type: Literal["label_initial_prompt"] = "label_initial_prompt"
|
||||
prompt: str
|
||||
|
||||
|
||||
class LabelConversationReplyTask(AbstractLabelTask):
|
||||
"""A task to label a reply to a conversation."""
|
||||
|
||||
type: Literal["label_conversation_reply"] = "label_conversation_reply"
|
||||
conversation: Conversation # the conversation so far
|
||||
message_id: UUID
|
||||
reply: str
|
||||
valid_labels: list[str]
|
||||
mandatory_labels: Optional[list[str]]
|
||||
mode: Optional[LabelTaskMode]
|
||||
|
||||
|
||||
class LabelPrompterReplyTask(LabelConversationReplyTask):
|
||||
|
||||
Reference in New Issue
Block a user