Skip to content

Commit c0715c5

Browse files
committed
chore: integrate prompt generation in value conversion
--bug=1056308 --user=ๅˆ˜็‘žๆ–Œ ใ€ๅบ”็”จใ€‘ๅ‡ฝๆ•ฐ็š„่‡ชๅฎšไน‰็ฑปๅž‹ๅ‚ๆ•ฐไธๆ”ฏๆŒ้€š่ฟ‡ๅ˜้‡ๅๅผ•็”จๅ˜้‡ https://www.tapd.cn/62980211/s/1783859
1 parent 9d9ae43 commit c0715c5

File tree

2 files changed

+14
-8
lines changed

2 files changed

+14
-8
lines changed

โ€Žapps/application/flow/step_node/tool_lib_node/impl/base_tool_lib_node.pyโ€Ž

Lines changed: 13 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -93,6 +93,7 @@ def convert_value(name: str, value, _type, is_required, source, node):
9393
return float(value)
9494
return value
9595
try:
96+
value = node.workflow_manage.generate_prompt(value)
9697
if _type == 'int':
9798
return int(value)
9899
if _type == 'float':
@@ -136,14 +137,18 @@ def execute(self, tool_lib_id, input_field_list, **kwargs) -> NodeResult:
136137
workspace_id = self.workflow_manage.get_body().get('workspace_id')
137138
tool_lib = QuerySet(Tool).filter(id=tool_lib_id).first()
138139
valid_function(tool_lib, workspace_id)
139-
params = {field.get('name'): convert_value(field.get('name'), field.get('value'), field.get('type'),
140-
field.get('is_required'),
141-
field.get('source'), self)
142-
for field in
143-
[{'value': get_field_value(input_field_list, field.get('name'), field.get('is_required'),
144-
), **field}
145-
for field in
146-
tool_lib.input_field_list]}
140+
params = {
141+
field.get('name'): convert_value(
142+
field.get('name'), field.get('value'), field.get('type'),
143+
field.get('is_required'),
144+
field.get('source'), self
145+
)
146+
for field in [
147+
{
148+
'value': get_field_value(input_field_list, field.get('name'), field.get('is_required'), ), **field
149+
} for field in tool_lib.input_field_list
150+
]
151+
}
147152

148153
self.context['params'] = params
149154
# ๅˆๅนถๅˆๅง‹ๅŒ–ๅ‚ๆ•ฐ

โ€Žapps/application/flow/step_node/tool_node/impl/base_tool_node.pyโ€Ž

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -79,6 +79,7 @@ def convert_value(name: str, value, _type, is_required, source, node):
7979
return float(value)
8080
return value
8181
try:
82+
value = node.workflow_manage.generate_prompt(value)
8283
if _type == 'int':
8384
return int(value)
8485
if _type == 'float':

0 commit comments

Comments
ย (0)