diff options
Diffstat (limited to 'g4f/Provider/AItianhu.py')
| -rw-r--r-- | g4f/Provider/AItianhu.py | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/g4f/Provider/AItianhu.py b/g4f/Provider/AItianhu.py index 42631d7e..1d8653b7 100644 --- a/g4f/Provider/AItianhu.py +++ b/g4f/Provider/AItianhu.py @@ -38,6 +38,8 @@ class AItianhu(AsyncGeneratorProvider): async with session.post(f"{cls.url}/api/chat-process", json=data) as response: response.raise_for_status() async for line in response.iter_lines(): + if line == b"<script>": + raise RuntimeError("Solve Challenge") if b"platform's risk control" in line: raise RuntimeError("Platform's Risk Control") line = json.loads(line) |