MCPcopy Index your code
hub / github.com/OpenBMB/ToolBench / set_init_node

Method set_init_node

toolbench/tooleval/evaluation/dataclass.py:181–192  ·  view source on GitHub ↗
(self,node:Union[GID,ExecutionNode])

Source from the content-addressed store, hash-verified

179 return count
180
181 def set_init_node(self,node:Union[GID,ExecutionNode]):
182 if isinstance(node,ExecutionNode):
183 self.init_node = node.node_id
184 if node.node_id not in self.nodes:
185 self.nodes[node.node_id] = node
186 elif isinstance(node,GID):
187 if node not in self.nodes:
188 raise KeyError('node not in graph!')
189 else:
190 self.init_node = node
191 else:
192 raise TypeError('node must be instance of ExecutionNode!')
193
194 def get_init_node(self):
195 return self.nodes[self.init_node]

Callers 3

Calls

no outgoing calls

Tested by

no test coverage detected