mirror of
https://github.com/mii443/openai-api-rs.git
synced 2025-08-22 15:15:34 +00:00
Add vision
This commit is contained in:
@ -10,7 +10,7 @@ fn main() -> Result<(), Box<dyn std::error::Error>> {
|
||||
GPT4.to_string(),
|
||||
vec![chat_completion::ChatCompletionMessage {
|
||||
role: chat_completion::MessageRole::user,
|
||||
content: String::from("What is Bitcoin?"),
|
||||
content: chat_completion::Content::Text(String::from("What is bitcoin?")),
|
||||
name: None,
|
||||
}],
|
||||
);
|
||||
|
@ -31,7 +31,7 @@ fn main() -> Result<(), Box<dyn std::error::Error>> {
|
||||
GPT3_5_TURBO_0613.to_string(),
|
||||
vec![chat_completion::ChatCompletionMessage {
|
||||
role: chat_completion::MessageRole::user,
|
||||
content: String::from("What is the price of Ethereum?"),
|
||||
content: chat_completion::Content::Text(String::from("What is the price of Ethereum?")),
|
||||
name: None,
|
||||
}],
|
||||
)
|
||||
|
@ -31,7 +31,7 @@ fn main() -> Result<(), Box<dyn std::error::Error>> {
|
||||
GPT3_5_TURBO_0613.to_string(),
|
||||
vec![chat_completion::ChatCompletionMessage {
|
||||
role: chat_completion::MessageRole::user,
|
||||
content: String::from("What is the price of Ethereum?"),
|
||||
content: chat_completion::Content::Text(String::from("What is the price of Ethereum?")),
|
||||
name: None,
|
||||
}],
|
||||
)
|
||||
@ -83,15 +83,17 @@ fn main() -> Result<(), Box<dyn std::error::Error>> {
|
||||
vec![
|
||||
chat_completion::ChatCompletionMessage {
|
||||
role: chat_completion::MessageRole::user,
|
||||
content: String::from("What is the price of Ethereum?"),
|
||||
content: chat_completion::Content::Text(String::from(
|
||||
"What is the price of Ethereum?",
|
||||
)),
|
||||
name: None,
|
||||
},
|
||||
chat_completion::ChatCompletionMessage {
|
||||
role: chat_completion::MessageRole::function,
|
||||
content: {
|
||||
content: chat_completion::Content::Text({
|
||||
let price = get_coin_price(&coin);
|
||||
format!("{{\"price\": {}}}", price)
|
||||
},
|
||||
}),
|
||||
name: Some(String::from("get_coin_price")),
|
||||
},
|
||||
],
|
||||
|
39
examples/vision.rs
Normal file
39
examples/vision.rs
Normal file
@ -0,0 +1,39 @@
|
||||
use openai_api_rs::v1::api::Client;
|
||||
use openai_api_rs::v1::chat_completion::{self, ChatCompletionRequest};
|
||||
use openai_api_rs::v1::common::GPT4_VISION_PREVIEW;
|
||||
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 = ChatCompletionRequest::new(
|
||||
GPT4_VISION_PREVIEW.to_string(),
|
||||
vec![chat_completion::ChatCompletionMessage {
|
||||
role: chat_completion::MessageRole::user,
|
||||
content: chat_completion::Content::ImageUrl(vec![
|
||||
chat_completion::ImageUrl {
|
||||
r#type: chat_completion::ContentType::text,
|
||||
text: Some(String::from("What’s in this image?")),
|
||||
image_url: None,
|
||||
},
|
||||
chat_completion::ImageUrl {
|
||||
r#type: chat_completion::ContentType::image_url,
|
||||
text: None,
|
||||
image_url: Some(chat_completion::ImageUrlType {
|
||||
url: String::from(
|
||||
"https://upload.wikimedia.org/wikipedia/commons/5/50/Bitcoin.png",
|
||||
),
|
||||
}),
|
||||
},
|
||||
]),
|
||||
name: None,
|
||||
}],
|
||||
);
|
||||
|
||||
let result = client.chat_completion(req)?;
|
||||
println!("{:?}", result.choices[0].message.content);
|
||||
|
||||
Ok(())
|
||||
}
|
||||
|
||||
// OPENAI_API_KEY=xxxx cargo run --package openai-api-rs --example chat_completion
|
Reference in New Issue
Block a user