POC class

tools/http_search.py

@@ -1,60 +1,77 @@
 from typing import Dict, Any
 from smolagents.tools import Tool
 import requests
-import json
 
 
 class HttpSearchTool(Tool):
     name = "http_search"
     description = (
-        "Appelle une API HTTP de recherche et renvoie un résumé lisible des résultats. "
-        "Configure l'URL de base et le paramètre de requête dans le code."
+        "Effectue une recherche de pages Wikipédia et renvoie une liste de résultats "
+        "lisibles (titre, lien, extrait)."
     )
     inputs = {
         "query": {
             "type": "string",
-            "description": "La requête de recherche à envoyer à l'API HTTP.",
+            "description": "Le terme à rechercher sur Wikipédia.",
         }
     }
     output_type = "string"
 
-    def __init__(self, base_url: str, query_param: str = "q", extra_params: Dict[str, Any] = None, headers: Dict[str, str] = None):
+    def __init__(self, language: str = "en"):
         """
         Args:
-            base_url: URL de l'API de recherche (ex: 'https://api.serpapi.com/search').
-            query_param: nom du paramètre utilisé pour la requête (ex: 'q', 'query', ...).
-            extra_params: dict de paramètres fixes à ajouter à chaque requête (clé API, type de moteur, etc.).
-            headers: headers HTTP à envoyer (optionnels).
+            language: code langue de Wikipédia ('en', 'fr', ...).
         """
         super().__init__()
-        self.base_url = base_url
-        self.query_param = query_param
-        self.extra_params = extra_params or {}
-        self.headers = headers or {}
+        self.language = language
+        self.base_url = f"https://{language}.wikipedia.org/w/api.php"
 
     def forward(self, query: str) -> str:
-        params = {self.query_param: query}
-        params.update(self.extra_params)
+        params = {
+            "action": "opensearch",
+            "search": query,
+            "limit": 5,
+            "namespace": 0,
+            "format": "json",
+        }
 
         try:
-            resp = requests.get(self.base_url, params=params, headers=self.headers, timeout=20)
+            resp = requests.get(self.base_url, params=params, timeout=15)
             resp.raise_for_status()
         except Exception as e:
-            return f"Erreur lors de l'appel HTTP à l'API de recherche : {e}"
+            return f"Erreur lors de l'appel HTTP à Wikipédia : {e}"
 
-        # On suppose une réponse JSON
         try:
             data = resp.json()
         except Exception:
-            # Si ce n'est pas du JSON, on renvoie le texte brut tronqué
-            text = resp.text
-            if len(text) > 1000:
-                text = text[:997] + "..."
-            return f"Réponse non JSON de l'API :\n{text}"
-
-        # Résumé générique, l'agent pourra ensuite analyser le JSON plus finement si besoin
-        pretty = json.dumps(data, indent=2, ensure_ascii=False)
-        if len(pretty) > 2000:
-            pretty = pretty[:1997] + "..."
-
-        return f"Résultats bruts de l'API HTTP pour la requête \"{query}\" :\n{pretty}"
+            return "Impossible de décoder la réponse de Wikipédia en JSON."
+
+        # Format opensearch: [search_term, titles[], descriptions[], urls[]]
+        if len(data) != 4 or not data[1]:
+            return (
+                f'Aucun résultat trouvé sur Wikipédia ({self.language}) '
+                f'pour la requête "{query}".'
+            )
+
+        titles = data[1]
+        descriptions = data[2]
+        urls = data[3]
+
+        lines = [
+            f'Voici quelques résultats Wikipédia ({self.language}) pour la requête "{query}" :',
+            "",
+        ]
+
+        for idx, (title, desc, url) in enumerate(
+            zip(titles, descriptions, urls), start=1
+        ):
+            desc = (desc or "").strip()
+            if len(desc) > 220:
+                desc = desc[:217].rstrip() + "..."
+            lines.append(f"{idx}. {title}")
+            lines.append(f"   Lien : {url}")
+            if desc:
+                lines.append(f"   Résumé : {desc}")
+            lines.append("")
+
+        return "\n".join(lines).strip()