From 650275db8a860ec4a2d046e65ca2dfaa1c48cd25 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Mon, 10 Jul 2023 08:05:18 -0300 Subject: [PATCH] =?UTF-8?q?=F0=9F=94=A7=20chore(base.py):=20add=20=5F=5Fre?= =?UTF-8?q?pr=5F=5F=20method=20to=20Graph=20class=20for=20better=20debuggi?= =?UTF-8?q?ng=20and=20representation?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/backend/langflow/graph/graph/base.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/src/backend/langflow/graph/graph/base.py b/src/backend/langflow/graph/graph/base.py index 65fff3239..86a2f98a9 100644 --- a/src/backend/langflow/graph/graph/base.py +++ b/src/backend/langflow/graph/graph/base.py @@ -214,3 +214,10 @@ class Graph: if node_type in node_types: children.append(node) return children + + def __repr__(self): + node_ids = [node.id for node in self.nodes] + edges_repr = "\n".join( + [f"{edge.source.id} --> {edge.target.id}" for edge in self.edges] + ) + return f"Graph:\nNodes: {node_ids}\nConnections:\n{edges_repr}"