fix

app.py

@@ -35,95 +35,36 @@ except Exception as e:
 
 # --- Core Tools ---
 
-def web_search(query: str) -> str:
-    """Web search with fallbacks"""
-    try:
-        time.sleep(random.uniform(0.5, 1.5))
-        
-        # Try Serper API if available
-        serper_key = os.getenv("SERPER_API_KEY")
-        if serper_key:
-            try:
-                url = "https://google.serper.dev/search"
-                payload = json.dumps({"q": query, "num": 5})
-                headers = {
-                    'X-API-KEY': serper_key,
-                    'Content-Type': 'application/json'
-                }
-                response = requests.post(url, headers=headers, data=payload, timeout=10)
-                
-                if response.status_code == 200:
-                    data = response.json()
-                    results = []
-                    
-                    # Get direct answer if available
-                    if 'answerBox' in data:
-                        answer = data['answerBox'].get('answer', '')
-                        if answer:
-                            results.append(answer)
-                    
-                    # Get knowledge graph info
-                    if 'knowledgeGraph' in data:
-                        kg = data['knowledgeGraph']
-                        title = kg.get('title', '')
-                        desc = kg.get('description', '')
-                        if title and desc:
-                            results.append(f"{title}: {desc}")
-                    
-                    # Get organic results
-                    if 'organic' in data:
-                        for item in data['organic'][:2]:
-                            title = item.get('title', '')
-                            snippet = item.get('snippet', '')
-                            if title and snippet:
-                                results.append(f"{title} | {snippet}")
-                    
-                    if results:
-                        return " | ".join(results[:2])  # Return top 2 most relevant
-                        
-            except Exception as e:
-                print(f"Serper API failed: {e}")
-        
-        # Fallback to Wikipedia
-        return wikipedia_search(query)
-        
-    except Exception as e:
-        return f"Search error: {str(e)}"
-
+@tool
 def wikipedia_search(query: str) -> str:
-    """Wikipedia search"""
-    try:
-        clean_query = re.sub(r'[^a-zA-Z0-9 ]', '', query)[:100]
-        
-        params = {
-            'action': 'query',
-            'format': 'json',
-            'list': 'search',
-            'srsearch': clean_query,
-            'srlimit': 3,
-            'srprop': 'snippet'
-        }
-        
-        response = requests.get(
-            "https://en.wikipedia.org/w/api.php",
-            params=params,
-            timeout=8,
-            headers={'User-Agent': 'GAIA-Agent/1.0'}
-        )
-        
-        if response.status_code == 200:
-            data = response.json()
-            
-            for item in data.get('query', {}).get('search', []):
-                title = item.get('title', '')
-                snippet = re.sub(r'<[^>]+>', '', item.get('snippet', ''))
-                if title and snippet:
-                    return f"{title}: {snippet}"
-        
-        return f"No Wikipedia results for: {clean_query}"
+    """Search Wikipedia for a query and return maximum 2 results.
     
-    except Exception as e:
-        return f"Wikipedia error: {str(e)}"
+    Args:
+        query: The search query."""
+    search_docs = WikipediaLoader(query=query, load_max_docs=2).load()
+    formatted_search_docs = "\n\n---\n\n".join(
+        [
+            f'<Document source="{doc.metadata["source"]}" page="{doc.metadata.get("page", "")}"/>\n{doc.page_content}\n</Document>'
+            for doc in search_docs
+        ])
+    return {"wiki_results": formatted_search_docs}
+
+@tool
+def web_search(query: str) -> str:
+    """Search Tavily for a query and return maximum 3 results.
+    
+    Args:
+        query: The search query."""
+    search_docs = TavilySearchResults(max_results=3).invoke(query=query)
+    formatted_search_docs = "\n\n---\n\n".join(
+        [
+            f'<Document source="{doc.metadata["source"]}" page="{doc.metadata.get("page", "")}"/>\n{doc.page_content}\n</Document>'
+            for doc in search_docs
+        ])
+    return {"web_results": formatted_search_docs}
+
+
+
 
 def extract_youtube_info(url: str) -> str:
     """Extract YouTube video information"""