diff options
Diffstat (limited to '')
| -rw-r--r-- | g4f/Provider/Aivvm.py (renamed from g4f/Provider/deprecated/Aivvm.py) | 10 | ||||
| -rw-r--r-- | g4f/Provider/deprecated/__init__.py | 3 |
2 files changed, 8 insertions, 5 deletions
diff --git a/g4f/Provider/deprecated/Aivvm.py b/g4f/Provider/Aivvm.py index bceb6faf..1a3b6f0b 100644 --- a/g4f/Provider/deprecated/Aivvm.py +++ b/g4f/Provider/Aivvm.py @@ -1,8 +1,8 @@ from __future__ import annotations -from ...requests import StreamSession -from ..base_provider import AsyncGeneratorProvider -from ...typing import AsyncGenerator +from ..requests import StreamSession +from .base_provider import AsyncGeneratorProvider +from ..typing import AsyncGenerator # to recreate this easily, send a post request to https://chat.aivvm.com/api/models models = { @@ -20,6 +20,7 @@ class Aivvm(AsyncGeneratorProvider): url = 'https://chat.aivvm.com' supports_gpt_35_turbo = True supports_gpt_4 = True + working = True @classmethod async def create_async_generator( @@ -51,6 +52,9 @@ class Aivvm(AsyncGeneratorProvider): async with session.post(f"{cls.url}/api/chat", json=json_data) as response: response.raise_for_status() async for chunk in response.iter_content(): + if b'Access denied | chat.aivvm.com used Cloudflare' in chunk: + raise ValueError("Rate Limit | use another provider") + yield chunk.decode() @classmethod diff --git a/g4f/Provider/deprecated/__init__.py b/g4f/Provider/deprecated/__init__.py index d6b93e8d..5c66c87f 100644 --- a/g4f/Provider/deprecated/__init__.py +++ b/g4f/Provider/deprecated/__init__.py @@ -11,5 +11,4 @@ from .Equing import Equing from .Wuguokai import Wuguokai from .V50 import V50 from .FastGpt import FastGpt -from .ChatgptLogin import ChatgptLogin -from .Aivvm import Aivvm
\ No newline at end of file +from .ChatgptLogin import ChatgptLogin
\ No newline at end of file |