pt/gpt4free
1
0
Fork 0
mirror of https://github.com/xtekky/gpt4free.git synced 2024-09-12 08:29:30 -04:00
Code Issues Packages Projects Releases Wiki Activity

~ | Merge pull request #838

Browse Source 
fix easychat stream : remove whitspace from data: and raise exception
This commit is contained in:
Tekky 2023-08-25 17:18:54 +01:00 committed by GitHub
commit 19a13c240a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
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