summaryrefslogtreecommitdiffstats
path: root/g4f/Provider/ChatBase.py
diff options
context:
space:
mode:
authorTekky <98614666+xtekky@users.noreply.github.com>2023-09-12 11:37:20 +0200
committerGitHub <noreply@github.com>2023-09-12 11:37:20 +0200
commitcf7ba76e7eb27a3e62bea900f5f7fbd3381783ba (patch)
tree0a1f1eb3639c34472884d8d3d7bd102b12c891ff /g4f/Provider/ChatBase.py
parent~ | 0.0.2.8 (diff)
parentDisable not working providers: (diff)
downloadgpt4free-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.py8
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