From 41e00dde266ed90e4993c68692101a73000a7be8 Mon Sep 17 00:00:00 2001 From: LopeKinz <48872005+LopeKinz@users.noreply.github.com> Date: Tue, 13 Jun 2023 15:34:14 +0200 Subject: Update __init__.py --- gpt4free/usesless/__init__.py | 11 ++++++++--- 1 file changed, 8 insertions(+), 3 deletions(-) diff --git a/gpt4free/usesless/__init__.py b/gpt4free/usesless/__init__.py index bd7879eb..00f7f75d 100644 --- a/gpt4free/usesless/__init__.py +++ b/gpt4free/usesless/__init__.py @@ -144,10 +144,15 @@ class Completion: response = Completion.__response_to_json(content) return response + @classmethod def __response_to_json(cls, text) -> str: text = str(text.decode("utf-8")) - split_text = text.rsplit("\n", 1)[1] - to_json = json.loads(split_text) - return to_json + split_text = text.rsplit("\n", 1) + if len(split_text) > 1: + to_json = json.loads(split_text[1]) + return to_json + else: + return None + -- cgit v1.2.3