fix: improved robustness of cache wrapper in module contexts

[dmadisetti]

📝 Summary

Follow up to #5758

Solves reference dependency issues in an external lib, by creating a partial graph of toplevel functions on invocation. Previously, the "graph" of an external function was allowed to be blank. This is incorrect, since both setup cell values, and other top level definitions can be exposed to the cached function.

For example in tests/_save/external_decorators/transitive_wrappers_1.py

with app.setup:
    ...
    impure_state = ...

@app.function
def my_impure_decorator(func):
    """An impure decorator that depends on impure_state"""

    @functools.wraps(func)
    def wrapper(*args: Any, **kwargs: Any):
        # Decorator depends on impure_state
        wrapper._value = ... # dependent on impure state ...
        return func(*args, **kwargs)

    return wrapper

...

@app.function
@my_impure_decorator
def pure_function():
    # This function itself is pure (no external dependencies)
    return 42

...

on from external_decorators.transitive_wrappers_1 import pure_function a partial graph is constructed to invalidate pure_function based on its relationship pure_function -> impure_function -> impure_state. Whereas before, we just ignored these relationships because pure_function was determined to be an external function and thus 'pure'.

[vercel]

The latest updates on your projects. Learn more about Vercel for Git ↗︎

Name	Status	Preview	Comments	Updated (UTC)
marimo-docs	✅ Ready (Inspect)	Visit Preview	💬 Add feedback	Aug 4, 2025 11:24pm

[dmadisetti]

I wouldn't mind a bit of a sanity check here.

This is the setup:

• we are loading a marimo notebook as a module
• a toplevel cache decorated function has dependencies on other top level dependencies.

This function is called on first invocation, where we do a best effort to create the graph thus far with top level definitions.

Notably, function definitions can be out of order (solved by waiting for invocation), but also invoked even before the app is done being constructed (via decorators). We grab app from the scope of the function declaration, and construct our own version of the graph with only top level deps.

[dmadisetti]

Woops. Saw the fastforward requested everyone. Sorry about that. rebasing

[github-actions]

🚀 Development release published. You may be able to view the changes at https://marimo.app?v=0.14.17-dev38