diff options
| author | Tekky <98614666+xtekky@users.noreply.github.com> | 2023-09-12 11:37:20 +0200 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-09-12 11:37:20 +0200 |
| commit | cf7ba76e7eb27a3e62bea900f5f7fbd3381783ba (patch) | |
| tree | 0a1f1eb3639c34472884d8d3d7bd102b12c891ff /g4f/Provider/ChatBase.py | |
| parent | ~ | 0.0.2.8 (diff) | |
| parent | Disable not working providers: (diff) | |
| download | gpt4free-cf7ba76e7eb27a3e62bea900f5f7fbd3381783ba.tar gpt4free-cf7ba76e7eb27a3e62bea900f5f7fbd3381783ba.tar.gz gpt4free-cf7ba76e7eb27a3e62bea900f5f7fbd3381783ba.tar.bz2 gpt4free-cf7ba76e7eb27a3e62bea900f5f7fbd3381783ba.tar.lz gpt4free-cf7ba76e7eb27a3e62bea900f5f7fbd3381783ba.tar.xz gpt4free-cf7ba76e7eb27a3e62bea900f5f7fbd3381783ba.tar.zst gpt4free-cf7ba76e7eb27a3e62bea900f5f7fbd3381783ba.zip | |
Diffstat (limited to 'g4f/Provider/ChatBase.py')
| -rw-r--r-- | g4f/Provider/ChatBase.py | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/g4f/Provider/ChatBase.py b/g4f/Provider/ChatBase.py index 7d73fd2f..b98fe565 100644 --- a/g4f/Provider/ChatBase.py +++ b/g4f/Provider/ChatBase.py @@ -21,8 +21,10 @@ class ChatBase(AsyncGeneratorProvider): ) -> AsyncGenerator: if model == "gpt-4": chat_id = "quran---tafseer-saadi-pdf-wbgknt7zn" - elif model == "gpt-3.5-turbo" or True: + elif model == "gpt-3.5-turbo" or not model: chat_id = "chatbase--1--pdf-p680fxvnm" + else: + raise ValueError(f"Model are not supported: {model}") headers = { "User-Agent" : "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/116.0.0.0 Safari/537.36", "Accept" : "*/*", @@ -45,9 +47,7 @@ class ChatBase(AsyncGeneratorProvider): async with session.post("https://www.chatbase.co/api/fe/chat", json=data) as response: response.raise_for_status() async for stream in response.content.iter_any(): - stream = stream.decode() - if stream: - yield stream + yield stream.decode() @classmethod |