AI/dify
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git synced 2025-08-12 00:58:57 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: improved error handling for spider tool (#6835)

Browse Source
This commit is contained in:
William Espegren 2024-07-31 11:11:52 +02:00 committed by GitHub
parent dd64e65ea0
commit 4be12b29b9
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 9 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
api/core/tools/provider/builtin/spider/spider.py
View File

@ -8,7 +8,13 @@ from core.tools.provider.builtin_tool_provider import BuiltinToolProviderControl
class SpiderProvider(BuiltinToolProviderController): class SpiderProvider(BuiltinToolProviderController):
def _validate_credentials(self, credentials: dict[str, Any]) -> None: def _validate_credentials(self, credentials: dict[str, Any]) -> None:
try: try:
app = Spider(api_key=credentials["spider_api_key"]) app = Spider(api_key=credentials['spider_api_key'])
app.scrape_url(url="https://spider.cloud") app.scrape_url(url='https://spider.cloud')
except AttributeError as e:
# Handle cases where NoneType is not iterable, which might indicate API issues
if 'NoneType' in str(e) and 'not iterable' in str(e):
raise ToolProviderCredentialValidationError('API is currently down, try again in 15 minutes', str(e))
else:
raise ToolProviderCredentialValidationError('An unexpected error occurred.', str(e))
except Exception as e: except Exception as e:
raise ToolProviderCredentialValidationError(str(e)) raise ToolProviderCredentialValidationError('An unexpected error occurred.', str(e))