diff options
-rw-r--r-- | gpt4free/usesless/__init__.py | 11 |
1 files 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 + |