diff options
Diffstat (limited to 'g4f/Provider/Feedough.py')
-rw-r--r-- | g4f/Provider/Feedough.py | 52 |
1 files changed, 52 insertions, 0 deletions
diff --git a/g4f/Provider/Feedough.py b/g4f/Provider/Feedough.py new file mode 100644 index 00000000..2ce4561e --- /dev/null +++ b/g4f/Provider/Feedough.py @@ -0,0 +1,52 @@ +from __future__ import annotations + +import json +from aiohttp import ClientSession + +from ..typing import AsyncResult, Messages +from .base_provider import AsyncGeneratorProvider +from .helper import format_prompt + + +class Feedough(AsyncGeneratorProvider): + url = "https://www.feedough.com" + working = True + supports_gpt_35_turbo = True + + @classmethod + async def create_async_generator( + cls, + model: str, + messages: Messages, + proxy: str = None, + **kwargs + ) -> AsyncResult: + headers = { + "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/122.0.0.0 Safari/537.36", + "Accept": "*/*", + "Accept-Language": "en-US,en;q=0.5", + "Accept-Encoding": "gzip, deflate, br", + "Referer": "https://www.feedough.com/ai-prompt-generator/", + "Content-Type": "application/x-www-form-urlencoded;charset=UTF-8", + "Origin": "https://www.feedough.com", + "DNT": "1", + "Sec-GPC": "1", + "Connection": "keep-alive", + "Sec-Fetch-Dest": "empty", + "Sec-Fetch-Mode": "cors", + "Sec-Fetch-Site": "same-origin", + "TE": "trailers", + } + async with ClientSession(headers=headers) as session: + prompt = format_prompt(messages) + data = { + "action": "aixg_generate", + "prompt": prompt, + } + async with session.post(f"{cls.url}/wp-admin/admin-ajax.php", data=data, proxy=proxy) as response: + response.raise_for_status() + response_text = await response.text() + response_json = json.loads(response_text) + if response_json["success"]: + message = response_json["data"]["message"] + yield message |