diff options
| author | H Lohaus <hlohaus@users.noreply.github.com> | 2023-11-16 12:12:37 +0100 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-11-16 12:12:37 +0100 |
| commit | 6a793e782ca44e381f510812fd7b8f33f5866cbf (patch) | |
| tree | b6f8436ceab16e13b07586fe0145ceaacdbfcc14 /g4f/Provider/ChatBase.py | |
| parent | Merge pull request #1259 from Lin-jun-xiang/test (diff) | |
| parent | Add back crutch of incorrect responses to ChatBase (diff) | |
| download | gpt4free-6a793e782ca44e381f510812fd7b8f33f5866cbf.tar gpt4free-6a793e782ca44e381f510812fd7b8f33f5866cbf.tar.gz gpt4free-6a793e782ca44e381f510812fd7b8f33f5866cbf.tar.bz2 gpt4free-6a793e782ca44e381f510812fd7b8f33f5866cbf.tar.lz gpt4free-6a793e782ca44e381f510812fd7b8f33f5866cbf.tar.xz gpt4free-6a793e782ca44e381f510812fd7b8f33f5866cbf.tar.zst gpt4free-6a793e782ca44e381f510812fd7b8f33f5866cbf.zip | |
Diffstat (limited to 'g4f/Provider/ChatBase.py')
| -rw-r--r-- | g4f/Provider/ChatBase.py | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/g4f/Provider/ChatBase.py b/g4f/Provider/ChatBase.py index 9f11e1ac..ccc20244 100644 --- a/g4f/Provider/ChatBase.py +++ b/g4f/Provider/ChatBase.py @@ -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 |