Improved context for large modules

Currently, we add docs for all submodules as context. This can quickly get expensive or even run out of context window. We need to do smart context engineering by

using RAG/Summarization/Selective filtering or a combination of these.