minor improvements for release

kqlio67 · Nov 15, 2024 · 6ac5454 · 6ac5454
1 parent 66d3cec
commit 6ac5454
Show file tree

Hide file tree

Showing 8 changed files with 188 additions and 179 deletions.
diff --git 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
@@ -12,7 +12,6 @@
 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
@@ -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
@@ -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
@@ -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 @@ def create_completion(
         **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
@@ -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 → g4f/Provider/not_working/Allyfy.py b/g4f/Provider/Allyfy.py → g4f/Provider/not_working/Allyfy.py
@@ -4,15 +4,15 @@
 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