completion/
completion.rs

1use openai_api_rs::v1::api::OpenAIClient;
2use openai_api_rs::v1::completion::{self, CompletionRequest};
3use std::env;
4
5#[tokio::main]
6async fn main() -> Result<(), Box<dyn std::error::Error>> {
7    let api_key = env::var("OPENAI_API_KEY").unwrap().to_string();
8    let mut client = OpenAIClient::builder().with_api_key(api_key).build()?;
9
10    let req = CompletionRequest::new(
11        completion::GPT3_TEXT_DAVINCI_003.to_string(),
12        String::from("What is Bitcoin?"),
13    )
14    .max_tokens(3000)
15    .temperature(0.9)
16    .top_p(1.0)
17    .stop(vec![String::from(" Human:"), String::from(" AI:")])
18    .presence_penalty(0.6)
19    .frequency_penalty(0.0);
20
21    let result = client.completion(req).await?;
22    println!("{:}", result.choices[0].text);
23
24    Ok(())
25}
26
27// OPENAI_API_KEY=xxxx cargo run --package openai-api-rs --example completion
completion/completion.rs

completion/
completion.rs
