From bfc601e679b1e02d52f669cc2cd498e5ae581c30 Mon Sep 17 00:00:00 2001 From: linyqh Date: Thu, 21 Nov 2024 03:21:18 +0800 Subject: [PATCH] =?UTF-8?q?refactor(webui):=20=E9=87=8D=E6=9E=84=20API=20?= =?UTF-8?q?=E5=8F=82=E6=95=B0=E6=A0=BC=E5=BC=8F?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit - 移除了 config.example.toml 中的多余注释 - 更新了 script_settings.py 中的 API 参数格式，简化为更通用的结构 - 调整了参数名称，使其更加清晰和一致 --- config.example.toml | 1 - webui/components/script_settings.py | 12 +++++------- 2 files changed, 5 insertions(+), 8 deletions(-) diff --git a/config.example.toml b/config.example.toml index 46f8319..52432d2 100644 --- a/config.example.toml +++ b/config.example.toml @@ -12,7 +12,6 @@ vision_gemini_model_name = "gemini-1.5-flash" ########### Vision NarratoAPI Key - # NarratoAPI 是为了便捷访问不了 Gemini API 的用户， 提供的代理服务 narrato_api_key = "" narrato_api_url = "https://narratoapi.scsmtech.cn/api/v1" narrato_vision_model = "gemini-1.5-flash" diff --git a/webui/components/script_settings.py b/webui/components/script_settings.py index 67a2e16..50bf4a9 100644 --- a/webui/components/script_settings.py +++ b/webui/components/script_settings.py @@ -500,14 +500,12 @@ def update_progress(progress: float, message: str = ""): update_progress(90, "正在生成文案...") # 校验配置 api_params = { - 'batch_size': st.session_state.get('narrato_batch_size', 10), - 'use_ai': False, - 'start_offset': 0, - 'vision_model': vision_model, + 'vision_model_name': vision_model, 'vision_api_key': vision_api_key, - 'llm_model': text_model, - 'llm_api_key': text_api_key, - 'custom_prompt': st.session_state.get('custom_prompt', '') + 'vision_base_url': vision_base_url, + 'text_model_name': text_model, + 'text_api_key': text_api_key, + 'text_base_url': text_base_url } session = requests.Session() retry_strategy = Retry(