Update news_tools.py

news_tools.py

@@ -1,58 +1,59 @@
+# news_tools.py - CrewAI Native Version
 import os
-from langchain.tools import tool
 import requests
+from crewai.tools import BaseTool
+from typing import Type
+from pydantic import BaseModel, Field
 
-@tool("Search Crypto News")
-def search_crypto_news(query: str) -> str:
-    """Searches for cryptocurrency news articles based on a given query."""
-    try:
-        # Since you might not have the CryptoNews API key, let's use a free alternative
-        # Using NewsAPI (free tier) or fall back to mock data
-        
-        api_key = os.getenv("NEWS_API_KEY")  # You can get this free from newsapi.org
-        
-        if api_key:
-            url = "https://newsapi.org/v2/everything"
-            params = {
-                'q': f"{query} cryptocurrency",
-                'sortBy': 'publishedAt',
-                'pageSize': 3,
-                'apiKey': api_key,
-                'language': 'en'
-            }
-            
-            response = requests.get(url, params=params, timeout=10)
-            response.raise_for_status()
-            
-            data = response.json()
-            articles = data.get('articles', [])
+class NewsSearchInput(BaseModel):
+    """Input schema for NewsSearchTool."""
+    query: str = Field(..., description="Query string to search for cryptocurrency news")
+
+class NewsSearchTool(BaseTool):
+    name: str = "Search Crypto News"
+    description: str = "Searches for cryptocurrency news articles based on a given query"
+    args_schema: Type[NewsSearchInput] = NewsSearchInput
+
+    def _run(self, query: str) -> str:
+        try:
+            # Using NewsAPI (free tier) or fall back to mock data
+            api_key = os.getenv("NEWS_API_KEY")
             
-            if articles:
-                news_summary = []
-                for article in articles[:3]:
-                    title = article.get('title', 'No title')
-                    source = article.get('source', {}).get('name', 'Unknown source')
-                    news_summary.append(f"• {title} - {source}")
+            if api_key:
+                url = "https://newsapi.org/v2/everything"
+                params = {
+                    'q': f"{query} cryptocurrency",
+                    'sortBy': 'publishedAt',
+                    'pageSize': 3,
+                    'apiKey': api_key,
+                    'language': 'en'
+                }
+                
+                response = requests.get(url, params=params, timeout=10)
+                response.raise_for_status()
+                
+                data = response.json()
+                articles = data.get('articles', [])
                 
-                return f"Recent news for {query}:\n" + "\n".join(news_summary)
+                if articles:
+                    news_summary = []
+                    for article in articles[:3]:
+                        title = article.get('title', 'No title')
+                        source = article.get('source', {}).get('name', 'Unknown source')
+                        news_summary.append(f"• {title} - {source}")
+                    
+                    return f"Recent news for {query}:\n" + "\n".join(news_summary)
+                else:
+                    return f"No recent news found for {query}"
             else:
-                return f"No recent news found for {query}"
-        else:
-            # Mock news data when API key is not available
-            mock_headlines = [
-                f"{query.title()} shows strong technical indicators amid market volatility",
-                f"Institutional adoption of {query.title()} continues to grow",
-                f"Market analysts remain optimistic about {query.title()}'s long-term prospects"
-            ]
-            
-            return f"Recent news for {query}:\n" + "\n".join([f"• {headline}" for headline in mock_headlines])
-            
-    except requests.exceptions.RequestException as e:
-        return f"Error fetching news: {str(e)}"
-    except Exception as e:
-        return f"News search error: {str(e)}"
-
-class NewsTools:
-    """Container class for news tools - for backward compatibility"""
-    def __init__(self):
-        self.search_crypto_news = search_crypto_news
+                # Mock news data when API key is not available
+                mock_headlines = [
+                    f"{query.title()} shows strong technical indicators amid market volatility",
+                    f"Institutional adoption of {query.title()} continues to grow",
+                    f"Market analysts remain optimistic about {query.title()}'s long-term prospects"
+                ]
+                
+                return f"Recent news for {query}:\n" + "\n".join([f"• {headline}" for headline in mock_headlines])
+                
+        except Exception as e:
+            return f"Error fetching news: {str(e)}"