mirror of
https://github.com/xtekky/gpt4free.git
synced 2025-12-05 18:20:35 -08:00
46 lines
1.4 KiB
Python
46 lines
1.4 KiB
Python
from .base_provider import AsyncGeneratorProvider, ProviderModelMixin
|
|
from ..typing import AsyncResult, Messages
|
|
from ..requests import DEFAULT_HEADERS
|
|
from aiohttp import ClientSession
|
|
|
|
class ItalyGPT(AsyncGeneratorProvider, ProviderModelMixin):
|
|
label = "ItalyGPT"
|
|
url = "https://italygpt.it"
|
|
working = True
|
|
supports_system_message = True
|
|
supports_message_history = True
|
|
|
|
default_model = "gpt-4o"
|
|
models = [default_model]
|
|
|
|
@classmethod
|
|
async def create_async_generator(
|
|
cls,
|
|
model: str,
|
|
messages: Messages,
|
|
stream: bool = True,
|
|
proxy: str = None,
|
|
**kwargs
|
|
) -> AsyncResult:
|
|
model = cls.get_model(model)
|
|
headers = {
|
|
**DEFAULT_HEADERS,
|
|
"content-type": "application/json",
|
|
"origin": "https://italygpt.it",
|
|
"referer": "https://italygpt.it/",
|
|
}
|
|
payload = {
|
|
"messages": messages,
|
|
"stream": stream,
|
|
}
|
|
async with ClientSession() as session:
|
|
async with session.post(
|
|
f"{cls.url}/api/chat",
|
|
json=payload,
|
|
headers=headers,
|
|
proxy=proxy,
|
|
) as resp:
|
|
resp.raise_for_status()
|
|
async for chunk in resp.content.iter_any():
|
|
if chunk:
|
|
yield chunk.decode()
|