diff --git a/api/core/tools/tool_manager.py b/api/core/tools/tool_manager.py index 3342300eb4..d7ddb40e6b 100644 --- a/api/core/tools/tool_manager.py +++ b/api/core/tools/tool_manager.py @@ -574,7 +574,7 @@ class ToolManager: provider: ApiToolProvider = db.session.query(ApiToolProvider).filter( ApiToolProvider.tenant_id == tenant_id, ApiToolProvider.id == provider_id - ) + ).first() return json.loads(provider.icon) except: return { diff --git a/api/services/app_service.py b/api/services/app_service.py index 069f766793..f88e824b07 100644 --- a/api/services/app_service.py +++ b/api/services/app_service.py @@ -346,7 +346,7 @@ class AppService: try: provider: ApiToolProvider = db.session.query(ApiToolProvider).filter( ApiToolProvider.id == provider_id - ) + ).first() meta['tool_icons'][tool_name] = json.loads(provider.icon) except: meta['tool_icons'][tool_name] = {