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