gemini-client-api 7.4.5

Library to use Google Gemini API. Automatic context management, schema generation, function calling and more.
Documentation 
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
50
51
52
53
54
55
56
57
58
59
60
61
62

use serde_json::{Value, json};

/// Trait for types that can generate a Gemini-compatible JSON schema.
pub trait GeminiSchema {
    fn gemini_schema() -> Value;
    /// Returns none if schema is of primitive type or name not provided.
    fn name(gemini_schema: &Value) -> Option<&str> {
        gemini_schema["name"].as_str()
    }
}

macro_rules! impl_primitive {
    ($ty:ty, $schema_type:expr) => {
        impl GeminiSchema for $ty {
            fn gemini_schema() -> Value {
                json!({ "type": $schema_type })
            }
        }
    };
}

impl_primitive!(String, "STRING");
impl_primitive!(bool, "BOOLEAN");
impl_primitive!(f32, "NUMBER");
impl_primitive!(f64, "NUMBER");
impl_primitive!(i8, "INTEGER");
impl_primitive!(i16, "INTEGER");
impl_primitive!(i32, "INTEGER");
impl_primitive!(i64, "INTEGER");
impl_primitive!(i128, "INTEGER");
impl_primitive!(isize, "INTEGER");
impl_primitive!(u8, "INTEGER");
impl_primitive!(u16, "INTEGER");
impl_primitive!(u32, "INTEGER");
impl_primitive!(u64, "INTEGER");
impl_primitive!(u128, "INTEGER");
impl_primitive!(usize, "INTEGER");

impl GeminiSchema for &str {
    fn gemini_schema() -> Value {
        json!({ "type": "STRING" })
    }
}

impl<T: GeminiSchema> GeminiSchema for Vec<T> {
    fn gemini_schema() -> Value {
        json!({
            "type": "ARRAY",
            "items": T::gemini_schema()
        })
    }
}

impl<T: GeminiSchema> GeminiSchema for Option<T> {
    fn gemini_schema() -> Value {
        let mut schema = T::gemini_schema();
        if let Some(obj) = schema.as_object_mut() {
            obj.insert("nullable".to_string(), json!(true));
        }
        schema
    }
}