✨ 添加记忆功能命令：实现查看和重置用户记忆的指令

nonebot_plugin_marshoai/plugins/marshoai_basic/__init__.py

@@ -1,6 +1,6 @@
 import os
 
-from zhDateTime import DateTime
+from zhDateTime import DateTime  # type: ignore
 
 from nonebot_plugin_marshoai.plugin import PluginMetadata, String, on_function_call
 

nonebot_plugin_marshoai/plugins/marshoai_memory/__init__.py

@@ -13,6 +13,7 @@ from nonebot_plugin_marshoai.instances import client
 from nonebot_plugin_marshoai.plugin import PluginMetadata, on_function_call
 from nonebot_plugin_marshoai.plugin.func_call.params import String
 
+from .command import *
 from .config import plugin_config
 
 __marsho_meta__ = PluginMetadata(

nonebot_plugin_marshoai/plugins/marshoai_memory/command.py

@@ -0,0 +1,47 @@
+import json
+
+from arclet.alconna import Alconna, Args, Subcommand
+from nonebot import logger
+from nonebot.adapters import Bot, Event
+from nonebot.matcher import Matcher
+from nonebot.typing import T_State
+from nonebot_plugin_alconna import on_alconna
+from nonebot_plugin_localstore import get_plugin_data_file
+
+from nonebot_plugin_marshoai.config import config
+
+marsho_memory_cmd = on_alconna(
+    Alconna(
+        f"{config.marshoai_default_name}.memory",
+        Subcommand("view", alias={"v"}),
+        Subcommand("reset", alias={"r"}),
+    ),
+    priority=10,
+    block=True,
+)
+
+memory_path = get_plugin_data_file("memory.json")
+
+
+@marsho_memory_cmd.assign("view")
+async def view_memory(matcher: Matcher, state: T_State, event: Event):
+    user_id = str(event.get_user_id())
+    with open(memory_path, "r", encoding="utf-8") as f:
+        memory_data = json.load(f)
+    memorys = memory_data.get(user_id, [])
+    if not memorys:
+        await matcher.finish("好像对ta还没有任何记忆呢~")
+    await matcher.finish("这些是有关ta的记忆：" + "\n".join(memorys))
+
+
+@marsho_memory_cmd.assign("reset")
+async def reset_memory(matcher: Matcher, state: T_State, event: Event):
+    user_id = str(event.get_user_id())
+    with open(memory_path, "r", encoding="utf-8") as f:
+        memory_data = json.load(f)
+    if user_id in memory_data:
+        del memory_data[user_id]
+        with open(memory_path, "w", encoding="utf-8") as f:
+            json.dump(memory_data, f, ensure_ascii=False, indent=4)
+        await matcher.finish("记忆已重置~")
+    await matcher.finish("没有找到该用户的记忆~")