From 15dc4347e01025e9335621751b38f13cf7e37981 Mon Sep 17 00:00:00 2001
From: Gabriel Luiz Freitas Almeida <gabriel@langflow.org>
Date: Wed, 14 Aug 2024 13:54:10 -0300
Subject: [PATCH] refactor: Simplify __repr__ method in Edge and ContractEdge
 classes. (#3333)

---
 src/backend/base/langflow/graph/edge/base.py | 16 +++++-----------
 1 file changed, 5 insertions(+), 11 deletions(-)

diff --git a/src/backend/base/langflow/graph/edge/base.py b/src/backend/base/langflow/graph/edge/base.py
index 80aa5519b..a97a7c919 100644
--- a/src/backend/base/langflow/graph/edge/base.py
+++ b/src/backend/base/langflow/graph/edge/base.py
@@ -159,10 +159,11 @@ class Edge:
             raise ValueError(f"Edge between {source.vertex_type} and {target.vertex_type} " f"has no matched type")
 
     def __repr__(self) -> str:
-        return (
-            f"Edge(source={self.source_id}, target={self.target_id}, target_param={self.target_param}"
-            f", matched_type={self.matched_type})"
-        )
+        if (hasattr(self, "source_handle") and self.source_handle) and (
+            hasattr(self, "target_handle") and self.target_handle
+        ):
+            return f"{self.source_id} -[{self.source_handle.name}->{self.target_handle.field_name}]-> {self.target_id}"
+        return f"{self.source_id} -[{self.target_param}]-> {self.target_id}"
 
     def __hash__(self) -> int:
         return hash(self.__repr__())
@@ -220,10 +221,3 @@ class ContractEdge(Edge):
             if target.params.get("message") == "":
                 return self.result
         return self.result
-
-    def __repr__(self) -> str:
-        if (hasattr(self, "source_handle") and self.source_handle) and (
-            hasattr(self, "target_handle") and self.target_handle
-        ):
-            return f"{self.source_id} -[{self.source_handle.name}->{self.target_handle.field_name}]-> {self.target_id}"
-        return f"{self.source_id} -[{self.target_param}]-> {self.target_id}"