Update app.py

app.py

@@ -13,7 +13,7 @@ from dataclasses import dataclass
 
 @dataclass
 class SearchResult:
-    """Data class to store search results"""
+    """Data class to store search results from news searches"""
     title: str
     link: str
     date: str
@@ -25,6 +25,7 @@ class GroqLLM:
         self.model_name = model_name
     
     def __call__(self, prompt: Union[str, dict, List[Dict]]) -> str:
+        """Process the prompt and return the LLM's response"""
         try:
             prompt_str = str(prompt) if isinstance(prompt, (dict, list)) else prompt
             completion = self.client.chat.completions.create(
@@ -47,13 +48,16 @@ class NewsAnalysisAgent(CodeAgent):
     
     @property
     def articles(self) -> List[Dict]:
+        """Access stored article data"""
         return self._articles
     
     @property
     def search_results(self) -> List[SearchResult]:
+        """Access stored search results"""
         return self._search_results
     
     def run(self, prompt: str) -> str:
+        """Execute the agent with the given prompt"""
         enhanced_prompt = f"""
         You are a news analysis assistant that can:
         - Search for recent news articles
@@ -68,7 +72,14 @@ class NewsAnalysisAgent(CodeAgent):
         return super().run(enhanced_prompt)
 
 def extract_text_from_url(url: str) -> str:
-    """Helper function to extract text content from a URL using BeautifulSoup"""
+    """Extract main text content from a given URL using BeautifulSoup.
+    
+    Args:
+        url: The URL of the webpage to extract text from
+        
+    Returns:
+        str: Extracted and cleaned text content from the webpage
+    """
     try:
         headers = {
             'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36'
@@ -91,24 +102,17 @@ def extract_text_from_url(url: str) -> str:
     except Exception as e:
         return f"Error extracting text: {str(e)}"
 
-@tool(
-    name="search_news",
-    description="Search for recent news articles using DuckDuckGo",
-    inputs={
-        "query": {
-            "type": "string",
-            "description": "Search query string"
-        },
-        "max_results": {
-            "type": "integer",
-            "description": "Maximum number of results to return",
-            "default": 5,
-            "nullable": True
-        }
-    }
-)
+@tool
 def search_news(query: str, max_results: Optional[int] = 5) -> str:
-    """Search for recent news articles using DuckDuckGo."""
+    """Search for recent news articles using DuckDuckGo.
+    
+    Args:
+        query: Search query string to find relevant news articles
+        max_results: Maximum number of results to return (default: 5)
+        
+    Returns:
+        str: Formatted string containing search results with titles and URLs
+    """
     try:
         with DDGS() as ddgs:
             search_results = list(ddgs.news(
@@ -134,18 +138,16 @@ def search_news(query: str, max_results: Optional[int] = 5) -> str:
     except Exception as e:
         return f"Error searching news: {str(e)}"
 
-@tool(
-    name="analyze_article",
-    description="Extract and analyze content from a news article URL",
-    inputs={
-        "url": {
-            "type": "string",
-            "description": "URL of the news article to analyze"
-        }
-    }
-)
+@tool
 def analyze_article(url: str) -> str:
-    """Extract and analyze content from a news article URL."""
+    """Extract and analyze content from a news article URL.
+    
+    Args:
+        url: URL of the news article to analyze
+        
+    Returns:
+        str: Analysis of the article including summary, key points, and main themes
+    """
     try:
         content = extract_text_from_url(url)
         
@@ -174,19 +176,16 @@ def analyze_article(url: str) -> str:
     except Exception as e:
         return f"Error analyzing article: {str(e)}"
 
-@tool(
-    name="identify_trends",
-    description="Identify common themes and trends across analyzed articles",
-    inputs={
-        "articles": {
-            "type": "array",
-            "description": "List of analyzed articles",
-            "nullable": True
-        }
-    }
-)
+@tool
 def identify_trends(articles: Optional[List[Dict]] = None) -> str:
-    """Identify common themes and trends across analyzed articles."""
+    """Identify common themes and trends across analyzed articles.
+    
+    Args:
+        articles: Optional list of analyzed article data. If None, uses stored articles.
+        
+    Returns:
+        str: Analysis of trends and patterns found across the articles
+    """
     articles = articles or tool.agent._articles
     
     if not articles: