mirror of
https://github.com/snailyp/gemini-balance.git
synced 2026-05-07 05:52:59 +08:00
e9d19de7c6
将音频/视频相关的配置（支持格式、大小限制、MIME类型）从 `config.py` 移动到 `core/constants.py`，以集中管理常量。 更新 `message_converter.py`： - 从 `core.constants` 导入媒体常量。 - 添加并使用 `message_converter` 的专用日志记录器。 - 清理导入和代码格式。 更新 `openai_chat_service.py`： - 调整 `_has_media_parts` 函数以正确检测 `inline_data`。 - 清理导入和代码格式。 在 `log/logger.py` 中添加 `get_message_converter_logger` 函数。 对 `config.py` 和 `response_handler.py` 进行了相关的移除和微小的代码清理。
63 lines
1.7 KiB
Python
63 lines
1.7 KiB
Python
"""
|
|
常量定义模块
|
|
"""
|
|
|
|
# API相关常量
|
|
API_VERSION = "v1beta"
|
|
DEFAULT_TIMEOUT = 300 # 秒
|
|
MAX_RETRIES = 3 # 最大重试次数
|
|
|
|
# 模型相关常量
|
|
SUPPORTED_ROLES = ["user", "model", "system"]
|
|
DEFAULT_MODEL = "gemini-1.5-flash"
|
|
DEFAULT_TEMPERATURE = 0.7
|
|
DEFAULT_MAX_TOKENS = 8192
|
|
DEFAULT_TOP_P = 0.9
|
|
DEFAULT_TOP_K = 40
|
|
DEFAULT_FILTER_MODELS = [
|
|
"gemini-1.0-pro-vision-latest",
|
|
"gemini-pro-vision",
|
|
"chat-bison-001",
|
|
"text-bison-001",
|
|
"embedding-gecko-001"
|
|
]
|
|
DEFAULT_CREATE_IMAGE_MODEL = "imagen-3.0-generate-002"
|
|
|
|
# 图像生成相关常量
|
|
VALID_IMAGE_RATIOS = ["1:1", "3:4", "4:3", "9:16", "16:9"]
|
|
|
|
# 上传提供商
|
|
UPLOAD_PROVIDERS = ["smms", "picgo", "cloudflare_imgbed"]
|
|
DEFAULT_UPLOAD_PROVIDER = "smms"
|
|
|
|
# 流式输出相关常量
|
|
DEFAULT_STREAM_MIN_DELAY = 0.016
|
|
DEFAULT_STREAM_MAX_DELAY = 0.024
|
|
DEFAULT_STREAM_SHORT_TEXT_THRESHOLD = 10
|
|
DEFAULT_STREAM_LONG_TEXT_THRESHOLD = 50
|
|
DEFAULT_STREAM_CHUNK_SIZE = 5
|
|
|
|
# 正则表达式模式
|
|
IMAGE_URL_PATTERN = r'!\[(.*?)\]\((.*?)\)'
|
|
DATA_URL_PATTERN = r'data:([^;]+);base64,(.+)'
|
|
|
|
# Audio/Video Settings
|
|
SUPPORTED_AUDIO_FORMATS = ["wav", "mp3", "flac", "ogg"]
|
|
SUPPORTED_VIDEO_FORMATS = ["mp4", "mov", "avi", "webm"]
|
|
MAX_AUDIO_SIZE_BYTES = 50 * 1024 * 1024 # Example: 50MB limit for Base64 payload
|
|
MAX_VIDEO_SIZE_BYTES = 200 * 1024 * 1024 # Example: 200MB limit
|
|
|
|
# Optional: Define MIME type mappings if needed, or handle directly in converter
|
|
AUDIO_FORMAT_TO_MIMETYPE = {
|
|
"wav": "audio/wav",
|
|
"mp3": "audio/mpeg",
|
|
"flac": "audio/flac",
|
|
"ogg": "audio/ogg",
|
|
}
|
|
|
|
VIDEO_FORMAT_TO_MIMETYPE = {
|
|
"mp4": "video/mp4",
|
|
"mov": "video/quicktime",
|
|
"avi": "video/x-msvideo",
|
|
"webm": "video/webm",
|
|
} |