Use return models for responses.

We should use return_models as soon as we return data, would be best practice. For example see src/automtated_screening/router/get_llm_progress.

This ensures that our data do have the form we expect them to have. Thus this should be used as soon as we return more then a status code I think.