Is it possible to utilise Node Expansion when returning objects from nodes?

[jonathansandman]

Hi everybody,

I am utilizing Node Expansion (docs). This is a really powerful feature that allows my custom node to return a subgraph of nodes on the fly, taking advantage of the built-in caching functionality in ComfyUI.

However, when returning values from the node:

params = {
    "model": merge_model_node.out(0),
    "clip": merge_clip_node.out(0),
    "vae": checkpoint_node1.out(2)
}

return {
    "result": (params, merge_model_node.out(0), merge_clip_node.out(0), checkpoint_node1.out(2)),
    "expand": graph.finalize(),
}

If I log the returned value from the node, outputs directly in the tuple appear to be correctly expanded, but those in the object are still the raw links, e.g.:

# Example output
(params: {
    "model": ["68.0.0.1", 0],
    "clip": ["72.0.0.1", 0],
    "vae": ["44.0.0.1", 0]
}, Model, CLIP, VAE)

where I would expect:

(params: {
    "model": Model
    "clip": CLIP,
    "vae": VAE
}, Model, CLIP, VAE)

I was hoping that I might be able to utilize Node Expansion and then combine outputs into single object to easily "pipe" it to another node. Is there any way to achieve this?