From 9f73dfa8d593af194150af37f51011346dab2a9f Mon Sep 17 00:00:00 2001 From: Nuno Campos Date: Wed, 4 Dec 2024 17:43:05 -0800 Subject: [PATCH] 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 @@ def add_edge( 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