Prompt changes

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 FunctionAgent
+from llama_index.core.agent.workflow import ReActAgent
 
 from llama_index.tools.duckduckgo import DuckDuckGoSearchToolSpec
 from llama_index.core.tools import FunctionTool
@@ -24,13 +24,15 @@ class BasicAgent:
         tool_spec = DuckDuckGoSearchToolSpec()
         search_tool = FunctionTool.from_defaults(tool_spec.duckduckgo_full_search)
         llm = Groq(model="deepseek-r1-distill-llama-70b", api_key=api_key)
-        self.agent = FunctionAgent(tools=[search_tool], llm=llm, system_prompt='''You are a helpful assistant that has access to a seach tool to retrieve appropriate information to fulfill a task. 
-        Also, if needed, you can generate code and execute it to verify.
-        After receiving a request from a customer identify relevant subtasks to complete before you can produce the final response. 
-        If you cannot derive information from the context provided within a request, use the search tool.
-        Process each subtask separately and then aggregate the final response.
-        Respond as concisely as possible. 
-        Always follow the instructions about the final response provided in a customer's request.''')
+        self.agent = ReActAgent(tools=[search_tool], llm=llm, system_prompt='''
+        You are a general AI assistant.
+        I will ask you a question. You can use the search tool to help you with an answer.
+        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.
+        If you are asked for a string, don't use articles, neither abbreviations (e.g. for cities), and write the digits in plain text unless specified otherwise.
+        If you are asked for a comma separated list, apply the above rules depending of whether the element to be put in the list is a number or a string.'''
+        )
         print("BasicAgent initialized.")
     async def __call__(self, question: str) -> str:
         print(f"Agent received question (first 50 chars): {question[:50]}...")