Skip to content

Commit 1c803ac

Browse files
committed
Update Gemini model to gemini-3 and adjust thinking budget logic
1 parent baa1a28 commit 1c803ac

File tree

3 files changed

+3
-3
lines changed

3 files changed

+3
-3
lines changed

cmd/main.go

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -100,7 +100,7 @@ func init() {
100100
case "openai":
101101
cfg.ModelName = "gpt-4o"
102102
case "gemini":
103-
cfg.ModelName = "gemini-2.5-pro"
103+
cfg.ModelName = "gemini-3-flash-preview"
104104
}
105105
}
106106

internal/providers/gemini.go

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -198,7 +198,7 @@ func (g *GeminiProvider) TranslateBatch(ctx context.Context, batch []srt.Subtitl
198198
if config.Thinking {
199199
thinkingBudget = int32(config.ThinkingBudget)
200200
} else {
201-
if strings.Contains(config.ModelName, "gemini-2.5-pro") {
201+
if strings.Contains(config.ModelName, "-pro") {
202202
thinkingBudget = 128
203203
} else {
204204
thinkingBudget = 0

pkg/config/config.go

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -66,7 +66,7 @@ func NewConfig() *Config {
6666
Provider: "gemini", // Default to Gemini for backward compatibility
6767
APIKeys: parseAPIKeys("GEMINI_API_KEY"), // Default to Gemini env var
6868
BaseURL: os.Getenv("GOOGLE_GEMINI_BASE_URL"), // Default to Gemini base URL
69-
ModelName: "gemini-2.5-pro",
69+
ModelName: "gemini-3-pro-preview",
7070
BatchSize: 300,
7171
RetryCount: 3,
7272
Streaming: true,

0 commit comments

Comments
 (0)