US to tell partners they must pick sides in AI race with China