Update app.py

app.py

@@ -22,13 +22,6 @@ class BasicAgent:
         print(f"Agent returning fixed answer: {fixed_answer}")
         return fixed_answer
 
-import os
-from langgraph.graph import StateGraph, END
-from langchain_core.runnables import Runnable
-from langchain_core.messages import HumanMessage
-from langchain_core.runnables import RunnableLambda
-from langchain_openai import ChatOpenAI
-
 from langgraph.graph import StateGraph
 from langchain_core.runnables import RunnableLambda
 from langchain_core.messages import HumanMessage, SystemMessage
@@ -58,7 +51,8 @@ class LangGraphAgent:
             final_answer = self._extract_final_answer(response)
             return {"answer": final_answer}
 
-        builder = StateGraph(state={"question": str, "answer": str})
+        builder = StateGraph()
+        builder = builder.with_input_types(question=str, answer=str)
         builder.add_node("answer_question", RunnableLambda(answer_fn))
         builder.set_entry_point("answer_question")
         builder.set_finish_point("answer_question")