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

This commit is contained in:
Leo Q
2024-03-30 14:44:50 +08:00
committed by GitHub
parent 94d04934b3
commit fc5ed17fe9
2 changed files with 52 additions and 84 deletions

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' }