Spaces:
Paused
Paused
File size: 7,743 Bytes
fd21f34 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 |
#!/usr/bin/env python
# -*- coding: utf-8 -*-
import os
from typing import Dict, List, Optional
from pydantic_settings import BaseSettings
from app.utils.logger import logger
class Settings(BaseSettings):
"""Application settings"""
# API Configuration
API_ENDPOINT: str = "https://chat.z.ai/api/chat/completions"
AUTH_TOKEN: str = os.getenv("AUTH_TOKEN", "sk-your-api-key")
# 认证token文件路径（可选）
AUTH_TOKENS_FILE: Optional[str] = os.getenv("AUTH_TOKENS_FILE")
# Token池配置
TOKEN_HEALTH_CHECK_INTERVAL: int = int(os.getenv("TOKEN_HEALTH_CHECK_INTERVAL", "300")) # 5分钟
TOKEN_FAILURE_THRESHOLD: int = int(os.getenv("TOKEN_FAILURE_THRESHOLD", "3")) # 失败3次后标记为不可用
TOKEN_RECOVERY_TIMEOUT: int = int(os.getenv("TOKEN_RECOVERY_TIMEOUT", "1800")) # 30分钟后重试失败的token
def _load_tokens_from_file(self, file_path: str) -> List[str]:
"""
从文件加载token列表
支持多种格式的混合使用：
1. 每行一个token（换行分隔）
2. 逗号分隔的token
3. 混合格式（同时支持换行和逗号分隔）
"""
tokens = []
try:
if os.path.exists(file_path):
with open(file_path, 'r', encoding='utf-8') as f:
content = f.read().strip()
if not content:
logger.debug(f"📄 Token文件为空: {file_path}")
return tokens
# 智能解析：同时支持换行和逗号分隔
# 1. 先按换行符分割处理每一行
lines = content.split('\n')
for line in lines:
line = line.strip()
# 跳过空行和注释行
if not line or line.startswith('#'):
continue
# 2. 检查当前行是否包含逗号分隔
if ',' in line:
# 按逗号分割当前行
comma_tokens = line.split(',')
for token in comma_tokens:
token = token.strip()
if token: # 跳过空token
tokens.append(token)
else:
# 整行作为一个token
tokens.append(line)
logger.info(f"📄 从文件加载了 {len(tokens)} 个token: {file_path}")
else:
logger.debug(f"📄 Token文件不存在: {file_path}")
except Exception as e:
logger.error(f"❌ 读取token文件失败 {file_path}: {e}")
return tokens
@property
def auth_token_list(self) -> List[str]:
"""
解析认证token列表
从AUTH_TOKENS_FILE指定的文件加载token（如果配置了文件路径）
"""
# 如果未配置token文件路径，返回空列表
if not self.AUTH_TOKENS_FILE:
logger.debug("📄 未配置AUTH_TOKENS_FILE，跳过token文件加载")
return []
# 从文件加载token
tokens = self._load_tokens_from_file(self.AUTH_TOKENS_FILE)
# 去重，保持顺序
if tokens:
seen = set()
unique_tokens = []
for token in tokens:
if token not in seen:
unique_tokens.append(token)
seen.add(token)
# 记录去重信息
duplicate_count = len(tokens) - len(unique_tokens)
if duplicate_count > 0:
logger.warning(f"⚠️ 检测到 {duplicate_count} 个重复token，已自动去重")
return unique_tokens
return []
@property
def longcat_token_list(self) -> List[str]:
"""
解析 LongCat token 列表
从 LONGCAT_TOKENS_FILE 指定的文件加载 token（如果配置了文件路径）
"""
# 如果未配置token文件路径，返回空列表
if not self.LONGCAT_TOKENS_FILE:
logger.debug("📄 未配置LONGCAT_TOKENS_FILE，跳过LongCat token文件加载")
return []
# 从文件加载token
tokens = self._load_tokens_from_file(self.LONGCAT_TOKENS_FILE)
# 去重，保持顺序
if tokens:
seen = set()
unique_tokens = []
for token in tokens:
if token not in seen:
unique_tokens.append(token)
seen.add(token)
# 记录去重信息
duplicate_count = len(tokens) - len(unique_tokens)
if duplicate_count > 0:
logger.warning(f"⚠️ 检测到 {duplicate_count} 个重复LongCat token，已自动去重")
return unique_tokens
return []
# Model Configuration
PRIMARY_MODEL: str = os.getenv("PRIMARY_MODEL", "GLM-4.5")
THINKING_MODEL: str = os.getenv("THINKING_MODEL", "GLM-4.5-Thinking")
SEARCH_MODEL: str = os.getenv("SEARCH_MODEL", "GLM-4.5-Search")
AIR_MODEL: str = os.getenv("AIR_MODEL", "GLM-4.5-Air")
GLM46_MODEL: str = os.getenv("GLM46_MODEL", "GLM-4.6")
GLM46_THINKING_MODEL: str = os.getenv("GLM46_THINKING_MODEL", "GLM-4.6-Thinking")
GLM46_SEARCH_MODEL: str = os.getenv("GLM46_SEARCH_MODEL", "GLM-4.6-Search")
# Provider Model Mapping
@property
def provider_model_mapping(self) -> Dict[str, str]:
"""模型到提供商的映射"""
return {
# Z.AI models
"GLM-4.5": "zai",
"GLM-4.5-Thinking": "zai",
"GLM-4.5-Search": "zai",
"GLM-4.5-Air": "zai",
"GLM-4.6": "zai",
"GLM-4.6-Thinking": "zai",
"GLM-4.6-Search": "zai",
# K2Think models
"MBZUAI-IFM/K2-Think": "k2think",
# LongCat models
"LongCat-Flash": "longcat",
"LongCat": "longcat",
"LongCat-Search": "longcat",
}
# Server Configuration
LISTEN_PORT: int = int(os.getenv("LISTEN_PORT", "8080"))
DEBUG_LOGGING: bool = os.getenv("DEBUG_LOGGING", "true").lower() == "true"
SERVICE_NAME: str = os.getenv("SERVICE_NAME", "z-ai2api-server")
ANONYMOUS_MODE: bool = os.getenv("ANONYMOUS_MODE", "true").lower() == "true"
TOOL_SUPPORT: bool = os.getenv("TOOL_SUPPORT", "true").lower() == "true"
SCAN_LIMIT: int = int(os.getenv("SCAN_LIMIT", "200000"))
SKIP_AUTH_TOKEN: bool = os.getenv("SKIP_AUTH_TOKEN", "false").lower() == "true"
# LongCat Configuration
LONGCAT_PASSPORT_TOKEN: Optional[str] = os.getenv("LONGCAT_PASSPORT_TOKEN")
LONGCAT_TOKENS_FILE: Optional[str] = os.getenv("LONGCAT_TOKENS_FILE")
# Retry Configuration
MAX_RETRIES: int = int(os.getenv("MAX_RETRIES", "5"))
RETRY_DELAY: float = float(os.getenv("RETRY_DELAY", "1.0")) # 初始重试延迟（秒）
# Browser Headers
CLIENT_HEADERS: Dict[str, str] = {
"Content-Type": "application/json",
"Accept": "application/json, text/event-stream",
"User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/139.0.0.0 Safari/537.36 Edg/139.0.0.0",
"Accept-Language": "zh-CN",
"sec-ch-ua": '"Not;A=Brand";v="99", "Microsoft Edge";v="139", "Chromium";v="139"',
"sec-ch-ua-mobile": "?0",
"sec-ch-ua-platform": '"Windows"',
"X-FE-Version": "prod-fe-1.0.70",
"Origin": "https://chat.z.ai",
}
class Config:
env_file = ".env"
settings = Settings()
|