RRRosmontis
34089d0042
Include utility modules, shared channels, prompts, read-only skills, wardrobe memory and regression tests. Externalize legacy credentials and exclude private runtime state.
87 lines
4.4 KiB
Python
87 lines
4.4 KiB
Python
"""Per-user persona overlays must be shared by chat, help, and notifications."""
|
|
|
|
from pathlib import Path
|
|
import sys
|
|
import unittest
|
|
from unittest.mock import AsyncMock, patch
|
|
|
|
BASE = Path(__file__).resolve().parents[1]
|
|
sys.path[:0] = [str(BASE / "bot-utility"), str(BASE)]
|
|
|
|
import help_ui
|
|
import notification_render
|
|
import notifications
|
|
from chat import engine
|
|
|
|
|
|
class UserPersonaTests(unittest.IsolatedAsyncioTestCase):
|
|
def test_overlay_uses_long_term_preferences_with_latest_priority(self):
|
|
rows = [
|
|
(2, "称呼用户为主人", 5, "个人设定,表达偏好", 2, 0),
|
|
(1, "用户不喜欢书面腔", 4, "表达偏好", 1, 0),
|
|
]
|
|
with patch.object(engine.memory_store, "list_facts", return_value=rows) as facts:
|
|
prompt = engine.user_persona_prompt(42)
|
|
facts.assert_called_once_with(42, limit=30)
|
|
self.assertIn("当前用户的个人设定", prompt)
|
|
self.assertIn("称呼用户为主人", prompt)
|
|
self.assertIn("最新、最明确", prompt)
|
|
self.assertTrue(prompt.startswith(engine.BASE_PROMPT))
|
|
self.assertIs(engine.user_persona_prompt(None), engine.BASE_PROMPT)
|
|
|
|
def test_private_chat_prompt_uses_same_user_overlay(self):
|
|
custom = engine.BASE_PROMPT + "\n【当前用户的个人设定】称呼用户为主人"
|
|
with patch.object(engine, "user_persona_prompt", return_value=custom) as persona, \
|
|
patch.object(engine, "load_user_status", return_value={"pause": False, "sleep": False}), \
|
|
patch.object(engine.memory_store, "get_relation", return_value=None), \
|
|
patch.object(engine.memory_store, "list_entities", return_value=[]), \
|
|
patch.object(engine.memory_store, "recall_facts", return_value=[]), \
|
|
patch.object(engine.memory_store, "migrate_legacy"), \
|
|
patch.object(engine, "get_history_within_budget", return_value=[]), \
|
|
patch.object(engine.notebook.notebook, "context_for_chat", return_value=""), \
|
|
patch.object(engine, "_sticker_prompt_summary", return_value=""):
|
|
prompt = engine.build_system_prompt(42, query="你好")
|
|
persona.assert_called_once_with(42)
|
|
self.assertTrue(prompt.startswith(custom))
|
|
|
|
async def test_notification_renderer_uses_same_user_overlay(self):
|
|
event = notifications.Event(
|
|
kind="komari", dedupe_key="persona-test", uid=42,
|
|
legacy_text="raw", payload={"server": "Ama-10", "status": "offline"},
|
|
)
|
|
custom = engine.BASE_PROMPT + "\n【当前用户的个人设定】称呼用户为主人"
|
|
with patch.object(engine, "user_persona_prompt", return_value=custom) as persona, \
|
|
patch.object(engine, "call_deepseek", new=AsyncMock(return_value="主人，Ama-10 掉线了。")) as model:
|
|
rendered = await notification_render.render_event(event, "chat", 10001)
|
|
persona.assert_called_once_with(42)
|
|
self.assertTrue(model.await_args.args[0].startswith(custom))
|
|
self.assertEqual(rendered, "主人，Ama-10 掉线了。")
|
|
|
|
async def test_help_renderer_uses_same_user_overlay(self):
|
|
custom = engine.BASE_PROMPT + "\n【当前用户的个人设定】称呼用户为主人"
|
|
with patch.object(engine, "user_persona_prompt", return_value=custom) as persona, \
|
|
patch.object(engine, "call_deepseek", new=AsyncMock(return_value="主人，可以用这些功能。")):
|
|
await help_ui.chat_help(42)
|
|
persona.assert_called_once_with(42)
|
|
|
|
def test_deterministic_notification_fallback_has_no_hardcoded_address(self):
|
|
mail = notifications.Event(
|
|
kind="mail", dedupe_key="mail-persona", uid=42, legacy_text="raw",
|
|
payload={"mid": 1, "subject": "test", "sender": "a@example.test"},
|
|
)
|
|
ginkgo = notifications.Event(
|
|
kind="ginkgo:course", dedupe_key="ginkgo-persona", uid=42, legacy_text="raw",
|
|
)
|
|
self.assertNotIn("主人", notification_render._fallback(mail))
|
|
self.assertNotIn("主人", notification_render._fallback(ginkgo))
|
|
|
|
def test_memory_instruction_preserves_expression_preferences(self):
|
|
prompt = engine.load_instruction("agent_instructions.txt")
|
|
self.assertIn("个人设定", prompt)
|
|
self.assertIn("表达偏好", prompt)
|
|
self.assertIn("importance=5", prompt)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|