xiaolv commited on
Commit
30abb48
·
1 Parent(s): 880cff7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +12 -11
app.py CHANGED
@@ -244,20 +244,21 @@ def request_to_v2(message, cookie, user_id,attachment=None,delst=True,context=[]
244
  print("******不使用代理.")
245
  response = requests.post(post_msg_url, headers=headers, data=post_msg_data, verify=False,timeout = timeout,impersonate="chrome110")
246
 
247
- bots = ""
248
- print("a"*100)
249
- for data in response.iter_lines():
250
- if data:
251
- # 将字节字符串转换为字符串
252
- data_str = data.decode('utf-8')
253
- # 删除开头的 "data: " 部分
254
- data_str = data_str.replace("data: ", "")
255
- data_json = json.loads(data_str)
256
- # print(data_json)
257
- bots += data_json["completion"]
258
  print(bots)
259
  context += [bots]
260
  responses = [(u, b) for u, b in zip(context[::2], context[1::2])]
 
 
261
  return responses, context
262
 
263
  except Exception as e:
 
244
  print("******不使用代理.")
245
  response = requests.post(post_msg_url, headers=headers, data=post_msg_data, verify=False,timeout = timeout,impersonate="chrome110")
246
 
247
+ decoded_data = response.content.decode("utf-8")
248
+ decoded_data = re.sub('\n+', '\n', decoded_data).strip()
249
+ data_strings = decoded_data.split('\n')
250
+ completions = []
251
+ for data_string in data_strings:
252
+ json_str = data_string[6:].strip()
253
+ data = json.loads(json_str)
254
+ if 'completion' in data:
255
+ completions.append(data['completion'])
256
+ bots = ''.join(completions)
 
257
  print(bots)
258
  context += [bots]
259
  responses = [(u, b) for u, b in zip(context[::2], context[1::2])]
260
+ print("a"*100)
261
+
262
  return responses, context
263
 
264
  except Exception as e: