web_search update

app.py

@@ -1,4 +1,6 @@
 from smolagents import CodeAgent, HfApiModel, load_tool, tool
+from typing import Any, Optional
+from smolagents.tools import Tool
 import datetime
 import pytz
 import yaml
@@ -37,6 +39,67 @@ def get_current_time_in_timezone(timezone: str) -> str:
     except Exception as e:
         return f"Erreur pour le fuseau '{timezone}' : {str(e)}"
 
+class DuckDuckGoSearchTool(Tool):
+    name = "web_search"
+    description = (
+        "Effectue une recherche web DuckDuckGo et renvoie un court résumé en français "
+        "des principaux résultats."
+    )
+    inputs = {
+        "query": {
+            "type": "string",
+            "description": "La requête de recherche à effectuer.",
+        }
+    }
+    output_type = "string"
+
+    def __init__(self, max_results: int = 5, **kwargs):
+        super().__init__()
+        self.max_results = max_results
+        try:
+            from duckduckgo_search import DDGS
+        except ImportError as e:
+            raise ImportError(
+                "Vous devez installer le paquet `duckduckgo-search` pour utiliser ce tool "
+                "(ex: `pip install duckduckgo-search`)."
+            ) from e
+        self.ddgs = DDGS(**kwargs)
+
+    def forward(self, query: str) -> str:
+        # Récupération des résultats bruts
+        results = self.ddgs.text(query, max_results=self.max_results)
+
+        # Aucun résultat trouvé → message clair plutôt qu'exception
+        if not results:
+            return (
+                f'Aucun résultat trouvé pour la requête "{query}". '
+                "Essaie une formulation plus courte ou plus générale."
+            )
+
+        # On ne garde que les premiers résultats les plus utiles
+        top_results = results[:3]
+
+        # Construction d'un résumé lisible en français
+        lines = [f'Voici quelques résultats pour la requête "{query}" :', ""]
+        for idx, r in enumerate(top_results, start=1):
+            title = r.get("title", "").strip()
+            url = r.get("href", "").strip()
+            snippet = r.get("body", "").strip()
+
+            # On coupe les extraits trop longs
+            if len(snippet) > 220:
+                snippet = snippet[:217].rstrip() + "..."
+
+            lines.append(f"{idx}. {title or '(titre manquant)'}")
+            if url:
+                lines.append(f"   Lien : {url}")
+            if snippet:
+                lines.append(f"   Résumé : {snippet}")
+            lines.append("")  # ligne vide entre les résultats
+
+        return "\n".join(lines).strip()
+
+
 
 # ==== Instanciation des tools existants ====