siumai 0.10.3

A unified LLM interface library for Rust
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
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249

//! Google Gemini Model Constants
//!
//! This module provides convenient constants for Google Gemini models, making it easy
//! for developers to reference specific models without hardcoding strings.

/// Gemini 2.5 Pro model family constants (latest flagship)
pub mod gemini_2_5_pro {
    /// Gemini 2.5 Pro - Most powerful thinking model
    pub const GEMINI_2_5_PRO: &str = "gemini-2.5-pro";

    /// All Gemini 2.5 Pro models
    pub const ALL: &[&str] = &[GEMINI_2_5_PRO];
}

/// Gemini 2.5 Flash model family constants
pub mod gemini_2_5_flash {
    /// Gemini 2.5 Flash - Excellent cost-performance ratio
    pub const GEMINI_2_5_FLASH: &str = "gemini-2.5-flash";
    /// Gemini 2.5 Flash Preview (05-20)
    pub const GEMINI_2_5_FLASH_PREVIEW_05_20: &str = "gemini-2.5-flash-preview-05-20";

    /// All Gemini 2.5 Flash models
    pub const ALL: &[&str] = &[GEMINI_2_5_FLASH, GEMINI_2_5_FLASH_PREVIEW_05_20];
}

/// Gemini 2.5 Flash-Lite model family constants
pub mod gemini_2_5_flash_lite {
    /// Gemini 2.5 Flash-Lite - Optimized for cost-effectiveness and high throughput
    pub const GEMINI_2_5_FLASH_LITE: &str = "gemini-2.5-flash-lite";
    /// Gemini 2.5 Flash-Lite Preview (06-17)
    pub const GEMINI_2_5_FLASH_LITE_06_17: &str = "gemini-2.5-flash-lite-06-17";

    /// All Gemini 2.5 Flash-Lite models
    pub const ALL: &[&str] = &[GEMINI_2_5_FLASH_LITE, GEMINI_2_5_FLASH_LITE_06_17];
}

/// Gemini 2.5 Flash Live model family constants
pub mod gemini_2_5_flash_live {
    /// Gemini 2.5 Flash Live - Low-latency bidirectional voice and video interaction
    pub const GEMINI_LIVE_2_5_FLASH_PREVIEW: &str = "gemini-live-2.5-flash-preview";

    /// All Gemini 2.5 Flash Live models
    pub const ALL: &[&str] = &[GEMINI_LIVE_2_5_FLASH_PREVIEW];
}

/// Gemini 2.5 Flash Native Audio model family constants
pub mod gemini_2_5_flash_native_audio {
    /// Gemini 2.5 Flash Native Audio Dialog
    pub const GEMINI_2_5_FLASH_PREVIEW_NATIVE_AUDIO_DIALOG: &str =
        "gemini-2.5-flash-preview-native-audio-dialog";
    /// Gemini 2.5 Flash Native Audio Thinking Dialog (experimental)
    pub const GEMINI_2_5_FLASH_EXP_NATIVE_AUDIO_THINKING_DIALOG: &str =
        "gemini-2.5-flash-exp-native-audio-thinking-dialog";

    /// All Gemini 2.5 Flash Native Audio models
    pub const ALL: &[&str] = &[
        GEMINI_2_5_FLASH_PREVIEW_NATIVE_AUDIO_DIALOG,
        GEMINI_2_5_FLASH_EXP_NATIVE_AUDIO_THINKING_DIALOG,
    ];
}

/// Gemini 2.5 TTS model family constants
pub mod gemini_2_5_tts {
    /// Gemini 2.5 Flash Preview TTS
    pub const GEMINI_2_5_FLASH_PREVIEW_TTS: &str = "gemini-2.5-flash-preview-tts";
    /// Gemini 2.5 Pro Preview TTS
    pub const GEMINI_2_5_PRO_PREVIEW_TTS: &str = "gemini-2.5-pro-preview-tts";

    /// All Gemini 2.5 TTS models
    pub const ALL: &[&str] = &[GEMINI_2_5_FLASH_PREVIEW_TTS, GEMINI_2_5_PRO_PREVIEW_TTS];
}

/// Gemini 2.0 Flash model family constants
pub mod gemini_2_0_flash {
    /// Gemini 2.0 Flash - Next-generation features and improved performance
    pub const GEMINI_2_0_FLASH: &str = "gemini-2.0-flash";
    /// Gemini 2.0 Flash (001)
    pub const GEMINI_2_0_FLASH_001: &str = "gemini-2.0-flash-001";
    /// Gemini 2.0 Flash Experimental
    pub const GEMINI_2_0_FLASH_EXP: &str = "gemini-2.0-flash-exp";

    /// All Gemini 2.0 Flash models
    pub const ALL: &[&str] = &[GEMINI_2_0_FLASH, GEMINI_2_0_FLASH_001, GEMINI_2_0_FLASH_EXP];
}

/// Gemini 2.0 Flash Image Generation model family constants
pub mod gemini_2_0_flash_image_gen {
    /// Gemini 2.0 Flash Preview Image Generation
    pub const GEMINI_2_0_FLASH_PREVIEW_IMAGE_GENERATION: &str =
        "gemini-2.0-flash-preview-image-generation";

    /// All Gemini 2.0 Flash Image Generation models
    pub const ALL: &[&str] = &[GEMINI_2_0_FLASH_PREVIEW_IMAGE_GENERATION];
}

/// Gemini 2.0 Flash-Lite model family constants
pub mod gemini_2_0_flash_lite {
    /// Gemini 2.0 Flash-Lite - Optimized for cost-effectiveness and reduced latency
    pub const GEMINI_2_0_FLASH_LITE: &str = "gemini-2.0-flash-lite";
    /// Gemini 2.0 Flash-Lite (001)
    pub const GEMINI_2_0_FLASH_LITE_001: &str = "gemini-2.0-flash-lite-001";

    /// All Gemini 2.0 Flash-Lite models
    pub const ALL: &[&str] = &[GEMINI_2_0_FLASH_LITE, GEMINI_2_0_FLASH_LITE_001];
}

/// Gemini 2.0 Flash Live model family constants
pub mod gemini_2_0_flash_live {
    /// Gemini 2.0 Flash Live - Low-latency bidirectional voice and video interaction
    pub const GEMINI_2_0_FLASH_LIVE_001: &str = "gemini-2.0-flash-live-001";

    /// All Gemini 2.0 Flash Live models
    pub const ALL: &[&str] = &[GEMINI_2_0_FLASH_LIVE_001];
}

/// Gemini 1.5 Flash model family constants (deprecated)
pub mod gemini_1_5_flash {
    /// Gemini 1.5 Flash - Fast and versatile multimodal model
    pub const GEMINI_1_5_FLASH: &str = "gemini-1.5-flash";
    /// Gemini 1.5 Flash Latest
    pub const GEMINI_1_5_FLASH_LATEST: &str = "gemini-1.5-flash-latest";
    /// Gemini 1.5 Flash (001)
    pub const GEMINI_1_5_FLASH_001: &str = "gemini-1.5-flash-001";
    /// Gemini 1.5 Flash (002)
    pub const GEMINI_1_5_FLASH_002: &str = "gemini-1.5-flash-002";

    /// All Gemini 1.5 Flash models
    pub const ALL: &[&str] = &[
        GEMINI_1_5_FLASH,
        GEMINI_1_5_FLASH_LATEST,
        GEMINI_1_5_FLASH_001,
        GEMINI_1_5_FLASH_002,
    ];
}

/// Gemini 1.5 Flash-8B model family constants (deprecated)
pub mod gemini_1_5_flash_8b {
    /// Gemini 1.5 Flash-8B - Small model for high-volume, lower-intelligence tasks
    pub const GEMINI_1_5_FLASH_8B: &str = "gemini-1.5-flash-8b";
    /// Gemini 1.5 Flash-8B Latest
    pub const GEMINI_1_5_FLASH_8B_LATEST: &str = "gemini-1.5-flash-8b-latest";
    /// Gemini 1.5 Flash-8B (001)
    pub const GEMINI_1_5_FLASH_8B_001: &str = "gemini-1.5-flash-8b-001";

    /// All Gemini 1.5 Flash-8B models
    pub const ALL: &[&str] = &[
        GEMINI_1_5_FLASH_8B,
        GEMINI_1_5_FLASH_8B_LATEST,
        GEMINI_1_5_FLASH_8B_001,
    ];
}

/// Gemini 1.5 Pro model family constants (deprecated)
pub mod gemini_1_5_pro {
    /// Gemini 1.5 Pro - Mid-size multimodal model optimized for reasoning tasks
    pub const GEMINI_1_5_PRO: &str = "gemini-1.5-pro";
    /// Gemini 1.5 Pro Latest
    pub const GEMINI_1_5_PRO_LATEST: &str = "gemini-1.5-pro-latest";
    /// Gemini 1.5 Pro (001)
    pub const GEMINI_1_5_PRO_001: &str = "gemini-1.5-pro-001";
    /// Gemini 1.5 Pro (002)
    pub const GEMINI_1_5_PRO_002: &str = "gemini-1.5-pro-002";

    /// All Gemini 1.5 Pro models
    pub const ALL: &[&str] = &[
        GEMINI_1_5_PRO,
        GEMINI_1_5_PRO_LATEST,
        GEMINI_1_5_PRO_001,
        GEMINI_1_5_PRO_002,
    ];
}

/// Popular model recommendations
pub mod popular {
    use super::*;

    /// Most capable model
    pub const FLAGSHIP: &str = gemini_2_5_pro::GEMINI_2_5_PRO;
    /// Best balance of capability and cost
    pub const BALANCED: &str = gemini_2_5_flash::GEMINI_2_5_FLASH;
    /// Most cost-effective
    pub const ECONOMICAL: &str = gemini_2_5_flash_lite::GEMINI_2_5_FLASH_LITE;
    /// Best for real-time interaction
    pub const REALTIME: &str = gemini_2_5_flash_live::GEMINI_LIVE_2_5_FLASH_PREVIEW;
    /// Latest and most advanced
    pub const LATEST: &str = gemini_2_5_pro::GEMINI_2_5_PRO;
}

/// Model capabilities by family
pub mod capabilities {
    /// Models with thinking capability
    pub const THINKING_MODELS: &[&str] = &[
        super::gemini_2_5_pro::GEMINI_2_5_PRO,
        super::gemini_2_5_flash::GEMINI_2_5_FLASH,
        super::gemini_2_5_flash_lite::GEMINI_2_5_FLASH_LITE,
    ];

    /// Models with image generation capability
    pub const IMAGE_GENERATION_MODELS: &[&str] =
        &[super::gemini_2_0_flash_image_gen::GEMINI_2_0_FLASH_PREVIEW_IMAGE_GENERATION];

    /// Models with audio generation capability
    pub const AUDIO_GENERATION_MODELS: &[&str] = &[
        super::gemini_2_5_tts::GEMINI_2_5_FLASH_PREVIEW_TTS,
        super::gemini_2_5_tts::GEMINI_2_5_PRO_PREVIEW_TTS,
        super::gemini_2_5_flash_live::GEMINI_LIVE_2_5_FLASH_PREVIEW,
        super::gemini_2_0_flash_live::GEMINI_2_0_FLASH_LIVE_001,
    ];

    /// Models with Live API support
    pub const LIVE_API_MODELS: &[&str] = &[
        super::gemini_2_5_flash_live::GEMINI_LIVE_2_5_FLASH_PREVIEW,
        super::gemini_2_0_flash_live::GEMINI_2_0_FLASH_LIVE_001,
    ];
}

/// Get all chat models
pub fn all_chat_models() -> Vec<&'static str> {
    let mut models = Vec::new();
    models.extend_from_slice(gemini_2_5_pro::ALL);
    models.extend_from_slice(gemini_2_5_flash::ALL);
    models.extend_from_slice(gemini_2_5_flash_lite::ALL);
    models.extend_from_slice(gemini_2_0_flash::ALL);
    models.extend_from_slice(gemini_2_0_flash_lite::ALL);
    models.extend_from_slice(gemini_1_5_flash::ALL);
    models.extend_from_slice(gemini_1_5_flash_8b::ALL);
    models.extend_from_slice(gemini_1_5_pro::ALL);
    models
}

/// Get all models with thinking capability
pub fn all_thinking_models() -> Vec<&'static str> {
    capabilities::THINKING_MODELS.to_vec()
}

/// Get all models with image generation capability
pub fn all_image_generation_models() -> Vec<&'static str> {
    capabilities::IMAGE_GENERATION_MODELS.to_vec()
}

/// Get all models with audio generation capability
pub fn all_audio_generation_models() -> Vec<&'static str> {
    capabilities::AUDIO_GENERATION_MODELS.to_vec()
}

/// Get all models with Live API support
pub fn all_live_api_models() -> Vec<&'static str> {
    capabilities::LIVE_API_MODELS.to_vec()
}