diff options
Diffstat (limited to 'g4f/requests_aiohttp.py')
| -rw-r--r-- | g4f/requests_aiohttp.py | 30 |
1 files changed, 0 insertions, 30 deletions
diff --git a/g4f/requests_aiohttp.py b/g4f/requests_aiohttp.py deleted file mode 100644 index 0da8973b..00000000 --- a/g4f/requests_aiohttp.py +++ /dev/null @@ -1,30 +0,0 @@ -from __future__ import annotations - -from aiohttp import ClientSession, ClientResponse, ClientTimeout -from typing import AsyncGenerator, Any - -from .Provider.helper import get_connector -from .defaults import DEFAULT_HEADERS - -class StreamResponse(ClientResponse): - async def iter_lines(self) -> AsyncGenerator[bytes, None]: - async for line in self.content: - yield line.rstrip(b"\r\n") - - async def json(self) -> Any: - return await super().json(content_type=None) - -class StreamSession(ClientSession): - def __init__(self, headers: dict = {}, timeout: int = None, proxies: dict = {}, impersonate = None, **kwargs): - if impersonate: - headers = { - **DEFAULT_HEADERS, - **headers - } - super().__init__( - **kwargs, - timeout=ClientTimeout(timeout) if timeout else None, - response_class=StreamResponse, - connector=get_connector(kwargs.get("connector"), proxies.get("https")), - headers=headers - )
\ No newline at end of file |