panyy
/
UnisKB
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix: The application export loop node cannot be used (#4268)

v3.2
shaohuzhang1 2025-10-27 17:31:52 +08:00 committed by GitHub
parent aaa0dab747
commit f3bf9e2388
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 16 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
apps/application/serializers/application.py
View File

@ -58,6 +58,15 @@ def get_base_node_work_flow(work_flow):
return None return None
def hand_node(node, update_tool_map):
if node.get('type') == 'tool-lib-node':
tool_lib_id = (node.get('properties', {}).get('node_data', {}).get('tool_lib_id') or '')
node.get('properties', {}).get('node_data', {})['tool_lib_id'] = update_tool_map.get(tool_lib_id,
tool_lib_id)
if node.get('type') == 'search-knowledge-node':
node.get('properties', {}).get('node_data', {})['knowledge_id_list'] = []
class MKInstance: class MKInstance:
def __init__(self, application: dict, function_lib_list: List[dict], version: str, tool_list: List[dict]): def __init__(self, application: dict, function_lib_list: List[dict], version: str, tool_list: List[dict]):
@ -348,9 +357,9 @@ class Query(serializers.Serializer):
application_query_set = application_query_set.order_by("-create_time") application_query_set = application_query_set.order_by("-create_time")
resource_and_folder_query_set = QuerySet(WorkspaceUserResourcePermission).filter( resource_and_folder_query_set = QuerySet(WorkspaceUserResourcePermission).filter(
auth_target_type="APPLICATION", auth_target_type="APPLICATION",
workspace_id=workspace_id, workspace_id=workspace_id,
user_id=user_id) user_id=user_id)
return {'application_query_set': application_query_set, return {'application_query_set': application_query_set,
'workspace_user_resource_permission_query_set': resource_and_folder_query_set, 'workspace_user_resource_permission_query_set': resource_and_folder_query_set,
@ -582,12 +591,10 @@ class ApplicationSerializer(serializers.Serializer):
def to_application(application, workspace_id, user_id, update_tool_map, folder_id): def to_application(application, workspace_id, user_id, update_tool_map, folder_id):
work_flow = application.get('work_flow') work_flow = application.get('work_flow')
for node in work_flow.get('nodes', []): for node in work_flow.get('nodes', []):
if node.get('type') == 'tool-lib-node': hand_node(node, update_tool_map)
tool_lib_id = (node.get('properties', {}).get('node_data', {}).get('tool_lib_id') or '') if node.get('type') == 'loop-node':
node.get('properties', {}).get('node_data', {})['tool_lib_id'] = update_tool_map.get(tool_lib_id, for n in node.get('properties', {}).get('node_data', {}).get('loop_body', {}).get('nodes', []):
tool_lib_id) hand_node(n, update_tool_map)
if node.get('type') == 'search-knowledge-node':
node.get('properties', {}).get('node_data', {})['knowledge_id_list'] = []
return Application(id=uuid.uuid7(), return Application(id=uuid.uuid7(),
user_id=user_id, user_id=user_id,
name=application.get('name'), name=application.get('name'),