fix new gpt4 header

xqdoo00o · Feb 19, 2024 · 48475df · 48475df
1 parent 9e79bf9
commit 48475df
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/internal/chatgpt/request.go b/internal/chatgpt/request.go
@@ -59,6 +59,9 @@ func POSTconversation(message chatgpt_types.ChatGPTRequest, access_token string,
 	request.Header.Set("Content-Type", "application/json")
 	request.Header.Set("User-Agent", "Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/117.0.0.0 Safari/537.36")
 	request.Header.Set("Accept", "text/event-stream")
+	if message.Model == "gpt-4" {
+		request.Header.Set("Openai-Sentinel-Arkose-Token", message.ArkoseToken)
+	}
 	if access_token != "" {
 		request.Header.Set("Authorization", "Bearer "+access_token)
 	}