Update agent.py

agent.py

@@ -102,18 +102,7 @@ class LangGraphAgent:
         print("LangGraphAgent initialized.")
 
     def __call__(self, question: str) -> str:
-        # prepare the initial user message
-        input_state = {
-            "messages": [
-                {
-                    "role": "user",
-                    "content": [
-                        {"type": "text", "text": question}
-                    ]
-                }
-            ]
-        }
-            
+        input_state = {"messages": [HumanMessage(content=question)]}  # prepare the initial user message
         print(f"Running LangGraphAgent with input: {question[:150]}...")
         result = gaia_agent.invoke(input_state, {"recursion_limit": 30})  # prevents infinite looping when the LLM keeps calling tools over and over
         final_response = result["messages"][-1].content