Prompt updated

app.py

@@ -8,7 +8,7 @@ import asyncio
 from llama_index.core import SimpleDirectoryReader
 
 from llama_index.llms.groq import Groq
-from llama_index.core.agent.workflow import ReActAgent
+from llama_index.core.agent.workflow import FunctionAgent
 
 from llama_index.tools.duckduckgo import DuckDuckGoSearchToolSpec
 from llama_index.tools.wikipedia import WikipediaToolSpec
@@ -30,9 +30,10 @@ class BasicAgent:
         arxiv_spec = ArxivToolSpec()
         arxiv_tool = FunctionTool.from_defaults(arxiv_spec.arxiv_query)
         llm = Groq(model="deepseek-r1-distill-llama-70b", api_key=api_key)
-        self.agent = ReActAgent(tools=[search_tool, wiki_tool, arxiv_tool], llm=llm, system_prompt='''
+        self.agent = FunctionAgent(tools=[search_tool, wiki_tool, arxiv_tool], llm=llm, system_prompt='''
         You are a general AI assistant.
         I will ask you a question. You can use the tools at your disposal to help you with an answer.
+        After receiving a question identify relevant subtasks to complete before you can produce the final response.
         Report your thoughts, and finish your answer with the following template: FINAL ANSWER: [YOUR FINAL ANSWER].
         YOUR FINAL ANSWER should be a number OR as few words as possible OR a comma separated list of numbers and/or strings.
         If you are asked for a number, don't use comma to write your number neither use units such as $ or percent sign unless specified otherwise.
@@ -44,10 +45,10 @@ class BasicAgent:
         print(f"Agent received question (first 50 chars): {question[:50]}...")
         fixed_answer = await self.agent.run(question)
         result = fixed_answer.response.content
-        # if "FINAL ANSWER:" in result:
-        #     result = result.split("FINAL ANSWER: ")[1]
-        if "Answer" in result:
-            result = result.split("Answer")[1]
+        if "FINAL ANSWER:" in result:
+            result = result.split("FINAL ANSWER: ")[1]
+        # if "Answer" in result:
+        #     result = result.split("Answer")[1]
         print(result)
         return result