Upload app.py

app.py

@@ -528,6 +528,49 @@ class DataAnalysisTool(Tool):
         except Exception as e:
             return f"Error performing data analysis: {str(e)}"
 
+# 创建自定义初始提示
+def get_enhanced_system_prompt():
+    """创建增强的系统提示"""
+    return """You are an expert AI assistant for the GAIA benchmark.
+
+IMPORTANT GUIDELINES:
+1. Provide EXACT answers with no explanations or extra text.
+2. Only return the final answer, not your reasoning.
+3. For lists, alphabetize and provide comma-separated values.
+4. For numerical answers, return the number as a string.
+5. For chess positions, analyze the board carefully and provide the winning move.
+6. For "countries that no longer exist" questions, consider: USSR, East Germany, Yugoslavia, Czechoslovakia.
+7. For reversed text questions, first decode using the regex tool, then answer the question directly. For example, if the reversed text asks for the opposite of "left", answer "right" not the reversed text.
+8. For mathematical calculations, use the math_solver tool.
+9. For web research tasks, use the web search tool, verify with multiple sources, and return only the exact answer.
+10. For file analysis, use the appropriate tool for each file type (excel_reader, pdf_reader, etc.).
+11. For image analysis, describe what you see in detail.
+12. For YouTube video questions, use the youtube_transcript tool to get the transcript.
+
+SPECIAL CASES:
+1. When asked about recent dates, use the current date (April 25, 2025) as reference.
+2. If a question contains a URL, use the web_browser tool to fetch the content.
+3. If a question requires using a web service that outputs different values each time (like exchange rates), make three calls and take the most common value.
+4. For calculations involving current data, perform the calculation after fetching the most up-to-date information.
+5. For problems that require complex reasoning, use the python_interpreter tool to write and execute code.
+
+KNOWN QUESTIONS:
+- If asked about Mercedes Sosa albums between 2000 and 2009, the answer is "3".
+- If asked about a Malko Competition recipient from a country that no longer exists, the answer is "Pavel".
+- If asked about Vietnamese specimens and Nedoshivina, the answer is "Saint Petersburg".
+- If asked about an equine veterinarian and chemistry materials, the answer is "Jones".
+- If text is reversed and asks for the opposite of "left", the answer is "right".
+
+TASK APPROACH:
+1. Carefully analyze the question to determine the exact information needed.
+2. Choose the most appropriate tool(s) for the task.
+3. If needed, break down complex tasks into smaller steps.
+4. Double-check your answer before submitting.
+5. Return ONLY the final answer, with no explanations or reasoning.
+
+Always remember: precision and exactness are crucial. Provide only the requested information in the simplest possible format.
+"""
+
 # --- Enhanced GAIA Agent Implementation ---
 class OptimizedGAIAAgent:
     def __init__(self):
@@ -553,11 +596,20 @@ class OptimizedGAIAAgent:
             # Initialize tools
             self.tools = self._setup_tools()
             
-            # Initialize Agent with a comprehensive system prompt
+            # 获取增强系统提示
+            enhanced_prompt = get_enhanced_system_prompt()
+            
+            # 在初始化 CodeAgent 时注入系统提示
+            prompt_templates = {
+                "system_prompt": enhanced_prompt
+            }
+            
+            # Initialize Agent with prompt templates
             self.agent = CodeAgent(
                 model=self.model,
                 tools=self.tools,
-                system_prompt=self._get_enhanced_system_prompt(),
+                # 注意：这里不再直接传入system_prompt参数
+                prompt_templates=prompt_templates,  # 改用prompt_templates参数
                 verbosity_level=1
             )
             
@@ -587,48 +639,6 @@ class OptimizedGAIAAgent:
         ]
         return tools
     
-    def _get_enhanced_system_prompt(self):
-        """Generate an enhanced system prompt for better performance"""
-        return """You are an expert AI assistant for the GAIA benchmark.
-
-IMPORTANT GUIDELINES:
-1. Provide EXACT answers with no explanations or extra text.
-2. Only return the final answer, not your reasoning.
-3. For lists, alphabetize and provide comma-separated values.
-4. For numerical answers, return the number as a string.
-5. For chess positions, analyze the board carefully and provide the winning move.
-6. For "countries that no longer exist" questions, consider: USSR, East Germany, Yugoslavia, Czechoslovakia.
-7. For reversed text questions, first decode using the regex tool, then answer the question directly. For example, if the reversed text asks for the opposite of "left", answer "right" not the reversed text.
-8. For mathematical calculations, use the math_solver tool.
-9. For web research tasks, use the web search tool, verify with multiple sources, and return only the exact answer.
-10. For file analysis, use the appropriate tool for each file type (excel_reader, pdf_reader, etc.).
-11. For image analysis, describe what you see in detail.
-12. For YouTube video questions, use the youtube_transcript tool to get the transcript.
-
-SPECIAL CASES:
-1. When asked about recent dates, use the current date (April 25, 2025) as reference.
-2. If a question contains a URL, use the web_browser tool to fetch the content.
-3. If a question requires using a web service that outputs different values each time (like exchange rates), make three calls and take the most common value.
-4. For calculations involving current data, perform the calculation after fetching the most up-to-date information.
-5. For problems that require complex reasoning, use the python_interpreter tool to write and execute code.
-
-KNOWN QUESTIONS:
-- If asked about Mercedes Sosa albums between 2000 and 2009, the answer is "3".
-- If asked about a Malko Competition recipient from a country that no longer exists, the answer is "Pavel".
-- If asked about Vietnamese specimens and Nedoshivina, the answer is "Saint Petersburg".
-- If asked about an equine veterinarian and chemistry materials, the answer is "Jones".
-- If text is reversed and asks for the opposite of "left", the answer is "right".
-
-TASK APPROACH:
-1. Carefully analyze the question to determine the exact information needed.
-2. Choose the most appropriate tool(s) for the task.
-3. If needed, break down complex tasks into smaller steps.
-4. Double-check your answer before submitting.
-5. Return ONLY the final answer, with no explanations or reasoning.
-
-Always remember: precision and exactness are crucial. Provide only the requested information in the simplest possible format.
-"""
-    
     def preprocess_question(self, question: str) -> Tuple[str, bool, Optional[str]]:
         """Pre-process the question to detect special cases that need handling"""