diff --git a/api/chatgpt/api.go b/api/chatgpt/api.go index 258869aaf..ad564c5eb 100644 --- a/api/chatgpt/api.go +++ b/api/chatgpt/api.go @@ -6,7 +6,9 @@ import ( "encoding/json" "fmt" "io" + "math/rand" "strings" + "time" "github.com/PuerkitoBio/goquery" "github.com/gin-gonic/gin" @@ -41,21 +43,7 @@ func CreateConversation(c *gin.Context) { } if request.Model == gpt4Model || request.Model == gpt4BrowsingModel || request.Model == gpt4PluginsModel { - formParams := fmt.Sprintf( - "public_key=%s", - gpt4PublicKey, - ) - req, _ := http.NewRequest(http.MethodPost, gpt4TokenUrl, strings.NewReader(formParams)) - req.Header.Set("Content-Type", api.ContentType) - resp, err := api.Client.Do(req) - if err != nil { - c.AbortWithStatusJSON(http.StatusInternalServerError, api.ReturnMessage(err.Error())) - return - } - - responseMap := make(map[string]string) - json.NewDecoder(resp.Body).Decode(&responseMap) - request.ArkoseToken = responseMap["token"] + request.ArkoseToken = fmt.Sprintf(arkoseTokenTemplate, generateRandomString(17), generateRandomString(10), gpt4ArkoseTokenPublicKey, generateRandomNumber()) } resp, done := sendConversationRequest(c, request) @@ -373,3 +361,23 @@ func handlePostOrPatch(c *gin.Context, req *http.Request, errorMessage string) { io.Copy(c.Writer, resp.Body) } + +//goland:noinspection SpellCheckingInspection +func generateRandomString(length int) string { + rand.NewSource(time.Now().UnixNano()) + + charset := "0123456789abcdefghijklmnopqrstuvwxyz" + result := make([]byte, length) + + for i := 0; i < length; i++ { + randomIndex := rand.Intn(len(charset)) + result[i] = charset[randomIndex] + } + + return string(result) +} + +func generateRandomNumber() int { + rand.NewSource(time.Now().UnixNano()) + return rand.Intn(100) + 1 +} diff --git a/api/chatgpt/constant.go b/api/chatgpt/constant.go index 6dac28f37..6139d3b74 100644 --- a/api/chatgpt/constant.go +++ b/api/chatgpt/constant.go @@ -1,5 +1,6 @@ package chatgpt +//goland:noinspection SpellCheckingInspection const ( apiPrefix = "https://chat.openai.com/backend-api" getConversationsErrorMessage = "Failed to get conversations." @@ -19,11 +20,11 @@ const ( getCsrfTokenErrorMessage = "Failed to get CSRF token." authSessionUrl = "https://chat.openai.com/api/auth/session" - gpt4Model = "gpt-4" - gpt4BrowsingModel = "gpt-4-browsing" - gpt4PluginsModel = "gpt-4-plugins" - gpt4PublicKey = "35536E1E-65B4-4D96-9D97-6ADB7EFF8147" - gpt4TokenUrl = "https://tcr9i.chat.openai.com/fc/gt2/public_key/" + gpt4PublicKey + gpt4Model = "gpt-4" + gpt4BrowsingModel = "gpt-4-browsing" + gpt4PluginsModel = "gpt-4-plugins" + gpt4ArkoseTokenPublicKey = "35536E1E-65B4-4D96-9D97-6ADB7EFF8147" + arkoseTokenTemplate = `%s.%s|r=us-east-1|meta=3|meta_width=300|metabgclr=transparent|metaiconclr=%%23555555|guitextcolor=%%23000000|pk=%s|at=40|rid=%d|ag=101|cdn_url=https%%3A%%2F%%2Ftcr9i.chat.openai.com%%2Fcdn%%2Ffc|lurl=https%%3A%%2F%%2Faudio-us-east-1.arkoselabs.com|surl=https%%3A%%2F%%2Ftcr9i.chat.openai.com|smurl=https%%3A%%2F%%2Ftcr9i.chat.openai.com%%2Fcdn%%2Ffc%%2Fassets%%2Fstyle-manager` actionContinue = "continue" responseTypeMaxTokens = "max_tokens"