Skip to content

Commit f808db5

Browse files
committed
fix: Tool execution parameters
1 parent 0634a4f commit f808db5

File tree

3 files changed

+10
-21
lines changed

3 files changed

+10
-21
lines changed

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

Lines changed: 5 additions & 19 deletions
Original file line numberDiff line numberDiff line change
@@ -35,24 +35,10 @@ def get_field_value(value, kwargs):
3535
return get_reference(value.get('value'), kwargs)
3636

3737

38-
def get_application_execute_parameters(parameter_setting, kwargs):
39-
parameters = {'form_data': {}}
40-
question_setting = parameter_setting.get('question')
41-
if question_setting:
42-
parameters['message'] = get_field_value(question_setting, kwargs)
43-
filed_list = ['image_list', 'document_list', 'audio_list', 'video_list', 'other_list']
44-
for field in filed_list:
45-
field_setting = parameter_setting.get(field)
46-
if field_setting:
47-
parameters[field] = get_field_value(field_setting, kwargs)
48-
api_input_field_list = parameter_setting.get('api_input_field_list')
49-
if api_input_field_list:
50-
for key, value in api_input_field_list.items():
51-
parameters['form_data'][key] = get_field_value(value, kwargs)
52-
user_input_field_list = parameter_setting.get('user_input_field_list')
53-
if user_input_field_list:
54-
for key, value in user_input_field_list.items():
55-
parameters['form_data'][key] = get_field_value(value, kwargs)
38+
def get_tool_execute_parameters(parameter_setting, kwargs):
39+
parameters = {}
40+
for key, value in parameter_setting.items():
41+
parameters[key] = get_field_value(value, kwargs)
5642
return parameters
5743

5844

@@ -81,7 +67,7 @@ def support(self, trigger_task, **kwargs):
8167

8268
def execute(self, trigger_task, **kwargs):
8369
parameter_setting = trigger_task.get('parameter')
84-
parameters = get_application_execute_parameters(parameter_setting, kwargs)
70+
parameters = get_tool_execute_parameters(parameter_setting, kwargs)
8571
tool_id = trigger_task.get('source_id')
8672
task_record_id = uuid.uuid7()
8773

apps/trigger/handler/simple_tools.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -9,10 +9,11 @@
99
from threading import Thread
1010

1111
from trigger.handler.impl.task.application_task import ApplicationTask
12+
from trigger.handler.impl.task.tool_task import ToolTask
1213
from trigger.handler.impl.trigger.event_trigger import EventTrigger
1314
from trigger.handler.impl.trigger.scheduled_trigger import ScheduledTrigger
1415

15-
simple_task_handlers = [ApplicationTask()]
16+
simple_task_handlers = [ApplicationTask(), ToolTask()]
1617

1718
simple_trigger_handlers = [ScheduledTrigger(), EventTrigger()]
1819

apps/trigger/sql/get_trigger_task_record_page_list.sql

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -20,8 +20,10 @@ WITH source_data_cte AS (SELECT 'APPLICATION' as source_type,
2020
"folder_id"
2121
FROM tool)
2222
select ett.*,
23+
ett.meta::json as meta,
2324
sdc.name as source_name,
24-
sdc.icon as source_icon
25+
sdc.icon as source_icon,
26+
sdc.type as type
2527
from event_trigger_task_record ett
2628
left join source_data_cte sdc
2729
on ett.source_id = sdc.id and ett.source_type = sdc.source_type

0 commit comments

Comments
 (0)