Update __init__.py

This commit is contained in:
LopeKinz 2023-06-13 15:34:14 +02:00 committed by GitHub
parent b9478049b3
commit 41e00dde26
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -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