gpt4free/g4f/Provider/Blackbox.py

78 lines
2.7 KiB
Python
Raw Normal View History

from __future__ import annotations
import uuid
import secrets
2024-07-08 23:49:38 +03:00
import re
from aiohttp import ClientSession, ClientResponse
from typing import AsyncGenerator, Optional
from ..typing import AsyncResult, Messages, ImageType
from ..image import to_data_uri
2024-07-08 23:49:38 +03:00
from .base_provider import AsyncGeneratorProvider, ProviderModelMixin
2024-07-08 23:49:38 +03:00
class Blackbox(AsyncGeneratorProvider, ProviderModelMixin):
url = "https://www.blackbox.ai"
working = True
2024-07-08 23:49:38 +03:00
default_model = 'blackbox'
@classmethod
async def create_async_generator(
cls,
model: str,
messages: Messages,
2024-07-08 23:49:38 +03:00
proxy: Optional[str] = None,
image: Optional[ImageType] = None,
image_name: Optional[str] = None,
**kwargs
2024-07-08 23:49:38 +03:00
) -> AsyncGenerator[str, None]:
if image is not None:
messages[-1]["data"] = {
2024-07-08 23:49:38 +03:00
"fileText": image_name,
"imageBase64": to_data_uri(image)
}
2024-07-08 23:49:38 +03:00
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": cls.url,
"Content-Type": "application/json",
"Origin": cls.url,
"DNT": "1",
"Sec-GPC": "1",
"Alt-Used": "www.blackbox.ai",
"Connection": "keep-alive",
}
2024-07-08 23:49:38 +03:00
async with ClientSession(headers=headers) as session:
random_id = secrets.token_hex(16)
random_user_id = str(uuid.uuid4())
2024-07-08 23:49:38 +03:00
data = {
"messages": messages,
"id": random_id,
"userId": random_user_id,
"codeModelMode": True,
"agentMode": {},
"trendingAgentMode": {},
"isMicMode": False,
"isChromeExt": False,
"playgroundMode": False,
"webSearchMode": False,
"userSystemPrompt": "",
2024-07-08 23:49:38 +03:00
"githubToken": None,
"maxTokens": None
}
2024-07-08 23:49:38 +03:00
async with session.post(
f"{cls.url}/api/chat", json=data, proxy=proxy
) as response: # type: ClientResponse
response.raise_for_status()
2024-07-08 23:49:38 +03:00
async for chunk in response.content.iter_any():
if chunk:
2024-07-08 23:49:38 +03:00
# Decode the chunk and clean up unwanted prefixes using a regex
decoded_chunk = chunk.decode()
cleaned_chunk = re.sub(r'\$@\$.+?\$@\$|\$@\$', '', decoded_chunk)
yield cleaned_chunk