jyapayne/dify
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

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
commit 4be12b29b9
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 9 additions and 3 deletions
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))