Initial commit

This commit is contained in:
Dongri Jin
2022-12-12 11:41:43 +09:00
commit c2ce8b4133
12 changed files with 599 additions and 0 deletions

32
examples/completion.rs Normal file
View 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