add yi generation module