Skip to content

Commit a2c7fda

Browse files
committed
feat: update parameter handling in tool.py and shared_tool.py for improved code generation
1 parent fe7e42c commit a2c7fda

2 files changed

Lines changed: 6 additions & 7 deletions

File tree

apps/tools/serializers/tool.py

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1134,14 +1134,13 @@ def generate_code(self):
11341134
init_field_list = self.data.get('init_field_list')
11351135
input_field_list = self.data.get('input_field_list')
11361136

1137-
init_params = list({i["field"]: i.get('default_value') for i in init_field_list}.keys())
1138-
input_params = list({field.get('name'): field.get('value') for field in input_field_list}.keys())
1139-
11401137
message = messages[-1]['content']
11411138
q = prompt.replace(
11421139
"{userInput}", message
11431140
).replace(
1144-
"{params}", ','.join(init_params + input_params)
1141+
"{initFieldList}", json.dumps(init_field_list)
1142+
).replace(
1143+
"{inputFieldList}", json.dumps(input_field_list)
11451144
)
11461145

11471146
messages[-1]['content'] = q

ui/src/views/tool/component/GenerateCodeDialog.vue

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -177,9 +177,9 @@ const promptTemplates = {
177177
请为 MaxKB 工具 生成 Python 代码,需求如下:
178178
179179
- 核心功能:用户输入的主题 / 功能需求
180-
- 启动参数:平台配置的启动参数,如 API 密钥、数据库地址、账号密码等不需要默认值
181-
- 输入参数:平台配置的输入参数,格式为 “参数名:数据类型:是否必填”不需要默认值
182-
- 函数定义:列举所有启动参数和输入参数并声明返回类型 {params}
180+
- 启动参数:平台配置的启动参数,如 API 密钥、数据库地址、账号密码等, 已声明参数:{initFieldList}
181+
- 输入参数:平台配置的输入参数,已声明参数:{inputFieldList}
182+
- 函数定义:列举所有启动参数和输入参数并声明返回类型
183183
- 输出要求:代码需接收输入参数,启动参数完成业务逻辑,仅输出函数定义
184184
`,
185185
}

0 commit comments

Comments
 (0)