Update app.py

app.py

@@ -64,54 +64,54 @@ class BasicAgent:
 
     # ── TOOL: Wikipedia ──────────────────────────────────────────────
     def wiki_search(self, query):
-    try:
-        query = query.strip(' ".,')
-        # Step 1: Search for page title
-        search_resp = requests.get(
-            "https://en.wikipedia.org/w/api.php",
-            params={
-                "action": "query",
-                "list": "search", 
-                "srsearch": query,
-                "format": "json",
-                "srlimit": 2
-            },
-            headers={"User-Agent": "GAIA-Agent/1.0"},
-            timeout=15
-        )
-        print(f"    [wiki status] {search_resp.status_code}, len={len(search_resp.text)}")
-        
-        if search_resp.status_code != 200 or not search_resp.text.strip():
-            # Fallback: try REST summary directly
-            slug = query.replace(" ", "_")
+        try:
+            query = query.strip(' ".,')
+            # Step 1: Search for page title
+            search_resp = requests.get(
+                "https://en.wikipedia.org/w/api.php",
+                params={
+                    "action": "query",
+                    "list": "search", 
+                    "srsearch": query,
+                    "format": "json",
+                    "srlimit": 2
+                },
+                headers={"User-Agent": "GAIA-Agent/1.0"},
+                timeout=15
+            )
+            print(f"    [wiki status] {search_resp.status_code}, len={len(search_resp.text)}")
+            
+            if search_resp.status_code != 200 or not search_resp.text.strip():
+                # Fallback: try REST summary directly
+                slug = query.replace(" ", "_")
+                rest = requests.get(
+                    f"https://en.wikipedia.org/api/rest_v1/page/summary/{slug}",
+                    headers={"User-Agent": "GAIA-Agent/1.0"},
+                    timeout=15
+                )
+                if rest.status_code == 200:
+                    data = rest.json()
+                    return f"WIKI [{data.get('title')}]: {data.get('extract','')[:2000]}"
+                return f"Wiki unavailable (status {search_resp.status_code})"
+    
+            data = search_resp.json()
+            results = data.get("query", {}).get("search", [])
+            if not results:
+                return f"No Wikipedia results for: {query}"
+            
+            title = results[0]["title"]
+            # Step 2: Get full extract
             rest = requests.get(
-                f"https://en.wikipedia.org/api/rest_v1/page/summary/{slug}",
+                f"https://en.wikipedia.org/api/rest_v1/page/summary/{requests.utils.quote(title)}",
                 headers={"User-Agent": "GAIA-Agent/1.0"},
                 timeout=15
             )
             if rest.status_code == 200:
-                data = rest.json()
-                return f"WIKI [{data.get('title')}]: {data.get('extract','')[:2000]}"
-            return f"Wiki unavailable (status {search_resp.status_code})"
-
-        data = search_resp.json()
-        results = data.get("query", {}).get("search", [])
-        if not results:
-            return f"No Wikipedia results for: {query}"
-        
-        title = results[0]["title"]
-        # Step 2: Get full extract
-        rest = requests.get(
-            f"https://en.wikipedia.org/api/rest_v1/page/summary/{requests.utils.quote(title)}",
-            headers={"User-Agent": "GAIA-Agent/1.0"},
-            timeout=15
-        )
-        if rest.status_code == 200:
-            d = rest.json()
-            return f"WIKI [{d.get('title')}]: {d.get('extract','')[:2500]}"
-        return f"Wiki fetch failed for: {title}"
-    except Exception as e:
-        return f"Wiki error: {e}"
+                d = rest.json()
+                return f"WIKI [{d.get('title')}]: {d.get('extract','')[:2500]}"
+            return f"Wiki fetch failed for: {title}"
+        except Exception as e:
+            return f"Wiki error: {e}"
 
     def web_search(self, query):
         """General web search using DuckDuckGo"""