mirror of
https://github.com/wassname/Open-Assistant.git
synced 2026-07-09 00:20:03 +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>
43 lines
1.3 KiB
Python
43 lines
1.3 KiB
Python
from uuid import UUID
|
|
|
|
from oasst_backend.models import Message
|
|
from oasst_shared.schemas import protocol
|
|
|
|
|
|
def prepare_message(m: Message) -> protocol.Message:
|
|
return protocol.Message(
|
|
id=m.id,
|
|
parent_id=m.parent_id,
|
|
text=m.text,
|
|
is_assistant=(m.role == "assistant"),
|
|
created_date=m.created_date,
|
|
)
|
|
|
|
|
|
def prepare_message_list(messages: list[Message]) -> list[protocol.Message]:
|
|
return [prepare_message(m) for m in messages]
|
|
|
|
|
|
def prepare_conversation_message_list(messages: list[Message]) -> list[protocol.ConversationMessage]:
|
|
return [
|
|
protocol.ConversationMessage(
|
|
text=message.text,
|
|
is_assistant=(message.role == "assistant"),
|
|
message_id=message.id,
|
|
frontend_message_id=message.frontend_message_id,
|
|
)
|
|
for message in messages
|
|
]
|
|
|
|
|
|
def prepare_conversation(messages: list[Message]) -> protocol.Conversation:
|
|
return protocol.Conversation(messages=prepare_conversation_message_list(messages))
|
|
|
|
|
|
def prepare_tree(tree: list[Message], tree_id: UUID) -> protocol.MessageTree:
|
|
tree_messages = []
|
|
for message in tree:
|
|
tree_messages.append(prepare_message(message))
|
|
|
|
return protocol.MessageTree(id=tree_id, messages=tree_messages)
|