update gemini model

2024-12-03 16:28:21 +08:00
parent 5181ba4552
commit fde0e46cbd
2 changed files with 3 additions and 1 deletions
--- a/router/chat.go
+++ b/router/chat.go
@@ -28,7 +28,7 @@ func ChatHandler(c *gin.Context) {
 		return
 	}

-	if strings.HasPrefix(chatreq.Model, "gemini") {
+	if strings.HasPrefix(chatreq.Model, "gemini") || strings.HasPrefix(chatreq.Model, "learnlm") {
 		google.ChatProxy(c, &chatreq)
 		return
 	}