Update tools/arxiv_tool.py

tools/arxiv_tool.py

@@ -4,25 +4,19 @@ from datetime import datetime, timedelta
 import json
 import os
 from typing import List, Dict
-from smolagents import Tool
+from smolagents.tools import Tool
 
 class ArxivSearchTool(Tool):
     name = "search_arxiv"
     description = "Search ArXiv for papers matching the query"
-    input_types = {"query": str, "max_results": int}
+    inputs = {
+        'query': {'type': 'string', 'description': 'The search query for papers'},
+        'max_results': {'type': 'integer', 'description': 'Maximum number of results to return'}
+    }
     output_type = List[Dict]
 
-    def __call__(self, query: str = "artificial intelligence", 
-                max_results: int = 50) -> List[Dict]:
-        """Search ArXiv using their API.
-        
-        Args:
-            query: Search query string
-            max_results: Maximum number of results to return
-            
-        Returns:
-            List[Dict]: List of paper results with metadata
-        """
+    def forward(self, query: str = "artificial intelligence", 
+              max_results: int = 50) -> List[Dict]:
         try:
             # Construct the API URL
             base_url = 'http://export.arxiv.org/api/query?'
@@ -69,10 +63,10 @@ class ArxivSearchTool(Tool):
 class LatestPapersTool(Tool):
     name = "get_latest_papers"
     description = "Get papers from the last N days from saved results"
-    input_types = {"days_back": int}
+    inputs = {'days_back': {'type': 'integer', 'description': 'Number of days to look back'}}
     output_type = List[Dict]
 
-    def __call__(self, days_back: int = 1) -> List[Dict]:
+    def forward(self, days_back: int = 1) -> List[Dict]:
         papers = []
         base_dir = "daily_papers"
         
@@ -90,27 +84,4 @@ class LatestPapersTool(Tool):
                     day_papers = json.load(f)
                     papers.extend(day_papers)
         
-        return papers
-
-def save_daily_papers(output_dir: str = "daily_papers") -> List[Dict]:
-    """Helper function to save daily papers - not exposed as a tool"""
-    os.makedirs(output_dir, exist_ok=True)
-    today = datetime.now().strftime("%Y-%m-%d")
-    
-    arxiv_tool = ArxivSearchTool()
-    papers = arxiv_tool(
-        query='cat:cs.AI OR cat:cs.LG OR cat:cs.CL OR "artificial intelligence"',
-        max_results=100
-    )
-    
-    # Filter for papers published today
-    today_papers = [
-        paper for paper in papers 
-        if paper.get('published', '').startswith(today)
-    ]
-    
-    output_file = os.path.join(output_dir, f"ai_papers_{today}.json")
-    with open(output_file, 'w', encoding='utf-8') as f:
-        json.dump(today_papers, f, indent=2)
-    
-    return today_papers
+        return papers