mirror of
https://github.com/mii443/openai-api-rs.git
synced 2025-08-22 23:25:39 +00:00
feat: add chain method to create Request instance
This commit is contained in:
@ -4,24 +4,18 @@ 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 req = CompletionRequest::new(
|
||||
completion::GPT3_TEXT_DAVINCI_003.to_string(),
|
||||
String::from("What is Bitcoin?"),
|
||||
)
|
||||
.max_tokens(3000)
|
||||
.temperature(0.9)
|
||||
.top_p(1.0)
|
||||
.stop(vec![String::from(" Human:"), String::from(" AI:")])
|
||||
.presence_penalty(0.6)
|
||||
.frequency_penalty(0.0);
|
||||
|
||||
let result = client.completion(req)?;
|
||||
println!("{:}", result.choices[0].text);
|
||||
|
||||
|
Reference in New Issue
Block a user