~ | `v-0.1.5.5`

Aiivm working again

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

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',

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
+from .ChatgptLogin  import ChatgptLogin

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