From 66d3cecc59de11970770777a14195ba065e7498a Mon Sep 17 00:00:00 2001 From: Tekky <98614666+xtekky@users.noreply.github.com> Date: Fri, 15 Nov 2024 11:24:56 +0100 Subject: Ai4Chat not working --- g4f/Provider/Ai4Chat.py | 89 ------------------------------------- g4f/Provider/not_working/Ai4Chat.py | 89 +++++++++++++++++++++++++++++++++++++ g4f/models.py | 3 +- 3 files changed, 91 insertions(+), 90 deletions(-) delete mode 100644 g4f/Provider/Ai4Chat.py create mode 100644 g4f/Provider/not_working/Ai4Chat.py diff --git a/g4f/Provider/Ai4Chat.py b/g4f/Provider/Ai4Chat.py deleted file mode 100644 index 56b47e5c..00000000 --- a/g4f/Provider/Ai4Chat.py +++ /dev/null @@ -1,89 +0,0 @@ -from __future__ import annotations - -import json -import re -import logging -from aiohttp import ClientSession - -from ..typing import AsyncResult, Messages -from .base_provider import AsyncGeneratorProvider, ProviderModelMixin -from .helper import format_prompt - -logger = logging.getLogger(__name__) - -class Ai4Chat(AsyncGeneratorProvider, ProviderModelMixin): - label = "AI4Chat" - url = "https://www.ai4chat.co" - api_endpoint = "https://www.ai4chat.co/generate-response" - working = False - supports_stream = True - supports_system_message = True - supports_message_history = True - - default_model = 'gpt-4' - models = [default_model] - - model_aliases = {} - - @classmethod - def get_model(cls, model: str) -> str: - if model in cls.models: - return model - elif model in cls.model_aliases: - return cls.model_aliases[model] - else: - return cls.default_model - - @classmethod - async def create_async_generator( - cls, - model: str, - messages: Messages, - proxy: str = None, - **kwargs - ) -> AsyncResult: - model = cls.get_model(model) - - headers = { - "accept": "*/*", - "accept-language": "en-US,en;q=0.9", - "cache-control": "no-cache", - "content-type": "application/json", - "origin": "https://www.ai4chat.co", - "pragma": "no-cache", - "priority": "u=1, i", - "referer": "https://www.ai4chat.co/gpt/talkdirtytome", - "sec-ch-ua": '"Chromium";v="129", "Not=A?Brand";v="8"', - "sec-ch-ua-mobile": "?0", - "sec-ch-ua-platform": '"Linux"', - "sec-fetch-dest": "empty", - "sec-fetch-mode": "cors", - "sec-fetch-site": "same-origin", - "user-agent": "Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/129.0.0.0 Safari/537.36" - } - - async with ClientSession(headers=headers) as session: - data = { - "messages": [ - { - "role": "user", - "content": format_prompt(messages) - } - ] - } - - try: - async with session.post(cls.api_endpoint, json=data, proxy=proxy) as response: - response.raise_for_status() - result = await response.text() - - json_result = json.loads(result) - - message = json_result.get("message", "") - - clean_message = re.sub(r'<[^>]+>', '', message) - - yield clean_message - except Exception as e: - logger.exception("Error while calling AI 4Chat API: %s", e) - yield f"Error: {e}" diff --git a/g4f/Provider/not_working/Ai4Chat.py b/g4f/Provider/not_working/Ai4Chat.py new file mode 100644 index 00000000..9b55e4ff --- /dev/null +++ b/g4f/Provider/not_working/Ai4Chat.py @@ -0,0 +1,89 @@ +from __future__ import annotations + +import json +import re +import logging +from aiohttp import ClientSession + +from ...typing import AsyncResult, Messages +from ..base_provider import AsyncGeneratorProvider, ProviderModelMixin +from ..helper import format_prompt + +logger = logging.getLogger(__name__) + +class Ai4Chat(AsyncGeneratorProvider, ProviderModelMixin): + label = "AI4Chat" + url = "https://www.ai4chat.co" + api_endpoint = "https://www.ai4chat.co/generate-response" + working = False + supports_stream = True + supports_system_message = True + supports_message_history = True + + default_model = 'gpt-4' + models = [default_model] + + model_aliases = {} + + @classmethod + def get_model(cls, model: str) -> str: + if model in cls.models: + return model + elif model in cls.model_aliases: + return cls.model_aliases[model] + else: + return cls.default_model + + @classmethod + async def create_async_generator( + cls, + model: str, + messages: Messages, + proxy: str = None, + **kwargs + ) -> AsyncResult: + model = cls.get_model(model) + + headers = { + "accept": "*/*", + "accept-language": "en-US,en;q=0.9", + "cache-control": "no-cache", + "content-type": "application/json", + "origin": "https://www.ai4chat.co", + "pragma": "no-cache", + "priority": "u=1, i", + "referer": "https://www.ai4chat.co/gpt/talkdirtytome", + "sec-ch-ua": '"Chromium";v="129", "Not=A?Brand";v="8"', + "sec-ch-ua-mobile": "?0", + "sec-ch-ua-platform": '"Linux"', + "sec-fetch-dest": "empty", + "sec-fetch-mode": "cors", + "sec-fetch-site": "same-origin", + "user-agent": "Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/129.0.0.0 Safari/537.36" + } + + async with ClientSession(headers=headers) as session: + data = { + "messages": [ + { + "role": "user", + "content": format_prompt(messages) + } + ] + } + + try: + async with session.post(cls.api_endpoint, json=data, proxy=proxy) as response: + response.raise_for_status() + result = await response.text() + + json_result = json.loads(result) + + message = json_result.get("message", "") + + clean_message = re.sub(r'<[^>]+>', '', message) + + yield clean_message + except Exception as e: + logger.exception("Error while calling AI 4Chat API: %s", e) + yield f"Error: {e}" diff --git a/g4f/models.py b/g4f/models.py index 6d3ef2ad..9c6178be 100644 --- a/g4f/models.py +++ b/g4f/models.py @@ -2,9 +2,10 @@ from __future__ import annotations from dataclasses import dataclass +from .Provider.not_working import Ai4Chat + from .Provider import IterListProvider, ProviderType from .Provider import ( - Ai4Chat, AIChatFree, Airforce, AIUncensored, -- cgit v1.2.3