Youssef-El-SaYed commited on
Commit
cd44f7e
·
verified ·
1 Parent(s): 6dde09a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +16 -3
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
- Retrieve Wikipedia article titles for a given topic.
 
 
 
 
 
 
 
42
  """
43
- import wikipedia
44
  search_results = wikipedia.search(search_term, results=results)
45
- return search_results
 
 
 
 
 
 
 
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