diff --git a/integrations/langgraph/python/ag_ui_langgraph/agent.py b/integrations/langgraph/python/ag_ui_langgraph/agent.py
index 0e260bc9a..a67deae3b 100644
--- a/integrations/langgraph/python/ag_ui_langgraph/agent.py
+++ b/integrations/langgraph/python/ag_ui_langgraph/agent.py
@@ -4,7 +4,13 @@
 import inspect
 
 from langgraph.graph.state import CompiledStateGraph
-from langchain.schema import BaseMessage, SystemMessage
+
+try:
+    from langchain.schema import BaseMessage, SystemMessage
+except ImportError:
+    # Langchain >= 1.0.0
+    from langchain_core.messages import BaseMessage, SystemMessage
+    
 from langchain_core.runnables import RunnableConfig, ensure_config
 from langchain_core.messages import HumanMessage
 from langgraph.types import Command