MCPcopy
hub / github.com/LAION-AI/Open-Assistant / import_tree

Method import_tree

backend/import.py:83–101  ·  view source on GitHub ↗
(
        self, tree: ExportMessageTree, state: TreeState = TreeState.BACKLOG_RANKING
    )

Source from the content-addressed store, hash-verified

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
104def import_file(

Callers 1

import_txFunction · 0.95

Calls 3

import_messageMethod · 0.95
MessageTreeStateClass · 0.90
addMethod · 0.80

Tested by

no test coverage detected