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/GptGod.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) (limited to 'g4f/Provider/GptGod.py') diff --git a/g4f/Provider/GptGod.py b/g4f/Provider/GptGod.py index 060500a9..8a575746 100644 --- a/g4f/Provider/GptGod.py +++ b/g4f/Provider/GptGod.py @@ -33,7 +33,7 @@ class GptGod(AsyncGeneratorProvider): "Pragma": "no-cache", "Cache-Control": "no-cache", } - + async with ClientSession(headers=headers) as session: prompt = format_prompt(messages) data = { @@ -45,12 +45,11 @@ class GptGod(AsyncGeneratorProvider): event = None async for line in response.content: print(line) - + if line.startswith(b'event: '): event = line[7:-1] elif event == b"data" and line.startswith(b"data: "): - data = json.loads(line[6:-1]) - if data: + if data := json.loads(line[6:-1]): yield data elif event == b"done": break \ No newline at end of file -- cgit v1.2.3