ChatGPT/RefreshToV1Api
1
0
Fork 0
mirror of https://github.com/Yanyutin753/RefreshToV1Api.git synced 2025-12-11 02:01:26 +08:00
Code Issues Packages Projects Releases Wiki Activity

🌟 效率更快

Browse Source
This commit is contained in:
Clivia
2024-09-29 12:10:06 +08:00
parent 46fc26db57
commit 013b8b43a9
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
main.py
View File

@@ -1301,7 +1301,7 @@ def data_fetcher(upstream_response, data_queue, stop_event, last_data_time, api_
end_index = buffer.index('\n\n') + 2 end_index = buffer.index('\n\n') + 2
complete_data, buffer = buffer[:end_index], buffer[end_index:] complete_data, buffer = buffer[:end_index], buffer[end_index:]
try: try:
data_content = re.sub(r'^data:\s*', '', complete_data).strip() data_content = complete_data[6:].strip()
if not data_content: if not data_content:
continue continue
data_json = json.loads(data_content) data_json = json.loads(data_content)
@@ -2154,7 +2154,7 @@ def images_generations():
complete_data, buffer = buffer[:end_index], buffer[end_index:] complete_data, buffer = buffer[:end_index], buffer[end_index:]
# 解析 data 块 # 解析 data 块
try: try:
data_json = json.loads(re.sub(r'^data:\s*', '', complete_data).strip()) data_json = json.loads(complete_data[6:].strip())
# print(f"data_json: {data_json}") # print(f"data_json: {data_json}")
message = data_json.get("message", {}) message = data_json.get("message", {})