How to use AI in a large codebase if it still a Jr Engineer?

The idea here is to breakdown code into stateless function blocks that become a sandbox for AI to work on. By breaking down the entire project in such blocks, AI could be more usuful because it needs less context to work.