oai did

lixplor · Apr 2, 2024 · 38eaba0 · 38eaba0
1 parent cadc027
commit 38eaba0
Show file tree

Hide file tree

Showing 4 changed files with 15 additions and 234 deletions.
diff --git a/internal/bard/lib.go b/internal/bard/lib.go
diff --git a/internal/bard/request.go b/internal/bard/request.go
diff --git a/internal/bard/utilities.go b/internal/bard/utilities.go
diff --git a/internal/chatgpt/request.go b/internal/chatgpt/request.go
@@ -6,6 +6,7 @@ import (
 	"context"
 	"encoding/base64"
 	"encoding/json"
+	"fmt"
 	"freechatgpt/typings"
 	chatgpt_types "freechatgpt/typings/chatgpt"
 	"io"
@@ -209,27 +210,35 @@ func getOidDid() string {
 	for _, cookie := range getResp.Cookies() {
 		if cookie.Name == "oai-did" {
 			oaiDid = cookie.Value
+			fmt.Print(oaiDid)
 			break
 		}
 	}
 
 	return oaiDid
 }
 
-func CheckRequire(access_token string, puid string, proxy string) *ChatRequire {
-	if proxy != "" {
-		client.SetProxy(proxy)
-	}
+var oaiDid string
 
+func init() {
 	oaiDid := ""
-	for i := 0; i < 10; i++ {
+	for i := 0; i < 20; i++ {
 		oaiDid = getOidDid()
 		if oaiDid != "" {
 			break
 		}
+		time.Sleep(time.Second)
 	}
 
-	print(oaiDid)
+	if oaiDid == "" {
+		panic("Failed to get oai-did")
+	}
+}
+
+func CheckRequire(access_token string, puid string, proxy string) *ChatRequire {
+	if proxy != "" {
+		client.SetProxy(proxy)
+	}
 
 	request, err := http.NewRequest(http.MethodPost, "https://chat.openai.com/backend-anon/sentinel/chat-requirements", bytes.NewBuffer([]byte(`{}`)))
 	if err != nil {