summaryrefslogtreecommitdiffstats
path: root/g4f/Provider/needs_auth/OpenaiChat.py
diff options
context:
space:
mode:
authorH Lohaus <hlohaus@users.noreply.github.com>2024-04-07 19:59:28 +0200
committerGitHub <noreply@github.com>2024-04-07 19:59:28 +0200
commite724079d0fea382e89fedcf1c7cdb55b24cb56e3 (patch)
treeb9190a5df25ba2ed303ce92c6a8ff885db6b882b /g4f/Provider/needs_auth/OpenaiChat.py
parentUpdate You.py (diff)
parentAdd nodriver to OpenaiChat (diff)
downloadgpt4free-e724079d0fea382e89fedcf1c7cdb55b24cb56e3.tar
gpt4free-e724079d0fea382e89fedcf1c7cdb55b24cb56e3.tar.gz
gpt4free-e724079d0fea382e89fedcf1c7cdb55b24cb56e3.tar.bz2
gpt4free-e724079d0fea382e89fedcf1c7cdb55b24cb56e3.tar.lz
gpt4free-e724079d0fea382e89fedcf1c7cdb55b24cb56e3.tar.xz
gpt4free-e724079d0fea382e89fedcf1c7cdb55b24cb56e3.tar.zst
gpt4free-e724079d0fea382e89fedcf1c7cdb55b24cb56e3.zip
Diffstat (limited to 'g4f/Provider/needs_auth/OpenaiChat.py')
-rw-r--r--g4f/Provider/needs_auth/OpenaiChat.py38
1 files changed, 38 insertions, 0 deletions
diff --git a/g4f/Provider/needs_auth/OpenaiChat.py b/g4f/Provider/needs_auth/OpenaiChat.py
index 921d7394..64e3aeac 100644
--- a/g4f/Provider/needs_auth/OpenaiChat.py
+++ b/g4f/Provider/needs_auth/OpenaiChat.py
@@ -338,6 +338,8 @@ class OpenaiChat(AsyncGeneratorProvider, ProviderModelMixin):
impersonate="chrome",
timeout=timeout
) as session:
+ if cls._expires is not None and cls._expires < time.time():
+ cls._headers = cls._api_key = None
if cls._headers is None or cookies is not None:
cls._create_request_args(cookies)
api_key = kwargs["access_token"] if "access_token" in kwargs else api_key
@@ -365,6 +367,8 @@ class OpenaiChat(AsyncGeneratorProvider, ProviderModelMixin):
cls._set_api_key(api_key)
except NoValidHarFileError:
...
+ if cls._api_key is None:
+ await cls.nodriver_access_token()
cls.default_model = cls.get_model(await cls.get_default_model(session, cls._headers))
async with session.post(
@@ -567,6 +571,40 @@ this.fetch = async (url, options) => {
cls._update_cookie_header()
@classmethod
+ async def nodriver_access_token(cls):
+ try:
+ import nodriver as uc
+ except ImportError:
+ return
+ try:
+ from platformdirs import user_config_dir
+ user_data_dir = user_config_dir("g4f-nodriver")
+ except:
+ user_data_dir = None
+ browser = await uc.start(user_data_dir=user_data_dir)
+ page = await browser.get("https://chat.openai.com/")
+ while await page.query_selector("#prompt-textarea") is None:
+ await asyncio.sleep(1)
+ api_key = await page.evaluate(
+ "(async () => {"
+ "let session = await fetch('/api/auth/session');"
+ "let data = await session.json();"
+ "let accessToken = data['accessToken'];"
+ "let expires = new Date(); expires.setTime(expires.getTime() + 60 * 60 * 4 * 1000);"
+ "document.cookie = 'access_token=' + accessToken + ';expires=' + expires.toUTCString() + ';path=/';"
+ "return accessToken;"
+ "})();",
+ await_promise=True
+ )
+ cookies = {}
+ for c in await page.browser.cookies.get_all():
+ if c.domain.endswith("chat.openai.com"):
+ cookies[c.name] = c.value
+ await page.close()
+ cls._create_request_args(cookies)
+ cls._set_api_key(api_key)
+
+ @classmethod
def browse_access_token(cls, proxy: str = None, timeout: int = 1200) -> None:
"""
Browse to obtain an access token.