fix: Application task parameters

Author: shaohuzhang1

apps/trigger/handler/impl/task/application_task.py

@@ -6,16 +6,17 @@
     @date：2026/1/14 19:14
     @desc:
 """
+import json
+from functools import reduce
 
 import uuid_utils.compat as uuid
 from django.db.models import QuerySet
 
-from application.models import ChatUserType, Chat, ChatRecord, ChatSourceChoices
+from application.models import ChatUserType, Chat, ChatRecord, ChatSourceChoices, Application
 from chat.serializers.chat import ChatSerializers
 from knowledge.models.knowledge_action import State
-
 from trigger.handler.base_task import BaseTriggerTask
-from trigger.models import TaskRecord
+from trigger.models import TaskRecord, TriggerTask
 
 
 def get_reference(fields, obj):
@@ -28,32 +29,82 @@ def get_reference(fields, obj):
     return obj
 
 
-def get_field_value(value, kwargs):
+def conversion_custom_value(value, _type):
+    if ['array', 'dict', 'float', 'int', 'boolean'].__contains__(_type):
+        return json.loads(value)
+    if _type == 'any':
+        try:
+            return json.loads(value)
+        except Exception as e:
+            pass
+    return value
+
+
+def valid_value_type(value, _type):
+    if _type == 'array':
+        return isinstance(value, list)
+    if _type == 'dict':
+        return isinstance(value, dict)
+    if _type == 'float':
+        return isinstance(value, float)
+    if _type == 'int':
+        return isinstance(value, int)
+    if _type == 'boolean':
+        return isinstance(value, bool)
+    return isinstance(value, str)
+
+
+def get_field_value(value, kwargs, _type, required, default_value, field):
     source = value.get('source')
     if source == 'custom':
-        return value.get('value')
+        _value = value.get('value')
+        if _value:
+            _value = conversion_custom_value(_value, _type)
+        else:
+            if default_value:
+                return default_value
+            if required:
+                raise Exception(f'{field} is required')
+            else:
+                return None
     else:
-        return get_reference(value.get('value'), kwargs)
+        _value = get_reference(value.get('value'), kwargs)
+    valid = valid_value_type(_value, _type)
+    if not valid:
+        raise Exception(f'{field} type error')
+    return _value
 
 
-def get_application_execute_parameters(parameter_setting, kwargs):
+def get_application_execute_parameters(parameter_setting, application_parameters_setting, kwargs):
+    many_field = ['api_input_field_list', 'user_input_field_list']
     parameters = {'form_data': {}}
-    question_setting = parameter_setting.get('question')
-    if question_setting:
-        parameters['message'] = get_field_value(question_setting, kwargs)
-    filed_list = ['image_list', 'document_list', 'audio_list', 'video_list', 'other_list']
-    for field in filed_list:
-        field_setting = parameter_setting.get(field)
-        if field_setting:
-            parameters[field] = get_field_value(field_setting, kwargs)
-    api_input_field_list = parameter_setting.get('api_input_field_list')
-    if api_input_field_list:
-        for key, value in api_input_field_list.items():
-            parameters['form_data'][key] = get_field_value(value, kwargs)
-    user_input_field_list = parameter_setting.get('user_input_field_list')
-    if user_input_field_list:
-        for key, value in user_input_field_list.items():
-            parameters['form_data'][key] = get_field_value(value, kwargs)
+    for key, value in application_parameters_setting.items():
+        setting = parameter_setting.get(key)
+        if setting:
+            if many_field.__contains__(key):
+                for ck, cv in value.items():
+                    _setting = setting.get(ck)
+                    if _setting:
+                        _value = get_field_value(_setting, kwargs, cv.get('type'), cv.get('required'),
+                                                 cv.get('default_value'), cv.get('field'))
+                        parameters['form_data'][ck] = _value
+                    else:
+                        if cv.get('default_value'):
+                            parameters['form_data'][ck] = cv.get('default_value')
+                        else:
+                            if cv.get('required'):
+                                raise Exception(f'{ck} is required')
+            else:
+                value = get_field_value(setting, kwargs, value.get('type'), value.get('required'),
+                                        value.get('default_value'), value.get('field'))
+                parameters['message' if key == 'question' else key] = value
+        else:
+            if value.get('default_value'):
+                parameters['message' if key == 'question' else key] = value.get('default_value')
+            else:
+                if value.get('required'):
+                    raise Exception(f'{"message" if key == "question" else key} is required')
+
     return parameters
 
 
@@ -76,13 +127,63 @@ def get_workflow_state(details):
     return State.SUCCESS
 
 
+def get_user_field_component_input_type(input_type):
+    if input_type == "MultiRow":
+        return 'array'
+    if input_type == "SwitchInput":
+        return 'boolean'
+    return 'string'
+
+
+def get_application_parameters_setting(application):
+    application_parameter_setting = {'question': {
+        'required': True,
+        'type': 'string'
+    }}
+    if application.type == 'SIMPLE':
+        return application_parameter_setting
+    else:
+        base_node_list = [n for n in application.work_flow.get('nodes') if n.get('type') == "base-node"]
+        if len(base_node_list) == 0:
+            raise Exception('错误的应用工作流信息')
+        base_node = base_node_list[0]
+        api_input_field_list = base_node.get('properties').get('api_input_field_list') or []
+        api_input_field_list = {user_field.get('variable'): {
+            'required': user_field.get('is_required'),
+            'default_value': user_field.get('default_value'),
+            'type': 'string'
+        } for user_field in api_input_field_list}
+        user_input_field_list = base_node.get('properties').get('user_input_field_list') or []
+        user_input_field_list = {user_field.get('field'): {
+            'required': user_field.get('required'),
+            'default_value': user_field.get('default_value'),
+            'type': get_user_field_component_input_type(user_field.get('input_type'))
+        } for user_field in user_input_field_list}
+        application_parameter_setting['api_input_field_list'] = api_input_field_list
+        application_parameter_setting['user_input_field_list'] = user_input_field_list
+        node_data = base_node.get('properties').get('node_data') or {}
+        file_upload_enable = node_data.get('file_upload_enable')
+        if file_upload_enable:
+            file_upload_setting = node_data.get('file_upload_setting') or {}
+            for field in ['audio', 'document', 'image', 'other', 'video']:
+                v = file_upload_setting.get(field)
+                if v:
+                    application_parameter_setting[field] = {'required': False, 'default_value': [], 'type': 'array'}
+        return application_parameter_setting
+
+
 class ApplicationTask(BaseTriggerTask):
     def support(self, trigger_task, **kwargs):
         return trigger_task.get('source_type') == 'APPLICATION'
 
     def execute(self, trigger_task, **kwargs):
         parameter_setting = trigger_task.get('parameter')
-        parameters = get_application_execute_parameters(parameter_setting, kwargs)
+        application = QuerySet(Application).filter(id=trigger_task.get('source_id')).only('type', 'work_flow').first()
+        if application is None:
+            QuerySet(TriggerTask).filter(id=trigger_task.get('id')).delete()
+            return
+        application_parameters_setting = get_application_parameters_setting(application)
+        parameters = get_application_execute_parameters(parameter_setting, application_parameters_setting, kwargs)
         parameters['re_chat'] = False
         parameters['stream'] = True
         chat_id = uuid.uuid7()