mirror of
https://github.com/mii443/openai-api-rs.git
synced 2025-08-22 15:15:34 +00:00
32 lines
991 B
Rust
32 lines
991 B
Rust
use openai_api_rs::v1::api::Client;
|
|
use openai_api_rs::v1::completion::{self, CompletionRequest};
|
|
use std::env;
|
|
|
|
fn main() -> Result<(), Box<dyn std::error::Error>> {
|
|
let client = Client::new(env::var("OPENAI_API_KEY").unwrap().to_string());
|
|
let req = CompletionRequest {
|
|
model: completion::GPT3_TEXT_DAVINCI_003.to_string(),
|
|
prompt: String::from("What is Bitcoin?"),
|
|
suffix: None,
|
|
max_tokens: Some(3000),
|
|
temperature: Some(0.9),
|
|
top_p: Some(1.0),
|
|
n: None,
|
|
stream: None,
|
|
logprobs: None,
|
|
echo: None,
|
|
stop: Some(vec![String::from(" Human:"), String::from(" AI:")]),
|
|
presence_penalty: Some(0.6),
|
|
frequency_penalty: Some(0.0),
|
|
best_of: None,
|
|
logit_bias: None,
|
|
user: None,
|
|
};
|
|
let result = client.completion(req)?;
|
|
println!("{:}", result.choices[0].text);
|
|
|
|
Ok(())
|
|
}
|
|
|
|
// OPENAI_API_KEY=xxxx cargo run --package openai-api-rs --example completion
|