summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
Diffstat (limited to '')
-rw-r--r--g4f/Provider/Aivvm.py (renamed from g4f/Provider/deprecated/Aivvm.py)10
-rw-r--r--g4f/Provider/__init__.py2
-rw-r--r--g4f/Provider/deprecated/__init__.py3
-rw-r--r--g4f/__init__.py1
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