feat: add a function to quickly retrieve the context messages for a given message.

feat: add a function to quickly retrieve the context messages for a given message. #1267