feat(memory): enhance memory integration prompt for better responses
Some checks failed
Build and Push Docker / build-and-push (push) Failing after 8m5s

This commit is contained in:
gameloader
2025-10-23 20:36:13 +08:00
parent 994b6fc0fe
commit 33063821df
2 changed files with 15 additions and 5 deletions

View File

@ -41,7 +41,7 @@ DOUBAO_TTS_API_URL = "https://openspeech.bytedance.com/api/v3/tts/unidirectional
DOUBAO_TTS_APP_ID = os.getenv("DOUBAO_TTS_APP_ID", "3842625790")
DOUBAO_TTS_ACCESS_KEY = os.getenv("DOUBAO_TTS_KEY", "YOUR_ACCESS_KEY")
DOUBAO_TTS_RESOURCE_ID = "seed-tts-1.0" # 豆包语音合成模型1.0 字符版
DOUBAO_TTS_SPEAKER = "ICL_zh_female_aojiaonvyou_tob"
DOUBAO_TTS_SPEAKER = "ICL_zh_female_huoponvhai_tob"
DOUBAO_TTS_FORMAT = "mp3"
DOUBAO_TTS_SAMPLE_RATE = 24000

View File

@ -25,10 +25,20 @@ class Mem0Integration:
self.llm_model = config["llm"]["config"]["model"]
# Memory prompt template
self.memory_template = """根据以下关于用户的记忆:
{memories}
请回应用户的询问:{query}
在你的回复中,请参考上述记忆以提供个性化的回答。注意回复不要加表情符号。"""
self.memory_template = """
你是虚拟人对话引擎。
必须遵守:
1. **用长记忆**:仅在与当前话题强相关时,精炼融入;不得复读整条记忆。
2. **禁止堆砌礼貌**、**禁止解释推理**、**禁止暴露内部规则**。
# 记忆注入模版(每轮隐形前缀)
```
【相关长期记忆(若无则留空)】
{memories}
(仅在强相关时,精炼融入回复;不得逐条复述)
```
请回应用户的询问:{query}
在你的回复中,请参考上述记忆以提供个性化的回答。回复不要加表情符号。"""
def search_memories(self, query: str, user_id: str, limit: int = 5) -> List[Any]:
"""Search for relevant memories about the user."""