openai-api-rust

OpenAI API的简单Rust客户端。

提供一些便利，但主要在API级别。

安装

$ cargo add openai-api

快速入门

#[tokio::main]
async fn main() -> Result<(), Box<dyn std::error::Error>> {
    let api_token = std::env::var("OPENAI_SK")?;
    let client = openai_api::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::api::CompletionArgs::builder()
        .prompt("Once upon a time,")
        .engine(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/