如何更改IME选项(Return/Go/Done键盘按钮)在alertDialog编辑文本中？

AlertDialog.Builder builder = new AlertDialog.Builder(context);
builder.setTitle("编辑文本");
builder.setMessage("请输入文本内容：");

// 设置自定义布局文件
LayoutInflater inflater = LayoutInflater.from(context);
View view = inflater.inflate(R.layout.dialog_edit_text, null);
builder.setView(view);

EditText editText = view.findViewById(R.id.edit_text);
editText.setImeOptions(EditorInfo.IME_ACTION_DONE); // 设置IME选项为Done

editText.setOnEditorActionListener(new TextView.OnEditorActionListener() {
    @Override
    public boolean onEditorAction(TextView v, int actionId, KeyEvent event) {
        if (actionId == EditorInfo.IME_ACTION_DONE) {
            // 执行保存文本的操作
            String text = editText.getText().toString();
            // ...

            // 关闭对话框
            alertDialog.dismiss();
            return true;
        }
        return false;
    }
});

AlertDialog alertDialog = builder.create();
alertDialog.show();