Procházet zdrojové kódy

fix tool icon get failed (#6375)

Co-authored-by: songyawen <songyawen@zkme.xyz>
Songyawn před 9 měsíci
rodič
revize
218930c897
2 změnil soubory, kde provedl 2 přidání a 2 odebrání
  1. 1 1
      api/core/tools/tool_manager.py
  2. 1 1
      api/services/app_service.py

+ 1 - 1
api/core/tools/tool_manager.py

@@ -574,7 +574,7 @@ class ToolManager:
                 provider: ApiToolProvider = db.session.query(ApiToolProvider).filter(
                 provider: ApiToolProvider = db.session.query(ApiToolProvider).filter(
                     ApiToolProvider.tenant_id == tenant_id,
                     ApiToolProvider.tenant_id == tenant_id,
                     ApiToolProvider.id == provider_id
                     ApiToolProvider.id == provider_id
-                )
+                ).first()
                 return json.loads(provider.icon)
                 return json.loads(provider.icon)
             except:
             except:
                 return {
                 return {

+ 1 - 1
api/services/app_service.py

@@ -346,7 +346,7 @@ class AppService:
                     try:
                     try:
                         provider: ApiToolProvider = db.session.query(ApiToolProvider).filter(
                         provider: ApiToolProvider = db.session.query(ApiToolProvider).filter(
                             ApiToolProvider.id == provider_id
                             ApiToolProvider.id == provider_id
-                        )
+                        ).first()
                         meta['tool_icons'][tool_name] = json.loads(provider.icon)
                         meta['tool_icons'][tool_name] = json.loads(provider.icon)
                     except:
                     except:
                         meta['tool_icons'][tool_name] = {
                         meta['tool_icons'][tool_name] = {