change formatted result

agent.py

@@ -100,7 +100,7 @@ def search_wiki(query: str) -> Dict[str, str]:
     docs = WikipediaQueryRun(api_wrapper=WikipediaAPIWrapper())
     docs.run(query)
     formatted_result = f'<Document source="{docs.metadata["source"]}" page="{docs.metadata.get("page", "")}"/>\n{docs.page_content}\n</Document>'
-    return {"wiki_results": formatted_result}
+    return formatted_result
 
 
 # internet search tool
@@ -114,7 +114,7 @@ def search_web(query: str) -> Dict[str, str]:
     docs = DuckDuckGoSearchResults(api_wrapper=wrapper)
     docs.invoke(query)
     formatted_result = f'<Document source="{docs.metadata["source"]}" page="{docs.metadata.get("page", "")}"/>\n{docs.page_content}\n</Document>'
-    return {"web_results": formatted_result}
+    return formatted_result
 
 
 # ArXiv search tool
@@ -127,7 +127,7 @@ def search_arxiv(query: str) -> Dict[str, str]:
     arxiv = ArxivAPIWrapper()
     docs = arxiv.run(query)
     formatted_result = f'<Document source="{docs.metadata["source"]}" page="{docs.metadata.get("page", "")}"/>\n{docs.page_content}\n</Document>'
-    return {"arxiv_results": formatted_result}
+    return formatted_result
 
 
 # build retriever
@@ -155,7 +155,6 @@ tools = [
 ]
 
 
-
 # build graph function
 def build_graph():
     # llm

app.py

@@ -25,7 +25,7 @@ class BasicAgent:
         messages = [HumanMessage(content=question)]
         messages = self.graph.invoke({"messages": messages})
         answer = messages['messages'][-1].content
-        return answer[14:]
+        return answer
 
 
 def run_and_submit_all( profile: gr.OAuthProfile | None):