Class UpstreamClient

app/core/upstream.py:363–2236

当前服务使用的上游适配器。

class UpstreamClient:
    """当前服务使用的上游适配器。"""

    def __init__(self):
        self.name = "upstream"
        self.logger = logger
        self.api_endpoint = settings.API_ENDPOINT

        # 当前上游特定配置
        self.base_url = DEFAULT_ZAI_BASE_URL
        self.auth_url = f"{self.base_url}/api/v1/auths/"
        
        # 模型映射
        self.model_mapping = {
            settings.GLM45_MODEL: "0727-360B-API",  # GLM-4.5
            settings.GLM45_THINKING_MODEL: "0727-360B-API",  # GLM-4.5-Thinking
            settings.GLM45_SEARCH_MODEL: "0727-360B-API",  # GLM-4.5-Search
            settings.GLM45_AIR_MODEL: "0727-106B-API",  # GLM-4.5-Air
            settings.GLM46V_MODEL: "glm-4.6v",  # GLM-4.6V多模态
            settings.GLM5_MODEL: "glm-5",  # GLM-5
            settings.GLM47_MODEL: "glm-4.7",  # GLM-4.7
            settings.GLM47_THINKING_MODEL: "glm-4.7",  # GLM-4.7-Thinking
            settings.GLM47_SEARCH_MODEL: "glm-4.7",  # GLM-4.7-Search
            settings.GLM47_ADVANCED_SEARCH_MODEL: "glm-4.7",  # GLM-4.7-advanced-search
        }

    def _get_guest_retry_limit(self) -> int:
        """匿名号池可提供的最大重试预算。"""
        if not settings.ANONYMOUS_MODE:
            return 0

        guest_pool = get_guest_session_pool()
        if not guest_pool:
            return max(2, settings.GUEST_POOL_SIZE + 1)

        pool_status = guest_pool.get_pool_status()
        available_sessions = int(
            pool_status.get("valid_sessions")
            or pool_status.get("available_sessions")
            or 0
        )
        return max(2, available_sessions + 1)

    def _get_authenticated_retry_limit(self) -> int:
        """认证号池与静态 Token 可提供的最大重试预算。"""
        available_tokens = 0
        token_pool = get_token_pool()
        if token_pool:
            available_tokens = int(
                token_pool.get_pool_status().get("available_tokens", 0) or 0
            )

        return max(0, available_tokens)

    def _get_total_retry_limit(self) -> int:
        """综合认证号池与匿名号池的最大尝试次数。"""
        return max(
            1,