mirror of
https://github.com/xtekky/gpt4free.git
synced 2024-12-24 03:23:49 +03:00
8d5d522c4e
* refactor(g4f/Provider/Airforce.py): Enhance Airforce provider with dynamic model fetching * refactor(g4f/Provider/Blackbox.py): Enhance Blackbox AI provider configuration and streamline code * feat(g4f/Provider/RobocodersAPI.py): Add RobocodersAPI new async chat provider * refactor(g4f/client/__init__.py): Improve provider handling in async_generate method * refactor(g4f/models.py): Update provider configurations for multiple models * refactor(g4f/Provider/Blackbox.py): Streamline model configuration and improve response handling * feat(g4f/Provider/DDG.py): Enhance model support and improve conversation handling * refactor(g4f/Provider/Copilot.py): Enhance Copilot provider with model support * refactor(g4f/Provider/AmigoChat.py): update models and improve code structure * chore(g4f/Provider/not_working/AIUncensored.): move AIUncensored to not_working directory * chore(g4f/Provider/not_working/Allyfy.py): remove Allyfy provider * Update (g4f/Provider/not_working/AIUncensored.py g4f/Provider/not_working/__init__.py) * refactor(g4f/Provider/ChatGptEs.py): Implement format_prompt for message handling * refactor(g4f/Provider/Blackbox.py): Update message formatting and improve code structure * refactor(g4f/Provider/LLMPlayground.py): Enhance text generation and error handling * refactor(g4f/Provider/needs_auth/PollinationsAI.py): move PollinationsAI to needs_auth directory * refactor(g4f/Provider/Liaobots.py): Update Liaobots provider models and aliases * feat(g4f/Provider/DeepInfraChat.py): Add new DeepInfra models and aliases * Update (g4f/Provider/__init__.py) * Update (g4f/models.py) * g4f/models.py * Update g4f/models.py * Update g4f/Provider/LLMPlayground.py * Update (g4f/models.py g4f/Provider/Airforce.py g4f/Provider/__init__.py g4f/Provider/LLMPlayground.py) * Update g4f/Provider/__init__.py * Update (g4f/Provider/Airforce.py) --------- Co-authored-by: kqlio67 <kqlio67@users.noreply.github.com>
133 lines
5.0 KiB
Python
133 lines
5.0 KiB
Python
from __future__ import annotations
|
|
|
|
import json
|
|
import random
|
|
from aiohttp import ClientSession, ClientError
|
|
import asyncio
|
|
from itertools import cycle
|
|
|
|
from ...typing import AsyncResult, Messages
|
|
from ..base_provider import AsyncGeneratorProvider, ProviderModelMixin
|
|
from ...image import ImageResponse
|
|
|
|
class AIUncensored(AsyncGeneratorProvider, ProviderModelMixin):
|
|
url = "https://www.aiuncensored.info/ai_uncensored"
|
|
api_endpoints_text = [
|
|
"https://twitterclone-i0wr.onrender.com/api/chat",
|
|
"https://twitterclone-4e8t.onrender.com/api/chat",
|
|
"https://twitterclone-8wd1.onrender.com/api/chat",
|
|
]
|
|
api_endpoints_image = [
|
|
"https://twitterclone-4e8t.onrender.com/api/image",
|
|
"https://twitterclone-i0wr.onrender.com/api/image",
|
|
"https://twitterclone-8wd1.onrender.com/api/image",
|
|
]
|
|
working = False
|
|
supports_stream = True
|
|
supports_system_message = True
|
|
supports_message_history = True
|
|
|
|
default_model = 'TextGenerations'
|
|
text_models = [default_model]
|
|
image_models = ['ImageGenerations']
|
|
models = [*text_models, *image_models]
|
|
|
|
model_aliases = {
|
|
"flux": "ImageGenerations",
|
|
}
|
|
|
|
@staticmethod
|
|
def generate_cipher() -> str:
|
|
"""Generate a cipher in format like '3221229284179118'"""
|
|
return ''.join([str(random.randint(0, 9)) for _ in range(16)])
|
|
|
|
@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.aiuncensored.info',
|
|
'pragma': 'no-cache',
|
|
'priority': 'u=1, i',
|
|
'referer': 'https://www.aiuncensored.info/',
|
|
'sec-ch-ua': '"Not?A_Brand";v="99", "Chromium";v="130"',
|
|
'sec-ch-ua-mobile': '?0',
|
|
'sec-ch-ua-platform': '"Linux"',
|
|
'sec-fetch-dest': 'empty',
|
|
'sec-fetch-mode': 'cors',
|
|
'sec-fetch-site': 'cross-site',
|
|
'user-agent': 'Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/130.0.0.0 Safari/537.36'
|
|
}
|
|
|
|
async with ClientSession(headers=headers) as session:
|
|
if model in cls.image_models:
|
|
prompt = messages[-1]['content']
|
|
data = {
|
|
"prompt": prompt,
|
|
"cipher": cls.generate_cipher()
|
|
}
|
|
|
|
endpoints = cycle(cls.api_endpoints_image)
|
|
|
|
while True:
|
|
endpoint = next(endpoints)
|
|
try:
|
|
async with session.post(endpoint, json=data, proxy=proxy, timeout=10) as response:
|
|
response.raise_for_status()
|
|
response_data = await response.json()
|
|
image_url = response_data['image_url']
|
|
image_response = ImageResponse(images=image_url, alt=prompt)
|
|
yield image_response
|
|
return
|
|
except (ClientError, asyncio.TimeoutError):
|
|
continue
|
|
|
|
elif model in cls.text_models:
|
|
data = {
|
|
"messages": messages,
|
|
"cipher": cls.generate_cipher()
|
|
}
|
|
|
|
endpoints = cycle(cls.api_endpoints_text)
|
|
|
|
while True:
|
|
endpoint = next(endpoints)
|
|
try:
|
|
async with session.post(endpoint, json=data, proxy=proxy, timeout=10) as response:
|
|
response.raise_for_status()
|
|
full_response = ""
|
|
async for line in response.content:
|
|
line = line.decode('utf-8')
|
|
if line.startswith("data: "):
|
|
try:
|
|
json_str = line[6:]
|
|
if json_str != "[DONE]":
|
|
data = json.loads(json_str)
|
|
if "data" in data:
|
|
full_response += data["data"]
|
|
yield data["data"]
|
|
except json.JSONDecodeError:
|
|
continue
|
|
return
|
|
except (ClientError, asyncio.TimeoutError):
|
|
continue
|