Class ExecutionNode

toolbench/tooleval/evaluation/dataclass.py:39–52

    return str(uuid.uuid4())

class ExecutionNode(BaseModel):
    node_id:GID = Field(default_factory=assign_gid)
    role: Optional[Any] = None # System, User, Assistant, Tool
    message: Optional[Any] = None
    in_degree:int = 0
    out_degree:int = 0
    
    def __eq__(self, other) -> bool:
        if isinstance(other,ExecutionNode):
            return self.node_id == other.node_id
        raise NotImplementedError('Unsupported operation between {} and {}'.format(type(self),type(other)))
    
    def __str__(self) -> str:
        return str(self.node_id)

    
class DirectedEdge(BaseModel):