Method chat_completion

app/core/upstream.py:1601–1723

聊天完成接口。

(
        self,
        request: OpenAIRequest,
        **kwargs
    )

        }
    
    async def chat_completion(
        self,
        request: OpenAIRequest,
        **kwargs
    ) -> Union[Dict[str, Any], AsyncGenerator[str, None]]:
        """聊天完成接口。"""
        self.logger.info(f"🔄 {self.name} 处理请求: {request.model}")
        self.logger.debug(f"  消息数量: {len(request.messages)}")
        self.logger.debug(f"  流式模式: {request.stream}")

        try:
            transformed = await self.transform_request(request)

            if request.stream:
                return self._create_stream_response(request, transformed)

            proxies = self._get_proxy_config()
            max_attempts = self._get_total_retry_limit()
            excluded_tokens: Set[str] = set()
            excluded_guest_user_ids: Set[str] = set()

            for attempt in range(max_attempts):
                async with httpx.AsyncClient(
                    timeout=self._build_timeout(read_timeout=60.0),
                    limits=self._build_limits(),
                    proxy=proxies,
                ) as client:
                    response = await client.post(
                        transformed["url"],
                        headers=transformed["headers"],
                        json=transformed["body"],
                    )

                error_code, error_message = self._extract_upstream_error_details(
                    response.status_code,
                    response.text,
                )
                is_concurrency_limited = self._is_concurrency_limited(
                    response.status_code,
                    error_code,
                    error_message,
                )

                if self._should_retry_guest_session(
                    response.status_code,
                    is_concurrency_limited,
                    attempt,
                    max_attempts,
                    transformed,
                ):
                    guest_user_id = str(
                        transformed.get("guest_user_id")
                        or transformed.get("user_id")
                        or ""
                    )
                    if guest_user_id:
                        excluded_guest_user_ids.add(guest_user_id)
                    transformed = await self._refresh_guest_request(