fetch arkose token for each request

acheong08 · Jun 24, 2023 · 09303a9 · 09303a9
1 parent df2fb77
commit 09303a9
Showing 1 changed file with 4 additions and 11 deletions.
diff --git a/main.go b/main.go
@@ -72,17 +72,6 @@ func init() {
 			}
 		}()
 	}
-	go func() {
-		for {
-			var err error
-			arkose_token, err = get_arkose_token()
-			if err != nil {
-				println(err.Error())
-				time.Sleep(10 * time.Second)
-			}
-			time.Sleep(15 * time.Second)
-		}
-	}()
 }
 
 func main() {
@@ -178,6 +167,10 @@ func proxy(c *gin.Context) {
 			return
 		}
 		if strings.HasPrefix(request_body["model"].(string), "gpt-4") {
+			token, err := get_arkose_token()
+			if err == nil {
+				arkose_token = token
+			}
 			request_body["arkose_token"] = arkose_token
 			println("GPT-4!")
 		}