bring back PUID

bi1101 · May 31, 2024 · 66502eb · 66502eb
1 parent 28d86b1
commit 66502eb
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/internal/chatgpt/request.go b/internal/chatgpt/request.go
@@ -88,6 +88,9 @@ func newRequest(method string, url string, body io.Reader, secret *tokens.Secret
 	if secret.Token != "" {
 		request.Header.Set("Authorization", "Bearer "+secret.Token)
 	}
+	if secret.PUID == "" {
+		secret.PUID = os.Getenv("PUID")
+	}
 	if secret.PUID != "" {
 		request.Header.Set("Cookie", "_puid="+secret.PUID+";")
 	}