mirror of
https://github.com/mii443/openai-api-rs.git
synced 2025-08-22 23:25:39 +00:00
Initial commit
This commit is contained in:
32
examples/completion.rs
Normal file
32
examples/completion.rs
Normal file
@ -0,0 +1,32 @@
|
||||
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 completion_response = client.completion(req).await?;
|
||||
println!("{:?}", completion_response.choices[0].text);
|
||||
|
||||
Ok(())
|
||||
}
|
||||
|
||||
// cargo run --package openai-rs --example completion
|
Reference in New Issue
Block a user