Refactor AI Prompt form element to have '#tree' => TRUE
The related 1.x issue showed the problem of having form element with `#tree => FALSE`. It means that wherever the element is used the form parents should always be set manually. In 2.x it is allowed to do a breaking changes, so we should try to implement the form element in other way to avoid setting the parents manually all the time.
issue