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