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 generate_script(tr, params): 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(