Update app.py

app.py

@@ -64,7 +64,7 @@ class LangGraphAgent:
     def __init__(self):
         self.model = ZephyrAPI()
 
-        builder = StateGraph()
+        builder = StateGraph(dict)
 
         def call_model(state):
             messages = state.get("messages", [])
@@ -255,6 +255,7 @@ with gr.Blocks() as demo:
     def test_agent_response(question: str) -> str:
         # agent = BasicAgent()
         agent = LangGraphAgent()
+        print(agent("What's the capital of France?"))
         return agent(question)
 
     test_button.click(fn=test_agent_response, inputs=question_input, outputs=answer_output)