diff options
| -rw-r--r-- | g4f/Provider/Aivvm.py (renamed from g4f/Provider/deprecated/Aivvm.py) | 10 | ||||
| -rw-r--r-- | g4f/Provider/__init__.py | 2 | ||||
| -rw-r--r-- | g4f/Provider/deprecated/__init__.py | 3 | ||||
| -rw-r--r-- | g4f/__init__.py | 1 |
4 files changed, 10 insertions, 6 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/__init__.py b/g4f/Provider/__init__.py index 7609744e..c7513c35 100644 --- a/g4f/Provider/__init__.py +++ b/g4f/Provider/__init__.py @@ -4,6 +4,7 @@ from .AiAsk import AiAsk from .Aibn import Aibn from .Aichat import Aichat from .Ails import Ails +from .Aivvm import Aivvm from .AItianhu import AItianhu from .AItianhuSpace import AItianhuSpace from .Bing import Bing @@ -46,6 +47,7 @@ __all__ = [ 'Aibn', 'Aichat', 'Ails', + 'Aivvm', 'AiService', 'AItianhu', 'AItianhuSpace', 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 diff --git a/g4f/__init__.py b/g4f/__init__.py index 586a6d2e..bd4bfde0 100644 --- a/g4f/__init__.py +++ b/g4f/__init__.py @@ -1,5 +1,4 @@ from __future__ import annotations - from requests import get from g4f.models import Model, ModelUtils from .Provider import BaseProvider |