Skip to content

Commit cc02d79

Browse files
committed
chore: enhance tool parameter handling with JSON decryption
1 parent a7c09a5 commit cc02d79

File tree

1 file changed

+9
-1
lines changed

1 file changed

+9
-1
lines changed

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

Lines changed: 9 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,13 +6,15 @@
66
@date:2026/1/14 19:14
77
@desc:
88
"""
9+
import json
910
import time
1011
import traceback
1112

1213
import uuid_utils.compat as uuid
1314
from django.db.models import QuerySet
1415

1516
from common.utils.logger import maxkb_logger
17+
from common.utils.rsa_util import rsa_long_decrypt
1618
from common.utils.tool_code import ToolExecutor
1719
from knowledge.models.knowledge_action import State
1820
from tools.models import Tool
@@ -128,9 +130,15 @@ def execute(self, trigger_task, **kwargs):
128130
try:
129131
tool = QuerySet(Tool).filter(id=tool_id).first()
130132
parameters = get_tool_execute_parameters(tool.input_field_list, parameter_setting, kwargs)
133+
init_params_default_value = {i["field"]: i.get('default_value') for i in tool.init_field_list}
131134

135+
if tool.init_params is not None:
136+
all_params = init_params_default_value | json.loads(rsa_long_decrypt(tool.init_params)) | parameters
137+
else:
138+
all_params = init_params_default_value | parameters
132139
executor = ToolExecutor()
133-
result = executor.exec_code(tool.code, parameters)
140+
result = executor.exec_code(tool.code, all_params)
141+
134142
maxkb_logger.info(f"Tool execution result: {result}")
135143

136144
QuerySet(TaskRecord).filter(id=task_record_id).update(

0 commit comments

Comments
 (0)