better：优化做梦表现

src/dream/tools/__init__.py

@@ -0,0 +1,9 @@
+"""
+dream agent 工具实现模块。
+
+每个工具的具体实现放在独立文件中，通过 make_xxx(chat_id) 工厂函数
+生成绑定到特定 chat_id 的协程函数，由 dream_agent.init_dream_tools 统一注册。
+"""
+
+
+

src/dream/tools/create_chat_history_tool.py

@@ -0,0 +1,65 @@
+import time
+
+from src.common.logger import get_logger
+from src.common.database.database_model import ChatHistory
+
+logger = get_logger("dream_agent")
+
+
+def make_create_chat_history(chat_id: str):
+    async def create_chat_history(
+        theme: str,
+        summary: str,
+        keywords: str,
+        key_point: str,
+        start_time: float,
+        end_time: float,
+    ) -> str:
+        """创建一条新的 ChatHistory 概括记录（用于整理/合并后的新记忆）"""
+        try:
+            logger.info(
+                f"[dream][tool] 调用 create_chat_history("
+                f"theme={bool(theme)}, summary={bool(summary)}, "
+                f"keywords={bool(keywords)}, key_point={bool(key_point)}, "
+                f"start_time={start_time}, end_time={end_time}) (chat_id={chat_id})"
+            )
+
+            now_ts = time.time()
+
+            # 将传入的 start_time/end_time（如果有）解析为时间戳；否则回退为当前时间
+            def _parse_ts(value, default):
+                if value is None:
+                    return default
+                try:
+                    return float(value)
+                except (TypeError, ValueError):
+                    return default
+
+            start_ts = _parse_ts(start_time, now_ts)
+            end_ts = _parse_ts(end_time, now_ts)
+
+            record = ChatHistory.create(
+                chat_id=chat_id,
+                theme=theme,
+                summary=summary,
+                keywords=keywords,
+                key_point=key_point,
+                # 对于由 dream 整理产生的新概括，时间范围优先使用工具提供的时间，否则使用当前时间占位
+                start_time=start_ts,
+                end_time=end_ts,
+            )
+
+            msg = (
+                f"已创建新的 ChatHistory 记录，ID={record.id}，"
+                f"theme={record.theme or '无'}，summary={'有' if record.summary else '无'}。"
+            )
+            logger.info(f"[dream][tool] create_chat_history 完成: {msg}")
+            return msg
+        except Exception as e:
+            logger.error(f"create_chat_history 失败: {e}")
+            return f"create_chat_history 执行失败: {e}"
+
+    return create_chat_history
+
+
+

src/dream/tools/delete_chat_history_tool.py

@@ -0,0 +1,28 @@
+from src.common.logger import get_logger
+from src.common.database.database_model import ChatHistory
+
+logger = get_logger("dream_agent")
+
+
+def make_delete_chat_history(chat_id: str):  # chat_id 目前未直接使用，预留以备扩展
+    async def delete_chat_history(memory_id: int) -> str:
+        """删除一条 chat_history 记录"""
+        try:
+            logger.info(f"[dream][tool] 调用 delete_chat_history(memory_id={memory_id})")
+            record = ChatHistory.get_or_none(ChatHistory.id == memory_id)
+            if not record:
+                msg = f"未找到 ID={memory_id} 的 ChatHistory 记录，无法删除。"
+                logger.info(f"[dream][tool] delete_chat_history 未找到记录: {msg}")
+                return msg
+            rows = ChatHistory.delete().where(ChatHistory.id == memory_id).execute()
+            msg = f"已删除 ID={memory_id} 的 ChatHistory 记录，受影响行数={rows}。"
+            logger.info(f"[dream][tool] delete_chat_history 完成: {msg}")
+            return msg
+        except Exception as e:
+            logger.error(f"delete_chat_history 失败: {e}")
+            return f"delete_chat_history 执行失败: {e}"
+
+    return delete_chat_history
+
+
+

src/dream/tools/delete_jargon_tool.py

@@ -0,0 +1,28 @@
+from src.common.logger import get_logger
+from src.common.database.database_model import Jargon
+
+logger = get_logger("dream_agent")
+
+
+def make_delete_jargon(chat_id: str):  # chat_id 目前未直接使用，预留以备扩展
+    async def delete_jargon(jargon_id: int) -> str:
+        """删除一条 Jargon 记录"""
+        try:
+            logger.info(f"[dream][tool] 调用 delete_jargon(jargon_id={jargon_id})")
+            record = Jargon.get_or_none(Jargon.id == jargon_id)
+            if not record:
+                msg = f"未找到 ID={jargon_id} 的 Jargon 记录，无法删除。"
+                logger.info(f"[dream][tool] delete_jargon 未找到记录: {msg}")
+                return msg
+            rows = Jargon.delete().where(Jargon.id == jargon_id).execute()
+            msg = f"已删除 ID={jargon_id} 的 Jargon 记录（内容：{record.content}），受影响行数={rows}。"
+            logger.info(f"[dream][tool] delete_jargon 完成: {msg}")
+            return msg
+        except Exception as e:
+            logger.error(f"delete_jargon 失败: {e}")
+            return f"delete_jargon 执行失败: {e}"
+
+    return delete_jargon
+
+
+

src/dream/tools/finish_maintenance_tool.py

@@ -0,0 +1,19 @@
+from typing import Optional
+
+from src.common.logger import get_logger
+
+logger = get_logger("dream_agent")
+
+
+def make_finish_maintenance(chat_id: str):  # chat_id 目前未直接使用，预留以备扩展
+    async def finish_maintenance(reason: Optional[str] = None) -> str:
+        """结束本次 dream 维护任务。当你认为当前 chat_id 下的维护工作已经完成，没有更多需要整理的内容时，调用此工具来结束本次运行。"""
+        reason_text = f"，原因：{reason}" if reason else ""
+        msg = f"DREAM_MAINTENANCE_COMPLETE{reason_text}"
+        logger.info(f"[dream][tool] 调用 finish_maintenance，结束本次维护{reason_text}")
+        return msg
+
+    return finish_maintenance
+
+
+

src/dream/tools/get_chat_history_detail_tool.py

@@ -0,0 +1,54 @@
+import time
+from typing import Optional
+
+from src.common.logger import get_logger
+from src.common.database.database_model import ChatHistory
+
+logger = get_logger("dream_agent")
+
+
+def make_get_chat_history_detail(chat_id: str):  # chat_id 目前未直接使用，预留以备扩展
+    async def get_chat_history_detail(memory_id: int) -> str:
+        """获取单条 chat_history 的完整内容"""
+        try:
+            logger.info(f"[dream][tool] 调用 get_chat_history_detail(memory_id={memory_id})")
+            record = ChatHistory.get_or_none(ChatHistory.id == memory_id)
+            if not record:
+                msg = f"未找到 ID={memory_id} 的 ChatHistory 记录。"
+                logger.info(f"[dream][tool] get_chat_history_detail 未找到记录: {msg}")
+                return msg
+
+            # 将时间戳转换为可读时间格式
+            start_time_str = (
+                time.strftime("%Y-%m-%d %H:%M:%S", time.localtime(record.start_time))
+                if record.start_time
+                else "未知"
+            )
+            end_time_str = (
+                time.strftime("%Y-%m-%d %H:%M:%S", time.localtime(record.end_time))
+                if record.end_time
+                else "未知"
+            )
+
+            result = (
+                f"ID={record.id}\n"
+                # f"chat_id={record.chat_id}\n"
+                f"时间范围={start_time_str} 至 {end_time_str}\n"
+                f"主题={record.theme or '无'}\n"
+                f"关键词={record.keywords or '无'}\n"
+                f"参与者={record.participants or '无'}\n"
+                f"概括={record.summary or '无'}\n"
+                f"关键信息={record.key_point or '无'}"
+            )
+            logger.debug(
+                f"[dream][tool] get_chat_history_detail 成功，预览: {result[:200].replace(chr(10), ' ')}"
+            )
+            return result
+        except Exception as e:
+            logger.error(f"get_chat_history_detail 失败: {e}")
+            return f"get_chat_history_detail 执行失败: {e}"
+
+    return get_chat_history_detail
+
+
+

src/dream/tools/search_chat_history_tool.py

@@ -0,0 +1,225 @@
+import json
+from typing import List, Optional
+
+from src.common.logger import get_logger
+from src.common.database.database_model import ChatHistory
+from src.chat.utils.utils import parse_keywords_string
+
+logger = get_logger("dream_agent")
+
+
+def make_search_chat_history(chat_id: str):
+    async def search_chat_history(
+        keyword: Optional[str] = None,
+        participant: Optional[str] = None,
+    ) -> str:
+        """根据关键词或参与人查询记忆，返回匹配的记忆id、记忆标题theme和关键词keywords（dream 维护专用版本）"""
+        try:
+            # 检查参数
+            if not keyword and not participant:
+                return "未指定查询参数（需要提供keyword或participant之一）"
+
+            logger.info(
+                f"[dream][tool] 调用 search_chat_history(keyword={keyword}, participant={participant}) "
+                f"(作用域 chat_id={chat_id})"
+            )
+
+            # 构建查询条件
+            query = ChatHistory.select().where(ChatHistory.chat_id == chat_id)
+
+            # 执行查询（按时间倒序，最近的在前）
+            records = list(query.order_by(ChatHistory.start_time.desc()).limit(50))
+
+            filtered_records: List[ChatHistory] = []
+
+            for record in records:
+                participant_matched = True  # 如果没有participant条件，默认为True
+                keyword_matched = True  # 如果没有keyword条件，默认为True
+
+                # 检查参与人匹配
+                if participant:
+                    participant_matched = False
+                    participants_list: List[str] = []
+                    if record.participants:
+                        try:
+                            participants_data = (
+                                json.loads(record.participants)
+                                if isinstance(record.participants, str)
+                                else record.participants
+                            )
+                            if isinstance(participants_data, list):
+                                participants_list = [str(p).lower() for p in participants_data]
+                        except (json.JSONDecodeError, TypeError, ValueError):
+                            pass
+
+                    participant_lower = participant.lower().strip()
+                    if participant_lower and any(participant_lower in p for p in participants_list):
+                        participant_matched = True
+
+                # 检查关键词匹配
+                if keyword:
+                    keyword_matched = False
+                    # 解析多个关键词（支持空格、逗号等分隔符）
+                    keywords_list = parse_keywords_string(keyword)
+                    if not keywords_list:
+                        keywords_list = [keyword.strip()] if keyword.strip() else []
+
+                    # 转换为小写以便匹配
+                    keywords_lower = [kw.lower() for kw in keywords_list if kw.strip()]
+
+                    if keywords_lower:
+                        # 在theme、keywords、summary、original_text中搜索
+                        theme = (record.theme or "").lower()
+                        summary = (record.summary or "").lower()
+                        original_text = (record.original_text or "").lower()
+
+                        # 解析record中的keywords JSON
+                        record_keywords_list: List[str] = []
+                        if record.keywords:
+                            try:
+                                keywords_data = (
+                                    json.loads(record.keywords)
+                                    if isinstance(record.keywords, str)
+                                    else record.keywords
+                                )
+                                if isinstance(keywords_data, list):
+                                    record_keywords_list = [str(k).lower() for k in keywords_data]
+                            except (json.JSONDecodeError, TypeError, ValueError):
+                                pass
+
+                        # 有容错的全匹配：如果关键词数量>2，允许n-1个关键词匹配；否则必须全部匹配
+                        matched_count = 0
+                        for kw in keywords_lower:
+                            kw_matched = (
+                                kw in theme
+                                or kw in summary
+                                or kw in original_text
+                                or any(kw in k for k in record_keywords_list)
+                            )
+                            if kw_matched:
+                                matched_count += 1
+
+                        # 计算需要匹配的关键词数量
+                        total_keywords = len(keywords_lower)
+                        if total_keywords > 2:
+                            # 关键词数量>2，允许n-1个关键词匹配
+                            required_matches = total_keywords - 1
+                        else:
+                            # 关键词数量<=2，必须全部匹配
+                            required_matches = total_keywords
+
+                        keyword_matched = matched_count >= required_matches
+
+                # 两者都匹配（如果同时有participant和keyword，需要两者都匹配；如果只有一个条件，只需要该条件匹配）
+                matched = participant_matched and keyword_matched
+
+                if matched:
+                    filtered_records.append(record)
+
+            if not filtered_records:
+                if keyword and participant:
+                    keywords_str = "、".join(parse_keywords_string(keyword) if keyword else [])
+                    return f"未找到包含关键词'{keywords_str}'且参与人包含'{participant}'的聊天记录"
+                elif keyword:
+                    keywords_list = parse_keywords_string(keyword)
+                    keywords_str = "、".join(keywords_list)
+                    if len(keywords_list) > 2:
+                        required_count = len(keywords_list) - 1
+                        return (
+                            f"未找到包含至少{required_count}个关键词（共{len(keywords_list)}个）'{keywords_str}'的聊天记录"
+                        )
+                    else:
+                        return f"未找到包含所有关键词'{keywords_str}'的聊天记录"
+                elif participant:
+                    return f"未找到参与人包含'{participant}'的聊天记录"
+                else:
+                    return "未找到相关聊天记录"
+
+            # 如果匹配结果超过20条，不返回具体记录，只返回提示和所有相关关键词
+            if len(filtered_records) > 20:
+                all_keywords_set = set()
+                for record in filtered_records:
+                    if record.keywords:
+                        try:
+                            keywords_data = (
+                                json.loads(record.keywords)
+                                if isinstance(record.keywords, str)
+                                else record.keywords
+                            )
+                            if isinstance(keywords_data, list):
+                                for k in keywords_data:
+                                    k_str = str(k).strip()
+                                    if k_str:
+                                        all_keywords_set.add(k_str)
+                        except (json.JSONDecodeError, TypeError, ValueError):
+                            continue
+
+                search_label = keyword or participant or "当前条件"
+
+                if all_keywords_set:
+                    keywords_str = "、".join(sorted(all_keywords_set))
+                    response_text = (
+                        f"包含“{search_label}”的结果过多，请尝试更多关键词精确查找\n\n"
+                        f"有关\"{search_label}\"的关键词：\n"
+                        f"{keywords_str}"
+                    )
+                else:
+                    response_text = (
+                        f"包含“{search_label}”的结果过多，请尝试更多关键词精确查找\n\n"
+                        f"有关\"{search_label}\"的关键词信息为空"
+                    )
+
+                logger.info(
+                    f"[dream][tool] search_chat_history 匹配结果超过20条，返回关键词汇总提示，总数={len(filtered_records)}"
+                )
+                return response_text
+
+            # 构建结果文本，返回id、theme和keywords（最多20条）
+            results: List[str] = []
+            for record in filtered_records[:20]:
+                result_parts: List[str] = []
+
+                # 记忆ID
+                result_parts.append(f"记忆ID：{record.id}")
+
+                # 主题
+                if record.theme:
+                    result_parts.append(f"主题：{record.theme}")
+                else:
+                    result_parts.append("主题：（无）")
+
+                # 关键词
+                if record.keywords:
+                    try:
+                        keywords_data = (
+                            json.loads(record.keywords)
+                            if isinstance(record.keywords, str)
+                            else record.keywords
+                        )
+                        if isinstance(keywords_data, list) and keywords_data:
+                            keywords_str = "、".join([str(k) for k in keywords_data])
+                            result_parts.append(f"关键词：{keywords_str}")
+                        else:
+                            result_parts.append("关键词：（无）")
+                    except (json.JSONDecodeError, TypeError, ValueError):
+                        result_parts.append("关键词：（无）")
+                else:
+                    result_parts.append("关键词：（无）")
+
+                results.append("\n".join(result_parts))
+
+            if not results:
+                return "未找到相关聊天记录"
+
+            response_text = "\n\n---\n\n".join(results)
+
+            logger.info(f"[dream][tool] search_chat_history 返回 {len(filtered_records)} 条匹配记录")
+            return response_text
+        except Exception as e:
+            logger.error(f"search_chat_history 失败: {e}")
+            return f"search_chat_history 执行失败: {e}"
+
+    return search_chat_history
+
+
+

src/dream/tools/update_chat_history_tool.py

@@ -0,0 +1,54 @@
+from typing import Any, Dict, Optional
+
+from src.common.logger import get_logger
+from src.common.database.database_model import ChatHistory
+from src.plugin_system.apis import database_api
+
+logger = get_logger("dream_agent")
+
+
+def make_update_chat_history(chat_id: str):  # chat_id 目前未直接使用，预留以备扩展
+    async def update_chat_history(
+        memory_id: int,
+        theme: Optional[str] = None,
+        summary: Optional[str] = None,
+        keywords: Optional[str] = None,
+        key_point: Optional[str] = None,
+    ) -> str:
+        """按字段更新 chat_history（字符串字段要求 JSON 的字段须传入已序列化的字符串）"""
+        try:
+            logger.info(
+                f"[dream][tool] 调用 update_chat_history(memory_id={memory_id}, "
+                f"theme={bool(theme)}, summary={bool(summary)}, keywords={bool(keywords)}, key_point={bool(key_point)})"
+            )
+            record = ChatHistory.get_or_none(ChatHistory.id == memory_id)
+            if not record:
+                msg = f"未找到 ID={memory_id} 的 ChatHistory 记录，无法更新。"
+                logger.info(f"[dream][tool] update_chat_history 未找到记录: {msg}")
+                return msg
+
+            data: Dict[str, Any] = {}
+            if theme is not None:
+                data["theme"] = theme
+            if summary is not None:
+                data["summary"] = summary
+            if keywords is not None:
+                data["keywords"] = keywords
+            if key_point is not None:
+                data["key_point"] = key_point
+
+            if not data:
+                return "未提供任何需要更新的字段。"
+
+            await database_api.db_save(ChatHistory, data=data, key_field="id", key_value=memory_id)
+            msg = f"已更新 ChatHistory 记录 ID={memory_id}，更新字段={list(data.keys())}。"
+            logger.info(f"[dream][tool] update_chat_history 完成: {msg}")
+            return msg
+        except Exception as e:
+            logger.error(f"update_chat_history 失败: {e}")
+            return f"update_chat_history 执行失败: {e}"
+
+    return update_chat_history
+
+
+

src/dream/tools/update_jargon_tool.py

@@ -0,0 +1,54 @@
+from typing import Any, Dict, Optional
+
+from src.common.logger import get_logger
+from src.common.database.database_model import Jargon
+from src.plugin_system.apis import database_api
+
+logger = get_logger("dream_agent")
+
+
+def make_update_jargon(chat_id: str):  # chat_id 目前未直接使用，预留以备扩展
+    async def update_jargon(
+        jargon_id: int,
+        meaning: Optional[str] = None,
+        is_global: Optional[bool] = None,
+        is_jargon: Optional[bool] = None,
+        content: Optional[str] = None,
+    ) -> str:
+        """按字段更新 Jargon 记录，可用于修正含义、调整全局性、标记是否为黑话等"""
+        try:
+            logger.info(
+                f"[dream][tool] 调用 update_jargon(jargon_id={jargon_id}, "
+                f"meaning={bool(meaning)}, is_global={is_global}, is_jargon={is_jargon}, content={bool(content)})"
+            )
+            record = Jargon.get_or_none(Jargon.id == jargon_id)
+            if not record:
+                msg = f"未找到 ID={jargon_id} 的 Jargon 记录，无法更新。"
+                logger.info(f"[dream][tool] update_jargon 未找到记录: {msg}")
+                return msg
+
+            data: Dict[str, Any] = {}
+            if meaning is not None:
+                data["meaning"] = meaning
+            if is_global is not None:
+                data["is_global"] = is_global
+            if is_jargon is not None:
+                data["is_jargon"] = is_jargon
+            if content is not None:
+                data["content"] = content
+
+            if not data:
+                return "未提供任何需要更新的字段。"
+
+            await database_api.db_save(Jargon, data=data, key_field="id", key_value=jargon_id)
+            msg = f"已更新 Jargon 记录 ID={jargon_id}，更新字段={list(data.keys())}。"
+            logger.info(f"[dream][tool] update_jargon 完成: {msg}")
+            return msg
+        except Exception as e:
+            logger.error(f"update_jargon 失败: {e}")
+            return f"update_jargon 执行失败: {e}"
+
+    return update_jargon
+
+
+