How can I get a dataflow graph (DAG) from a notebook via CLI or API?

[Meng6]

I checked the PyPI page and it mentioned that we could convert a Jupyter notebook to a marimo notebook with the CLI:

marimo convert your_notebook.ipynb > your_notebook.py

Similarly, is it possible to get a dataflow graph (DAG) from a notebook via CLI or API? For example, the input is a Jupyter notebook and the output is a JSON string:

{
    "cell1": {
        "usesVariable": [],
        "declaresVariable": ["v1", "v2"],
    },
    "cell2"{
        "usesVariable": ["v1"],
        "declaresVariable": ["v3"]
   },
    ...
}

Any suggestions? Thank you!

[mscolnick]

this is not currently possible, but seems like a useful artifact. we definitely have this knowledge.

what is the use case? do you need the DAG by itself? if you are coming from a jupyter notebook, it is not guaranteed to be a DAG (could have cycles and re-declarations)

[mscolnick]

cell's are zero indexed. so it is used by cell1 and cell3, which is correct

[Meng6]

Isn't it used by val = test(1, 3)?

[mscolnick]

transitively, yes, but the DAG only shows "parent/children" not "ancestors/descendants". the later can be inferred

[Meng6]

I see, thank you!

[Meng6]

Thanks for all the help! I need to think about my project carefully & might contribute to Marimo in the future :)
Thanks for developing this fancy notebook!

[NukeMike]

I have created a notebook to access the dag graph within the notebook itself, and represent it trough a mo.mermaid component.
I had to handle the cells linear order as stored in mo._ast.cellmanaager
Cell_manager.py

And app.graph()
If I recall well (on mobile now)

I found some useful classes and methods also in the _ai repo tools.

From within the notebook I could also load an external notebook and show it's graph.

My goal was multiple:
I wanted a graph view like the one in the sidebar, but more compact, just with cell number, cell custom name, and possibly function defined.
Have an embedded view in the notebook
Be able to export the graph view either as mermaid source or markdown with a mermaid code block that GitHub and vscode can already render (in vscode with an extension)

I purposedly created a graph to mermaid syntax function.
I also created a parent child / selector, to show the graph only of selected cells.

Then I started using more the minimap, (that does not fill the full width of the panel as the graph does)

Different views in the Marino native graph panel could be button enabled to show only titles or cell number.

My notebook could easily be adapted to run as a cli script.