Update agent.py

agent.py

@@ -102,8 +102,18 @@ class LangGraphAgent:
         print("LangGraphAgent initialized.")
 
     def __call__(self, question: str) -> str:
-        input_state = {"messages": [HumanMessage(content=question)]}  # prepare the initial user message
-        print(f"Running LangGraphAgent with input: {question[:150]}...")
+        # prepare the initial user message
+        input_state = {
+            "messages": [
+                {
+                    "role: "user",
+                    "content": [
+                        {"type": "text", "text": question}
+                    ]
+                }
+            ]
+            
+        print(f"Running LangGraphAgent with input: {question[:200]}...")
         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