xtekky/gpt4free
1
1
Fork 0
mirror of https://github.com/xtekky/gpt4free.git synced 2024-12-24 19:53:11 +03:00
Code Issues Packages Projects Releases Wiki Activity

remove whitspace from data: and raise exception

Browse Source
This commit is contained in:
Bagus Indrayana 2023-08-24 18:13:20 +08:00
parent 847843d120
commit e56f9b7c0d
1 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
g4f/Provider/EasyChat.py
View File

@ -75,16 +75,18 @@ class EasyChat(BaseProvider):
if "choices" in json_data:
yield json_data["choices"][0]["message"]["content"]
else:
yield Exception("No response from server")
raise Exception("No response from server")
else:
for chunk in response.iter_lines():
if b"content" in chunk:
splitData = chunk.decode().split("data: ")
splitData = chunk.decode().split("data:")
if len(splitData) > 1:
yield json.loads(splitData[1])["choices"][0]["delta"]["content"]
else:
continue
else:
yield Exception(f"Error {response.status_code} from server")
raise Exception(f"Error {response.status_code} from server : {response.reason}")
@classmethod