fix(sdk-python): compatibility with langchain v1 (CopilotKit#2633) (CopilotKit#2634)

Author: alexforrestf

sdk-python/copilotkit/langgraph_agent.py

@@ -8,8 +8,15 @@
 from typing_extensions import NotRequired
 
 from langgraph.types import Command
-from langchain.load.dump import dumps as langchain_dumps
-from langchain.schema import BaseMessage, SystemMessage
+
+try:
+    from langchain.load.dump import dumps as langchain_dumps
+    from langchain.schema import BaseMessage, SystemMessage
+except ImportError:
+    # Langchain >= 1.0.0
+    from langchain_core.load import dumps as langchain_dumps
+    from langchain_core.messages import BaseMessage, SystemMessage
+    
 from langchain_core.runnables import RunnableConfig, ensure_config
 from langchain_core.messages import HumanMessage
 

sdk-python/copilotkit/langgraph_agui_agent.py

@@ -15,7 +15,12 @@
 )
 from langgraph.graph.state import CompiledStateGraph
 from langchain_core.runnables import RunnableConfig
-from langchain.schema import BaseMessage
+
+try:
+    from langchain.schema import BaseMessage
+except ImportError:
+    # Langchain >= 1.0.0
+    from langchain_core.messages import BaseMessage
 
 
 class CustomEventNames(Enum):