Возможно ли динамически подменять image в Dockerfile при использовании multi-stage?

У меня есть основной репозиторий проекта и отдельный репозиторий с тестами. Они билдятся в разные образы с разными тэгами. В связи с некоторыми ограничениями они я не могу запустить их, как разные образы с помощью docker-compose. Они обязательной должны быть одним образом. Я понял, что это можно сделать с помощью multi-stage, но не могу понять, можно ли при этом при билде подменять, допустим, тэг образов.