0.3.3.4

7 files changed, 16 insertions, 5 deletions

diff --git a/g4f/Provider/Pi.py b/g4f/Provider/Pi.py
index 266647ba..68a7357f 100644
--- a/g4f/Provider/Pi.py
+++ b/g4f/Provider/Pi.py
@@ -12,6 +12,7 @@ class Pi(AbstractProvider):
     supports_stream = True
     _session = None
     default_model = "pi"
+    models = [default_model]
 
     @classmethod
     def create_completion(
diff --git a/g4f/Provider/__init__.py b/g4f/Provider/__init__.py
index dcf9c352..da0eacfe 100644
--- a/g4f/Provider/__init__.py
+++ b/g4f/Provider/__init__.py
@@ -12,7 +12,6 @@ from .not_working      import *
 from .local            import *
 
 from .AIUncensored     import AIUncensored
-from .Allyfy           import Allyfy
 from .Airforce         import Airforce
 from .Bing             import Bing
 from .Blackbox         import Blackbox
diff --git a/g4f/Provider/needs_auth/Gemini.py b/g4f/Provider/needs_auth/Gemini.py
index 8d741476..dad54c84 100644
--- a/g4f/Provider/needs_auth/Gemini.py
+++ b/g4f/Provider/needs_auth/Gemini.py
@@ -57,6 +57,7 @@ class Gemini(AsyncGeneratorProvider):
     default_model = 'gemini'
     image_models = ["gemini"]
     default_vision_model = "gemini"
+    models = ["gemini", "gemini-1.5-flash", "gemini-1.5-pro"]
     _cookies: Cookies = None
     _snlm0e: str = None
     _sid: str = None
diff --git a/g4f/Provider/needs_auth/Poe.py b/g4f/Provider/needs_auth/Poe.py
index 0c969d27..65fdbef9 100644
--- a/g4f/Provider/needs_auth/Poe.py
+++ b/g4f/Provider/needs_auth/Poe.py
@@ -26,6 +26,7 @@ class Poe(AbstractProvider):
     needs_auth = True
     supports_gpt_35_turbo = True
     supports_stream = True
+    models = models.keys()
 
     @classmethod
     def create_completion(
diff --git a/g4f/Provider/needs_auth/Raycast.py b/g4f/Provider/needs_auth/Raycast.py
index 07abeda3..b8ec5a97 100644
--- a/g4f/Provider/needs_auth/Raycast.py
+++ b/g4f/Provider/needs_auth/Raycast.py
@@ -16,6 +16,11 @@ class Raycast(AbstractProvider):
     needs_auth              = True
     working                 = True
 
+    models = [
+        "gpt-3.5-turbo",
+        "gpt-4"
+    ]
+
     @staticmethod
     def create_completion(
         model: str,
@@ -25,6 +30,9 @@ class Raycast(AbstractProvider):
         **kwargs,
     ) -> CreateResult:
         auth = kwargs.get('auth')
+        if not auth:
+            raise ValueError("Raycast needs an auth token, pass it with the `auth` parameter")
+
         headers = {
             'Accept': 'application/json',
             'Accept-Language': 'en-US,en;q=0.9',
diff --git a/g4f/Provider/needs_auth/Theb.py b/g4f/Provider/needs_auth/Theb.py
index af690063..c7d7d58e 100644
--- a/g4f/Provider/needs_auth/Theb.py
+++ b/g4f/Provider/needs_auth/Theb.py
@@ -38,6 +38,7 @@ class Theb(AbstractProvider):
     supports_gpt_35_turbo = True
     supports_gpt_4 = True
     supports_stream = True
+    models = models.keys()
 
     @classmethod
     def create_completion(
diff --git a/g4f/Provider/Allyfy.py b/g4f/Provider/not_working/Allyfy.py
index 51f44963..a1c73499 100644
--- a/g4f/Provider/Allyfy.py
+++ b/g4f/Provider/not_working/Allyfy.py
@@ -4,15 +4,15 @@ import asyncio
 import json
 import uuid
 from aiohttp import ClientSession
-from ..typing import AsyncResult, Messages
-from .base_provider import AsyncGeneratorProvider, ProviderModelMixin
-from .helper import format_prompt
+from ...typing import AsyncResult, Messages
+from ..base_provider import AsyncGeneratorProvider, ProviderModelMixin
+from ..helper import format_prompt
 
 
 class Allyfy(AsyncGeneratorProvider, ProviderModelMixin):
     url = "https://allyfy.chat"
     api_endpoint = "https://chatbot.allyfy.chat/api/v1/message/stream/super/chat"
-    working = True
+    working = False
     supports_stream = True
     supports_system_message = True
     supports_message_history = True