Update gaia_agent.py

gaia_agent.py

@@ -12,6 +12,9 @@ from tools import *
 load_dotenv()
 
 class GaiaAgent:
+    """
+    A agent that leverages available tools to answer questions.
+    """
     def __init__(self):
         self.llm = ChatLiteLLM(
             model="openai/gemini-2.5-pro", 
@@ -109,5 +112,5 @@ class GaiaAgent:
         messages: list[HumanMessage] = [HumanMessage(content=question)]
         result = graph.invoke({"messages": messages})
 
-        last_message = self.extract_answer(result["messages"][-1].content)
+        last_message = result["messages"][-1].content
         return self.extract_answer(last_message)