Add shortcuts for commonly used functions

examples/system_instruction.rs

@@ -18,17 +18,10 @@ async fn main() -> Result<(), Box<dyn std::error::Error>> {
     let system_instruction = "Answer as if you were Yoda";
     let prompt = "What is the airspeed of an unladen swallow?";
 
-    let request = GenerateContentRequest {
+    let request = GenerateContentRequest::builder()
-        contents: vec![Content {
+        .add_text_content(Role::User, prompt)
-            role: Some(Role::User),
+        .system_instruction_text(system_instruction)
-            parts: Some(vec![Part::Text(prompt.to_string())]),
+        .build();
-        }],
-        system_instruction: Some(Content {
-            role: None,
-            parts: Some(vec![Part::Text(system_instruction.to_string())]),
-        }),
-        ..Default::default()
-    };
 
     let result = gemini
         .generate_content(&request, "gemini-1.0-pro-002")

examples/text-from-text-streaming.rs

@@ -16,12 +16,7 @@ async fn main() -> Result<(), Box<dyn std::error::Error>> {
 
     let prompt = "Tell me the story of the genesis of the universe as a bedtime story.";
     let request = GenerateContentRequest::builder()
-        .add_content(
+        .add_text_content(Role::User, prompt)
-            Content::builder()
-                .role(Role::User)
-                .add_part(Part::Text(prompt.to_string()))
-                .build(),
-        )
         .build();
 
     let queue = gemini.stream_generate_content(&request, "gemini-pro").await;

examples/text-from-text.rs

@@ -16,12 +16,7 @@ async fn main() -> Result<(), Box<dyn std::error::Error>> {
 
     let prompt = "What is the airspeed of an unladen swallow?";
     let request = GenerateContentRequest::builder()
-        .add_content(
+        .add_text_content(Role::User, prompt)
-            Content::builder()
-                .role(Role::User)
-                .add_part(Part::Text(prompt.to_string()))
-                .build(),
-        )
         .build();
     let response = gemini.generate_content(&request, "gemini-pro").await?;
     println!("Response: {:?}", response.candidates[0].get_text().unwrap());

src/types/common.rs

@@ -37,6 +37,10 @@ impl ContentBuilder {
         }
     }
 
+    pub fn add_text_part<T: Into<String>>(self, text: T) -> Self {
+        self.add_part(Part::Text(text.into()))
+    }
+
     pub fn add_part(mut self, part: Part) -> Self {
         match &mut self.content.parts {
             Some(parts) => parts.push(part),

src/types/generate_content.rs

@@ -3,7 +3,7 @@ use std::collections::HashMap;
 use serde::{Deserialize, Serialize};
 use serde_json::Value;
 
-use super::{Content, VertexApiError};
+use super::{Content, Role, VertexApiError};
 use crate::error::Result;
 
 #[derive(Clone, Default, Serialize, Deserialize)]
@@ -37,6 +37,11 @@ impl GenerateContentRequestBuilder {
         }
     }
 
+    pub fn add_text_content<T: Into<String>>(self, role: Role, text: T) -> Self {
+        let content = Content::builder().role(role).add_text_part(text).build();
+        self.add_content(content)
+    }
+
     pub fn add_content(mut self, content: Content) -> Self {
         self.request.contents.push(content);
         self
@@ -57,6 +62,10 @@ impl GenerateContentRequestBuilder {
         self
     }
 
+    pub fn system_instruction_text<T: Into<String>>(self, text: T) -> Self {
+        self.system_instruction(Content::builder().add_text_part(text).build())
+    }
+
     pub fn system_instruction(mut self, system_instruction: Content) -> Self {
         self.request.system_instruction = Some(system_instruction);
         self