feat：非常好的早期聊天记录压缩系统，麦麦现在有5倍上文记忆量（真的吗？

src/do_tool/tool_can_use/get_current_task.py

@@ -1,7 +1,8 @@
-from src.do_tool.tool_can_use.base_tool import BaseTool, register_tool
+from src.do_tool.tool_can_use.base_tool import BaseTool
 from src.plugins.schedule.schedule_generator import bot_schedule
 from src.common.logger import get_module_logger
 from typing import Dict, Any
+from datetime import datetime
 
 logger = get_module_logger("get_current_task_tool")
 
@@ -9,19 +10,19 @@ logger = get_module_logger("get_current_task_tool")
 class GetCurrentTaskTool(BaseTool):
     """获取当前正在做的事情/最近的任务工具"""
 
-    name = "get_current_task"
-    description = "获取当前正在做的事情/最近的任务"
+    name = "get_schedule"
+    description = "获取当前正在做的事情，或者某个时间点/时间段的日程信息"
     parameters = {
         "type": "object",
         "properties": {
-            "num": {"type": "integer", "description": "要获取的任务数量"},
-            "time_info": {"type": "boolean", "description": "是否包含时间信息"},
+            "start_time": {"type": "string", "description": "开始时间，格式为'HH:MM'，填写current则获取当前任务"},
+            "end_time": {"type": "string", "description": "结束时间，格式为'HH:MM'，填写current则获取当前任务"},
         },
-        "required": [],
+        "required": ["start_time", "end_time"],
     }
 
     async def execute(self, function_args: Dict[str, Any], message_txt: str = "") -> Dict[str, Any]:
-        """执行获取当前任务
+        """执行获取当前任务或指定时间段的日程信息
 
         Args:
             function_args: 工具参数
@@ -30,25 +31,31 @@ class GetCurrentTaskTool(BaseTool):
         Returns:
             Dict: 工具执行结果
         """
-        try:
-            # 获取参数，如果没有提供则使用默认值
-            num = function_args.get("num", 1)
-            time_info = function_args.get("time_info", False)
+        start_time = function_args.get("start_time")
+        end_time = function_args.get("end_time")
 
-            # 调用日程系统获取当前任务
-            current_task = bot_schedule.get_current_num_task(num=num, time_info=time_info)
-
-            # 格式化返回结果
+        # 如果 start_time 或 end_time 为 "current"，则获取当前任务
+        if start_time == "current" or end_time == "current":
+            current_task = bot_schedule.get_current_num_task(num=1, time_info=True)
+            current_time = datetime.now().strftime("%H:%M:%S")
+            current_date = datetime.now().strftime("%Y-%m-%d")
             if current_task:
-                task_info = current_task
+                task_info = f"{current_date} {current_time}，你在{current_task}"
             else:
-                task_info = "当前没有正在进行的任务"
+                task_info = f"{current_time} {current_date}，没在做任何事情"
+        # 如果提供了时间范围，则获取该时间段的日程信息
+        elif start_time and end_time:
+            tasks = await bot_schedule.get_task_from_time_to_time(start_time, end_time)
+            if tasks:
+                task_list = []
+                for task in tasks:
+                    task_time = task[0].strftime("%H:%M")
+                    task_content = task[1]
+                    task_list.append(f"{task_time}时，{task_content}")
+                task_info = "\n".join(task_list)
+            else:
+                task_info = f"在 {start_time} 到 {end_time} 之间没有找到日程信息"
 
-            return {"name": "get_current_task", "content": f"当前任务信息: {task_info}"}
-        except Exception as e:
-            logger.error(f"获取当前任务工具执行失败: {str(e)}")
-            return {"name": "get_current_task", "content": f"获取当前任务失败: {str(e)}"}
+        return {"name": "get_current_task", "content": f"日程信息: {task_info}"}
 
 
-# 注册工具
-# register_tool(GetCurrentTaskTool)

src/do_tool/tool_can_use/get_memory.py

@@ -3,13 +3,13 @@ from src.plugins.memory_system.Hippocampus import HippocampusManager
 from src.common.logger import get_module_logger
 from typing import Dict, Any
 
-logger = get_module_logger("get_memory_tool")
+logger = get_module_logger("mid_chat_mem_tool")
 
 
 class GetMemoryTool(BaseTool):
     """从记忆系统中获取相关记忆的工具"""
 
-    name = "get_memory"
+    name = "mid_chat_mem"
     description = "从记忆系统中获取相关记忆"
     parameters = {
         "type": "object",
@@ -49,10 +49,10 @@ class GetMemoryTool(BaseTool):
             else:
                 content = f"你不太记得有关{text}的记忆，你对此不太了解"
 
-            return {"name": "get_memory", "content": content}
+            return {"name": "mid_chat_mem", "content": content}
         except Exception as e:
             logger.error(f"记忆获取工具执行失败: {str(e)}")
-            return {"name": "get_memory", "content": f"记忆获取失败: {str(e)}"}
+            return {"name": "mid_chat_mem", "content": f"记忆获取失败: {str(e)}"}
 
 
 # 注册工具

src/do_tool/tool_can_use/get_time_date.py

@@ -0,0 +1,39 @@
+from src.do_tool.tool_can_use.base_tool import BaseTool
+from src.common.logger import get_module_logger
+from typing import Dict, Any
+from datetime import datetime
+
+logger = get_module_logger("get_time_date")
+
+class GetCurrentDateTimeTool(BaseTool):
+    """获取当前时间、日期、年份和星期的工具"""
+
+    name = "get_current_date_time"
+    description = "当有人询问或者涉及到具体时间或者日期的时候，必须使用这个工具"
+    parameters = {
+        "type": "object",
+        "properties": {},
+        "required": [],
+    }
+
+    async def execute(self, function_args: Dict[str, Any], message_txt: str = "") -> Dict[str, Any]:
+        """执行获取当前时间、日期、年份和星期
+
+        Args:
+            function_args: 工具参数（此工具不使用）
+            message_txt: 原始消息文本（此工具不使用）
+
+        Returns:
+            Dict: 工具执行结果
+        """
+        current_time = datetime.now().strftime("%H:%M:%S")
+        current_date = datetime.now().strftime("%Y-%m-%d")
+        current_year = datetime.now().strftime("%Y")
+        current_weekday = datetime.now().strftime("%A")
+
+        return {
+            "name": "get_current_date_time",
+            "content": f"当前时间: {current_time}, 日期: {current_date}, 年份: {current_year}, 星期: {current_weekday}"
+        }
+
+

src/do_tool/tool_can_use/mid_chat_mem.py

@@ -0,0 +1,40 @@
+from src.do_tool.tool_can_use.base_tool import BaseTool
+from src.common.logger import get_module_logger
+from typing import Dict, Any
+
+logger = get_module_logger("get_mid_memory_tool")
+
+
+class GetMidMemoryTool(BaseTool):
+    """从记忆系统中获取相关记忆的工具"""
+
+    name = "mid_chat_mem"
+    description = "之前的聊天内容中获取具体信息，当最新消息提到，或者你需要回复的消息中提到，你可以使用这个工具"
+    parameters = {
+        "type": "object",
+        "properties": {
+            "id": {"type": "integer", "description": "要查询的聊天记录id"},
+        },
+        "required": ["id"],
+    }
+
+    async def execute(self, function_args: Dict[str, Any], message_txt: str = "") -> Dict[str, Any]:
+        """执行记忆获取
+
+        Args:
+            function_args: 工具参数
+            message_txt: 原始消息文本
+
+        Returns:
+            Dict: 工具执行结果
+        """
+        try:
+            id = function_args.get("id")
+            return {"name": "mid_chat_mem", "content": str(id)}
+        except Exception as e:
+            logger.error(f"聊天记录获取工具执行失败: {str(e)}")
+            return {"name": "mid_chat_mem", "content": f"聊天记录获取失败: {str(e)}"}
+
+
+# 注册工具
+# register_tool(GetMemoryTool)