openai-api-rust-fork

OpenAI API的简单Rust客户端。

有一些便利之处，但主要是API级别。

注意！

这是openai-api的一个分支。这个分支移除了同步和异步功能，只提供异步API。同时，由于http-rs的维护者未响应，因此删除了surf。这个分支现在使用reqwest，由hyper的作者维护。

安装

$ cargo add openai-api-fork

快速入门

#[tokio::main]
async fn main() -> Result<(), Box<dyn std::error::Error>> {
    let api_token = std::env::var("OPENAI_SK")?;
    let client = openai_api_fork::Client::new(&api_token);
    let prompt = String::from("Once upon a time,");
    println!(
        "{}{}",
        prompt,
        client.complete(prompt.as_str()).await?
    );
    Ok(())
}

基本用法

创建完成

对于简单的演示和调试，您可以创建完成，并使用Display实例将Completion对象转换为字符串

let response = client.complete_prompt("Once upon a time").await?;
println!("{}", response);

为了更明确地配置提示，您可以使用CompletionArgs构建器

let args = openai_api_fork::api::CompletionArgs::builder()
        .prompt("Once upon a time,")
        .engine("davinci")
        .max_tokens(20)
        .temperature(0.7)
        .top_p(0.9)
        .stop(vec!["\n".into()]);
let completion = client.complete_prompt(args).await?;
println!("Response: {}", response.choices[0].text);
println!("Model used: {}", response.model);

请参阅 examples/