Method _request

app/clients/bigmodel_client.py:144–197

(
        self,
        account: AccountRecord,
        session: AccountSessionState,
        method: str,
        path: str,
        *,
        json_body: Any | None = None,
        params: dict[str, Any] | None = None,
        allow_fallback_proxy: bool = False,
    )

    def _request(
        self,
        account: AccountRecord,
        session: AccountSessionState,
        method: str,
        path: str,
        *,
        json_body: Any | None = None,
        params: dict[str, Any] | None = None,
    ) -> ApiCallResult:
        token = account.token.strip() or account.cookies.get("bigmodel_token_production", "").strip()
        if not token:
            raise BadRequestError("账号缺少 token，没法请求 BigModel")

        headers = {
            "Authorization": token,
            "Bigmodel-Organization": session.org_id or account.org_id,
            "Bigmodel-Project": session.project_id or account.project_id,
        }
        headers = {key: value for key, value in headers.items() if value}
        payload = self.http_client.request_json(
            method,
            self._build_url(path),
            headers=headers,
            cookies=account.cookies,
            json_body=json_body,
            params=params,
            proxy_url=account.proxy_url or None,
            user_agent=resolve_user_agent(account.user_agent, account.browser_impersonate),
            browser_impersonate=account.browser_impersonate or None,
        )
        if not isinstance(payload, dict):
            raise UpstreamRequestError(
                "上游返回结构异常",
                details={"path": path, "payload_type": type(payload).__name__},
            )

        code = payload.get("code")
        if code not in (None, 0, 200):
            # preview 555 (系统繁忙) 不抛异常，让上层循环重试
            if code == 555 and path.endswith("/preview"):
                return ApiCallResult(data=payload.get("data"), raw=payload)
            raise UpstreamRequestError(
                str(payload.get("msg") or payload.get("message") or "上游业务返回失败"),
                details={"path": path, "payload": payload},
            )

        if "data" in payload:
            return ApiCallResult(data=payload.get("data"), raw=payload)
        if "result" in payload:
            return ApiCallResult(data=payload.get("result"), raw=payload)
        return ApiCallResult(data=payload, raw=payload)

    def _build_url(self, path: str) -> str:
        return f"{self.settings.bigmodel_api_base}/{path.lstrip('/')}"


@lru_cache(maxsize=1)
def get_bigmodel_client() -> BigModelClient: