后，其家属被宋羁縻随州的时候，元昊派间谍矫宋朝的诏令释放他们，人未有知者，后乃闻西人临境作乐，迎此二家而去。入夏后颇得信任，元昊称帝建国后不久，即任命张元为中书令，之后吴昊也获重用。
任官.西夏天授礼法延祚四年（1041年）好水川之战中张元辅助元昊大败韩琦等人率领的宋军，宋军阵亡七万多人，张元在界上寺墙壁上题诗一首讥讽宋人：「夏竦何曾耸，韩琦未足奇。满川龙虎辇，犹自说兵机。」又在诗后题言讥讽韩琦，署名时写了一大串官衔：「（西夏）太师、尚书令、兼中书令张元随大驾至此。」同年官至国相。