mirror of
https://github.com/ClovertaTheTrilobita/SanYeCao-Nonebot.git
synced 2026-04-01 22:04:51 +00:00
222e6f788e
- 在全局范围内添加 silicon_flow_key 配置 - 在调用 chat.completions.create 方法前设置 openai.api_key 和 openai.base_url- 更新请求 headers 中的 Authorization 字段，使用 silicon_flow_key
75 lines
2.6 KiB
Python
75 lines
2.6 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,silicon_flow_key
|
||
import aiohttp
|
||
|
||
"""
|
||
来源：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:
|
||
"""
|
||
openai.api_key = deepseek_key
|
||
openai.base_url = deepseek_url
|
||
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
|
||
|
||
async def silicon_flow(group_openid, content):
|
||
await GroupChatRole.save_chat_history(group_openid, {"role": "user", "content": content})
|
||
messages = await GroupChatRole.get_chat_history(group_openid)
|
||
url = "https://api.siliconflow.cn/v1/chat/completions"
|
||
payload = {
|
||
"model": "Pro/deepseek-ai/DeepSeek-V3",
|
||
"stream": False,
|
||
"messages": messages
|
||
}
|
||
headers = {
|
||
"Authorization": f"Bearer {silicon_flow_key}",
|
||
"Content-Type": "application/json"
|
||
}
|
||
|
||
async with aiohttp.ClientSession() as session:
|
||
async with session.post(url, json=payload, headers=headers) as response:
|
||
result = await response.json()
|
||
print(result)
|
||
reply_content = result["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("你拽什么啊？"))
|