Skip to content

Commit b47599e

Browse files
committed
feat: enhance parameter handling in tool.py for improved code generation
1 parent 27fe6e2 commit b47599e

1 file changed

Lines changed: 4 additions & 3 deletions

File tree

apps/tools/serializers/tool.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1134,13 +1134,14 @@ 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+
11371140
message = messages[-1]['content']
11381141
q = prompt.replace(
11391142
"{userInput}", message
11401143
).replace(
1141-
"{initFieldList}", json.dumps(init_field_list)
1142-
).replace(
1143-
"{inputFieldList}", json.dumps(input_field_list)
1144+
"{params}", ','.join(init_params + input_params)
11441145
)
11451146

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

0 commit comments

Comments
 (0)