Merge pull request #4 from M97Chahboun/main

Adds GenerationConfig struct to enhance content generation options

Author: josh-tracey

Cargo.lock

@@ -1,6 +1,6 @@
 [package]
 name = "gemini_client_rs"
-version = "0.3.0"
+version = "0.4.0"
 edition = "2021"
 description = "A Rust client for Google Gemini API"
 license = "MIT"

Cargo.toml

@@ -0,0 +1,63 @@
+use dotenvy::dotenv;
+use gemini_client_rs::{
+    types::{GenerateContentRequest, PartResponse},
+    GeminiClient,
+};
+use serde_json::json;
+
+#[tokio::main]
+async fn main() -> Result<(), Box<dyn std::error::Error>> {
+    dotenv().ok();
+
+    let api_key = std::env::var("GEMINI_API_KEY").expect("GEMINI_API_KEY must be set");
+
+    let client = GeminiClient::new(api_key.to_string());
+    let model_name = "gemini-1.5-flash";
+
+    let req_json = json!({
+        "contents": [
+            {
+                "parts": [
+                    {
+                        "text": "Generate a happy greeting message"
+                    }
+                ],
+                "role": "user"
+            }
+        ],
+        "generationConfig": {
+            "responseMimeType": "application/json",
+            "responseSchema": {
+                "type": "object",
+                "properties": {
+                    "emoji": {
+                        "type": "string",
+                        "description": "A single emoji representing the mood of the message"
+                    },
+                    "message": {
+                        "type": "string",
+                        "description": "A greeting message"
+                    }
+                },
+                "required": ["emoji", "message"]
+            }
+        }
+    });
+
+    let request: GenerateContentRequest = serde_json::from_value(req_json)?;
+
+    let response = client.generate_content(model_name, &request).await?;
+
+    if let Some(candidates) = response.candidates {
+        for candidate in &candidates {
+            for part in &candidate.content.parts {
+                match part {
+                    PartResponse::Text(text) => println!("Response: {}", text),
+                    _ => println!("Unexpected response type"),
+                }
+            }
+        }
+    }
+
+    Ok(())
+}

examples/generation_config.rs

@@ -120,4 +120,3 @@ impl GeminiClient {
         }
     }
 }
-

src/lib.rs

@@ -19,6 +19,8 @@ pub struct GenerateContentRequest {
     pub system_instruction: Option<Content>,
     pub contents: Vec<Content>,
     pub tools: Option<Vec<ToolConfig>>,
+    #[serde(rename = "generationConfig")]
+    pub generation_config: Option<GenerationConfig>,
 }
 
 #[derive(Debug, Serialize, Deserialize)]
@@ -50,6 +52,49 @@ pub struct Content {
     pub role: Role,
 }
 
+#[derive(Debug, Serialize, Deserialize)]
+pub struct GenerationConfig {
+    #[serde(rename = "stopSequences", skip_serializing_if = "Option::is_none")]
+    pub stop_sequences: Option<Vec<String>>,
+    #[serde(rename = "responseMimeType", skip_serializing_if = "Option::is_none")]
+    pub response_mime_type: Option<String>,
+    #[serde(rename = "responseSchema", skip_serializing_if = "Option::is_none")]
+    pub response_schema: Option<serde_json::Value>,
+    #[serde(rename = "responseModalities", skip_serializing_if = "Option::is_none")]
+    pub response_modalities: Option<Vec<String>>,
+    #[serde(rename = "candidateCount", skip_serializing_if = "Option::is_none")]
+    pub candidate_count: Option<i32>,
+    #[serde(rename = "maxOutputTokens", skip_serializing_if = "Option::is_none")]
+    pub max_output_tokens: Option<i32>,
+    #[serde(rename = "temperature", skip_serializing_if = "Option::is_none")]
+    pub temperature: Option<f64>,
+    #[serde(rename = "topP", skip_serializing_if = "Option::is_none")]
+    pub top_p: Option<f64>,
+    #[serde(rename = "topK", skip_serializing_if = "Option::is_none")]
+    pub top_k: Option<i32>,
+    #[serde(rename = "seed", skip_serializing_if = "Option::is_none")]
+    pub seed: Option<i64>,
+    #[serde(rename = "presencePenalty", skip_serializing_if = "Option::is_none")]
+    pub presence_penalty: Option<f64>,
+    #[serde(rename = "frequencyPenalty", skip_serializing_if = "Option::is_none")]
+    pub frequency_penalty: Option<f64>,
+    #[serde(rename = "responseLogprobs", skip_serializing_if = "Option::is_none")]
+    pub response_logprobs: Option<bool>,
+    #[serde(rename = "logprobs", skip_serializing_if = "Option::is_none")]
+    pub logprobs: Option<i32>,
+    #[serde(
+        rename = "enableEnhancedCivicAnswers",
+        skip_serializing_if = "Option::is_none"
+    )]
+    pub enable_enhanced_civic_answers: Option<bool>,
+    #[serde(rename = "speechConfig", skip_serializing_if = "Option::is_none")]
+    pub speech_config: Option<serde_json::Value>,
+    #[serde(rename = "thinkingConfig", skip_serializing_if = "Option::is_none")]
+    pub thinking_config: Option<serde_json::Value>,
+    #[serde(rename = "mediaResolution", skip_serializing_if = "Option::is_none")]
+    pub media_resolution: Option<String>,
+}
+
 #[derive(Debug, Serialize, Deserialize)]
 pub enum ContentPart {
     #[serde(rename = "text")]
@@ -68,9 +113,6 @@ pub enum ContentPart {
     CodeExecutionResult(Value),
 }
 
-
-
-
 #[derive(Debug, Serialize, Deserialize)]
 pub struct ToolConfigFunctionDeclaration {
     pub function_declarations: Vec<FunctionDeclaration>,
@@ -169,7 +211,7 @@ pub struct ExecutableCode {
 pub struct InlineData {
     #[serde(rename = "mimeType")]
     mime_type: String,
-    data: String
+    data: String,
 }
 
 #[derive(Debug, Serialize, Deserialize)]