Update agent.py

agent.py

@@ -21,7 +21,7 @@ def use_search_tool(query: str) -> str:
     Returns: str: The search result.
     """
     search_result = DuckDuckGoSearchRun(verbose=0).run(query)
-    return {"search_result": search_result}
+    return {"messages": search_result}
 
 @tool
 def use_wikipedia_tool(query: str) -> str:
@@ -34,9 +34,9 @@ def use_wikipedia_tool(query: str) -> str:
     """
     result = WikipediaLoader(query=query, load_max_docs=2).load()
     if result:
-        return {"Wikipedia_summary": result}
+        return {"messages": result}
     else:
-        return {"response" : f"Sorry, I couldn't find any information on '{query}' in Wikipedia."}
+        return {"messages" : f"Sorry, I couldn't find any information on '{query}' in Wikipedia."}
 
 def build_agent():
     # llm = ChatOllama(model="llama3.1")