CC/gpt_academic
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Merge branch 'master' of github.com:binary-husky/chatgpt_academic

Browse Source
This commit is contained in:
binary-husky
2024-07-09 03:18:19 +00:00
parent 83961c1002 114192e025
commit 47fb81cfde
1 changed files with 8 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
request_llms/oai_std_model_template.py
View File

@@ -44,7 +44,7 @@ def decode_chunk(chunk):
try: try:
chunk = json.loads(chunk[6:]) chunk = json.loads(chunk[6:])
except: except:
respose = "API_ERROR" respose = ""
finish_reason = chunk finish_reason = chunk
# 错误处理部分 # 错误处理部分
if "error" in chunk: if "error" in chunk:
@@ -200,10 +200,13 @@ def get_predict_function(
stream_response = response.iter_lines() stream_response = response.iter_lines()
result = "" result = ""
finish_reason = ""
while True: while True:
try: try:
chunk = next(stream_response) chunk = next(stream_response)
except StopIteration: except StopIteration:
if result == "":
raise RuntimeError(f"获得空的回复，可能原因:{finish_reason}")
break break
except requests.exceptions.ConnectionError: except requests.exceptions.ConnectionError:
chunk = next(stream_response) # 失败了，重试一次？再失败就没办法了。 chunk = next(stream_response) # 失败了，重试一次？再失败就没办法了。
@@ -351,6 +354,10 @@ def get_predict_function(
response_text, finish_reason = decode_chunk(chunk) response_text, finish_reason = decode_chunk(chunk)
# 返回的数据流第一次为空，继续等待 # 返回的数据流第一次为空，继续等待
if response_text == "" and finish_reason != "False": if response_text == "" and finish_reason != "False":
status_text = f"finish_reason: {finish_reason}"
yield from update_ui(
chatbot=chatbot, history=history, msg=status_text
)
continue continue
if chunk: if chunk:
try: try: