RemVdH commited on
Commit
faafc81
·
verified ·
1 Parent(s): 2abba01

Update app.py

Browse files

Improved error handling

Files changed (1) hide show
  1. app.py +8 -8
app.py CHANGED
@@ -18,10 +18,10 @@ def search_wikipedia_pages(search_string: str) -> List[str]:
18
  try:
19
  page_titles = wikipedia.search(search_string, results=10)
20
  return page_titles
21
- except wikipedia.exceptions.PageError:
22
- s_error = "PageError"
23
- except wikipedia.exceptions.DisambiguationError:
24
- s_error = "DisambiguationError"
25
  print (f"An error occured {s_error}")
26
 
27
  @tool
@@ -35,10 +35,10 @@ def get_wikipedia_page_content(page_name: str) -> str:
35
  try:
36
  a_page = wikipedia.page(page_name)
37
  return a_page.content
38
- except wikipedia.exceptions.PageError:
39
- s_error = "PageError"
40
- except wikipedia.exceptions.DisambiguationError:
41
- s_error = "DisambiguationError"
42
  print (f"An error occured {s_error}")
43
 
44
  final_answer = FinalAnswerTool()
 
18
  try:
19
  page_titles = wikipedia.search(search_string, results=10)
20
  return page_titles
21
+ except wikipedia.exceptions.PageError as err:
22
+ s_error = f"PageError: {err}"
23
+ except wikipedia.exceptions.DisambiguationError as err:
24
+ s_error = f"DisambiguationError {err}"
25
  print (f"An error occured {s_error}")
26
 
27
  @tool
 
35
  try:
36
  a_page = wikipedia.page(page_name)
37
  return a_page.content
38
+ except wikipedia.exceptions.PageError as err:
39
+ s_error = f"PageError: {err}"
40
+ except wikipedia.exceptions.DisambiguationError as err:
41
+ s_error = f"DisambiguationError {err}"
42
  print (f"An error occured {s_error}")
43
 
44
  final_answer = FinalAnswerTool()