diff options
author | hs_junxiang <jimmy871117@gmail.com> | 2023-11-16 10:13:35 +0100 |
---|---|---|
committer | hs_junxiang <jimmy871117@gmail.com> | 2023-11-16 10:13:35 +0100 |
commit | 5c16856c3f7d358f109045fcaf7de835b216c772 (patch) | |
tree | ad0d2922a60c0bbb71b05219c31ca6aca8fb4a0f /g4f/Provider/GPTalk.py | |
parent | Update README.md (diff) | |
download | gpt4free-5c16856c3f7d358f109045fcaf7de835b216c772.tar gpt4free-5c16856c3f7d358f109045fcaf7de835b216c772.tar.gz gpt4free-5c16856c3f7d358f109045fcaf7de835b216c772.tar.bz2 gpt4free-5c16856c3f7d358f109045fcaf7de835b216c772.tar.lz gpt4free-5c16856c3f7d358f109045fcaf7de835b216c772.tar.xz gpt4free-5c16856c3f7d358f109045fcaf7de835b216c772.tar.zst gpt4free-5c16856c3f7d358f109045fcaf7de835b216c772.zip |
Diffstat (limited to '')
-rw-r--r-- | g4f/Provider/GPTalk.py | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/g4f/Provider/GPTalk.py b/g4f/Provider/GPTalk.py index b5881e5d..5749ff2e 100644 --- a/g4f/Provider/GPTalk.py +++ b/g4f/Provider/GPTalk.py @@ -13,6 +13,7 @@ class GPTalk(AsyncGeneratorProvider): working = True supports_gpt_35_turbo = True _auth = None + used_times = 0 @classmethod async def create_async_generator( @@ -44,7 +45,7 @@ class GPTalk(AsyncGeneratorProvider): 'x-auth-timestamp': f"{timestamp}", } async with ClientSession(headers=headers) as session: - if not cls._auth or cls._auth["expires_at"] < timestamp: + if not cls._auth or cls._auth["expires_at"] < timestamp or cls.used_times == 5: data = { "fingerprint": secrets.token_hex(16).zfill(32), "platform": "fingerprint" @@ -52,6 +53,7 @@ class GPTalk(AsyncGeneratorProvider): async with session.post(f"{cls.url}/api/chatgpt/user/login", json=data, proxy=proxy) as response: response.raise_for_status() cls._auth = (await response.json())["data"] + cls.used_times = 0 data = { "content": format_prompt(messages), "accept": "stream", @@ -72,6 +74,7 @@ class GPTalk(AsyncGeneratorProvider): async with session.post(f"{cls.url}/api/chatgpt/chatapi/text", json=data, headers=headers, proxy=proxy) as response: response.raise_for_status() token = (await response.json())["data"]["token"] + cls.used_times += 1 last_message = "" async with session.get(f"{cls.url}/api/chatgpt/chatapi/stream", params={"token": token}, proxy=proxy) as response: response.raise_for_status() |