Upload 5 files

config/__init__.py

@@ -0,0 +1,88 @@
+"""
+配置模块统一入口
+导出所有配置项，便于其他模块导入使用
+"""
+
+# 从各个配置文件导入所有配置项
+from .constants import *
+from .timeouts import *
+from .selectors import *
+from .settings import *
+
+# 显式导出主要配置项（用于IDE自动完成和类型检查）
+__all__ = [
+    # 常量配置
+    'MODEL_NAME',
+    'CHAT_COMPLETION_ID_PREFIX', 
+    'DEFAULT_FALLBACK_MODEL_ID',
+    'DEFAULT_TEMPERATURE',
+    'DEFAULT_MAX_OUTPUT_TOKENS',
+    'DEFAULT_TOP_P',
+    'DEFAULT_STOP_SEQUENCES',
+    'AI_STUDIO_URL_PATTERN',
+    'MODELS_ENDPOINT_URL_CONTAINS',
+    'USER_INPUT_START_MARKER_SERVER',
+    'USER_INPUT_END_MARKER_SERVER',
+    'EXCLUDED_MODELS_FILENAME',
+    'STREAM_TIMEOUT_LOG_STATE',
+    
+    # 超时配置
+    'RESPONSE_COMPLETION_TIMEOUT',
+    'INITIAL_WAIT_MS_BEFORE_POLLING',
+    'POLLING_INTERVAL',
+    'POLLING_INTERVAL_STREAM',
+    'SILENCE_TIMEOUT_MS',
+    'POST_SPINNER_CHECK_DELAY_MS',
+    'FINAL_STATE_CHECK_TIMEOUT_MS',
+    'POST_COMPLETION_BUFFER',
+    'CLEAR_CHAT_VERIFY_TIMEOUT_MS',
+    'CLEAR_CHAT_VERIFY_INTERVAL_MS',
+    'CLICK_TIMEOUT_MS',
+    'CLIPBOARD_READ_TIMEOUT_MS',
+    'WAIT_FOR_ELEMENT_TIMEOUT_MS',
+    'PSEUDO_STREAM_DELAY',
+    
+    # 选择器配置
+    'PROMPT_TEXTAREA_SELECTOR',
+    'INPUT_SELECTOR',
+    'INPUT_SELECTOR2',
+    'SUBMIT_BUTTON_SELECTOR',
+    'CLEAR_CHAT_BUTTON_SELECTOR',
+    'CLEAR_CHAT_CONFIRM_BUTTON_SELECTOR',
+    'RESPONSE_CONTAINER_SELECTOR',
+    'RESPONSE_TEXT_SELECTOR',
+    'LOADING_SPINNER_SELECTOR',
+    'OVERLAY_SELECTOR',
+    'ERROR_TOAST_SELECTOR',
+    'EDIT_MESSAGE_BUTTON_SELECTOR',
+    'MESSAGE_TEXTAREA_SELECTOR',
+    'FINISH_EDIT_BUTTON_SELECTOR',
+    'MORE_OPTIONS_BUTTON_SELECTOR',
+    'COPY_MARKDOWN_BUTTON_SELECTOR',
+    'COPY_MARKDOWN_BUTTON_SELECTOR_ALT',
+    'MAX_OUTPUT_TOKENS_SELECTOR',
+    'STOP_SEQUENCE_INPUT_SELECTOR',
+    'MAT_CHIP_REMOVE_BUTTON_SELECTOR',
+    'TOP_P_INPUT_SELECTOR',
+    'TEMPERATURE_INPUT_SELECTOR',
+    
+    # 设置配置
+    'DEBUG_LOGS_ENABLED',
+    'TRACE_LOGS_ENABLED',
+    'AUTO_SAVE_AUTH',
+    'AUTH_SAVE_TIMEOUT',
+    'AUTO_CONFIRM_LOGIN',
+    'AUTH_PROFILES_DIR',
+    'ACTIVE_AUTH_DIR',
+    'SAVED_AUTH_DIR',
+    'LOG_DIR',
+    'APP_LOG_FILE_PATH',
+    'NO_PROXY_ENV',
+    'ENABLE_SCRIPT_INJECTION',
+    'USERSCRIPT_PATH',
+
+    # 工具函数
+    'get_environment_variable',
+    'get_boolean_env',
+    'get_int_env',
+] 

config/constants.py

@@ -0,0 +1,48 @@
+"""
+常量配置模块
+包含所有固定的常量定义，如模型名称、标记符、文件名等
+"""
+
+import os
+import json
+from dotenv import load_dotenv
+
+# 加载 .env 文件
+load_dotenv()
+
+# --- 模型相关常量 ---
+MODEL_NAME = os.environ.get('MODEL_NAME', 'AI-Studio_Proxy_API')
+CHAT_COMPLETION_ID_PREFIX = os.environ.get('CHAT_COMPLETION_ID_PREFIX', 'chatcmpl-')
+DEFAULT_FALLBACK_MODEL_ID = os.environ.get('DEFAULT_FALLBACK_MODEL_ID', "no model list")
+
+# --- 默认参数值 ---
+DEFAULT_TEMPERATURE = float(os.environ.get('DEFAULT_TEMPERATURE', '1.0'))
+DEFAULT_MAX_OUTPUT_TOKENS = int(os.environ.get('DEFAULT_MAX_OUTPUT_TOKENS', '65536'))
+DEFAULT_TOP_P = float(os.environ.get('DEFAULT_TOP_P', '0.95'))
+
+# 默认停止序列 - 支持 JSON 格式配置
+try:
+    DEFAULT_STOP_SEQUENCES = json.loads(os.environ.get('DEFAULT_STOP_SEQUENCES', '["用户:"]'))
+except (json.JSONDecodeError, TypeError):
+    DEFAULT_STOP_SEQUENCES = ["用户:"]  # 回退到默认值
+
+# --- URL模式 ---
+AI_STUDIO_URL_PATTERN = os.environ.get('AI_STUDIO_URL_PATTERN', 'aistudio.google.com/')
+MODELS_ENDPOINT_URL_CONTAINS = os.environ.get('MODELS_ENDPOINT_URL_CONTAINS', "MakerSuiteService/ListModels")
+
+# --- 输入标记符 ---
+USER_INPUT_START_MARKER_SERVER = os.environ.get('USER_INPUT_START_MARKER_SERVER', "__USER_INPUT_START__")
+USER_INPUT_END_MARKER_SERVER = os.environ.get('USER_INPUT_END_MARKER_SERVER', "__USER_INPUT_END__")
+
+# --- 文件名常量 ---
+EXCLUDED_MODELS_FILENAME = os.environ.get('EXCLUDED_MODELS_FILENAME', "excluded_models.txt")
+
+# --- 流状态配置 ---
+STREAM_TIMEOUT_LOG_STATE = {
+    "consecutive_timeouts": 0,
+    "last_error_log_time": 0.0,  # 使用 time.monotonic()
+    "suppress_until_time": 0.0,  # 使用 time.monotonic()
+    "max_initial_errors": int(os.environ.get('STREAM_MAX_INITIAL_ERRORS', '3')),
+    "warning_interval_after_suppress": float(os.environ.get('STREAM_WARNING_INTERVAL_AFTER_SUPPRESS', '60.0')),
+    "suppress_duration_after_initial_burst": float(os.environ.get('STREAM_SUPPRESS_DURATION_AFTER_INITIAL_BURST', '400.0')),
+}

config/selectors.py

@@ -0,0 +1,42 @@
+"""
+CSS选择器配置模块
+包含所有用于页面元素定位的CSS选择器
+"""
+
+# --- 输入相关选择器 ---
+PROMPT_TEXTAREA_SELECTOR = 'ms-prompt-input-wrapper ms-autosize-textarea textarea'
+INPUT_SELECTOR = PROMPT_TEXTAREA_SELECTOR
+INPUT_SELECTOR2 = PROMPT_TEXTAREA_SELECTOR
+
+# --- 按钮选择器 ---
+SUBMIT_BUTTON_SELECTOR = 'button[aria-label="Run"].run-button'
+CLEAR_CHAT_BUTTON_SELECTOR = 'button[data-test-clear="outside"][aria-label="Clear chat"]'
+CLEAR_CHAT_CONFIRM_BUTTON_SELECTOR = 'button.mdc-button:has-text("Continue")'
+
+# --- 响应相关选择器 ---
+RESPONSE_CONTAINER_SELECTOR = 'ms-chat-turn .chat-turn-container.model'
+RESPONSE_TEXT_SELECTOR = 'ms-cmark-node.cmark-node'
+
+# --- 加载和状态选择器 ---
+LOADING_SPINNER_SELECTOR = 'button[aria-label="Run"].run-button svg .stoppable-spinner'
+OVERLAY_SELECTOR = 'div.cdk-overlay-backdrop'
+
+# --- 错误提示选择器 ---
+ERROR_TOAST_SELECTOR = 'div.toast.warning, div.toast.error'
+
+# --- 编辑相关选择器 ---
+EDIT_MESSAGE_BUTTON_SELECTOR = 'ms-chat-turn:last-child .actions-container button.toggle-edit-button'
+MESSAGE_TEXTAREA_SELECTOR = 'ms-chat-turn:last-child ms-text-chunk ms-autosize-textarea'
+FINISH_EDIT_BUTTON_SELECTOR = 'ms-chat-turn:last-child .actions-container button.toggle-edit-button[aria-label="Stop editing"]'
+
+# --- 菜单和复制相关选择器 ---
+MORE_OPTIONS_BUTTON_SELECTOR = 'div.actions-container div ms-chat-turn-options div > button'
+COPY_MARKDOWN_BUTTON_SELECTOR = 'button.mat-mdc-menu-item:nth-child(4)'
+COPY_MARKDOWN_BUTTON_SELECTOR_ALT = 'div[role="menu"] button:has-text("Copy Markdown")'
+
+# --- 设置相关选择器 ---
+MAX_OUTPUT_TOKENS_SELECTOR = 'input[aria-label="Maximum output tokens"]'
+STOP_SEQUENCE_INPUT_SELECTOR = 'input[aria-label="Add stop token"]'
+MAT_CHIP_REMOVE_BUTTON_SELECTOR = 'mat-chip-set mat-chip-row button[aria-label*="Remove"]'
+TOP_P_INPUT_SELECTOR = 'div.settings-item-column:has(h3:text-is("Top P")) input[type="number"].slider-input'
+TEMPERATURE_INPUT_SELECTOR = 'div[data-test-id="temperatureSliderContainer"] input[type="number"].slider-input' 

config/settings.py

@@ -0,0 +1,54 @@
+"""
+主要设置配置模块
+包含环境变量配置、路径配置、代理配置等运行时设置
+"""
+
+import os
+from dotenv import load_dotenv
+
+# 加载 .env 文件
+load_dotenv()
+
+# --- 全局日志控制配置 ---
+DEBUG_LOGS_ENABLED = os.environ.get('DEBUG_LOGS_ENABLED', 'false').lower() in ('true', '1', 'yes')
+TRACE_LOGS_ENABLED = os.environ.get('TRACE_LOGS_ENABLED', 'false').lower() in ('true', '1', 'yes')
+
+# --- 认证相关配置 ---
+AUTO_SAVE_AUTH = os.environ.get('AUTO_SAVE_AUTH', '').lower() in ('1', 'true', 'yes')
+AUTH_SAVE_TIMEOUT = int(os.environ.get('AUTH_SAVE_TIMEOUT', '30'))
+AUTO_CONFIRM_LOGIN = os.environ.get('AUTO_CONFIRM_LOGIN', 'true').lower() in ('1', 'true', 'yes')
+
+# --- 路径配置 ---
+AUTH_PROFILES_DIR = os.path.join(os.path.dirname(__file__), '..', 'auth_profiles')
+ACTIVE_AUTH_DIR = os.path.join(AUTH_PROFILES_DIR, 'active')
+SAVED_AUTH_DIR = os.path.join(AUTH_PROFILES_DIR, 'saved')
+LOG_DIR = os.path.join(os.path.dirname(__file__), '..', 'logs')
+APP_LOG_FILE_PATH = os.path.join(LOG_DIR, 'app.log')
+
+def get_environment_variable(key: str, default: str = '') -> str:
+    """获取环境变量值"""
+    return os.environ.get(key, default)
+
+def get_boolean_env(key: str, default: bool = False) -> bool:
+    """获取布尔型环境变量"""
+    value = os.environ.get(key, '').lower()
+    if default:
+        return value not in ('false', '0', 'no', 'off')
+    else:
+        return value in ('true', '1', 'yes', 'on')
+
+def get_int_env(key: str, default: int = 0) -> int:
+    """获取整型环境变量"""
+    try:
+        return int(os.environ.get(key, str(default)))
+    except (ValueError, TypeError):
+        return default
+
+# --- 代理配置 ---
+# 注意：代理配置现在在 api_utils/app.py 中动态设置，根据 STREAM_PORT 环境变量决定
+NO_PROXY_ENV = os.environ.get('NO_PROXY')
+
+# --- 脚本注入配置 ---
+ENABLE_SCRIPT_INJECTION = get_boolean_env('ENABLE_SCRIPT_INJECTION', True)
+USERSCRIPT_PATH = get_environment_variable('USERSCRIPT_PATH', 'browser_utils/more_modles.js')
+# 注意：MODEL_CONFIG_PATH 已废弃，现在直接从油猴脚本解析模型数据

config/timeouts.py

@@ -0,0 +1,40 @@
+"""
+超时和时间配置模块
+包含所有超时时间、轮询间隔等时间相关配置
+"""
+
+import os
+from dotenv import load_dotenv
+
+# 加载 .env 文件
+load_dotenv()
+
+# --- 响应等待配置 ---
+RESPONSE_COMPLETION_TIMEOUT = int(os.environ.get('RESPONSE_COMPLETION_TIMEOUT', '300000'))  # 5 minutes total timeout (in ms)
+INITIAL_WAIT_MS_BEFORE_POLLING = int(os.environ.get('INITIAL_WAIT_MS_BEFORE_POLLING', '500'))  # ms, initial wait before polling for response completion
+
+# --- 轮询间隔配置 ---
+POLLING_INTERVAL = int(os.environ.get('POLLING_INTERVAL', '300'))  # ms
+POLLING_INTERVAL_STREAM = int(os.environ.get('POLLING_INTERVAL_STREAM', '180'))  # ms
+
+# --- 静默超时配置 ---
+SILENCE_TIMEOUT_MS = int(os.environ.get('SILENCE_TIMEOUT_MS', '60000'))  # ms
+
+# --- 页面操作超时配置 ---
+POST_SPINNER_CHECK_DELAY_MS = int(os.environ.get('POST_SPINNER_CHECK_DELAY_MS', '500'))
+FINAL_STATE_CHECK_TIMEOUT_MS = int(os.environ.get('FINAL_STATE_CHECK_TIMEOUT_MS', '1500'))
+POST_COMPLETION_BUFFER = int(os.environ.get('POST_COMPLETION_BUFFER', '700'))
+
+# --- 清理聊天相关超时 ---
+CLEAR_CHAT_VERIFY_TIMEOUT_MS = int(os.environ.get('CLEAR_CHAT_VERIFY_TIMEOUT_MS', '5000'))
+CLEAR_CHAT_VERIFY_INTERVAL_MS = int(os.environ.get('CLEAR_CHAT_VERIFY_INTERVAL_MS', '2000'))
+
+# --- 点击和剪贴板操作超时 ---
+CLICK_TIMEOUT_MS = int(os.environ.get('CLICK_TIMEOUT_MS', '3000'))
+CLIPBOARD_READ_TIMEOUT_MS = int(os.environ.get('CLIPBOARD_READ_TIMEOUT_MS', '3000'))
+
+# --- 元素等待超时 ---
+WAIT_FOR_ELEMENT_TIMEOUT_MS = int(os.environ.get('WAIT_FOR_ELEMENT_TIMEOUT_MS', '10000'))  # Timeout for waiting for elements like overlays
+
+# --- 流相关配置 ---
+PSEUDO_STREAM_DELAY = float(os.environ.get('PSEUDO_STREAM_DELAY', '0.01'))