gpt3_macro

Rust 宏，使用 GPT3 codex 在编译时生成代码。

只需描述您想要的功能，并（可选）定义一个函数头。宏将在编译时为您生成源代码。

它还可以为您生成测试。（见示例 3）

示例 1

create_function!("checks if number is prime" fn is_prime(num: u64) -> bool);

通常将扩展为类似

// A rust function that checks if number is prime
fn is_prime(num: u64) -> bool {
    if num == 2 {
        return true;
    }
    if num % 2 == 0 {
        return false;
    }
    let mut i = 3;
    while i * i <= num {
        if num % i == 0 {
            return false;
        }
        i += 2;
    }
    true
}

示例 2

create_function!("prints n elements of the fibonnacci sequence to stdout" fn fib(n: u64));

有时扩展为

// prints n elements of the fibonnacci sequence to stdout
fn fib (n : u64) {
    let mut a = 0;
    let mut b = 1;
    let mut c = 0;
    for _ in 0..n {
        c = a + b;
        a = b;
        b = c;
        println!("{}", c);
    }
}

示例 3：代码和测试生成

create_function_and_tests!("fizzbuzz", fn fizzbuzz(n: u64) -> String)

通常扩展为

// A rust function that fizzbuzz
fn fizzbuzz (n : u64) -> String {
    if n % 15 == 0 {
        "FizzBuzz".to_string()
    } else if n % 3 == 0 {
        "Fizz".to_string()
    } else if n % 5 == 0 {
        "Buzz".to_string()
    } else {
        n.to_string()
    }
}
// 5 tests for the function
#[test]
fn test_fizzbuzz_1() {
    assert_eq!(fizzbuzz(1), "1");
}
#[test]
fn test_fizzbuzz_2() {
    assert_eq!(fizzbuzz(2), "2");
}
#[test]
fn test_fizzbuzz_3() {
    assert_eq!(fizzbuzz(3), "Fizz");
}
#[test]
fn test_fizzbuzz_4() {
    assert_eq!(fizzbuzz(4), "4");
}
#[test]
fn test_fizzbuzz_5() {
    assert_eq!(fizzbuzz(5), "Buzz");
}

优点和缺点

安装

在 OpenAI 账户页面 生成 OpenAI API 密钥，并设置 $OPENAI_KEY 环境变量

然后执行

cargo add gpt3_macro

或手动添加

gpt3_macro = "0.3.1"

到您的 Cargo.toml