Import error: cannot import name 'TOOL_MESSAGE_BLOCK_TYPES'

[beatriz1215]

The following code:

from langgraph.prebuilt.tool_node import ToolNode

Error Message and Stack Trace (if applicable):

ImportError                               Traceback (most recent call last)
Cell In[38], line 1
----> 1 from langgraph.prebuilt.tool_node import ToolNode

File ~/.local/lib/python3.10/site-packages/langgraph/prebuilt/__init__.py:3
      1 """langgraph.prebuilt exposes a higher-level API for creating and executing agents and tools."""
----> 3 from langgraph.prebuilt.chat_agent_executor import create_react_agent
      4 from langgraph.prebuilt.tool_node import (
      5     InjectedState,
      6     InjectedStore,
      7     ToolNode,
      8     tools_condition,
      9 )
     10 from langgraph.prebuilt.tool_validator import ValidationNode

File ~/.local/lib/python3.10/site-packages/langgraph/prebuilt/chat_agent_executor.py:47
     45 from langgraph.managed import RemainingSteps
     46 from langgraph.prebuilt._internal import ToolCallWithContext
---> 47 from langgraph.prebuilt.tool_node import ToolNode
     48 from langgraph.runtime import Runtime
     49 from langgraph.store.base import BaseStore

File ~/.local/lib/python3.10/site-packages/langgraph/prebuilt/tool_node.py:67
     65 from langchain_core.tools import BaseTool, InjectedToolArg
     66 from langchain_core.tools import tool as create_tool
---> 67 from langchain_core.tools.base import (
     68     TOOL_MESSAGE_BLOCK_TYPES,
     69     get_all_basemodel_annotations,
     70 )
     71 from pydantic import BaseModel
     72 from typing_extensions import Annotated, get_args, get_origin

ImportError: cannot import name 'TOOL_MESSAGE_BLOCK_TYPES' from 'langchain_core.tools.base' (/opt/micromamba/lib/python3.10/site-packages/langchain_core/tools/base.py)

Description:

I am having issues using langchain_core, these are the versions I'm using:

[vofronte]

Hey @beatriz1215!
This looks like a classic version mismatch between langgraph and langchain-core.

Have you tried just upgrading both packages? This usually fixes these kinds of import errors.

pip install --upgrade langgraph langchain-core