From 9140541179e1c2fe855acf1c2743e1800fd5052e Mon Sep 17 00:00:00 2001 From: abc <98614666+xtekky@users.noreply.github.com> Date: Mon, 20 Nov 2023 18:40:55 +0000 Subject: ~ | improve compatibility --- g4f/Provider/deprecated/CodeLinkAva.py | 4 +++- g4f/Provider/deprecated/Equing.py | 4 +++- g4f/Provider/deprecated/FastGpt.py | 6 ++++-- g4f/Provider/deprecated/Lockchat.py | 5 ++++- g4f/Provider/deprecated/Vitalentum.py | 4 +++- 5 files changed, 17 insertions(+), 6 deletions(-) (limited to 'g4f/Provider/deprecated') diff --git a/g4f/Provider/deprecated/CodeLinkAva.py b/g4f/Provider/deprecated/CodeLinkAva.py index a909ab97..22f4468a 100644 --- a/g4f/Provider/deprecated/CodeLinkAva.py +++ b/g4f/Provider/deprecated/CodeLinkAva.py @@ -46,5 +46,7 @@ class CodeLinkAva(AsyncGeneratorProvider): if line.startswith("data: [DONE]"): break line = json.loads(line[6:-1]) - if content := line["choices"][0]["delta"].get("content"): + + content = line["choices"][0]["delta"].get("content") + if content: yield content \ No newline at end of file diff --git a/g4f/Provider/deprecated/Equing.py b/g4f/Provider/deprecated/Equing.py index 076b5ac5..9f510e50 100644 --- a/g4f/Provider/deprecated/Equing.py +++ b/g4f/Provider/deprecated/Equing.py @@ -65,5 +65,7 @@ class Equing(BaseProvider): if line: if b'content' in line: line_json = json.loads(line.decode('utf-8').split('data: ')[1]) - if token := line_json['choices'][0]['delta'].get('content'): + + token = line_json['choices'][0]['delta'].get('content') + if token: yield token \ No newline at end of file diff --git a/g4f/Provider/deprecated/FastGpt.py b/g4f/Provider/deprecated/FastGpt.py index ef69e892..3af8c213 100644 --- a/g4f/Provider/deprecated/FastGpt.py +++ b/g4f/Provider/deprecated/FastGpt.py @@ -69,9 +69,11 @@ class FastGpt(BaseProvider): try: if b'content' in line: line_json = json.loads(line.decode('utf-8').split('data: ')[1]) - if token := line_json['choices'][0]['delta'].get( + token = line_json['choices'][0]['delta'].get( 'content' - ): + ) + + if token: yield token except: continue \ No newline at end of file diff --git a/g4f/Provider/deprecated/Lockchat.py b/g4f/Provider/deprecated/Lockchat.py index d93c9f8a..f885672d 100644 --- a/g4f/Provider/deprecated/Lockchat.py +++ b/g4f/Provider/deprecated/Lockchat.py @@ -38,6 +38,7 @@ class Lockchat(BaseProvider): for token in response.iter_lines(): if b"The model: `gpt-4` does not exist" in token: print("error, retrying...") + Lockchat.create_completion( model = model, messages = messages, @@ -47,5 +48,7 @@ class Lockchat(BaseProvider): if b"content" in token: token = json.loads(token.decode("utf-8").split("data: ")[1]) - if token := token["choices"][0]["delta"].get("content"): + token = token["choices"][0]["delta"].get("content") + + if token: yield (token) \ No newline at end of file diff --git a/g4f/Provider/deprecated/Vitalentum.py b/g4f/Provider/deprecated/Vitalentum.py index 13160d94..8f466a52 100644 --- a/g4f/Provider/deprecated/Vitalentum.py +++ b/g4f/Provider/deprecated/Vitalentum.py @@ -49,5 +49,7 @@ class Vitalentum(AsyncGeneratorProvider): if line.startswith("data: [DONE]"): break line = json.loads(line[6:-1]) - if content := line["choices"][0]["delta"].get("content"): + content = line["choices"][0]["delta"].get("content") + + if content: yield content \ No newline at end of file -- cgit v1.2.3