浏览代码

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 1 年之前
父节点
当前提交
a6fca53b51
共有 1 个文件被更改,包括 1 次插入1 次删除
  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}')