mirror of
https://github.com/xtekky/gpt4free.git
synced 2024-11-03 03:40:12 +00:00
901595b10f
Use `from __future__ import annotations avoid `dict` and `list` cause "TypeErro: 'type' object is not subscriptable". Refer to the following Stack Overflow discussions for more information: 1. https://stackoverflow.com/questions/75202610/typeerror-type-object-is-not-subscriptable-python 2. https://stackoverflow.com/questions/59101121/type-hint-for-a-dict-gives-typeerror-type-object-is-not-subscriptable
50 lines
1.2 KiB
Python
50 lines
1.2 KiB
Python
from __future__ import annotations
|
|
|
|
import time
|
|
|
|
import requests
|
|
|
|
from ..typing import Any, CreateResult
|
|
from .base_provider import BaseProvider
|
|
|
|
|
|
class Acytoo(BaseProvider):
|
|
url = 'https://chat.acytoo.com/'
|
|
working = True
|
|
supports_gpt_35_turbo = True
|
|
|
|
@classmethod
|
|
def create_completion(
|
|
cls,
|
|
model: str,
|
|
messages: list[dict[str, str]],
|
|
stream: bool, **kwargs: Any) -> CreateResult:
|
|
|
|
response = requests.post(f'{cls.url}api/completions',
|
|
headers=_create_header(), json=_create_payload(messages, kwargs.get('temperature', 0.5)))
|
|
|
|
response.raise_for_status()
|
|
response.encoding = 'utf-8'
|
|
|
|
yield response.text
|
|
|
|
|
|
def _create_header():
|
|
return {
|
|
'accept': '*/*',
|
|
'content-type': 'application/json',
|
|
}
|
|
|
|
|
|
def _create_payload(messages: list[dict[str, str]], temperature):
|
|
payload_messages = [
|
|
message | {'createdAt': int(time.time()) * 1000} for message in messages
|
|
]
|
|
|
|
return {
|
|
'key' : '',
|
|
'model' : 'gpt-3.5-turbo',
|
|
'messages' : payload_messages,
|
|
'temperature' : temperature,
|
|
'password' : ''
|
|
} |