fix: support int64 indices for embedding (#4447)