From 6e91c1a4bd2de27020a7a2b93c8c1f278a48c8d4 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Andreas=20K=C3=B6pf?= Date: Mon, 30 Jan 2023 10:36:58 +0100 Subject: [PATCH] export messages ordered by rank --- backend/oasst_backend/utils/tree_export.py | 1 + 1 file changed, 1 insertion(+) diff --git a/backend/oasst_backend/utils/tree_export.py b/backend/oasst_backend/utils/tree_export.py index ce69edad..e6bea0d9 100644 --- a/backend/oasst_backend/utils/tree_export.py +++ b/backend/oasst_backend/utils/tree_export.py @@ -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