feat: add examples

Signed-off-by: Xin Liu <[email protected]>

Author: apepkuss

Cargo.toml

@@ -29,3 +29,26 @@ image = []
 [package.metadata.docs.rs]
 all-features = true
 rustdoc-args = ["--cfg", "docsrs"]
+
+[[example]]
+name = "chat"
+path = "examples/chat.rs"
+crate-type = ["bin"]
+
+[[example]]
+name = "transcribe_audio"
+path = "examples/transcribe_audio.rs"
+crate-type = ["bin"]
+required-features = ["audio"]
+
+[[example]]
+name = "translate_audio"
+path = "examples/translate_audio.rs"
+crate-type = ["bin"]
+required-features = ["audio"]
+
+[[example]]
+name = "create_image"
+path = "examples/create_image.rs"
+crate-type = ["bin"]
+required-features = ["image"]

examples/chat.rs

@@ -0,0 +1,31 @@
+use endpoints::chat::{
+    ChatCompletionRequestMessage, ChatCompletionSystemMessage, ChatCompletionUserMessage,
+    ChatCompletionUserMessageContent,
+};
+use llamaedge::{params::ChatParams, Client};
+
+#[tokio::main]
+async fn main() {
+    const SERVER_BASE_URL: &str = "http://localhost:8080";
+
+    // Create a client
+    let client = Client::new(SERVER_BASE_URL).unwrap();
+
+    // create messages
+    let mut messages = Vec::new();
+    let system_message = ChatCompletionRequestMessage::System(ChatCompletionSystemMessage::new(
+        "You are a helpful assistant. Answer questions as concisely and accurately as possible.",
+        None,
+    ));
+    messages.push(system_message);
+    let user_message = ChatCompletionRequestMessage::User(ChatCompletionUserMessage::new(
+        ChatCompletionUserMessageContent::Text("What is the capital of France?".to_string()),
+        None,
+    ));
+    messages.push(user_message);
+
+    // send chat completion request
+    if let Ok(generation) = client.chat(&messages[..], &ChatParams::default()).await {
+        println!("AI response: {}", generation);
+    }
+}

examples/create_image.rs

@@ -0,0 +1,21 @@
+use llamaedge::{params::ImageCreateParams, Client};
+
+#[tokio::main]
+async fn main() {
+    const SERVER_BASE_URL: &str = "http://localhost:8080";
+
+    let client = Client::new(SERVER_BASE_URL).unwrap();
+
+    let image_object_vec = match client
+        .create_image("A lovely dog", ImageCreateParams::default())
+        .await
+    {
+        Ok(image_object_vec) => image_object_vec,
+        Err(e) => {
+            println!("Error: {}", e);
+            return;
+        }
+    };
+
+    println!("{:?}", image_object_vec[0]);
+}

examples/transcribe_audio.rs

@@ -0,0 +1,25 @@
+use llamaedge::{params::TranscriptionParams, Client};
+
+#[tokio::main]
+async fn main() {
+    const SERVER_BASE_URL: &str = "http://localhost:8080";
+
+    let client = Client::new(SERVER_BASE_URL).unwrap();
+
+    let transcription_object = match client
+        .transcribe(
+            "tests/assets/test.wav",
+            "en",
+            TranscriptionParams::default(),
+        )
+        .await
+    {
+        Ok(to) => to,
+        Err(e) => {
+            println!("Error: {}", e);
+            return;
+        }
+    };
+
+    println!("{}", transcription_object.text);
+}

examples/translate_audio.rs

@@ -0,0 +1,25 @@
+use llamaedge::{params::TranslationParams, Client};
+
+#[tokio::main]
+async fn main() {
+    const SERVER_BASE_URL: &str = "http://localhost:8080";
+
+    let client = Client::new(SERVER_BASE_URL).unwrap();
+
+    let translation_object = match client
+        .translate(
+            "tests/assets/test_zh.wav",
+            "zh",
+            TranslationParams::default(),
+        )
+        .await
+    {
+        Ok(to) => to,
+        Err(e) => {
+            println!("Error: {}", e);
+            return;
+        }
+    };
+
+    println!("{}", translation_object.text);
+}