fix: stream need query alt=sse

2024-07-14 10:16:59 +08:00 · 2024-07-14 10:16:59 +08:00 · ab8a56848f
commit ab8a56848f
parent 005fa5fc4a
1 changed files with 2 additions and 2 deletions
--- a/relay/adaptor/vertexai/adaptor.go
+++ b/relay/adaptor/vertexai/adaptor.go
@ -63,13 +63,13 @@ func (a *Adaptor) GetRequestURL(meta *meta.Meta) (string, error) {
 	suffix := ""
 	if strings.HasPrefix(meta.ActualModelName, "gemini") {
 		if meta.IsStream {
-			suffix = "streamGenerateContent"
+			suffix = "streamGenerateContent?alt=sse"
 		} else {
 			suffix = "generateContent"
 		}
 	} else {
 		if meta.IsStream {
-			suffix = "streamRawPredict"
+			suffix = "streamRawPredict?alt=sse"
 		} else {
 			suffix = "rawPredict"
 		}