mirror of
https://github.com/wassname/Open-Assistant.git
synced 2026-06-29 16:30:24 +08:00
Andreas Köpf
14fa08e2e7
* add query_incomplete_rankings() * Add SQL queries for TreeManager task selection * first working version of TreeManager.next_task() * remove old generate_task(), add mandatory_labels to text_labels task * Add ConversationMessage list to Ranking tasks * add more sophisticated sql queries to find extendible trees * add TreeManager.query_extendible_parents() * fix task validation, seed data insertion (reviewed) * provide user for task selection in text-frontend * enter 'growing' state * enter 'aborted_low_grade' state * enter 'ranking' state * check tree 'growing' state upon relpy insertion * exclude user from labeling their own messages (added DEBUG_ALLOW_SELF_LABELING setting) * add DEBUG_ALLOW_SELF_LABELING to docker-compose.yaml * fix ranking submission * add query_tree_ranking_results() * add ranked_message_ids to RankingReactionPayload * fix reply_messages instead of prompt_messages * incorment 'ranking_count' of ranked replies * added logic to check_condition_for_scoring_state * changes to msg_tree_state_machine * pre-commit changes * enter 'ready_for_scoring' state * re-add HF embedding call (lost during merge) * use prepare_conversation() helper for seed-data creation * Partially add user specified task selection Co-authored-by: Daniel Hug <danielpatrickhug@gmail.com>
56 lines
2.7 KiB
Python
56 lines
2.7 KiB
Python
from datetime import datetime
|
|
from http import HTTPStatus
|
|
from typing import Optional
|
|
from uuid import UUID, uuid4
|
|
|
|
import sqlalchemy as sa
|
|
import sqlalchemy.dialects.postgresql as pg
|
|
from oasst_backend.models.db_payload import MessagePayload
|
|
from oasst_shared.exceptions.oasst_api_error import OasstError, OasstErrorCode
|
|
from sqlalchemy import false
|
|
from sqlmodel import Field, Index, SQLModel
|
|
|
|
from .payload_column_type import PayloadContainer, payload_column_type
|
|
|
|
|
|
class Message(SQLModel, table=True):
|
|
__tablename__ = "message"
|
|
__table_args__ = (Index("ix_message_frontend_message_id", "api_client_id", "frontend_message_id", unique=True),)
|
|
|
|
id: Optional[UUID] = Field(
|
|
sa_column=sa.Column(
|
|
pg.UUID(as_uuid=True), primary_key=True, default=uuid4, server_default=sa.text("gen_random_uuid()")
|
|
),
|
|
)
|
|
parent_id: Optional[UUID] = Field(nullable=True)
|
|
message_tree_id: UUID = Field(nullable=False, index=True)
|
|
task_id: Optional[UUID] = Field(nullable=True, index=True)
|
|
user_id: Optional[UUID] = Field(nullable=True, foreign_key="user.id", index=True)
|
|
role: str = Field(nullable=False, max_length=128, regex="^prompter|assistant$")
|
|
api_client_id: UUID = Field(nullable=False, foreign_key="api_client.id")
|
|
frontend_message_id: str = Field(max_length=200, nullable=False)
|
|
created_date: Optional[datetime] = Field(
|
|
sa_column=sa.Column(sa.DateTime(), nullable=False, server_default=sa.func.current_timestamp())
|
|
)
|
|
payload_type: str = Field(nullable=False, max_length=200)
|
|
payload: Optional[PayloadContainer] = Field(
|
|
sa_column=sa.Column(payload_column_type(PayloadContainer), nullable=True)
|
|
)
|
|
lang: str = Field(nullable=False, max_length=200, default="en-US")
|
|
depth: int = Field(sa_column=sa.Column(sa.Integer, default=0, server_default=sa.text("0"), nullable=False))
|
|
children_count: int = Field(sa_column=sa.Column(sa.Integer, default=0, server_default=sa.text("0"), nullable=False))
|
|
deleted: bool = Field(sa_column=sa.Column(sa.Boolean, nullable=False, server_default=false()))
|
|
|
|
review_count: int = Field(sa_column=sa.Column(sa.Integer, default=0, server_default=sa.text("0"), nullable=False))
|
|
review_result: bool = Field(sa_column=sa.Column(sa.Boolean, default=False, server_default=false(), nullable=False))
|
|
ranking_count: int = Field(sa_column=sa.Column(sa.Integer, default=0, server_default=sa.text("0"), nullable=False))
|
|
|
|
def ensure_is_message(self) -> None:
|
|
if not self.payload or not isinstance(self.payload.payload, MessagePayload):
|
|
raise OasstError("Invalid message", OasstErrorCode.INVALID_MESSAGE, HTTPStatus.INTERNAL_SERVER_ERROR)
|
|
|
|
@property
|
|
def text(self) -> str:
|
|
self.ensure_is_message()
|
|
return self.payload.payload.text
|