Fix: release WorkflowTool database sessions promptly (#26893)

This commit is contained in:
-LAN-
2025-10-21 15:17:17 +08:00
committed by GitHub
parent fb6f05c267
commit 759a932bb7
12 changed files with 123 additions and 65 deletions

View File

@@ -6,6 +6,7 @@ from faker import Faker
from core.tools.entities.api_entities import ToolProviderApiEntity
from core.tools.entities.common_entities import I18nObject
from core.tools.entities.tool_entities import ToolProviderType
from libs.uuid_utils import uuidv7
from models.tools import ApiToolProvider, BuiltinToolProvider, MCPToolProvider, WorkflowToolProvider
from services.tools.tools_transform_service import ToolTransformService
@@ -66,6 +67,7 @@ class TestToolTransformService:
)
elif provider_type == "workflow":
provider = WorkflowToolProvider(
id=str(uuidv7()),
name=fake.company(),
description=fake.text(max_nb_chars=100),
icon='{"background": "#FF6B6B", "content": "🔧"}',
@@ -758,6 +760,7 @@ class TestToolTransformService:
# Create workflow tool provider
provider = WorkflowToolProvider(
id=str(uuidv7()),
name=fake.company(),
description=fake.text(max_nb_chars=100),
icon='{"background": "#FF6B6B", "content": "🔧"}',