Skip to content

Commit 750e4ff

Browse files
committed
fix: simplify tool result handling and improve JSON processing
1 parent 4756710 commit 750e4ff

File tree

1 file changed

+5
-9
lines changed

1 file changed

+5
-9
lines changed

apps/application/flow/tools.py

Lines changed: 5 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -675,26 +675,22 @@ def _upsert_fragment(key, raw_id, func_name, part_args):
675675
chunk[0].content) > 0 else {}
676676
else:
677677
tool_result = {}
678-
text = tool_result.pop(
679-
'text') if 'text' in tool_result else None
678+
text = tool_result.get('text') if 'text' in tool_result else None
680679
text_result = json.loads(text) if text else tool_result
681680
if text:
682-
tool_lib_id = text_result.pop(
683-
'tool_id') if 'tool_id' in text_result else None
681+
tool_lib_id = text_result.pop('tool_id') if 'tool_id' in text_result else None
684682
else:
685-
tool_lib_id = tool_result.pop(
686-
'tool_id') if 'tool_id' in tool_result else None
683+
tool_lib_id = tool_result.pop('tool_id') if 'tool_id' in tool_result else None
687684
if tool_lib_id:
688685
await save_tool_record(tool_lib_id, tool_info, tool_result, source_id, source_type)
689-
tool_result = json.dumps(
690-
text_result, ensure_ascii=False)
686+
tool_result = json.dumps(text_result, ensure_ascii=False)
691687
except Exception as e:
692688
tool_result = chunk[0].content
693689
content = generate_tool_message_complete(
694690
tool_info.get('icon', ''),
695691
tool_info['name'],
696692
tool_info['input'],
697-
tool_result + '\n\n'
693+
tool_result
698694
)
699695
chunk[0].content = content
700696
else:

0 commit comments

Comments
 (0)