langchain versions

app.py

@@ -8,7 +8,7 @@ from langchain_google_genai import ChatGoogleGenerativeAI
 from typing import IO, Dict
 from io import BytesIO
 from langchain_core.messages import HumanMessage, SystemMessage
-from langgraph.graph import MessagesState, START, StateGraph
+from langgraph.graph import START, StateGraph
 from langgraph.prebuilt import tools_condition
 from langgraph.prebuilt import ToolNode
 import base64
@@ -198,7 +198,7 @@ class BasicAgent:
                         ''')
 
         # Graph
-        self.builder = StateGraph(MessagesState)
+        self.builder = StateGraph(START)
 
         # Define nodes: these do the work
         self.builder.add_node("assistant", self.assistant)
@@ -217,7 +217,7 @@ class BasicAgent:
         
         print("BasicAgent initialized.")
     
-    def assistant(self, state: MessagesState):
+    def assistant(self, state: StateGraph):
         return {"messages": [self.agent.invoke([self.sys_msg] + state["messages"])]}
     
     async def __call__(self, question: str, task_id: str) -> str: