diff --git a/request_llm/bridge_chatgpt.py b/request_llm/bridge_chatgpt.py index 719a14f..9c4c3d2 100644 --- a/request_llm/bridge_chatgpt.py +++ b/request_llm/bridge_chatgpt.py @@ -169,7 +169,7 @@ def predict(inputs, llm_kwargs, plugin_kwargs, chatbot, history=[], system_promp while True: chunk = next(stream_response) # print(chunk.decode()[6:]) - if is_head_of_the_stream: + if is_head_of_the_stream and (r'"object":"error"' not in chunk.decode()): # 数据流的第一帧不携带content is_head_of_the_stream = False; continue