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