the agent is working again

app.py

@@ -91,16 +91,44 @@ def research_topic(query: str) -> str:
     """
     try:
         # Create a fresh instance for each search
-        search_tool = get_search_tool(max_results=3)
+        search_tool = get_search_tool(max_results=5)  # Increased from 3 to 5
         
         # Add focus on tech and AI product development
         enhanced_query = f"{query} AI product development"
         results = search_tool.forward(enhanced_query)
         
         # Format the results in your style
-        response = f"Quick research on {query}:\n\n"
-        response += results
-        response += "\n\nNote: Always verify current info."
+        response = f"# Research on {query}\n\n"
+        
+        # Split content into examples and sources
+        examples = []
+        sources = []
+        current_example = []
+        
+        for line in results.split('\n'):
+            if line.startswith('http'):
+                sources.append(line.strip())
+                if current_example:
+                    examples.append('\n'.join(current_example))
+                    current_example = []
+            elif line.strip():
+                current_example.append(line.strip())
+        
+        if current_example:
+            examples.append('\n'.join(current_example))
+        
+        # Add examples with clear separation
+        response += "## Key Insights\n\n"
+        for i, example in enumerate(examples, 1):
+            response += f"### Example {i}\n"
+            response += f"{example}\n\n"
+        
+        # Add sources
+        response += "\n## Sources\n"
+        for source in sources:
+            response += f"- {source}\n"
+        
+        response += "\n\nNote: Always verify current information."
         
         return response
     except Exception as e:

tools/final_answer.py

@@ -1,14 +1,51 @@
-from typing import Any, Optional
+from typing import Any, Optional, Dict, Union
 from smolagents.tools import Tool
 
 class FinalAnswerTool(Tool):
     name = "final_answer"
-    description = "Provides a final answer to the given problem."
-    inputs = {'answer': {'type': 'any', 'description': 'The final answer to the problem'}}
-    output_type = "any"
+    description = "Provides a final answer to the given problem with proper formatting and sources."
+    inputs = {
+        'content': {'type': 'string', 'description': 'The main content of the answer'},
+        'sources': {
+            'type': 'array', 
+            'description': 'List of sources used in the research',
+            'nullable': True
+        },
+        'metadata': {
+            'type': 'object', 
+            'description': 'Additional metadata like readability score, SEO info, etc.',
+            'nullable': True
+        }
+    }
+    output_type = "string"
 
-    def forward(self, answer: Any) -> Any:
-        return answer
+    def format_output(self, content: str, sources: list = None, metadata: Dict = None) -> str:
+        """Format the output in a clean, readable way with proper sections."""
+        output_parts = []
+        
+        # Add main content
+        output_parts.append("# Content\n")
+        output_parts.append(content.strip())
+        output_parts.append("\n\n")
+        
+        # Add sources if available
+        if sources and len(sources) > 0:
+            output_parts.append("# Sources\n")
+            for idx, source in enumerate(sources, 1):
+                output_parts.append(f"{idx}. {source}\n")
+            output_parts.append("\n")
+        
+        # Add metadata if available
+        if metadata:
+            output_parts.append("# Additional Information\n")
+            for key, value in metadata.items():
+                output_parts.append(f"- {key}: {value}\n")
+        
+        return "".join(output_parts)
+
+    def forward(self, content: str, sources: list = None, metadata: Dict = None) -> str:
+        """Process the final answer with proper formatting."""
+        return self.format_output(content, sources, metadata)
 
     def __init__(self, *args, **kwargs):
         self.is_initialized = False