返回提交历史
Added
g4f/Provider/Aura.py
+56
-0
Modified
g4f/Provider/__init__.py
+1
-1
XFEstudio/gpt4free
Added OpenChat's Aura Model as a new Provider (#1371)
* Added File for new Provider: Aura * Added Aura Provider in init
bcd1cdf4
代码差异
2 个文件
+57
-1
@@ -0,0 +1,56 @@
1
from __future__ import annotations
2
from aiohttp import ClientSession
3
from ..typing import AsyncResult, Messages
4
from .base_provider import AsyncGeneratorProvider
5
6
class Aura(AsyncGeneratorProvider):
7
url = "https://openchat.team"
8
working = True
9
supports_gpt_35_turbo = True
10
11
@classmethod
12
async def create_async_generator(
13
cls,
14
model: str,
15
messages: Messages,
16
proxy: str = None,
17
**kwargs
18
) -> AsyncResult:
19
headers = {
20
"Accept": "*/*",
21
"Accept-Encoding": "gzip, deflate, br",
22
"Accept-Language": "en-GB,en-US;q=0.9,en;q=0.8",
23
"Content-Type": "application/json",
24
"Origin": f"{cls.url}",
25
"Referer": f"{cls.url}/",
26
"Sec-Ch-Ua": '"Not_A Brand";v="8", "Chromium";v="120", "Google Chrome";v="120"',
27
"Sec-Ch-Ua-Mobile": "?0",
28
"Sec-Ch-Ua-Platform": '"Linux"',
29
"Sec-Fetch-Dest": "empty",
30
"Sec-Fetch-Mode": "cors",
31
"Sec-Fetch-Site": "same-origin",
32
"User-Agent": "Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36",
33
}
34
async with ClientSession(headers=headers) as session:
35
system_prompt=""
36
system_message=[]
37
for message in messages:
38
if message["role"]=="system":
39
system_prompt +=message["content"]
40
else:
41
system_message.append(message)
42
43
data = {
44
"model": {
45
"id": "openchat_v3.2_mistral",
46
"name": "OpenChat Aura",
47
"maxLength": 24576,
48
"tokenLimit": 8192
49
},
50
"messages": system_message,
51
"key": "",
52
"prompt": f"{system_prompt}",
53
"temperature": 0.5
54
}
55
async with session.post(f"{cls.url}/api/chat",json=data,proxy=proxy) as response:
56
yield await response.text()
@@ -6,7 +6,7 @@ from .deprecated import *
6
6
from .needs_auth import *
7
7
from .unfinished import *
8
8
from .selenium import *
9
9
from .Aura import Aura
10
10
from .AiAsk import AiAsk
11
11
from .Aichat import Aichat
12
12
from .AiChatOnline import AiChatOnline