Gotcha! That makes sense. I would recommend looking at LangChain though, as it does a good job at modeling multi-stage learning / inference environments.