Tool calling in "worker" nodes

[yarjor]

Hello,
I am trying to create a multi-agent system that will include an orchestrator / dispatcher creating a number of short-lived agents that will handle a specific task. So far while I didn't need actual agentic behavior and tool use in the short-lived agents, I was able to implement this using the Send API and the map-reduce pattern described here.
However, I want to implement tool use loops for those sub-agents, and this is currently not working out of the box as there is no separate distinct node for each of the workers, so it's difficult to make sure each worker keeps its own state and receives the outputs of its own tool calls - I cannot use simple conditional edges like I would with a single agent with a tool node.
Has anyone here ever tried to do something similar and can suggest the best solution here?
Thanks in advance :)