mirror of
https://github.com/snailyp/gemini-balance.git
synced 2026-05-11 18:09:55 +08:00
2225a40bbe
- 新增 `SAFETY_SETTINGS` 配置项，允许用户通过环境变量或数据库配置 Gemini 模型的安全过滤级别。 - 更新后端服务 (`config.py`, `constants.py`, `gemini_routes.py`, `openai_routes.py`, `openai_chat_service.py`, `api_client.py`, `model_service.py`) 以支持和传递 `safety_settings` 参数。 - 在配置编辑器前端 (`config_editor.js`, `config_editor.html`) 添加了用于管理安全设置的用户界面。 - 将模型获取逻辑 (`model_service.py`, `api_client.py`) 改为异步。 - 优化 Service Worker (`service-worker.js`) 的缓存策略为 "cache then network"。 Bump version to 2.1.2
79 lines
2.5 KiB
Python
79 lines
2.5 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",
|
|
}
|
|
|
|
GEMINI_2_FLASH_EXP_SAFETY_SETTINGS = [
|
|
{"category": "HARM_CATEGORY_HARASSMENT", "threshold": "OFF"},
|
|
{"category": "HARM_CATEGORY_HATE_SPEECH", "threshold": "OFF"},
|
|
{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT", "threshold": "OFF"},
|
|
{"category": "HARM_CATEGORY_DANGEROUS_CONTENT", "threshold": "OFF"},
|
|
{"category": "HARM_CATEGORY_CIVIC_INTEGRITY", "threshold": "OFF"},
|
|
]
|
|
|
|
DEFAULT_SAFETY_SETTINGS = [
|
|
{"category": "HARM_CATEGORY_HARASSMENT", "threshold": "OFF"},
|
|
{"category": "HARM_CATEGORY_HATE_SPEECH", "threshold": "OFF"},
|
|
{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT", "threshold": "OFF"},
|
|
{"category": "HARM_CATEGORY_DANGEROUS_CONTENT", "threshold": "OFF"},
|
|
{"category": "HARM_CATEGORY_CIVIC_INTEGRITY", "threshold": "BLOCK_NONE"},
|
|
] |