Update app.py

app.py

@@ -47,7 +47,7 @@ prompt_recomendado = """You are a general AI assistant. I will ask you a questio
                         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.
                         To assist in your task, you can supervise other agents who perform specific tasks that could not be handled by tools, since they require the processing of another LLM. Below, I will inform you about your assistants:
-                        - web_research_agent. Assign web research related tasks to this agent
+                        - web_research_agent. Assign web research related tasks to this agent, prioritizing the use of Wikipedia sources
                         Assign work to one agent at a time, do not call agents in parallel.
                         Priorize the use of tools and another agents to help in reasoning.
                         When a file or URL is entered at the prompt, use it in tools or other agents, both are prepared to handle files and URLs."""
@@ -55,12 +55,13 @@ prompt_recomendado = """You are a general AI assistant. I will ask you a questio
 prompt_search = """You are a web research agent.
                    INSTRUCTIONS:
                    Assist ONLY with research-related tasks, DO NOT do any math
+                   If a source is provided in the task, you MUST use it as your primary source of information
                    After you're done with your tasks, respond to the supervisor directly
                    Respond ONLY with the results of your work, do NOT include ANY other text."""
 
 #TOOLS
 web_search = TavilySearch(
-        max_results=5,
+        max_results=10,
         topic="general",
     )
 
@@ -146,7 +147,7 @@ class BasicAgent:
         start_index = answer.find("FINAL ANSWER: ")
         substring=""
         if start_index != -1:
-            substring = answer[start_index+14:-1] 
+            substring = answer[start_index+14:] 
         
         final_answer = substring #answer.removeprefix("FINAL ANSWER: ") #re.search(r"FINAL ANSWER:\s*(.*)", answer)
         print(f"Agent returning answer: {final_answer}\n")