Parcourir la source

Improve Wikipedia error messages

Refined the exception details in the Wikipedia tool to include the search query when no results are found, enhancing the clarity of error outputs for end-users. This change helps in debugging by indicating the exact query that led to a no-results situation. Additionally, the existing failure-to-connect error message was left as-is, maintaining accurate API connectivity diagnostics.
Kumi il y a 1 an
Parent
commit
a6fca53b51
1 fichiers modifiés avec 1 ajouts et 1 suppressions
  1. 1 1
      src/gptbot/tools/wikipedia.py

+ 1 - 1
src/gptbot/tools/wikipedia.py

@@ -57,6 +57,6 @@ class Wikipedia(BaseTool):
                     elif 'revisions' in page:
                         return f"**{page['title']}**\n{page['revisions'][0]['*']}"
                     else:
-                        raise Exception('No results found.')
+                        raise Exception(f'No results for {query} found in Wikipedia.')
                 else:
                     raise Exception(f'Could not connect to Wikipedia API: {response.status} {response.reason}')