gmini 0.1.1

A modular and asynchronous Rust SDK for interacting with the Google Gemini API.
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49

use crate::models::*;

pub fn create_request() -> GenerateContentRequest {
    GenerateContentRequest {
        contents: vec![Content {
            parts: vec![Part {
                text: "Write a hello world program in Python".to_string(),
            }],
            role: "user".to_string(),
        }],
        generation_config: GenerationConfig {
            response_modalities: vec!["TEXT".to_string()],
            temperature: 2.0,
            max_output_tokens: 8192,
            top_p: 1.0,
            response_mime_type: "application/json".to_string(),
            response_schema: ResponseSchema {
                schema_type: "OBJECT".to_string(),
                properties: ResponseProperties {
                    response: ResponseProperty {
                        property_type: "STRING".to_string(),
                    },
                },
            },
        },
        safety_settings: create_safety_settings(),
    }
}

pub fn create_safety_settings() -> Vec<SafetySetting> {
    vec![
        SafetySetting {
            category: HarmCategory::HarmCategoryHateSpeech,
            threshold: HarmBlockThreshold::Off,
        },
        SafetySetting {
            category: HarmCategory::HarmCategoryDangerousContent,
            threshold: HarmBlockThreshold::Off,
        },
        SafetySetting {
            category: HarmCategory::HarmCategorySexuallyExplicit,
            threshold: HarmBlockThreshold::Off,
        },
        SafetySetting {
            category: HarmCategory::HarmCategoryHarassment,
            threshold: HarmBlockThreshold::Off,
        },
    ]
}