Update research_agent.py

research_agent.py

@@ -17,22 +17,22 @@ class ResearchAgent(BaseAgent):
 
     def __call__(self, question: str, **kwargs) -> str:
         print(f"Research agent processing question: {question[:100]}...")
-        
-        # First, try structured scraping if the question looks like a list/data question
-        # structured_answer = self.try_structured_wikipedia_scrape(question)
-        # if structured_answer:
-        #     return structured_answer
 
-        # Otherwise, fallback to classic web search
+        # First extract the search query using LLM
         search_query = self.extract_search_query(question)
         # print(f"Extracted search query: {search_query}")
+
+        # Then try to search Wikipedia
         answer = self.try_structured_wikipedia_scrape(search_query)
-        print(f"Structured answer: {answer}")
+        # print(f"Structured answer: {answer}")
         if not answer:
             # If structured answer is not found, perform a web search
             search_results = self.search_web(search_query)
             answer = self.format_search_results(search_results)
 
+
+        # Now over to LLM again to interpret the result and give final answer
+
         system_prompt = """
         You are a research expert tasked with answering factual questions based on search results.
         Analyze the search results carefully and extract the information needed to answer the question accurately.
@@ -40,8 +40,6 @@ class ResearchAgent(BaseAgent):
         Provide a concise, accurate answer based only on the provided information.
         You must provide only the exact answer requested by the user, in the correct type and minimal format, based purely on the provided information.
 
-        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.