web search fall back

app.py

@@ -111,7 +111,7 @@ def open_youtube_video(url: str, query: str) -> str:
     return response.text
 
 def google_search(query: str) -> str:
-    '''Performs a web search for the given query using DuckDuckGo.'''
+    '''Performs a web search for the given query using DuckDuckGo, and falls back to Wikipedia if no results are found.'''
     try:
         # Add delay between requests to avoid rate limiting
         time.sleep(1)
@@ -175,6 +175,30 @@ def google_search(query: str) -> str:
             if quoted_data.get("Answer"):
                 result.append(f"Answer: {quoted_data['Answer']}")
         
+        # If still no results, try Wikipedia as a fallback
+        if not result:
+            try:
+                wiki_response = requests.get(
+                    "https://en.wikipedia.org/w/api.php",
+                    params={
+                        "action": "query",
+                        "format": "json",
+                        "prop": "extracts",
+                        "exintro": True,
+                        "explaintext": True,
+                        "titles": query
+                    },
+                    timeout=10
+                )
+                wiki_data = wiki_response.json()
+                pages = wiki_data.get("query", {}).get("pages", {})
+                for page in pages.values():
+                    extract = page.get("extract")
+                    if extract:
+                        result.append(extract)
+            except Exception as e:
+                logger.error(f"Wikipedia fallback error for query {query}: {str(e)}")
+        
         return "\n".join(result) if result else "No results found."
         
     except requests.exceptions.Timeout:
@@ -508,6 +532,10 @@ class BasicAgent:
                             # Add tool result to messages
                             messages.append(AIMessage(content=f"Tool result: {result}"))
                             
+                            # Loop breaker: If the result is 'No results found.', end with 'Not found'
+                            if isinstance(result, str) and "no results found" in result.lower():
+                                return {"messages": [AIMessage(content="Not found")], "next": END}
+                            
                             # If this was a google_search, analyze the results
                             if tool_name == "google_search":
                                 # Add a prompt to analyze the search results