up

2024-04-21 01:40:42 +08:00
parent 752053f7c4
commit 08ce7e2a4b
1 changed files with 6 additions and 0 deletions
--- a/router/chat.go
+++ b/router/chat.go
@@ -5,6 +5,7 @@ import (
 	"strings"

 	"opencatd-open/pkg/claude"
+	"opencatd-open/pkg/google"
 	"opencatd-open/pkg/openai"

 	"github.com/gin-gonic/gin"
@@ -26,4 +27,9 @@ func ChatHandler(c *gin.Context) {
 		claude.ChatProxy(c, &chatreq)
 		return
 	}
+
+	if strings.HasPrefix(chatreq.Model, "gemini") {
+		google.ChatProxy(c, &chatreq)
+		return
+	}
 }