MCP client not injecting state into tool nodes

[samirbajaj]

I'm writing a simple MCP client/server app, which works fine until I try to access the graph state from within one or more of the tools.

The code is as follows:

class CustomState(AgentState):
    other_stuff: str

...

  # `client` is an instance of `MultiServerMCPClient`

  tools = client.get_tools()
  tool_node = ToolNode(tools)

  agent = create_react_agent(
      model,
      tool_node,
      checkpointer=MemorySaver(),
      state_schema=CustomState
  )

When I run the client, the following AI message appears:

================================== Ai Message ==================================

I can help you with that, but I need the state and tool_call_id... 

On the server, I have an @mcp.tool decoration on each of my tools:

@mcp.tool()
async def cool_tool(
    tool_call_id: Annotated[str, InjectedToolCallId],
    state: Annotated[dict, InjectedState]) -> Command:
...

What am I missing?

Thank you for any help.

[samirbajaj]

Not planned.

[mohsen-zendy]

I want to check if this issue is solved ? or a way to inject the state to tools that are on mcp client

[theshyPika]

Are there any updates on this issue?