Help with ansible.controller.controller_api using return_all=True

Hi all,
I need some help with formatting this correctly. I had this all working fine, until one of the workflow_jobs had more the 25 workflow_nodes. In that case, I am not finding the erroring node because it comes in the later pages.

I saw the the controller_api does have a return_all parameter. I tried adding it as below, but the get the error shown.

Task:

- name: Query API for failed jobs
  ansible.builtin.set_fact:
    failed_job_wf_nodes: >-
      {{ query('ansible.controller.controller_api',
              'workflow_jobs/' ~ failed_job_wf_id ~ '/workflow_nodes/'),
                  return_all=True }}

Error:

msg: >-
  template error while templating string: expected token 'end of print
  statement', got '='. String: {{ query('ansible.controller.controller_api',
          'workflow_jobs/' ~ failed_job_wf_id ~ '/workflow_nodes/'),
              return_all=True }}. expected token 'end of print statement', got '='
_ansible_no_log: false

I am guessing from the error that I don’t have the return_all=True in the correct location or format.

Any help greatly appreciated,
Cheers!