Merge pull request #1258 from AndPim4912/chatbase

Add back crutch of incorrect responses to ChatBase
pull/1271/head
H Lohaus 8 months ago committed by GitHub
commit 6a793e782c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -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…
Cancel
Save