From 81279853a0bf9d47f45bc9546cb713668ab85219 Mon Sep 17 00:00:00 2001
From: Gabriel Luiz Freitas Almeida <gabriel@logspace.ai>
Date: Mon, 27 Nov 2023 18:29:44 -0300
Subject: [PATCH] Refactor Graph class to use __getstate__ and __setstate__
 methods

---
 src/backend/langflow/graph/graph/base.py | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/src/backend/langflow/graph/graph/base.py b/src/backend/langflow/graph/graph/base.py
index e9431a5ec..48e2bac30 100644
--- a/src/backend/langflow/graph/graph/base.py
+++ b/src/backend/langflow/graph/graph/base.py
@@ -34,11 +34,11 @@ class Graph:
         self._edges = self._graph_data["edges"]
         self._build_graph()
 
+    def __getstate__(self):
+        return self.raw_graph_data
+
     def __setstate__(self, state):
-        self.__dict__.update(state)
-        for edge in self.edges:
-            edge.reset()
-            edge.validate_edge()
+        self.__init__(**state)
 
     @classmethod
     def from_payload(cls, payload: Dict) -> "Graph":