From ca7da2fc4188b7b4ddf0e5d378ed91c9a535038e Mon Sep 17 00:00:00 2001
From: Christopher Brooks <cab938@mail.usask.ca>
Date: Tue, 19 Nov 2024 11:18:53 -0500
Subject: [PATCH] feat: Make CompiledGraph displayable in Juypyter with
 display() (#2159)

---
 libs/langgraph/langgraph/graph/graph.py | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/libs/langgraph/langgraph/graph/graph.py b/libs/langgraph/langgraph/graph/graph.py
index e91ac4a47..a1cc029e8 100644
--- a/libs/langgraph/langgraph/graph/graph.py
+++ b/libs/langgraph/langgraph/graph/graph.py
@@ -628,3 +628,11 @@ class CompiledGraph(Pregel):
                     add_edge(key, end, conditional=True)
 
         return graph
+
+    def _repr_mimebundle_(self, **kwargs) -> dict[str, Any]:
+        """Mime bundle used by Jupyter to display the graph"""
+        output = {
+            "text/plain": repr(self),
+            "image/png": self.get_graph().draw_mermaid_png()
+        }
+        return output