From 3982f39424ea037aca1086d45c6f657b4bfc457c Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E2=B2=98r=E1=B9=A8h=E0=B8=AA=E2=88=82ow?= <71973368+MrShadowDev@users.noreply.github.com> Date: Mon, 23 Oct 2023 09:46:25 +0200 Subject: 'Refactored by Sourcery' (#1125) Co-authored-by: Sourcery AI <> --- g4f/Provider/ChatgptAi.py | 13 ++++++------- 1 file changed, 6 insertions(+), 7 deletions(-) (limited to 'g4f/Provider/ChatgptAi.py') diff --git a/g4f/Provider/ChatgptAi.py b/g4f/Provider/ChatgptAi.py index d5fd5bff..9783d868 100644 --- a/g4f/Provider/ChatgptAi.py +++ b/g4f/Provider/ChatgptAi.py @@ -38,18 +38,17 @@ class ChatgptAi(AsyncGeneratorProvider): "user-agent" : "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/114.0.0.0 Safari/537.36", } async with ClientSession( - headers=headers - ) as session: + headers=headers + ) as session: if not cls._system: async with session.get(cls.url, proxy=proxy) as response: response.raise_for_status() text = await response.text() - result = re.search(r"data-system='(.*?)'", text) - if result: + if result := re.search(r"data-system='(.*?)'", text): cls._system = json.loads(html.unescape(result.group(1))) - if not cls._system: - raise RuntimeError("System args not found") - + if not cls._system: + raise RuntimeError("System args not found") + data = { "botId": cls._system["botId"], "customId": cls._system["customId"], -- cgit v1.2.3