diff --git a/g4f/Provider/AItianhu.py b/g4f/Provider/AItianhu.py index 0a31e40b790..fcf9a4fb8a4 100644 --- a/g4f/Provider/AItianhu.py +++ b/g4f/Provider/AItianhu.py @@ -71,12 +71,11 @@ async def create_async_generator( if "detail" not in line: raise RuntimeError(f"Response: {line}") - if content := line["detail"]["choices"][0]["delta"].get( - "content" - ): + + content = line["detail"]["choices"][0]["delta"].get("content") + if content: yield content - @classmethod @property def params(cls): diff --git a/g4f/Provider/GptForLove.py b/g4f/Provider/GptForLove.py index e4787e5de5d..90e2096cd3b 100644 --- a/g4f/Provider/GptForLove.py +++ b/g4f/Provider/GptForLove.py @@ -55,9 +55,8 @@ async def create_async_generator( except: raise RuntimeError(f"Broken line: {line}") if "detail" in line: - if content := line["detail"]["choices"][0]["delta"].get( - "content" - ): + content = line["detail"]["choices"][0]["delta"].get("content") + if content: yield content elif "10分钟内提问超过了5次" in line: raise RuntimeError("Rate limit reached") diff --git a/g4f/Provider/GptGo.py b/g4f/Provider/GptGo.py index 6d477da07ca..ac3f7fe8946 100644 --- a/g4f/Provider/GptGo.py +++ b/g4f/Provider/GptGo.py @@ -62,7 +62,9 @@ async def create_async_generator( line = json.loads(line[len(start):-1]) if line["choices"][0]["finish_reason"] == "stop": break - if content := line["choices"][0]["delta"].get("content"): + + content = line["choices"][0]["delta"].get("content"): + if content: yield content diff --git a/g4f/Provider/Ylokh.py b/g4f/Provider/Ylokh.py index abf4d9c1471..136921390b7 100644 --- a/g4f/Provider/Ylokh.py +++ b/g4f/Provider/Ylokh.py @@ -50,9 +50,8 @@ async def create_async_generator( if line.startswith("data: [DONE]"): break line = json.loads(line[6:]) - if content := line["choices"][0]["delta"].get( - "content" - ): + content = line["choices"][0]["delta"].get("content") + if content: yield content else: chat = await response.json() diff --git a/g4f/Provider/deprecated/Ails.py b/g4f/Provider/deprecated/Ails.py index 93c63a69371..5244fd7570c 100644 --- a/g4f/Provider/deprecated/Ails.py +++ b/g4f/Provider/deprecated/Ails.py @@ -69,7 +69,9 @@ async def create_async_generator( if line.startswith(start) and line != "data: [DONE]": line = line[len(start):-1] line = json.loads(line) - if token := line["choices"][0]["delta"].get("content"): + token = line["choices"][0]["delta"].get("content") + + if token: if "ai.ls" in token or "ai.ci" in token: raise Exception(f"Response Error: {token}") yield token