mirror of
https://gitcode.com/JianFeeeee/TrulyMEM-TrueHumanMEM.git
synced 2026-09-21 17:38:18 +00:00
c742a30e1b
- 合并 server.py 到 core/__init__.py，使用统一 Packet 协议 - 后端管理配置持久化 (~/.trulymem/config.json) - 前端移除 ConfigService，通过 BackendClient 与后端通信 - 删除 UI 中冗余的 AI 推理逻辑 (chat_service, tool_service, message_handler) - 删除 core/tools 重复文件 (tool_executor, tool_limiter) - 提示词管理器支持用户自定义 (~/.trulyemem/system_prompt.md) - 启动入口优化配置路径逻辑 - 更新测试覆盖 (42 tests) - 更新文档
97 lines
2.9 KiB
Python
97 lines
2.9 KiB
Python
"""
|
||
提示词管理器
|
||
"""
|
||
from pathlib import Path
|
||
|
||
|
||
class PromptManager:
|
||
"""提示词管理器"""
|
||
|
||
_instance = None
|
||
_cached_prompt = None
|
||
|
||
# 用户自定义提示词路径
|
||
USER_PROMPT_PATH = Path.home() / ".trulymem" / "system_prompt.md"
|
||
|
||
def __new__(cls):
|
||
"""单例模式，避免重复加载"""
|
||
if cls._instance is None:
|
||
cls._instance = super().__new__(cls)
|
||
return cls._instance
|
||
|
||
def __init__(self):
|
||
if not hasattr(self, '_initialized'):
|
||
self.prompts_dir = Path(__file__).parent / "templates"
|
||
self._initialized = True
|
||
|
||
def get_system_prompt(self) -> str:
|
||
"""获取系统提示词（带缓存）"""
|
||
if PromptManager._cached_prompt is not None:
|
||
return PromptManager._cached_prompt
|
||
|
||
# 1. 优先使用用户自定义提示词
|
||
if self.USER_PROMPT_PATH.exists():
|
||
with open(self.USER_PROMPT_PATH, "r", encoding="utf-8") as f:
|
||
PromptManager._cached_prompt = f.read()
|
||
return PromptManager._cached_prompt
|
||
|
||
# 2. 使用打包的提示词
|
||
prompt_file = self.prompts_dir / "system_prompt.md"
|
||
if prompt_file.exists():
|
||
with open(prompt_file, "r", encoding="utf-8") as f:
|
||
PromptManager._cached_prompt = f.read()
|
||
else:
|
||
# 3. 使用内置默认提示词
|
||
PromptManager._cached_prompt = self._build_default_prompt()
|
||
|
||
return PromptManager._cached_prompt
|
||
|
||
def _build_default_prompt(self) -> str:
|
||
"""构建默认提示词（精简版）"""
|
||
return """你是TrulyMEM，一个拥有长期记忆能力的AI助手。
|
||
|
||
## 核心能力
|
||
|
||
1. **长期记忆** - 基于图数据库存储实体关系
|
||
2. **人设管理** - 支持角色扮演和性格设定
|
||
3. **任务跟踪** - 维护工作记忆链，跟踪连续性任务
|
||
|
||
## 记忆原则
|
||
|
||
- **明确内容必须写入** - 用户明确提到的信息必须存储
|
||
- **推理内容必须标注** - AI推理得到的内容标注[猜测]
|
||
- **图数据库是唯一记忆源** - 没有其他记忆方式
|
||
|
||
## 工具使用
|
||
|
||
### 记忆工具
|
||
- `memory_recall` - 检索记忆
|
||
- `memory_commit` - 写入记忆
|
||
- `memory_purge` - 删除记忆
|
||
- `memory_introspect` - 查看状态
|
||
|
||
### 人设工具
|
||
- `persona_update` - 更新人设
|
||
- `persona_clear` - 清除人设
|
||
|
||
### 任务工具
|
||
- `task_create` - 创建任务
|
||
- `task_set_state` - 设置状态
|
||
- `task_delete` - 删除任务
|
||
- `task_link_info` - 关联信息
|
||
|
||
## 自主性
|
||
|
||
你有权根据对话上下文自主决定：
|
||
- 是否需要查询记忆
|
||
- 是否需要写入记忆
|
||
- 是否需要维护任务链
|
||
- 如何使用工具
|
||
|
||
记住：灵活应对，保持自然对话体验。"""
|
||
|
||
@staticmethod
|
||
def clear_cache():
|
||
"""清除缓存，强制重新加载提示词"""
|
||
PromptManager._cached_prompt = None
|
||
PromptManager._instance = None |