[TASK] Add query boost support