up

app.py

@@ -73,6 +73,23 @@ def find_non_commutative_elements_from_table(table_markdown: str) -> str:
 
 # Định nghĩa các tools dùng
 
+@tool
+def wiki_search(query: str) -> str:
+    """Searches Wikipedia for a given query and returns a summary of the content."""
+    import wikipedia
+    try:
+        summary = wikipedia.summary(query, sentences=3, auto_suggest=False, redirect=True)
+        return summary
+    except wikipedia.exceptions.PageError:
+        return f"No Wikipedia page found for '{{query}}'."
+    except wikipedia.exceptions.DisambiguationError as e:
+        if e.options:
+            return f"Wikipedia search for '{{query}}' is ambiguous. Options include: {{', '.join(e.options[:3])}}..."
+        return f"Wikipedia search for '{{query}}' led to a disambiguation page with no clear options."
+    except Exception as e:
+        return f"An error occurred during Wikipedia search: {{str(e)}}"
+
+
 tools = [
     multiply,
     divide,