mirror of
https://github.com/mii443/openai-api-rs.git
synced 2025-08-22 15:15:34 +00:00
35 lines
1.1 KiB
Rust
35 lines
1.1 KiB
Rust
use openai_api_rs::v1::api::Client;
|
|
use openai_api_rs::v1::chat_completion::{self, ChatCompletionRequest};
|
|
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 = ChatCompletionRequest {
|
|
model: chat_completion::GPT4.to_string(),
|
|
messages: vec![chat_completion::ChatCompletionMessage {
|
|
role: chat_completion::MessageRole::user,
|
|
content: String::from("What is Bitcoin?"),
|
|
name: None,
|
|
function_call: None,
|
|
}],
|
|
functions: None,
|
|
function_call: None,
|
|
temperature: None,
|
|
top_p: None,
|
|
n: None,
|
|
stream: None,
|
|
stop: None,
|
|
max_tokens: None,
|
|
presence_penalty: None,
|
|
frequency_penalty: None,
|
|
logit_bias: None,
|
|
user: None,
|
|
};
|
|
let result = client.chat_completion(req).await?;
|
|
println!("{:?}", result.choices[0].message.content);
|
|
Ok(())
|
|
}
|
|
|
|
// OPENAI_API_KEY=xxxx cargo run --package openai-api-rs --example chat_completion
|