Spaces:
Paused
Paused
fix duplicate code in client.go
Browse files
internal/upstream/client.go
CHANGED
|
@@ -253,7 +253,7 @@ body := map[string]interface{}{
|
|
| 253 |
req.Header.Set("Connection", "keep-alive")
|
| 254 |
req.Header.Set("Origin", "https://chat.z.ai")
|
| 255 |
req.Header.Set("Referer", fmt.Sprintf("https://chat.z.ai/c/%s", uuid.New().String()))
|
| 256 |
-
|
| 257 |
if sess, err := auth.GetAnonymousSession(); err == nil {
|
| 258 |
var cookieStrs []string
|
| 259 |
for _, c := range sess.Cookies {
|
|
@@ -262,9 +262,6 @@ body := map[string]interface{}{
|
|
| 262 |
req.Header.Set("Cookie", strings.Join(cookieStrs, "; "))
|
| 263 |
}
|
| 264 |
|
| 265 |
-
client := proxy.GetHTTPClient()
|
| 266 |
-
}
|
| 267 |
-
|
| 268 |
client := proxy.GetHTTPClient()
|
| 269 |
resp, err := client.Do(req)
|
| 270 |
if err != nil {
|
|
@@ -274,6 +271,9 @@ body := map[string]interface{}{
|
|
| 274 |
return resp, targetModel, nil
|
| 275 |
}
|
| 276 |
|
|
|
|
|
|
|
|
|
|
| 277 |
if sess, err := auth.GetAnonymousSession(); err == nil {
|
| 278 |
var cookieStrs []string
|
| 279 |
for _, c := range sess.Cookies {
|
|
|
|
| 253 |
req.Header.Set("Connection", "keep-alive")
|
| 254 |
req.Header.Set("Origin", "https://chat.z.ai")
|
| 255 |
req.Header.Set("Referer", fmt.Sprintf("https://chat.z.ai/c/%s", uuid.New().String()))
|
| 256 |
+
req.Header.Set("User-Agent", uarand.GetRandom())
|
| 257 |
if sess, err := auth.GetAnonymousSession(); err == nil {
|
| 258 |
var cookieStrs []string
|
| 259 |
for _, c := range sess.Cookies {
|
|
|
|
| 262 |
req.Header.Set("Cookie", strings.Join(cookieStrs, "; "))
|
| 263 |
}
|
| 264 |
|
|
|
|
|
|
|
|
|
|
| 265 |
client := proxy.GetHTTPClient()
|
| 266 |
resp, err := client.Do(req)
|
| 267 |
if err != nil {
|
|
|
|
| 271 |
return resp, targetModel, nil
|
| 272 |
}
|
| 273 |
|
| 274 |
+
return resp, targetModel, nil
|
| 275 |
+
}
|
| 276 |
+
|
| 277 |
if sess, err := auth.GetAnonymousSession(); err == nil {
|
| 278 |
var cookieStrs []string
|
| 279 |
for _, c := range sess.Cookies {
|