返回提交历史
Modified
g4f/Provider/Aura.py
+11
-12
XFEstudio/gpt4free
Fix streaming in Aura
554609dc
代码差异
1 个文件
+11
-12
@@ -32,25 +32,24 @@ class Aura(AsyncGeneratorProvider):
32
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
33
}
34
34
async with ClientSession(headers=headers) as session:
35
system_prompt=""
36
system_message=[]
35
new_messages = []
36
system_message = []
37
37
for message in messages:
38
if message["role"]=="system":
39
system_prompt +=message["content"]
38
if message["role"] == "system":
39
system_message.append(message["content"])
40
40
else:
41
system_message.append(message)
42
41
new_messages.append(message)
43
42
data = {
44
43
"model": {
45
44
"id": "openchat_v3.2_mistral",
46
45
"name": "OpenChat Aura",
47
46
"maxLength": 24576,
48
47
"tokenLimit": 8192
49
},
50
"messages": system_message,
48
},
49
"messages": new_messages,
51
50
"key": "",
52
"prompt": f"{system_prompt}",
51
"prompt": "\n".join(system_message),
53
52
"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()
53
}
54
async with session.post(f"{cls.url}/api/chat", json=data, proxy=proxy) as response:
55
return response.content.iter_any()