LangBot/pkg/provider/modelmgr/token.py

from __future__ import annotations

import typing


class TokenManager:
    """鉴权 Token 管理器"""

    name: str

    tokens: list[str]

    using_token_index: typing.Optional[int] = 0

    def __init__(self, name: str, tokens: list[str]):
        self.name = name
        self.tokens = tokens
        self.using_token_index = 0

    def get_token(self) -> str:
        return self.tokens[self.using_token_index]

    def next_token(self):
        self.using_token_index = (self.using_token_index + 1) % len(self.tokens)