mirror of https://github.com/xtekky/gpt4free.git
Add back crutch of incorrect responses to ChatBase
This commit is contained in:
parent
2e46f1c074
commit
1906b5fd06
|
@ -12,6 +12,8 @@ class ChatBase(AsyncGeneratorProvider):
|
|||
supports_message_history = True
|
||||
working = True
|
||||
jailbreak = True
|
||||
list_incorrect_responses = ["support@chatbase",
|
||||
"about Chatbase"]
|
||||
|
||||
@classmethod
|
||||
async def create_async_generator(
|
||||
|
@ -53,6 +55,9 @@ class ChatBase(AsyncGeneratorProvider):
|
|||
response_data = ""
|
||||
async for stream in response.content.iter_any():
|
||||
response_data += stream.decode()
|
||||
for incorrect_response in cls.list_incorrect_responses:
|
||||
if incorrect_response in response_data:
|
||||
raise RuntimeError("Incorrect response")
|
||||
yield stream.decode()
|
||||
|
||||
@classmethod
|
||||
|
|
Loading…
Reference in New Issue