summaryrefslogtreecommitdiffstats
path: root/g4f/Provider/unfinished/ChatAiGpt.py
diff options
context:
space:
mode:
Diffstat (limited to 'g4f/Provider/unfinished/ChatAiGpt.py')
-rw-r--r--g4f/Provider/unfinished/ChatAiGpt.py8
1 files changed, 5 insertions, 3 deletions
diff --git a/g4f/Provider/unfinished/ChatAiGpt.py b/g4f/Provider/unfinished/ChatAiGpt.py
index c77afb48..9d050093 100644
--- a/g4f/Provider/unfinished/ChatAiGpt.py
+++ b/g4f/Provider/unfinished/ChatAiGpt.py
@@ -43,10 +43,12 @@ class ChatAiGpt(AsyncGeneratorProvider):
async with session.get(f"{cls.url}/", proxy=proxy) as response:
response.raise_for_status()
response = await response.text()
- result = re.search(r'data-nonce=(.*?) data-post-id=([0-9]+)', response)
- if not result:
+ if result := re.search(
+ r'data-nonce=(.*?) data-post-id=([0-9]+)', response
+ ):
+ cls._nonce, cls._post_id = result.group(1), result.group(2)
+ else:
raise RuntimeError("No nonce found")
- cls._nonce, cls._post_id = result.group(1), result.group(2)
prompt = format_prompt(messages)
data = {
"_wpnonce": cls._nonce,