Merge pull request #356 from langchain-ai/nc/31jul/langgraph-compat

Compat with future langgraph version
langchain-ai · Jul 31, 2024 · 541ae6f · 541ae6f
2 parents 9432c0b + 991802f
commit 541ae6f
Show file tree

Hide file tree

Showing 6 changed files with 6 additions and 6 deletions.
diff --git a/backend/app/agent_types/tools_agent.py b/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

diff --git a/backend/app/agent_types/xml_agent.py b/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

diff --git a/backend/app/chatbot.py b/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

diff --git a/backend/app/checkpoint.py b/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,

diff --git a/backend/app/retrieval.py b/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
 

diff --git a/tools/redis_to_postgres/migrate_data.py b/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,