mirror of
https://github.com/LiteyukiStudio/nonebot-plugin-marshoai.git
synced 2025-08-01 16:39:52 +00:00
✨ 支持保存历史对话, 规范代码
This commit is contained in:
@ -1,30 +1,32 @@
|
||||
from nonebot import on_command
|
||||
from nonebot.adapters import Message, Event
|
||||
from nonebot.params import CommandArg
|
||||
from nonebot.permission import SUPERUSER
|
||||
from nonebot_plugin_alconna import on_alconna, MsgTarget
|
||||
from nonebot_plugin_alconna.uniseg import UniMessage, UniMsg
|
||||
from arclet.alconna import Alconna, Args, AllParam
|
||||
from .util import *
|
||||
import traceback
|
||||
import contextlib
|
||||
from azure.ai.inference.aio import ChatCompletionsClient
|
||||
import traceback
|
||||
from typing import Optional
|
||||
|
||||
from arclet.alconna import Alconna, Args, AllParam
|
||||
from azure.ai.inference.models import (
|
||||
UserMessage,
|
||||
AssistantMessage,
|
||||
ContentItem,
|
||||
TextContentItem,
|
||||
ImageContentItem,
|
||||
ImageUrl,
|
||||
CompletionsFinishReason,
|
||||
)
|
||||
from azure.core.credentials import AzureKeyCredential
|
||||
from typing import Any, Optional
|
||||
from nonebot import on_command
|
||||
from nonebot.adapters import Message, Event
|
||||
from nonebot.params import CommandArg
|
||||
from nonebot.permission import SUPERUSER
|
||||
from nonebot_plugin_alconna import on_alconna, MsgTarget
|
||||
from nonebot_plugin_alconna.uniseg import UniMessage, UniMsg
|
||||
from nonebot import get_driver
|
||||
from nonebot_plugin_waiter import prompt
|
||||
|
||||
from .metadata import metadata
|
||||
from .config import config
|
||||
from .models import MarshoContext
|
||||
from .constants import *
|
||||
from .metadata import metadata
|
||||
from .models import MarshoContext
|
||||
from .util import *
|
||||
|
||||
driver = get_driver()
|
||||
|
||||
changemodel_cmd = on_command("changemodel", permission=SUPERUSER)
|
||||
resetmem_cmd = on_command("reset")
|
||||
@ -46,14 +48,16 @@ nickname_cmd = on_alconna(
|
||||
Alconna(
|
||||
"nickname",
|
||||
Args["name?", str],
|
||||
)
|
||||
)
|
||||
)
|
||||
refresh_data = on_alconna("refresh_data", permission=SUPERUSER)
|
||||
refresh_data_cmd = on_alconna("refresh_data", permission=SUPERUSER)
|
||||
|
||||
model_name = config.marshoai_default_model
|
||||
context = MarshoContext()
|
||||
token = config.marshoai_token
|
||||
endpoint = config.marshoai_azure_endpoint
|
||||
client = ChatCompletionsClient(endpoint=endpoint, credential=AzureKeyCredential(token))
|
||||
target_list = []
|
||||
|
||||
|
||||
@add_usermsg_cmd.handle()
|
||||
@ -84,9 +88,9 @@ async def contexts(target: MsgTarget):
|
||||
|
||||
@save_context_cmd.handle()
|
||||
async def save_context(target: MsgTarget, arg: Message = CommandArg()):
|
||||
contexts = context.build(target.id, target.private)[1:]
|
||||
contexts_data = context.build(target.id, target.private)[1:]
|
||||
if msg := arg.extract_plain_text():
|
||||
await save_context_to_json(msg, contexts)
|
||||
await save_context_to_json(msg, contexts_data, "context")
|
||||
await save_context_cmd.finish("已保存上下文")
|
||||
|
||||
|
||||
@ -94,7 +98,7 @@ async def save_context(target: MsgTarget, arg: Message = CommandArg()):
|
||||
async def load_context(target: MsgTarget, arg: Message = CommandArg()):
|
||||
if msg := arg.extract_plain_text():
|
||||
context.set_context(
|
||||
await load_context_from_json(msg), target.id, target.private
|
||||
await load_context_from_json(msg, "context"), target.id, target.private
|
||||
)
|
||||
await load_context_cmd.finish("已加载并覆盖上下文")
|
||||
|
||||
@ -128,29 +132,30 @@ async def nickname(event: Event, name=None):
|
||||
await set_nickname(user_id, name)
|
||||
await nickname_cmd.finish("已设置昵称为:" + name)
|
||||
|
||||
@refresh_data.handle()
|
||||
|
||||
@refresh_data_cmd.handle()
|
||||
async def refresh_data():
|
||||
await refresh_nickname_json()
|
||||
await refresh_data.finish("已刷新数据")
|
||||
await refresh_praises_json()
|
||||
await refresh_data_cmd.finish("已刷新数据")
|
||||
|
||||
|
||||
@marsho_cmd.handle()
|
||||
async def marsho(target: MsgTarget, event: Event, text: Optional[UniMsg] = None):
|
||||
global target_list
|
||||
if not text:
|
||||
# 发送说明
|
||||
await UniMessage(metadata.usage + "\n当前使用的模型:" + model_name).send()
|
||||
await marsho_cmd.finish(INTRODUCTION)
|
||||
return
|
||||
|
||||
try:
|
||||
|
||||
user_id = event.get_user_id()
|
||||
nicknames = await get_nicknames()
|
||||
nickname = nicknames.get(user_id, "")
|
||||
user_nickname = nicknames.get(user_id, "")
|
||||
if nickname != "":
|
||||
nickname_prompt = f"\n*此消息的说话者:{nickname}*"
|
||||
nickname_prompt = f"\n*此消息的说话者:{user_nickname}*"
|
||||
else:
|
||||
nickname_prompt = ""
|
||||
#user_nickname = event.sender.nickname
|
||||
#nickname_prompt = f"\n*此消息的说话者:{user_nickname}"
|
||||
user_nickname = event.sender.nickname # 未设置昵称时获取用户名
|
||||
nickname_prompt = f"\n*此消息的说话者:{user_nickname}"
|
||||
if config.marshoai_enable_nickname_tip:
|
||||
await UniMessage(
|
||||
"*你未设置自己的昵称。推荐使用'nickname [昵称]'命令设置昵称来获得个性化(可能)回答。"
|
||||
@ -167,7 +172,7 @@ async def marsho(target: MsgTarget, event: Event, text: Optional[UniMsg] = None)
|
||||
usermsg += str(i.data["text"] + nickname_prompt)
|
||||
elif i.type == "image":
|
||||
if is_support_image_model:
|
||||
usermsg.append(
|
||||
usermsg.append(
|
||||
ImageContentItem(
|
||||
image_url=ImageUrl(url=str(await get_image_b64(i.data["url"])))
|
||||
)
|
||||
@ -175,17 +180,25 @@ async def marsho(target: MsgTarget, event: Event, text: Optional[UniMsg] = None)
|
||||
elif config.marshoai_enable_support_image_tip:
|
||||
await UniMessage("*此模型不支持图片处理。").send()
|
||||
context_msg = context.build(target.id, target.private)
|
||||
if is_reasoning_model: context_msg = context_msg[1:] #o1等推理模型不支持系统提示词,故截断
|
||||
if not context_msg:
|
||||
context_msg = list(await load_context_from_json(f"back_up_context_{target.id}", "context/backup"))
|
||||
await save_context_to_json(f"back_up_context_{target.id}", [], "context/backup")
|
||||
msg_prompt = get_prompt()
|
||||
context_msg = [msg_prompt] + context_msg
|
||||
print(str(context_msg))
|
||||
target_list.append([target.id, target.private])
|
||||
if is_reasoning_model:
|
||||
context_msg = context_msg[1:]
|
||||
# o1等推理模型不支持系统提示词,故截断
|
||||
response = await make_chat(
|
||||
client=client,
|
||||
model_name=model_name,
|
||||
msg=context_msg
|
||||
+ [UserMessage(content=usermsg)],
|
||||
msg=context_msg + [UserMessage(content=usermsg)],
|
||||
)
|
||||
# await UniMessage(str(response)).send()
|
||||
choice = response.choices[0]
|
||||
if (
|
||||
choice["finish_reason"] == CompletionsFinishReason.STOPPED
|
||||
choice["finish_reason"] == CompletionsFinishReason.STOPPED
|
||||
): # 当对话成功时,将dict的上下文添加到上下文类中
|
||||
context.append(
|
||||
UserMessage(content=usermsg).as_dict(), target.id, target.private
|
||||
@ -207,12 +220,13 @@ with contextlib.suppress(ImportError): # 优化先不做()
|
||||
import nonebot.adapters.onebot.v11 # type: ignore
|
||||
from .azure_onebot import poke_notify
|
||||
|
||||
|
||||
@poke_notify.handle()
|
||||
async def poke(event: Event, target: MsgTarget):
|
||||
async def poke(event: Event):
|
||||
|
||||
user_id = event.get_user_id()
|
||||
nicknames = await get_nicknames()
|
||||
nickname = nicknames.get(user_id, "")
|
||||
user_nickname = nicknames.get(user_id, "")
|
||||
try:
|
||||
if config.marshoai_poke_suffix != "":
|
||||
response = await make_chat(
|
||||
@ -221,7 +235,7 @@ with contextlib.suppress(ImportError): # 优化先不做()
|
||||
msg=[
|
||||
get_prompt(),
|
||||
UserMessage(
|
||||
content=f"*{nickname}{config.marshoai_poke_suffix}"
|
||||
content=f"*{user_nickname}{config.marshoai_poke_suffix}"
|
||||
),
|
||||
],
|
||||
)
|
||||
@ -234,3 +248,11 @@ with contextlib.suppress(ImportError): # 优化先不做()
|
||||
await UniMessage(str(e) + suggest_solution(str(e))).send()
|
||||
traceback.print_exc()
|
||||
return
|
||||
|
||||
|
||||
@driver.on_shutdown
|
||||
async def save_context():
|
||||
for target_info in target_list:
|
||||
target_id, target_private = target_info
|
||||
contexts_data = context.build(target_id, target_private)[1:]
|
||||
await save_context_to_json(f"back_up_context_{target_id}", contexts_data, "context/backup")
|
||||
|
Reference in New Issue
Block a user