diff options
author | Heiner Lohaus <heiner@lohaus.eu> | 2023-10-01 10:36:09 +0200 |
---|---|---|
committer | Heiner Lohaus <heiner@lohaus.eu> | 2023-10-01 10:36:09 +0200 |
commit | bb481a03ab1a64e47731eb28e461e1dc1d655383 (patch) | |
tree | d067cdee641ace34713c77e28f3636a038d2baf1 /g4f/requests.py | |
parent | Disable PerplexityAi Provider (diff) | |
download | gpt4free-bb481a03ab1a64e47731eb28e461e1dc1d655383.tar gpt4free-bb481a03ab1a64e47731eb28e461e1dc1d655383.tar.gz gpt4free-bb481a03ab1a64e47731eb28e461e1dc1d655383.tar.bz2 gpt4free-bb481a03ab1a64e47731eb28e461e1dc1d655383.tar.lz gpt4free-bb481a03ab1a64e47731eb28e461e1dc1d655383.tar.xz gpt4free-bb481a03ab1a64e47731eb28e461e1dc1d655383.tar.zst gpt4free-bb481a03ab1a64e47731eb28e461e1dc1d655383.zip |
Diffstat (limited to '')
-rw-r--r-- | g4f/requests.py | 10 |
1 files changed, 7 insertions, 3 deletions
diff --git a/g4f/requests.py b/g4f/requests.py index f6f2383b..6f4b587c 100644 --- a/g4f/requests.py +++ b/g4f/requests.py @@ -11,7 +11,7 @@ from curl_cffi.requests import Response from curl_cffi import AsyncCurl is_newer_0_5_9 = hasattr(AsyncCurl, "remove_handle") - +is_newer_0_5_8 = hasattr(BaseSession, "_set_cookies") class StreamResponse: def __init__(self, inner: Response, content: StreamReader, request): @@ -46,6 +46,10 @@ class StreamRequest: ) self.method = method self.url = url + if "proxy" in kwargs: + proxy = kwargs.pop("proxy") + if proxy: + kwargs["proxies"] = {"http": proxy, "https": proxy} self.options = kwargs def on_content(self, data): @@ -75,7 +79,7 @@ class StreamRequest: self.handle = self.session.acurl._curl2future[self.curl] self.handle.add_done_callback(self.on_done) await self.enter - if is_newer_0_5_9: + if is_newer_0_5_8: response = self.session._parse_response(self.curl, _, header_buffer) response.request = request else: @@ -91,7 +95,7 @@ class StreamRequest: self.session.acurl.set_result(self.curl) self.curl.clean_after_perform() self.curl.reset() - self.session.push_curl(self.curl) + self.session.push_curl(self.curl) class AsyncSession(BaseSession): def request( |