"He knows that she’s done," a source tells PEOPLE about Kanye West and Kim Kardashian, who is working with top divorce lawyer Laura Wasser