Update model.go

Gemini 1.0 Pro Vision is deprecated. Use 1.5 Flash or 1.5 Pro instead.
hkint · Jul 12, 2024 · ea25df3 · ea25df3
1 parent 598871f
commit ea25df3
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pkg/model.go b/pkg/model.go
@@ -14,8 +14,8 @@ import (
 )
 
 const (
-	TextModel   = "gemini-pro"
-	VisionModel = "gemini-pro-vision"
+	TextModel   = "gemini-1.5-flash-latest"
+	VisionModel = "gemini-1.5-pro-latest"
 )
 
 var (