diff options
| -rw-r--r-- | g4f/Provider/GeminiPro.py | 2 | ||||
| -rw-r--r-- | g4f/Provider/Pizzagpt.py | 48 | ||||
| -rw-r--r-- | g4f/Provider/__init__.py | 1 |
3 files changed, 50 insertions, 1 deletions
diff --git a/g4f/Provider/GeminiPro.py b/g4f/Provider/GeminiPro.py index 56c211ef..b225c26c 100644 --- a/g4f/Provider/GeminiPro.py +++ b/g4f/Provider/GeminiPro.py @@ -18,7 +18,7 @@ class GeminiPro(AsyncGeneratorProvider, ProviderModelMixin): needs_auth = True default_model = "gemini-1.5-pro-latest" default_vision_model = default_model - models = [default_model, "gemini-pro", "gemini-pro-vision"] + models = [default_model, "gemini-pro", "gemini-pro-vision", "gemini-1.5-flash"] @classmethod async def create_async_generator( diff --git a/g4f/Provider/Pizzagpt.py b/g4f/Provider/Pizzagpt.py new file mode 100644 index 00000000..47e74ee3 --- /dev/null +++ b/g4f/Provider/Pizzagpt.py @@ -0,0 +1,48 @@ +import json +from aiohttp import ClientSession + +from ..typing import Messages, AsyncResult +from .base_provider import AsyncGeneratorProvider + +class Pizzagpt(AsyncGeneratorProvider): + url = "https://www.pizzagpt.it" + api_endpoint = "/api/chatx-completion" + supports_message_history = False + supports_gpt_35_turbo = True + working = True + + @classmethod + async def create_async_generator( + cls, + model: str, + messages: Messages, + proxy: str = None, + **kwargs + ) -> AsyncResult: + payload = { + "question": messages[-1]["content"] + } + headers = { + "Accept": "application/json", + "Accept-Encoding": "gzip, deflate, br, zstd", + "Accept-Language": "en-US,en;q=0.9", + "Content-Type": "application/json", + "Origin": cls.url, + "Referer": f"{cls.url}/en", + "Sec-Fetch-Dest": "empty", + "Sec-Fetch-Mode": "cors", + "Sec-Fetch-Site": "same-origin", + "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/126.0.0.0 Safari/537.36", + "X-Secret": "Marinara" + } + + async with ClientSession() as session: + async with session.post( + f"{cls.url}{cls.api_endpoint}", + json=payload, + proxy=proxy, + headers=headers + ) as response: + response.raise_for_status() + response_json = await response.json() + yield response_json["answer"]["content"] diff --git a/g4f/Provider/__init__.py b/g4f/Provider/__init__.py index 1c387087..dab6f5d5 100644 --- a/g4f/Provider/__init__.py +++ b/g4f/Provider/__init__.py @@ -44,6 +44,7 @@ from .MetaAIAccount import MetaAIAccount from .Ollama import Ollama from .PerplexityLabs import PerplexityLabs from .Pi import Pi +from .Pizzagpt import Pizzagpt from .Replicate import Replicate from .ReplicateImage import ReplicateImage from .Vercel import Vercel |