Update Agent.py

Agent.py

@@ -7,7 +7,7 @@ from langchain.memory import ConversationBufferMemory
 from langchain_community.tools import DuckDuckGoSearchRun
 #from langchain.schema import SystemMessage
 #from langchain.schema.messages import HumanMessage
-from langchain_core.messages import SystemMessage, HumanMessage
+from langchain_core.messages import SystemMessage, HumanMessage, AIMessage
 GAIA_SYSTEM_PROMPT = """
 You are an advanced autonomous agent designed to pass the GAIA benchmark.
 You must:
@@ -34,8 +34,8 @@ llm = ChatGoogleGenerativeAI(
     max_output_tokens=1024
 )
 messages = [
-    SystemMessage(content=GAIA_SYSTEM_PROMPT)  ,
-    HumanMessage(content="Get started, messages on the way.")
+    SystemMessage(content=GAIA_SYSTEM_PROMPT)  #,
+    #HumanMessage(content="Get started, messages on the way.")
 ]
 
 response =  llm.invoke(messages)  
@@ -100,10 +100,13 @@ agent = initialize_agent(
 )
 
 def run_agent(state: MessagesState) -> MessagesState:
-    result = agent.run(state.input)
-    state.output = result
-    state.intermediate_steps.append(result)
-    return state
+    # Let the agent respond based on the full message history
+    result = agent.invoke(state["messages"])
+
+    # Add the AI's response to the message history
+    state["messages"].append(AIMessage(content=result.content))
+
+    return state   
     
 graph = StateGraph(MessagesState)