From ca7da2fc4188b7b4ddf0e5d378ed91c9a535038e Mon Sep 17 00:00:00 2001 From: Christopher Brooks Date: Tue, 19 Nov 2024 11:18:53 -0500 Subject: [PATCH 1/3] 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 From 253090f34dbef6656ae2e1c98e40622a24b48fa6 Mon Sep 17 00:00:00 2001 From: Nuno Campos Date: Tue, 19 Nov 2024 10:29:32 -0800 Subject: [PATCH 2/3] lint --- libs/langgraph/langgraph/graph/graph.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/libs/langgraph/langgraph/graph/graph.py b/libs/langgraph/langgraph/graph/graph.py index a1cc029e8..11ef41820 100644 --- a/libs/langgraph/langgraph/graph/graph.py +++ b/libs/langgraph/langgraph/graph/graph.py @@ -633,6 +633,6 @@ class CompiledGraph(Pregel): """Mime bundle used by Jupyter to display the graph""" output = { "text/plain": repr(self), - "image/png": self.get_graph().draw_mermaid_png() + "image/png": self.get_graph().draw_mermaid_png(), } return output From 9f73dfa8d593af194150af37f51011346dab2a9f Mon Sep 17 00:00:00 2001 From: Nuno Campos Date: Wed, 4 Dec 2024 17:43:05 -0800 Subject: [PATCH 3/3] Fix --- libs/langgraph/langgraph/graph/graph.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/libs/langgraph/langgraph/graph/graph.py b/libs/langgraph/langgraph/graph/graph.py index 4ecb03f5a..5e2739f02 100644 --- a/libs/langgraph/langgraph/graph/graph.py +++ b/libs/langgraph/langgraph/graph/graph.py @@ -630,10 +630,9 @@ class CompiledGraph(Pregel): return graph - def _repr_mimebundle_(self, **kwargs) -> dict[str, Any]: + def _repr_mimebundle_(self, **kwargs: Any) -> dict[str, Any]: """Mime bundle used by Jupyter to display the graph""" - output = { + return { "text/plain": repr(self), "image/png": self.get_graph().draw_mermaid_png(), } - return output