Pass graph state to ToolNode along with arguments

[elupanov]

Checked

• I searched existing ideas and did not find a similar one
• I added a very descriptive title
• I've clearly described the feature request and motivation for it

Feature request

Pass the current graph state to the tool function along with the arguments using ToolNode.
Something like this:

from langgraph.prebuilt import ToolNode

@tool
def my_tool(first_arg: str, state):
  """Tool description"""
  return first_arg

graph.add_node("tool", ToolNode([my_tool]))

Motivation

I'm building a simple tools-based agent like on a screenshot.
In the state, I store some metadata that I would like to reference in a tool function. To do so, I have to write a custom implementation of the node and stop using ToolNode.

Proposal (If applicable)

No response

[bharath-muthineni]

@elupanov were you able to achieve what you are looking for in any way?
I have a similar use case where I need the state to be available in my custom tool run method.

[elupanov]

Hi!
It was an issue some time ago. Now you can pass state to the tools. Here is the article how to do it https://langchain-ai.github.io/langgraph/how-tos/pass-run-time-values-to-tools/#pass-graph-state-to-tools