Fix

app.py

@@ -13,6 +13,7 @@ from urllib.parse import urlparse, parse_qs
 
 # --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
+WIKIPEDIA_API_KEY = os.getenv("WIKIPEDIA_API_KEY", "default_key")  # Fallback key if needed
 
 # --- Enhanced Tools with Rate Limiting and Better Answers ---
 
@@ -78,6 +79,67 @@ def smart_web_search(query: str) -> str:
     except Exception as e:
         return f"Search error: {str(e)}"
 
+@tool
+def get_wikipedia_info(query: str) -> str:
+    """
+    Enhanced Wikipedia search with API key support and better result parsing.
+    
+    Args:
+        query: Search query string
+        
+    Returns:
+        Formatted Wikipedia information
+    """
+    try:
+        # Clean the query
+        clean_query = re.sub(r'[^a-zA-Z0-9 ]', '', query)[:100]
+        
+        # First try the Wikipedia API with our key
+        params = {
+            'action': 'query',
+            'format': 'json',
+            'list': 'search',
+            'srsearch': clean_query,
+            'srlimit': 3,
+            'srprop': 'snippet',
+            'utf8': 1
+        }
+        
+        if WIKIPEDIA_API_KEY and WIKIPEDIA_API_KEY != "default_key":
+            params['apikey'] = WIKIPEDIA_API_KEY
+            
+        response = requests.get(
+            "https://en.wikipedia.org/w/api.php",
+            params=params,
+            timeout=10
+        )
+        
+        if response.status_code == 200:
+            data = response.json()
+            results = []
+            
+            for item in data.get('query', {}).get('search', []):
+                title = item.get('title', '')
+                snippet = re.sub(r'<[^>]+>', '', item.get('snippet', ''))
+                results.append(f"TITLE: {title}\nSNIPPET: {snippet}")
+            
+            if results:
+                return "\n\n".join(results)
+        
+        # Fallback to page extracts for exact matches
+        page_title = clean_query.replace(' ', '_')
+        extract_url = f"https://en.wikipedia.org/api/rest_v1/page/summary/{page_title}"
+        extract_response = requests.get(extract_url, timeout=8)
+        
+        if extract_response.status_code == 200:
+            extract_data = extract_response.json()
+            return f"TITLE: {extract_data.get('title', '')}\nEXTRACT: {extract_data.get('extract', '')}"
+        
+        return f"No Wikipedia results found for: {clean_query}"
+    
+    except Exception as e:
+        return f"Wikipedia search error: {str(e)}"
+
 @tool
 def extract_youtube_details(url: str) -> str:
     """
@@ -299,67 +361,6 @@ def solve_advanced_math(problem: str) -> str:
     except Exception as e:
         return f"Math solver error: {str(e)}"
 
-@tool
-def get_detailed_wikipedia(topic: str) -> str:
-    """
-    Get detailed Wikipedia information with better parsing.
-    
-    Args:
-        topic: Wikipedia topic to search
-    
-    Returns:
-        Detailed Wikipedia information
-    """
-    try:
-        time.sleep(1)  # Rate limiting
-        
-        # Clean topic
-        topic_clean = topic.replace(" ", "_").strip()
-        
-        # Try direct page access
-        summary_url = f"https://en.wikipedia.org/api/rest_v1/page/summary/{topic_clean}"
-        response = requests.get(summary_url, timeout=12)
-        
-        if response.status_code == 200:
-            data = response.json()
-            results = []
-            results.append(f"TITLE: {data.get('title', '')}")
-            results.append(f"EXTRACT: {data.get('extract', '')}")
-            
-            # Get page URL for more details
-            page_url = data.get('content_urls', {}).get('desktop', {}).get('page', '')
-            if page_url:
-                results.append(f"URL: {page_url}")
-            
-            return "\n".join(results)
-        
-        # Fallback to search API
-        search_url = "https://en.wikipedia.org/w/api.php"
-        params = {
-            "action": "query",
-            "format": "json",
-            "list": "search",
-            "srsearch": topic,
-            "srlimit": 5
-        }
-        
-        search_response = requests.get(search_url, params=params, timeout=12)
-        if search_response.status_code == 200:
-            search_data = search_response.json()
-            
-            results = []
-            for item in search_data.get('query', {}).get('search', [])[:3]:
-                title = item['title']
-                snippet = re.sub(r'<[^>]+>', '', item['snippet'])
-                results.append(f"TITLE: {title}\nSNIPPET: {snippet}")
-            
-            return "\n\n".join(results) if results else "No Wikipedia results found"
-        
-        return f"Wikipedia lookup failed for: {topic}"
-        
-    except Exception as e:
-        return f"Wikipedia error: {str(e)}"
-
 # --- Optimized Agent Class ---
 class OptimizedGAIAAgent:
     def __init__(self):
@@ -367,16 +368,17 @@ class OptimizedGAIAAgent:
         
         self.tools = [
             smart_web_search,
+            get_wikipedia_info,
             extract_youtube_details,
             decode_reversed_text,
-            solve_advanced_math,
-            get_detailed_wikipedia
+            solve_advanced_math
         ]
         
         # Initialize CodeAgent with better error handling
         try:
             self.agent = CodeAgent(
                 tools=self.tools,
+                model="gpt-3.5-turbo",  # Added required model parameter
                 additional_authorized_imports=["math", "re", "json", "time"]
             )
             print("✅ CodeAgent initialized")
@@ -410,11 +412,11 @@ class OptimizedGAIAAgent:
         
         # Wikipedia-focused searches
         if any(term in question_lower for term in ["who", "what", "when", "where", "wikipedia", "article"]):
-            return get_detailed_wikipedia(question)
+            return get_wikipedia_info(question)
         
         # Olympics questions
         if "olympics" in question_lower or "1928" in question:
-            return get_detailed_wikipedia("1928 Summer Olympics")
+            return get_wikipedia_info("1928 Summer Olympics")
         
         # Default to smart search with delay
         return smart_web_search(question)
@@ -591,7 +593,7 @@ if __name__ == "__main__":
     print("🎯 Starting Optimized GAIA Agent...")
     
     # Environment check
-    env_vars = ["SPACE_ID", "SERPER_API_KEY"]
+    env_vars = ["SPACE_ID", "SERPER_API_KEY", "WIKIPEDIA_API_KEY"]
     for var in env_vars:
         status = "✅" if os.getenv(var) else "⚠️"
         print(f"{status} {var}")