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

provide a bit more info in logs when parsing api schema error (#3026)

Browse source
This commit is contained in:
Leo Q 2024-03-30 14:44:50 +08:00 • committed by GitHub
commit fc5ed17fe9
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 52 additions and 84 deletions
Download patch file Download diff file Expand all files Collapse all files

6
api/services/tools_manage_service.py
View file

@ -1,4 +1,5 @@
import json
import logging
from flask import current_app
from httpx import get
@ -24,6 +25,8 @@ from extensions.ext_database import db
from models.tools import ApiToolProvider, BuiltinToolProvider
from services.model_provider_service import ModelProviderService
logger = logging.getLogger(__name__)
class ToolManageService:
@staticmethod
@ -309,6 +312,7 @@ class ToolManageService:
# try to parse schema, avoid SSRF attack
ToolManageService.parser_api_schema(schema)
except Exception as e:
logger.error(f"parse api schema error: {str(e)}")
raise ValueError('invalid schema, please check the url you provided')
return {
@ -655,4 +659,4 @@ class ToolManageService:
except Exception as e:
return { 'error': str(e) }
return { 'result': result or 'empty response' }
return { 'result': result or 'empty response' }