cleanup

app.py

@@ -132,10 +132,10 @@ class SlpMultiAgent:
                 temperature=0.0,
                 max_tokens=500
             ),
-            tools=[KnowledgeBaseTool()],  # Remove Wikipedia to avoid timeouts
+            tools=[KnowledgeBaseTool(), WikipediaSearchTool(), DuckDuckGoSearchTool()],  # Pass all retrieval tools
             managed_agents=[research_agent, solver_agent],  # Only 2 agents
             name="ManagerAgent",
-            description="Efficient manager for quick problem solving.",
+            description="Efficient manager for quick problem solving. Uses Wikipedia and DuckDuckGo for factual questions.",
             additional_authorized_imports=["re", "math"],
             planning_interval=1,  # Faster planning
             verbosity_level=0,  # Reduce verbosity
@@ -188,20 +188,36 @@ class SlpMultiAgent:
         
         # Extract clean answer from result
         if result and isinstance(result, str):
-            # Look for final_answer pattern
             import re
-            final_answer_match = re.search(r'final_answer\(["\']([^"\']*)["\'\)]', result)  # Fixed regex
+            # Look for final_answer pattern
+            final_answer_match = re.search(r'final_answer\(["\']([^"\']*)["\']\)', result)
             if final_answer_match:
-                clean_answer = final_answer_match.group(1)
+                clean_answer = final_answer_match.group(1).strip()
+                # Post-process: remove code blocks, explanations, and keep only the direct answer
+                clean_answer = clean_answer.strip('`').strip()
+                # Remove common prefixes
+                for prefix in ["The answer is", "Based on", "According to", "Answer:"]:
+                    if clean_answer.lower().startswith(prefix.lower()):
+                        clean_answer = clean_answer[len(prefix):].strip(' :,.')
+                # If answer is a list, keep it comma separated and clean
+                if '\n' in clean_answer and ',' not in clean_answer:
+                    items = [x.strip() for x in clean_answer.split('\n') if x.strip()]
+                    clean_answer = ', '.join(items)
+                # If answer is a code block, extract the value
+                code_match = re.search(r'([A-Za-z0-9 ,.+-]+)', clean_answer)
+                if code_match:
+                    clean_answer = code_match.group(1).strip()
                 return clean_answer
-            
             # If no final_answer found, try to extract the last meaningful line
             lines = result.strip().split('\n')
             for line in reversed(lines):
-                line = line.strip()
+                line = line.strip('`').strip()
                 if line and not line.startswith('#') and not line.startswith('###') and len(line) < 200:
+                    # Remove common prefixes
+                    for prefix in ["The answer is", "Based on", "According to", "Answer:"]:
+                        if line.lower().startswith(prefix.lower()):
+                            line = line[len(prefix):].strip(' :,.')
                     return line
-        
         # Return the result from the agent
         return result if result else "Unable to determine answer."