Add support for Google Gemini

.gitignore

@@ -18,3 +18,4 @@ Cargo.lock
 # Jetbrains
 .idea
 *.iml
+/.vscode

Cargo.toml

@@ -11,13 +11,18 @@ keywords = ["ai", "machine-learning", "openai", "library"]
 [dependencies]
 serde_json = "1.0.94"
 derive_builder = "0.20.0"
-reqwest = { version = "0.12", default-features = false, features = ["json", "stream", "multipart"], optional = true }
+reqwest = { version = "0.12", default-features = false, features = [
+    "json",
+    "stream",
+    "multipart",
+], optional = true }
 serde = { version = "1.0.157", features = ["derive"] }
 reqwest-eventsource = "0.6"
 tokio = { version = "1.26.0", features = ["full"] }
 anyhow = "1.0.70"
 futures-util = "0.3.28"
 bytes = "1.4.0"
+tracing = "0.1.41"
 
 [dev-dependencies]
 dotenvy = "0.15.7"

examples/chat_cli.rs

@@ -1,6 +1,6 @@
 use dotenvy::dotenv;
 use openai::{
-    chat::{ChatCompletion, ChatCompletionMessage, ChatCompletionMessageRole},
+    chat::{ChatCompletion, ChatCompletionMessage, ChatCompletionMessageRole, Content},
     Credentials,
 };
 use std::io::{stdin, stdout, Write};
@@ -13,7 +13,7 @@ async fn main() {
 
     let mut messages = vec![ChatCompletionMessage {
         role: ChatCompletionMessageRole::System,
-        content: Some("You are a large language model built into a command line interface as an example of what the `openai` Rust library made by Valentine Briese can do.".to_string()),
+        content: Some(Content::new_str("You are a large language model built into a command line interface as an example of what the `openai` Rust library made by Valentine Briese can do.")),
         ..Default::default()
     }];
 
@@ -26,7 +26,7 @@ async fn main() {
         stdin().read_line(&mut user_message_content).unwrap();
         messages.push(ChatCompletionMessage {
             role: ChatCompletionMessageRole::User,
-            content: Some(user_message_content),
+            content: Some(Content::new_str(&user_message_content)),
             ..Default::default()
         });
 
@@ -40,7 +40,7 @@ async fn main() {
         println!(
             "{:#?}: {}",
             &returned_message.role,
-            &returned_message.content.clone().unwrap().trim()
+            &returned_message.content.clone().unwrap()
         );
 
         messages.push(returned_message);

examples/chat_simple.rs

@@ -1,6 +1,6 @@
 use dotenvy::dotenv;
 use openai::{
-    chat::{ChatCompletion, ChatCompletionMessage, ChatCompletionMessageRole},
+    chat::{ChatCompletion, ChatCompletionMessage, ChatCompletionMessageRole, Content},
     Credentials,
 };
 
@@ -13,12 +13,12 @@ async fn main() {
     let messages = vec![
         ChatCompletionMessage {
             role: ChatCompletionMessageRole::System,
-            content: Some("You are a helpful assistant.".to_string()),
+            content: Some(Content::new_str("You are a helpful assistant.")),
             ..Default::default()
         },
         ChatCompletionMessage {
             role: ChatCompletionMessageRole::User,
-            content: Some("Tell me a random crab fact".to_string()),
+            content: Some(Content::new_str("Tell me a random crab fact")),
             ..Default::default()
         },
     ];
@@ -32,6 +32,6 @@ async fn main() {
     println!(
         "{:#?}: {}",
         returned_message.role,
-        returned_message.content.unwrap().trim()
+        returned_message.content.unwrap()
     );
 }

examples/chat_stream_cli.rs

@@ -1,5 +1,5 @@
 use dotenvy::dotenv;
-use openai::chat::{ChatCompletion, ChatCompletionDelta};
+use openai::chat::{ChatCompletion, ChatCompletionDelta, Content};
 use openai::{
     chat::{ChatCompletionMessage, ChatCompletionMessageRole},
     Credentials,
@@ -15,7 +15,9 @@ async fn main() {
 
     let mut messages = vec![ChatCompletionMessage {
         role: ChatCompletionMessageRole::System,
-        content: Some("You're an AI that replies to each message verbosely.".to_string()),
+        content: Some(Content::new_str(
+            "You're an AI that replies to each message verbosely.",
+        )),
         ..Default::default()
     }];
 
@@ -28,7 +30,7 @@ async fn main() {
         stdin().read_line(&mut user_message_content).unwrap();
         messages.push(ChatCompletionMessage {
             role: ChatCompletionMessageRole::User,
-            content: Some(user_message_content),
+            content: Some(Content::new_str(&user_message_content)),
             ..Default::default()
         });