Merge pull request #2468 from cab938/issue2159

feat: Make CompiledGraph displayable in Jupyter with display()
This commit is contained in:
Nuno Campos
2024-12-05 10:03:40 -08:00
committed by GitHub
+7
View File
@@ -629,3 +629,10 @@ class CompiledGraph(Pregel):
add_edge(key, end, conditional=True)
return graph
def _repr_mimebundle_(self, **kwargs: Any) -> dict[str, Any]:
"""Mime bundle used by Jupyter to display the graph"""
return {
"text/plain": repr(self),
"image/png": self.get_graph().draw_mermaid_png(),
}