[TASK] Add a query to multi model and generator to the model