Tuchuanhuhuhu commited on
Commit
5e74754
·
1 Parent(s): 588dc20

如果读取超时,显示错误提示

Browse files
Files changed (2) hide show
  1. presets.py +1 -1
  2. utils.py +5 -1
presets.py CHANGED
@@ -32,7 +32,7 @@ pre code {
32
  """
33
 
34
  standard_error_msg = "☹️发生了错误:" # 错误信息的标准前缀
35
- error_retrieve_prompt = "连接超时,无法获取对话。请检查网络连接,或者API-Key是否有效。" # 获取对话时发生错误
36
  summarize_prompt = "请总结以上对话,不超过100字。" # 总结对话时的 prompt
37
  max_token_streaming = 3500 # 流式对话时的最大 token 数
38
  timeout_streaming = 5 # 流式对话时的超时时间
 
32
  """
33
 
34
  standard_error_msg = "☹️发生了错误:" # 错误信息的标准前缀
35
+ error_retrieve_prompt = "请检查网络连接,或者API-Key是否有效。" # 获取对话时发生错误
36
  summarize_prompt = "请总结以上对话,不超过100字。" # 总结对话时的 prompt
37
  max_token_streaming = 3500 # 流式对话时的最大 token 数
38
  timeout_streaming = 5 # 流式对话时的超时时间
utils.py CHANGED
@@ -144,7 +144,11 @@ def stream_predict(openai_api_key, system_prompt, history, inputs, chatbot, prev
144
  try:
145
  response = get_response(openai_api_key, system_prompt, history, temperature, top_p, True)
146
  except requests.exceptions.ConnectTimeout:
147
- status_text = standard_error_msg + error_retrieve_prompt
 
 
 
 
148
  yield get_return_value()
149
  return
150
 
 
144
  try:
145
  response = get_response(openai_api_key, system_prompt, history, temperature, top_p, True)
146
  except requests.exceptions.ConnectTimeout:
147
+ status_text = standard_error_msg + "连接超时,无法获取对话。" + error_retrieve_prompt
148
+ yield get_return_value()
149
+ return
150
+ except requests.exceptions.ReadTimeout:
151
+ status_text = standard_error_msg + "读取超时,无法获取对话。" + error_retrieve_prompt
152
  yield get_return_value()
153
  return
154