diff --git a/config.py b/config.py index 90e7fe7e..109ad4e2 100644 --- a/config.py +++ b/config.py @@ -121,6 +121,9 @@ AVAIL_FONTS = [ # 默认的系统提示词(system prompt) INIT_SYS_PROMPT = "Serve me as a writing and programming assistant." +# 任务失败时默认重试的次数 +RETRY_TIMES_AT_UNKNOWN_ERROR = 3 + # 对话窗的高度 (仅在LAYOUT="TOP-DOWN"时生效) CHATBOT_HEIGHT = 1115 diff --git a/crazy_functions/crazy_utils.py b/crazy_functions/crazy_utils.py index 5828fcf7..47090c93 100644 --- a/crazy_functions/crazy_utils.py +++ b/crazy_functions/crazy_utils.py @@ -69,7 +69,7 @@ def request_gpt_model_in_new_thread_with_ui_alive( inputs, inputs_show_user, llm_kwargs, chatbot, history, sys_prompt, refresh_interval=0.2, handle_token_exceed=True, - retry_times_at_unknown_error=2, + retry_times_at_unknown_error=get_conf('RETRY_TIMES_AT_UNKNOWN_ERROR'), ): """ Request GPT model,请求GPT模型同时维持用户界面活跃。 @@ -171,6 +171,9 @@ def can_multi_process(llm) -> bool: if llm.startswith('gpt-'): return True if llm.startswith('chatgpt-'): return True if llm.startswith('api2d-'): return True + if llm.startswith('qianfan'): return True + if llm.startswith('gemini-'): return True + if llm.startswith('moonshot-'): return True if llm.startswith('azure-'): return True if llm.startswith('spark'): return True if llm.startswith('zhipuai') or llm.startswith('glm-'): return True @@ -189,7 +192,7 @@ def request_gpt_model_multi_threads_with_very_awesome_ui_and_high_efficiency( chatbot, history_array, sys_prompt_array, refresh_interval=0.2, max_workers=-1, scroller_max_len=75, handle_token_exceed=True, show_user_at_complete=False, - retry_times_at_unknown_error=2, + retry_times_at_unknown_error=get_conf('RETRY_TIMES_AT_UNKNOWN_ERROR'), ): """ Request GPT model using multiple threads with UI and high efficiency diff --git a/crazy_functions/latex_fns/latex_actions.py b/crazy_functions/latex_fns/latex_actions.py index 45261dd0..c5df4eb9 100644 --- a/crazy_functions/latex_fns/latex_actions.py +++ b/crazy_functions/latex_fns/latex_actions.py @@ -348,7 +348,7 @@ def 编译Latex(chatbot, history, main_file_original, main_file_modified, work_f import os, time n_fix = 1 fixed_line = [] - max_try = 32 + max_try = 12 chatbot.append([f"正在编译PDF文档", f'编译已经开始。当前工作路径为{work_folder},如果程序停顿5分钟以上,请直接去该路径下取回翻译结果,或者重启之后再度尝试 ...']); yield from update_ui(chatbot=chatbot, history=history) chatbot.append([f"正在编译PDF文档", '...']); yield from update_ui(chatbot=chatbot, history=history); time.sleep(1); chatbot[-1] = list(chatbot[-1]) # 刷新界面 yield from update_ui_latest_msg('编译已经开始...', chatbot, history) # 刷新Gradio前端界面