(
self, tree: ExportMessageTree, state: TreeState = TreeState.BACKLOG_RANKING
)
| 81 | return msg |
| 82 | |
| 83 | def import_tree( |
| 84 | self, tree: ExportMessageTree, state: TreeState = TreeState.BACKLOG_RANKING |
| 85 | ) -> tuple[MessageTreeState, Message]: |
| 86 | assert tree.message_tree_id is not None and tree.message_tree_id == tree.prompt.message_id |
| 87 | root_msg = self.import_message(tree.prompt, message_tree_id=tree.prompt.message_id) |
| 88 | assert state == TreeState.BACKLOG_RANKING or state == TreeState.RANKING, f"{state} not supported for import" |
| 89 | active = state == TreeState.RANKING |
| 90 | mts = MessageTreeState( |
| 91 | message_tree_id=root_msg.id, |
| 92 | goal_tree_size=0, |
| 93 | max_depth=0, |
| 94 | max_children_count=0, |
| 95 | state=state, |
| 96 | origin=self.origin, |
| 97 | active=active, |
| 98 | lang=root_msg.lang or "en", |
| 99 | ) |
| 100 | self.db.add(mts) |
| 101 | return mts, root_msg |
| 102 | |
| 103 | |
| 104 | def import_file( |
no test coverage detected