编写一个API，以查找下一个质数
def next_prime(n):
    '''
    返回数字n之后的下一个质数
    输入：
        n（int）：API应从哪个数字开始
    输出：
        下一个质数
    '''
    num = n + 1
    while True:
        for i in range(2, num):
            if num % i == 0:
                num += 1
                break
        else:
            return num