Moge-Row commited on
Commit
a0108a8
·
1 Parent(s): 890dc51

fix duplicate code in client.go

Browse files
Files changed (1) hide show
  1. internal/upstream/client.go +4 -4
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
- 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,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 {