Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -38,11 +38,24 @@ model = HFModelWrapper(model_id=MODEL_ID, token=HF_TOKEN)
|
|
| 38 |
@tool
|
| 39 |
def search_wikipedia(search_term: str, results: int = 5) -> str:
|
| 40 |
"""
|
| 41 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 42 |
"""
|
| 43 |
-
import wikipedia
|
| 44 |
search_results = wikipedia.search(search_term, results=results)
|
| 45 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 46 |
|
| 47 |
|
| 48 |
@tool
|
|
|
|
| 38 |
@tool
|
| 39 |
def search_wikipedia(search_term: str, results: int = 5) -> str:
|
| 40 |
"""
|
| 41 |
+
Search Wikipedia for a given term and return a summary of results.
|
| 42 |
+
|
| 43 |
+
Args:
|
| 44 |
+
search_term (str): The term or topic to search for on Wikipedia.
|
| 45 |
+
results (int): The number of search results to return. Defaults to 5.
|
| 46 |
+
|
| 47 |
+
Returns:
|
| 48 |
+
str: A summary text of the top Wikipedia search results.
|
| 49 |
"""
|
|
|
|
| 50 |
search_results = wikipedia.search(search_term, results=results)
|
| 51 |
+
summaries = []
|
| 52 |
+
for title in search_results:
|
| 53 |
+
try:
|
| 54 |
+
page = wikipedia.page(title)
|
| 55 |
+
summaries.append(f"### {title}\n{page.summary[:400]}...")
|
| 56 |
+
except Exception:
|
| 57 |
+
pass
|
| 58 |
+
return "\n\n".join(summaries)
|
| 59 |
|
| 60 |
|
| 61 |
@tool
|