Update app.py

app.py

@@ -59,8 +59,8 @@ class LangGraphAgent:
             final_answer = self._extract_final_answer(response)
             return {"question": question, "answer": final_answer}
 
-        builder = StateGraph()
-        builder = builder.with_state_schema({"question": str, "answer": str})
+        # Fix: Use proper StateGraph configuration
+        builder = StateGraph(state_schema={"question": str, "answer": str})
         builder.add_node("answer_question", RunnableLambda(answer_fn))
         builder.set_entry_point("answer_question")
         builder.set_finish_point("answer_question")