Archives/gpt4free
2
0
Fork 0
mirror of https://github.com/xtekky/gpt4free.git synced 2024-11-19 03:25:32 +00: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: if "choices" in json_data:
yield json_data["choices"][0]["message"]["content"] yield json_data["choices"][0]["message"]["content"]
else: else:
yield Exception("No response from server") raise Exception("No response from server")
else: else:
for chunk in response.iter_lines(): for chunk in response.iter_lines():
if b"content" in chunk: if b"content" in chunk:
splitData = chunk.decode().split("data: ") splitData = chunk.decode().split("data:")
if len(splitData) > 1: if len(splitData) > 1:
yield json.loads(splitData[1])["choices"][0]["delta"]["content"] yield json.loads(splitData[1])["choices"][0]["delta"]["content"]
else:
continue
else: else:
yield Exception(f"Error {response.status_code} from server") raise Exception(f"Error {response.status_code} from server : {response.reason}")
@classmethod @classmethod