gemini text generation

2025-05-26 13:34:41 +08:00
parent 268be18baa
commit 75d859dce2
9 changed files with 353 additions and 2 deletions
@@ -43,6 +43,8 @@ const (
 	RelayModeResponses

 	RelayModeRealtime
+
+	RelayModeGemini
 )

 func Path2RelayMode(path string) int {
@@ -75,6 +77,8 @@ func Path2RelayMode(path string) int {
 		relayMode = RelayModeRerank
 	} else if strings.HasPrefix(path, "/v1/realtime") {
 		relayMode = RelayModeRealtime
+	} else if strings.HasPrefix(path, "/v1beta/models") {
+		relayMode = RelayModeGemini
 	}
 	return relayMode
 }