Update agent.py

agent.py

@@ -60,7 +60,16 @@ def init_agent():
             "Before giving out the final answer, please verify using available tools." \
             "When available, use the Wikipedia if it can be verifiable." \
             "Multiple tools can combine to answer the question as a whole."
-
     )
 
-    return agent
+    return agent
+
+# Run the agent
+async def run_agent(agent,query:str):
+    response = await agent.run(query)
+    return response
+
+# Await function for async
+async def await_result(agent,query:str):
+    response = await agent.run_agent(query)
+    return response.response.blocks[0].text