Fix edge source attribute in AgentVertex

This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-11-28 14:35:16 -03:00
commit dec36dbfb9

View file

@ -26,7 +26,7 @@ class AgentVertex(Vertex):
def _set_tools_and_chains(self) -> None: def _set_tools_and_chains(self) -> None:
for edge in self.edges: for edge in self.edges:
if not hasattr(edge, "source"): if not hasattr(edge, "source_id"):
continue continue
source_node = self.graph.get_vertex(edge.source_id) source_node = self.graph.get_vertex(edge.source_id)
if isinstance(source_node, (ToolVertex, ToolkitVertex)): if isinstance(source_node, (ToolVertex, ToolkitVertex)):