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

fix cookie injection position

Browse files
Files changed (1) hide show
  1. internal/upstream/client.go +10 -1
internal/upstream/client.go CHANGED
@@ -254,7 +254,16 @@ body := map[string]interface{}{
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
-
 
 
 
 
 
 
 
 
 
258
 
259
  client := proxy.GetHTTPClient()
260
  resp, err := client.Do(req)
 
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 {
260
+ cookieStrs = append(cookieStrs, c.Name+"="+c.Value)
261
+ }
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)