support multi ToolNode to have the function calls in parallel.

[Michael-YongWang]

I have multiple ToolNode, some need to be interrupted before calling. When have multiple function calls, how to send the tools to the ToolNode that contains the tool in parallel?

currently, I send all the function calls to all the ToolNode which gives error.

[vbarda]

@Michael-YongWang you would need custom node functions for each of your nodes, something like this:

search_tool_node = ToolNode(search_tools)
def call_search_tools(state):
    # update last AI message to contain only search tool calls
    filtered_messages = {"messages": <updated message>}
    return search_tool_node.invoke(filtered_messages)

let me know if this makes sense

[tgm185z]

I'm having the same problem.
Following the example in the initial post, let's say we have for example:

• sensitive_tools is a ToolNode with tools: update_bd and change_password.
• safe_tools is another ToolNode with the tools: look_up and calculator.

If I understend corretly, the way of handling this is:

1. Last message is an AI message with multiple tool calls: ["look_up", "calculator", "update_bd"].
2. A node function is necessary to change last AI message tool_calls to only have the tool_calls of one toolnode, safe_tools or sensitive_tools, let's say we start with the safe ones. And save in the state the pending tools from the other node.
3. Execute safe_tools node.
4. Return to node function (2) and check if the state have pending tools. In this case we still have a tool_call, so the node needs to add an AI message with the remaining tools.
5. Execute sensitive_tools node with the interrupt.

Do this makes sense? There is a better way to do this? And also, what happend if two tools have an interrupt?
Any help would be great!