Spaces:

markqiu
/

prinvest_mate

Sleeping

Tuchuanhuhuhu commited on Mar 13, 2023

Commit

05bf1ca

1 Parent(s): 108e387

修复GPT一次回复实在太长时导致的ERROR, fix #141

Files changed (1) hide show

utils.py CHANGED Viewed

@@ -172,7 +172,7 @@ def stream_predict(openai_api_key, system_prompt, history, inputs, chatbot, prev
                 try:
                     partial_words = partial_words + chunk['choices'][0]["delta"]["content"]
                 except KeyError:
-                    status_text = standard_error_msg + "API回复中找不到内容。很可能是Token计数达到上限了。当前Token计数: " + str(sum(previous_token_count)+token_counter+user_token_count)
                     yield get_return_value()
                     break
                 if token_counter == 0:

                 try:
                     partial_words = partial_words + chunk['choices'][0]["delta"]["content"]
                 except KeyError:
+                    status_text = standard_error_msg + "API回复中找不到内容。很可能是Token计数达到上限了。请重置对话。当前Token计数: " + str(sum(previous_token_count)+token_counter+user_token_count)
                     yield get_return_value()
                     break
                 if token_counter == 0: