summaryrefslogtreecommitdiffstats
path: root/g4f/Provider/ChatgptDemo.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/ChatgptDemo.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/ChatgptDemo.py')
-rw-r--r--g4f/Provider/ChatgptDemo.py11
1 files changed, 8 insertions, 3 deletions
diff --git a/g4f/Provider/ChatgptDemo.py b/g4f/Provider/ChatgptDemo.py
index bc592ca6..666b5753 100644
--- a/g4f/Provider/ChatgptDemo.py
+++ b/g4f/Provider/ChatgptDemo.py
@@ -37,10 +37,13 @@ class ChatgptDemo(AsyncGeneratorProvider):
async with session.get(f"{cls.url}/", 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:
user_id = result.group(1)
else:
raise RuntimeError("No user id found")
@@ -59,5 +62,7 @@ class ChatgptDemo(AsyncGeneratorProvider):
async for line in response.content:
if line.startswith(b"data: "):
line = json.loads(line[6:-1])
- if chunk := line["choices"][0]["delta"].get("content"):
+
+ chunk = line["choices"][0]["delta"].get("content")
+ if chunk:
yield chunk \ No newline at end of file