summaryrefslogtreecommitdiffstats
path: root/g4f/Provider/ChatgptLogin.py
diff options
context:
space:
mode:
authorabc <98614666+xtekky@users.noreply.github.com>2023-11-20 19:40:55 +0100
committerabc <98614666+xtekky@users.noreply.github.com>2023-11-20 19:40:55 +0100
commit9140541179e1c2fe855acf1c2743e1800fd5052e (patch)
tree774c460654335497ad272d14a6b5082717033651 /g4f/Provider/ChatgptLogin.py
parentMerge pull request #1274 from hlohaus/webdriver (diff)
downloadgpt4free-9140541179e1c2fe855acf1c2743e1800fd5052e.tar
gpt4free-9140541179e1c2fe855acf1c2743e1800fd5052e.tar.gz
gpt4free-9140541179e1c2fe855acf1c2743e1800fd5052e.tar.bz2
gpt4free-9140541179e1c2fe855acf1c2743e1800fd5052e.tar.lz
gpt4free-9140541179e1c2fe855acf1c2743e1800fd5052e.tar.xz
gpt4free-9140541179e1c2fe855acf1c2743e1800fd5052e.tar.zst
gpt4free-9140541179e1c2fe855acf1c2743e1800fd5052e.zip
Diffstat (limited to 'g4f/Provider/ChatgptLogin.py')
-rw-r--r--g4f/Provider/ChatgptLogin.py13
1 files changed, 8 insertions, 5 deletions
diff --git a/g4f/Provider/ChatgptLogin.py b/g4f/Provider/ChatgptLogin.py
index 206e4a89..037e0a6e 100644
--- a/g4f/Provider/ChatgptLogin.py
+++ b/g4f/Provider/ChatgptLogin.py
@@ -45,10 +45,12 @@ class ChatgptLogin(AsyncGeneratorProvider):
async with session.get(f"{cls.url}/chat/", proxy=proxy) as response:
response.raise_for_status()
response = await response.text()
- if result := re.search(
+ result = re.search(
r'<div id="USERID" style="display: none">(.*?)<\/div>',
response,
- ):
+ )
+
+ if result:
cls._user_id = result.group(1)
else:
raise RuntimeError("No user id found")
@@ -67,9 +69,10 @@ class ChatgptLogin(AsyncGeneratorProvider):
response.raise_for_status()
async for line in response.content:
if line.startswith(b"data: "):
- if content := json.loads(line[6:])["choices"][0][
- "delta"
- ].get("content"):
+
+ content = json.loads(line[6:])["choices"][0]["delta"].get("content")
+ if content:
yield content
+
async with session.post(f"{cls.url}/chat/delete_chat", json={"chat_id": chat_id}, proxy=proxy) as response:
response.raise_for_status() \ No newline at end of file