Skip to content

Commit a6236ef

Browse files
authored
fix: langgraph v1 compatibility (#586)
Add import fallback for langchain.schema for langchain >= 1.0.0 Fixes #585
1 parent 7020e06 commit a6236ef

File tree

1 file changed

+7
-1
lines changed
  • integrations/langgraph/python/ag_ui_langgraph

1 file changed

+7
-1
lines changed

integrations/langgraph/python/ag_ui_langgraph/agent.py

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,13 @@
44
import inspect
55

66
from langgraph.graph.state import CompiledStateGraph
7-
from langchain.schema import BaseMessage, SystemMessage
7+
8+
try:
9+
from langchain.schema import BaseMessage, SystemMessage
10+
except ImportError:
11+
# Langchain >= 1.0.0
12+
from langchain_core.messages import BaseMessage, SystemMessage
13+
814
from langchain_core.runnables import RunnableConfig, ensure_config
915
from langchain_core.messages import HumanMessage
1016
from langgraph.types import Command

0 commit comments

Comments
 (0)