Skip to main content

vtcode_config/constants/models/
google.rs

1/// Default model - using stable version for production reliability
2pub const DEFAULT_MODEL: &str = "gemini-3-flash-preview";
3
4pub const SUPPORTED_MODELS: &[&str] = &[
5    "gemini-3.1-pro-preview",             // Latest Gemini 3.1 Pro flagship
6    "gemini-3.1-pro-preview-customtools", // Optimized for custom tools & bash
7    "gemini-3.1-flash-lite-preview",      // Most cost-efficient, fastest for lightweight tasks
8    "gemini-3.5-flash",                   // Gemini 3.5 Flash - high-efficiency frontier model
9    "gemini-3-flash-preview",             // Fast version of Gemini 3 Pro with 3-level thinking
10    "gemini-3-pro-image-preview",         // Image generation model with 4K resolution
11];
12
13/// Models that support thinking/reasoning capability with configurable thinking_level
14/// Based on: <https://ai.google.dev/gemini-api/docs/gemini-3>
15/// Gemini 3 Pro/Flash: supports low, high (default)
16/// Gemini 3 Flash only: also supports minimal, medium
17pub const REASONING_MODELS: &[&str] = &[
18    "gemini-3.1-pro-preview",
19    "gemini-3.1-pro-preview-customtools",
20    "gemini-3.1-flash-lite-preview",
21    "gemini-3.5-flash",
22    "gemini-3-flash-preview",
23];
24
25/// Models that support Gemini 3 extended thinking levels (minimal, medium)
26/// Only Gemini 3 Flash supports these additional levels beyond low/high
27pub const EXTENDED_THINKING_MODELS: &[&str] =
28    &["gemini-3-flash-preview", "gemini-3.1-flash-lite-preview"];
29
30/// Models supporting image generation
31pub const IMAGE_GENERATION_MODELS: &[&str] = &["gemini-3-pro-image-preview"];
32
33/// Models that support context caching (min 2048 tokens required)
34/// Context caching reduces costs for repeated API calls with similar contexts
35/// Reference: <https://ai.google.dev/gemini-api/docs/caching>
36pub const CACHING_MODELS: &[&str] = &[
37    "gemini-3.1-pro-preview",
38    "gemini-3.1-pro-preview-customtools",
39    "gemini-3.1-flash-lite-preview",
40    "gemini-3.5-flash",
41    "gemini-3-flash-preview",
42];
43
44/// Models that support code execution (Python)
45/// Code execution allows models to write and execute Python code
46/// Reference: <https://ai.google.dev/gemini-api/docs/code-execution>
47pub const CODE_EXECUTION_MODELS: &[&str] = &[
48    "gemini-3.1-pro-preview",
49    "gemini-3.1-pro-preview-customtools",
50    "gemini-3.1-flash-lite-preview",
51    "gemini-3.5-flash",
52    "gemini-3-flash-preview",
53];
54
55// Convenience constants for commonly used models
56pub const GEMINI_3_1_PRO_PREVIEW: &str = "gemini-3.1-pro-preview";
57pub const GEMINI_3_1_PRO_PREVIEW_CUSTOMTOOLS: &str = "gemini-3.1-pro-preview-customtools";
58pub const GEMINI_3_1_FLASH_LITE_PREVIEW: &str = "gemini-3.1-flash-lite-preview";
59pub const GEMINI_3_5_FLASH: &str = "gemini-3.5-flash";
60pub const GEMINI_3_FLASH_PREVIEW: &str = "gemini-3-flash-preview";
61pub const GEMINI_3_PRO_IMAGE_PREVIEW: &str = "gemini-3-pro-image-preview";