mirror of
https://github.com/ClovertaTheTrilobita/SanYeCao-Nonebot.git
synced 2026-04-01 22:04:51 +00:00
53 lines
1.7 KiB
Python
53 lines
1.7 KiB
Python
import openai
|
||
import requests
|
||
from src.clover_sqlite.models.chat import GroupChatRole
|
||
from src.configs.api_config import v3url, v3key, deepseek_url, deepseek_key
|
||
|
||
openai.api_key = deepseek_key
|
||
openai.base_url = deepseek_url
|
||
|
||
"""
|
||
来源：https://api.v36.cm
|
||
"""
|
||
async def v3_chat(group_openid,content):
|
||
|
||
await GroupChatRole.save_chat_history(group_openid, {"role": "user", "content": content})
|
||
messages = await GroupChatRole.get_chat_history(group_openid)
|
||
headers = {"Content-Type": "application/json", "Authorization": v3key}
|
||
data = {
|
||
"model": "gpt-3.5-turbo-0125",
|
||
"messages": messages,
|
||
"max_tokens": 1688,
|
||
"temperature": 0.5,
|
||
"stream": False
|
||
}
|
||
response = requests.post(v3url, headers=headers, json=data)
|
||
reply_content = response.json().get('choices')[0].get('message').get('content')
|
||
await GroupChatRole.save_chat_history(group_openid, {"role": "assistant", "content": reply_content})
|
||
return reply_content
|
||
|
||
|
||
"""
|
||
来源:https://api.deepseek.com
|
||
"""
|
||
async def deepseek_chat(group_openid,content):
|
||
"""
|
||
ai 角色扮演聊天
|
||
:param group_openid:
|
||
:param content:
|
||
:return:
|
||
"""
|
||
|
||
await GroupChatRole.save_chat_history(group_openid, {"role": "user", "content": content})
|
||
messages = await GroupChatRole.get_chat_history(group_openid)
|
||
completion = openai.chat.completions.create(
|
||
model="deepseek-chat",
|
||
messages=messages,
|
||
stream=False
|
||
)
|
||
reply_content = completion.choices[0].message.content
|
||
await GroupChatRole.save_chat_history(group_openid, {"role": "assistant", "content": reply_content})
|
||
return reply_content
|
||
|
||
if __name__ == '__main__':
|
||
print(deepseek_chat("你拽什么啊？"))
|