Adding code to enable different document repositories based on training types