diff options
Diffstat (limited to 'g4f/Provider')
-rw-r--r-- | g4f/Provider/Pi.py | 1 | ||||
-rw-r--r-- | g4f/Provider/__init__.py | 1 | ||||
-rw-r--r-- | g4f/Provider/needs_auth/Gemini.py | 1 | ||||
-rw-r--r-- | g4f/Provider/needs_auth/Poe.py | 1 | ||||
-rw-r--r-- | g4f/Provider/needs_auth/Raycast.py | 8 | ||||
-rw-r--r-- | g4f/Provider/needs_auth/Theb.py | 1 | ||||
-rw-r--r-- | g4f/Provider/not_working/Allyfy.py (renamed from g4f/Provider/Allyfy.py) | 8 |
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 |