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 START, StateGraph
+from langgraph.graph import StateGraph
 from langgraph.prebuilt import tools_condition
 from langgraph.prebuilt import ToolNode
 import base64
@@ -198,14 +198,14 @@ class BasicAgent:
                         ''')
 
         # Graph
-        self.builder = StateGraph(START)
+        self.builder = StateGraph()
 
         # Define nodes: these do the work
         self.builder.add_node("assistant", self.assistant)
         self.builder.add_node("tools", ToolNode(self.tools))
 
         # Define edges: these determine how the control flow moves
-        self.builder.add_edge(START, "assistant")
+        self.builder.add_edge("start", "assistant")
         self.builder.add_conditional_edges(
             "assistant",
             # If the latest message (result) from assistant is a tool call -> tools_condition routes to tools
@@ -217,7 +217,7 @@ class BasicAgent:
         
         print("BasicAgent initialized.")
     
-    def assistant(self, state: StateGraph):
+    def assistant(self, state):
         return {"messages": [self.agent.invoke([self.sys_msg] + state["messages"])]}
     
     async def __call__(self, question: str, task_id: str) -> str: