mirror of
https://github.com/mii443/openai-api-rs.git
synced 2025-08-22 15:15:34 +00:00
33 lines
990 B
Rust
33 lines
990 B
Rust
use openai_rs::v1::completion::{self, CompletionRequest};
|
||
use openai_rs::v1::api::Client;
|
||
use std::env;
|
||
|
||
#[tokio::main]
|
||
async 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: Some(String::from("NFTとは？")),
|
||
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).await?;
|
||
println!("{:}", result.choices[0].text);
|
||
|
||
Ok(())
|
||
}
|
||
|
||
// cargo run --package openai-rs --example completion
|