初始化 Token 池 Args: tokens: Token 列表 [(token_id, token_value, token_type), ...] failure_threshold: 失败阈值，超过此次数将标记为不可用 recovery_timeout: 恢复超时时间（秒），失败 Token 在此时间后重新尝试
(
self,
tokens: List[Tuple[int, str, str]], # [(token_id, token_value, token_type), ...]
failure_threshold: int = 3,
recovery_timeout: int = 1800
)
| 195 | """Token 池管理器（数据库驱动）""" |
| 196 | |
| 197 | def __init__( |
| 198 | self, |
| 199 | tokens: List[Tuple[int, str, str]], # [(token_id, token_value, token_type), ...] |
| 200 | failure_threshold: int = 3, |
| 201 | recovery_timeout: int = 1800 |
| 202 | ): |
| 203 | """ |
| 204 | 初始化 Token 池 |
| 205 | |
| 206 | Args: |
| 207 | tokens: Token 列表 [(token_id, token_value, token_type), ...] |
| 208 | failure_threshold: 失败阈值，超过此次数将标记为不可用 |
| 209 | recovery_timeout: 恢复超时时间（秒），失败 Token 在此时间后重新尝试 |
| 210 | """ |
| 211 | self.failure_threshold = failure_threshold |
| 212 | self.recovery_timeout = recovery_timeout |
| 213 | self._lock = Lock() |
| 214 | self._current_index = 0 |
| 215 | |
| 216 | # 初始化 Token 状态（内存中） |
| 217 | self.token_statuses: Dict[str, TokenStatus] = {} |
| 218 | self.token_id_map: Dict[str, int] = {} # token -> token_id 映射 |
| 219 | |
| 220 | for token_id, token_value, token_type in tokens: |
| 221 | if token_value and token_value not in self.token_statuses: |
| 222 | self.token_statuses[token_value] = TokenStatus( |
| 223 | token=token_value, |
| 224 | token_id=token_id, |
| 225 | token_type=token_type |
| 226 | ) |
| 227 | self.token_id_map[token_value] = token_id |
| 228 | |
| 229 | if not self.token_statuses: |
| 230 | logger.warning("⚠️ Token 池为空，将依赖匿名模式") |
| 231 | |
| 232 | def get_next_token(self, exclude_tokens: Optional[Set[str]] = None) -> Optional[str]: |
| 233 | """ |
nothing calls this directly
no test coverage detected