New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

feat: Make CompiledGraph displayable in Jupyter with display() #2468

Merged

nfcampos merged 5 commits into langchain-ai:main from cab938:issue2159

Dec 5, 2024

libs/langgraph/langgraph/graph/graph.py

-Original file line number
+Diff line change
@@ -1,4 +1,4 @@
     import asyncio
     import logging
     from collections import defaultdict
     from typing import (
@@ Expand Down Expand Up / @@ -628,3 +628,11 @@ @@
                         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

Provide feedback