Update tools.py

tools.py

@@ -1,30 +1,26 @@
-# tools.py
-"""Tool set add-on per smolagents: Wikipedia e arXiv."""
+# tools.py  –  Wikipedia + arXiv helpers
 from smolagents import Tool
 import wikipedia, arxiv
 
+MAX_CHARS = 20_000   # evita prompt enormi
+
 class WikipediaTool(Tool):
     name = "wiki"
-    description = "Return full content of the most relevant Wikipedia page for the query."
+    description = "Return full content (truncated) of the most relevant Wikipedia page."
 
     def __call__(self, query: str) -> str:
         try:
             page = wikipedia.page(query, auto_suggest=True, redirect=True)
-            return page.content
-        except (wikipedia.PageError, wikipedia.DisambiguationError) as e:
-            return f"WIKI_ERROR: {e}"
+            return page.content[:MAX_CHARS]
+        except (wikipedia.PageError, wikipedia.DisambiguationError):
+            return ""
 
 class ArxivTool(Tool):
     name = "arxiv"
-    description = "Return title and abstract of the most recent arXiv paper matching the query."
+    description = "Return title + abstract of the most recent arXiv paper that matches the query."
 
     def __call__(self, query: str) -> str:
-        search = arxiv.Search(
-            query=query,
-            max_results=1,
-            sort_by=arxiv.SortCriterion.SubmittedDate
-        )
+        search = arxiv.Search(query=query, max_results=1,
+                              sort_by=arxiv.SortCriterion.SubmittedDate)
         paper = next(search.results(), None)
-        if paper:
-            return f"{paper.title}\n{paper.summary}"
-        return "ARXIV_NOTHING_FOUND"
+        return f"{paper.title}\n{paper.summary}" if paper else ""