Method _request_non_streaming

datastudio/models/openai_api.py:197–230

Make a non-streaming API request. Args: session: Shared aiohttp session. message: Preprocessed message list. **kwargs: Additional parameters. Returns: Tuple of (ret_code, response_content).

(
        self,
        session: aiohttp.ClientSession,
        message: List[dict],
        **kwargs,
    )

        return payload

    async def _request_non_streaming(
        self,
        session: aiohttp.ClientSession,
        message: List[dict],
        **kwargs,
    ) -> Tuple[int, Any]:
        """Make a non-streaming API request.

        Args:
            session: Shared aiohttp session.
            message: Preprocessed message list.
            **kwargs: Additional parameters.

        Returns:
            Tuple of (ret_code, response_content).
        """
        headers = self._build_openai_headers()
        payload = self._build_payload(message, **kwargs)

        async with session.post(
            self.api_base, json=payload, headers=headers
        ) as response:
            resp_status = response.status
            if resp_status == 200:
                data = await response.json(content_type=None)
            else:
                error_text = await response.text()

        if resp_status != 200:
            return 1, f"HTTP {resp_status}: {error_text[:200]}"

        # Post-process OUTSIDE connection scope
        answer = self._parse_openai_response(data)
        return 0, answer

    async def _make_request(
        self,