Merge pull request #1258 from AndPim4912/chatbase

Add back crutch of incorrect responses to ChatBase
This commit is contained in:
H Lohaus 2023-11-16 12:12:37 +01:00 committed by GitHub
commit 6a793e782c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 0 deletions

View File

@ -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