Dmitry Afanasyev a95403f594
Feat/fix start (#1)
format code
2023-09-16 20:09:40 +03:00

47 lines
1.3 KiB
Python

import os
import requests
from ...typing import get_type_hints
url = "https://chat9.yqcloud.top/"
model = [
"gpt-3.5-turbo",
]
supports_stream = True
needs_auth = False
def _create_completion(model: str, messages: list, stream: bool, chatId: str, **kwargs):
headers = {
"authority": "api.aichatos.cloud",
"origin": "https://chat9.yqcloud.top",
"referer": "https://chat9.yqcloud.top/",
"user-agent": "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/112.0.0.0 Safari/537.36",
}
json_data = {
"prompt": str(messages),
"userId": f"#/chat/{chatId}",
"network": True,
"apikey": "",
"system": "",
"withoutContext": False,
}
response = requests.post(
"https://api.aichatos.cloud/api/generateStream",
headers=headers,
json=json_data,
stream=True,
)
for token in response.iter_content(chunk_size=2046):
yield (token.decode("utf-8"))
params = f"g4f.Providers.{os.path.basename(__file__)[:-3]} supports: " + "(%s)" % ", ".join(
[
f"{name}: {get_type_hints(_create_completion)[name].__name__}"
for name in _create_completion.__code__.co_varnames[: _create_completion.__code__.co_argcount]
]
)