fix tool icon get failed (#6375)

Co-authored-by: songyawen <songyawen@zkme.xyz>
2 years ago · 218930c897
parent c8f5dfcf17
commit 218930c897
2 changed files with 2 additions and 2 deletions
--- 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 {
--- 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] = {