AIstudioProxyAPI / config /__init__.py
peijun1's picture
Deploy AI Studio Proxy API to Hugging Face Spaces
a5784e9
Raw
History Blame Contribute Delete
5.26 kB
"""
Configuration Module Entry Point
Exports all configuration items for easy import by other modules.
"""
# Import all configuration items from individual config files
from .constants import *
from .global_state import GlobalState
from .selector_utils import *
from .selectors import *
from .settings import *
from .timeouts import *
# Explicitly export main configuration items (for IDE autocomplete and type checking)
__all__ = [
# Constant Configuration
"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",
# Timeout Configuration
"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",
"SUBMIT_BUTTON_ENABLE_TIMEOUT_MS",
"SELECTOR_EXISTENCE_CHECK_TIMEOUT_MS",
"SELECTOR_VISIBILITY_TIMEOUT_MS",
"STARTUP_SELECTOR_VISIBILITY_TIMEOUT_MS",
# Selector Configuration
"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",
"QUOTA_EXCEEDED_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",
"USE_URL_CONTEXT_SELECTOR",
"UPLOAD_BUTTON_SELECTOR",
"MODEL_NAME_SELECTOR",
"CDK_OVERLAY_CONTAINER_SELECTOR",
"CHAT_TURN_SELECTOR",
"SCROLL_CONTAINER_SELECTOR",
"CHAT_SESSION_CONTENT_SELECTOR",
"LAST_CHAT_TURN_SELECTOR",
# Thinking Mode Related Selectors
"ENABLE_THINKING_MODE_TOGGLE_SELECTOR",
"SET_THINKING_BUDGET_TOGGLE_SELECTOR",
"THINKING_BUDGET_INPUT_SELECTOR",
"THINKING_LEVEL_DROPDOWN_SELECTOR",
"THINKING_LEVEL_SELECT_SELECTOR",
"THINKING_LEVEL_OPTION_LOW_SELECTOR",
"THINKING_LEVEL_OPTION_MEDIUM_SELECTOR",
"THINKING_LEVEL_OPTION_HIGH_SELECTOR",
"THINKING_LEVEL_OPTION_MINIMAL_SELECTOR",
"THINKING_MODE_TOGGLE_PARENT_SELECTOR",
"THINKING_MODE_TOGGLE_OLD_ROOT_SELECTOR",
"THINKING_BUDGET_TOGGLE_PARENT_SELECTOR",
"THINKING_BUDGET_TOGGLE_OLD_ROOT_SELECTOR",
# Google Search Related Selectors
"GROUNDING_WITH_GOOGLE_SEARCH_TOGGLE_SELECTOR",
# Function Calling Selectors
"FUNCTION_CALLING_CONTAINER_SELECTOR",
"FUNCTION_CALLING_TOGGLE_SELECTOR",
"FUNCTION_DECLARATIONS_EDIT_BUTTON_SELECTOR",
"FUNCTION_DECLARATIONS_DIALOG_SELECTOR",
"FUNCTION_DECLARATIONS_CODE_EDITOR_TAB_SELECTOR",
"FUNCTION_DECLARATIONS_VISUAL_EDITOR_TAB_SELECTOR",
"FUNCTION_DECLARATIONS_TEXTAREA_SELECTOR",
"FUNCTION_DECLARATIONS_SAVE_BUTTON_SELECTOR",
"FUNCTION_DECLARATIONS_RESET_BUTTON_SELECTOR",
"FUNCTION_DECLARATIONS_CLOSE_BUTTON_SELECTOR",
# Function Call Response Selectors (DOM parsing)
"FUNCTION_CALL_WIDGET_SELECTOR",
"FUNCTION_CALL_NAME_SELECTOR",
"FUNCTION_CALL_ARGS_SELECTOR",
"FUNCTION_CALL_CODE_BLOCK_SELECTOR",
# Thinking Visibility Selectors
"THINKING_CONTAINER_SELECTOR",
"THINKING_HEADER_SELECTOR",
"THINKING_CONTENT_SELECTOR",
"THINKING_DIV_SELECTOR",
"THINKING_ACCORDION_SELECTOR",
"FINAL_RESPONSE_SELECTOR",
"ANSWER_TEXT_SELECTOR",
"COMPLETE_RESPONSE_CONTAINER_SELECTOR",
"GENERATION_STATUS_SELECTOR",
# Settings Configuration
"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",
"NETWORK_INTERCEPTION_ENABLED",
"USERSCRIPT_PATH",
# Feature Toggle Settings
"ENABLE_THINKING_BUDGET",
"DEFAULT_THINKING_BUDGET",
"DEFAULT_THINKING_LEVEL",
"DEFAULT_THINKING_LEVEL_PRO",
"DEFAULT_THINKING_LEVEL_FLASH",
"ENABLE_GOOGLE_SEARCH",
"ENABLE_URL_CONTEXT",
# Cookie Refresh Settings
"COOKIE_REFRESH_ENABLED",
"COOKIE_REFRESH_INTERVAL_SECONDS",
"COOKIE_REFRESH_ON_REQUEST_ENABLED",
"COOKIE_REFRESH_REQUEST_INTERVAL",
"COOKIE_REFRESH_ON_SHUTDOWN",
# Utility Functions
"get_environment_variable",
"get_boolean_env",
"get_int_env",
# Global State
"GlobalState",
]