Update app.py
Browse filesImproved error handling
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()
|