1.加入qianfan与gemini-多线程请求

2.改变重试次数参数加入设置(config.py)RETRY_TIMES_AT_UNKNOWN_ERROR=3
3.改变tex重复编译次数,将32次改为12次(第一次编译不成功后面大概率编译不成功,没必要这么多次浪费时间)
This commit is contained in:
lin-qian123 2024-04-26 16:38:51 +08:00
parent 81df0aa210
commit d9fc21c234
3 changed files with 9 additions and 3 deletions

View File

@ -78,6 +78,9 @@ AVAIL_THEMES = ["Default", "Chuanhu-Small-and-Beautiful", "High-Contrast", "Gsta
# 默认的系统提示词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

View File

@ -41,7 +41,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模型同时维持用户界面活跃
@ -142,6 +142,9 @@ def can_multi_process(llm) -> bool:
# legacy condition
if llm.startswith('gpt-'): 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
@ -160,7 +163,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=30,
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

View File

@ -345,7 +345,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_lastest_msg('编译已经开始...', chatbot, history) # 刷新Gradio前端界面