wiki issue

app.py

@@ -48,47 +48,30 @@ class KnowledgeBaseTool(Tool):
                 return f"Knowledge base: {info}. Use this context to answer questions about {topic}."
         return f"No specific knowledge base entry for '{topic}'. Use general reasoning."
 
-class ReliableSearchTool(Tool):
-    name = "reliable_search"
-    description = "Search the web for information with built-in retry and fallback mechanisms"
-    inputs = {"query": {"type": "string", "description": "The search query to look up"}}
+class WikipediaSearchTool(Tool):
+    name = "wikipedia_search"
+    description = "Search Wikipedia for information"
+    inputs = {"query": {"type": "string", "description": "The search query for Wikipedia"}}
     output_type = "string"
     
     def __init__(self):
         super().__init__()
-        self.ddg_tool = DuckDuckGoSearchTool()
-        self.max_retries = 2
-        self.timeout = 8
         self.is_initialized = True
         
     def forward(self, query: str) -> str:
-        """Search the web with retry logic and fallbacks."""
-        # Try Wikipedia first (reduced timeout)
+        """Search Wikipedia with simple fallback."""
         try:
             import requests
             wiki_url = "https://en.wikipedia.org/api/rest_v1/page/summary/" + query.replace(" ", "_")
-            response = requests.get(wiki_url, timeout=3)
+            response = requests.get(wiki_url, timeout=2)
             if response.status_code == 200:
                 data = response.json()
                 if 'extract' in data and data['extract']:
-                    return f"Wikipedia: {data['extract']}"
+                    return f"Wikipedia: {data['extract'][:500]}"  # Limit length
         except Exception as e:
             print(f"Wikipedia search failed: {e}")
         
-        # Try Wikidata API as backup
-        try:
-            wikidata_url = f"https://www.wikidata.org/w/api.php?action=wbsearchentities&search={query.replace(' ', '%20')}&language=en&format=json"
-            response = requests.get(wikidata_url, timeout=3)
-            if response.status_code == 200:
-                data = response.json()
-                if 'search' in data and data['search']:
-                    return f"Wikidata: {data['search'][0].get('description', 'No description available')}"
-        except Exception as e:
-            print(f"Wikidata search failed: {e}")
-        
-        # Skip DuckDuckGo entirely due to consistent timeouts
-        # Return knowledge-based response
-        return f"External search unavailable. Based on available knowledge, I'll attempt to answer '{query}' using my training data. Note: This information may not be current."
+        return f"Wikipedia search unavailable for '{query}'. Use your knowledge to answer."
 
 # --- Basic Agent Definition ---
 # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
@@ -114,9 +97,9 @@ class SlpMultiAgent:
         
         # Create only essential agents with reduced complexity
         research_agent = CodeAgent(
-            tools=[KnowledgeBaseTool()],  # Remove search to avoid timeouts
+            tools=[KnowledgeBaseTool(), WikipediaSearchTool()],
             model=model,
-            additional_authorized_imports=["re", "datetime"],
+            additional_authorized_imports=["re", "datetime", "requests"],
             max_steps=2,  # Reduced steps for cost
             name="ResearchAgent",
             verbosity_level=0,
@@ -139,11 +122,11 @@ class SlpMultiAgent:
                 temperature=0.0,
                 max_tokens=500
             ),
-            tools=[KnowledgeBaseTool()],  # Only knowledge base
+            tools=[KnowledgeBaseTool(), WikipediaSearchTool()],
             managed_agents=[research_agent, solver_agent],  # Only 2 agents
             name="ManagerAgent",
             description="Efficient manager for quick problem solving.",
-            additional_authorized_imports=["re", "math"],
+            additional_authorized_imports=["re", "math", "requests"],
             planning_interval=1,  # Faster planning
             verbosity_level=0,  # Reduce verbosity
             max_steps=4,  # Drastically reduced steps
@@ -162,15 +145,16 @@ class SlpMultiAgent:
                     lambda: manager_agent.run(f"""
                     Question: {short_question}
                     
-                    You have ResearchAgent and SolverAgent. Be efficient:
-                    
-                    For factual questions: Use ResearchAgent
-                    For calculations/logic: Use SolverAgent
-                    For simple questions: Answer directly using knowledge_base()
+                    Available tools:
+                    - knowledge_base(topic): Get structured knowledge
+                    - wikipedia_search(query): Search Wikipedia
+                    - ResearchAgent: For factual research
+                    - SolverAgent: For calculations and logic
                     
-                    CRITICAL: Always end with <code>final_answer("answer")</code>
+                    Be efficient and direct. Always end with:
+                    <code>final_answer("your answer")</code>
                     
-                    Be fast and direct - no lengthy reasoning.
+                    No explanations - just the answer.
                     """)
                 )
                 break  # Success, exit retry loop