Compat with future langgraph version

backend/app/agent_types/tools_agent.py

@@ -9,7 +9,7 @@
     SystemMessage,
     ToolMessage,
 )
-from langgraph.checkpoint import BaseCheckpointSaver
+from langgraph.checkpoint.base import BaseCheckpointSaver
 from langgraph.graph import END
 from langgraph.graph.message import MessageGraph
 from langgraph.prebuilt import ToolExecutor, ToolInvocation

backend/app/agent_types/xml_agent.py

@@ -7,7 +7,7 @@
     HumanMessage,
     SystemMessage,
 )
-from langgraph.checkpoint import BaseCheckpointSaver
+from langgraph.checkpoint.base import BaseCheckpointSaver
 from langgraph.graph import END
 from langgraph.graph.message import MessageGraph
 from langgraph.prebuilt import ToolExecutor, ToolInvocation

backend/app/chatbot.py

@@ -2,7 +2,7 @@
 
 from langchain_core.language_models.base import LanguageModelLike
 from langchain_core.messages import BaseMessage, SystemMessage
-from langgraph.checkpoint import BaseCheckpointSaver
+from langgraph.checkpoint.base import BaseCheckpointSaver
 from langgraph.graph.state import StateGraph
 
 from app.message_types import add_messages_liberal

backend/app/checkpoint.py

@@ -4,7 +4,7 @@
 
 from langchain_core.messages import BaseMessage
 from langchain_core.runnables import ConfigurableFieldSpec, RunnableConfig
-from langgraph.checkpoint import BaseCheckpointSaver
+from langgraph.checkpoint.base import BaseCheckpointSaver
 from langgraph.checkpoint.base import (
     Checkpoint,
     CheckpointAt,

backend/app/retrieval.py

@@ -7,7 +7,7 @@
 from langchain_core.prompts import PromptTemplate
 from langchain_core.retrievers import BaseRetriever
 from langchain_core.runnables import chain
-from langgraph.checkpoint import BaseCheckpointSaver
+from langgraph.checkpoint.base import BaseCheckpointSaver
 from langgraph.graph import END
 from langgraph.graph.state import StateGraph
 

tools/redis_to_postgres/migrate_data.py

@@ -13,7 +13,7 @@
 import orjson
 from langchain.utilities.redis import get_client
 from langchain_core.runnables import ConfigurableFieldSpec, RunnableConfig
-from langgraph.checkpoint import BaseCheckpointSaver
+from langgraph.checkpoint.base import BaseCheckpointSaver
 from langgraph.checkpoint.base import (
     Checkpoint,
     empty_checkpoint,