gptrs

一个OpenAI代码生成CLI。

它向OpenAI API发送请求，解析输出并将代码生成到文件中。

这是一个主要用于帮助自己学习Rust和熟悉OpenAI API的小项目，一些初始代码是使用ChatGPT的帮助生成的！

使用文件和提示进行代码补全

gptrs completion --prompt=<PROMPT> --input=<PATH/TO/FILE> --output=</PATH/TO/OUTPUT>

示例

运行以下命令生成一个包含斐波那契序列的Python文件

gptrs completion --prompt="generate a fibonacci sequence in Python" --output=./fibonacci.py

这将生成一个Python文件

然后运行一个命令来重构此文件中的代码并输出到下面的新文件

gptrs completion --prompt="Refactor this code" --input=./fibonacci.py --output=./fibonacci_refactor.py 

您还可以设置不同的模型和温度

gptrs completion --prompt="Write unit tests for this fibonacci function in Python" --input=./fibonacci_refactor.py --output=./fibonacci_tests.py --model=code-davinci-002 --temperature=0

设置

这假设您有OpenAI开发者账户

生成API令牌

查看此处获取说明

生成令牌后，设置环境变量

export OPENAI_API_KEY={API_KEY}

通过Cargo安装

cargo install gptrs