diff options
Diffstat (limited to '')
-rw-r--r-- | g4f/requests/aiohttp.py | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/g4f/requests/aiohttp.py b/g4f/requests/aiohttp.py index cdbedef3..4b629963 100644 --- a/g4f/requests/aiohttp.py +++ b/g4f/requests/aiohttp.py @@ -24,6 +24,7 @@ class StreamSession(ClientSession): headers: dict = {}, timeout: int = None, connector: BaseConnector = None, + proxy: str = None, proxies: dict = {}, impersonate = None, **kwargs @@ -38,11 +39,13 @@ class StreamSession(ClientSession): connect, timeout = timeout; if timeout is not None: timeout = ClientTimeout(timeout, connect) + if proxy is None: + proxy = proxies.get("all", proxies.get("https")) super().__init__( **kwargs, timeout=timeout, response_class=StreamResponse, - connector=get_connector(connector, proxies.get("all", proxies.get("https"))), + connector=get_connector(connector, proxy), headers=headers ) |