mirror of
https://github.com/wassname/Open-Assistant.git
synced 2026-06-29 16:30:24 +08:00
export messages ordered by rank
This commit is contained in:
@@ -53,6 +53,7 @@ def build_export_tree(message_tree_id: str, messages: list[Message]) -> ExportMe
|
||||
|
||||
def assign_replies(node: ExportMessageNode) -> ExportMessageNode:
|
||||
node.replies = messages_by_parent[node.message_id]
|
||||
node.replies.sort(key=lambda x: x.rank if x.rank is not None else float("inf"))
|
||||
for child in node.replies:
|
||||
assign_replies(child)
|
||||
return node
|
||||
|
||||
Reference in New Issue
Block a user