LiteyukiStudio/nonebot-plugin-marshoai
4
4
Fork 0
mirror of https://github.com/LiteyukiStudio/nonebot-plugin-marshoai.git synced 2025-01-26 18:12:47 +08:00
Code Issues Packages Projects Releases Wiki Activity

⚡ 修正少量代码, 历史记录增加前缀

Browse Source
This commit is contained in:
MoeSnowyFox 2024-11-17 02:36:03 +08:00
parent 244153d4ac
commit 790f5a3dbe
1 changed files with 11 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
nonebot_plugin_marshoai/azure.py
View File

@ -49,7 +49,7 @@ nickname_cmd = on_alconna(
Args["name?", str], Args["name?", str],
) )
) )
refresh_data_cmd = on_alconna("refresh_data", permission=SUPERUSER) refresh_data_cmd = on_command("refresh_data", permission=SUPERUSER)
model_name = config.marshoai_default_model model_name = config.marshoai_default_model
context = MarshoContext() context = MarshoContext()
@ -181,7 +181,11 @@ async def marsho(target: MsgTarget, event: Event, text: Optional[UniMsg] = None)
await UniMessage("*此模型不支持图片处理。").send() await UniMessage("*此模型不支持图片处理。").send()
context_msg = context.build(target.id, target.private) context_msg = context.build(target.id, target.private)
if not context_msg and target.id not in loaded_target_list: if not context_msg and target.id not in loaded_target_list:
context_msg = list(await load_context_from_json(f"back_up_context_{target.id}", "contexts/backup")) if target.private:
channel_id = "private_" + target.id
else:
channel_id = "group_" + target.id
context_msg = list(await load_context_from_json(f"back_up_context_{channel_id}", "contexts/backup"))
loaded_target_list.append(target.id) loaded_target_list.append(target.id)
context_msg = [get_prompt()] + context_msg context_msg = [get_prompt()] + context_msg
target_list.append([target.id, target.private]) target_list.append([target.id, target.private])
@ -253,4 +257,8 @@ async def save_context():
for target_info in target_list: for target_info in target_list:
target_id, target_private = target_info target_id, target_private = target_info
contexts_data = context.build(target_id, target_private)[1:] contexts_data = context.build(target_id, target_private)[1:]
await save_context_to_json(f"back_up_context_{target_id}", contexts_data, "contexts/backup") if target_private:
channel_id = "private_" + target_id
else:
channel_id = "group_" + target_id
await save_context_to_json(f"back_up_context_{channel_id}", contexts_data, "contexts/backup")