Skip to content

Commit ad37400

Browse files
committed
fix: safely extract tool_id from tool_result to avoid KeyError
1 parent a4ae441 commit ad37400

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

apps/application/flow/tools.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -400,7 +400,7 @@ async def _yield_mcp_response(chat_model, message_list, mcp_servers, mcp_output_
400400
if tool_id in tool_calls_info:
401401
tool_info = tool_calls_info[tool_id]
402402
tool_result = json.loads(chunk[0].content)
403-
tool_lib_id = tool_result.get('tool_id')
403+
tool_lib_id = tool_result.pop('tool_id') if 'tool_id' in tool_result else None
404404
if tool_lib_id:
405405
await save_tool_record(tool_lib_id, tool_info, tool_result, source_id, source_type)
406406
content = generate_tool_message_complete(

0 commit comments

Comments
 (0)