The US Defense Department says Washington is working with Japan and the rest of its allies to maintain weapons supplies to Kiev.