Update Agent.py

Agent.py

@@ -4,8 +4,9 @@ from langgraph.graph import StateGraph, END
 from langchain.agents import AgentExecutor, initialize_agent
 from langchain.memory import ConversationBufferMemory
 from langchain_community.tools import DuckDuckGoSearchRun
-from langchain.schema import SystemMessage
-from langchain.schema.messages import HumanMessage
+#from langchain.schema import SystemMessage
+#from langchain.schema.messages import HumanMessage
+from langchain_core.messages import SystemMessage, HumanMessage
 GAIA_SYSTEM_PROMPT = """
 You are an advanced autonomous agent designed to pass the GAIA benchmark.
 You must: