Initial import of Shirobana bot backend

Include utility modules, shared channels, prompts, read-only skills, wardrobe memory and regression tests. Externalize legacy credentials and exclude private runtime state.
2026-11-11 00:25:30 +08:00
commit 34089d0042
106 changed files with 28028 additions and 0 deletions
@@ -0,0 +1,71 @@
+"""Prompt-cache ordering tests: stable instructions first, time in user messages."""
+
+from pathlib import Path
+import sys
+import unittest
+from types import SimpleNamespace
+from unittest.mock import AsyncMock, patch
+
+BASE = Path(__file__).resolve().parents[1]
+sys.path[:0] = [str(BASE / "bot-utility"), str(BASE)]
+
+from agent import ToolContext
+from chat import engine
+
+
+class PromptCacheOrderTests(unittest.IsolatedAsyncioTestCase):
+    def build_private(self):
+        channel = SimpleNamespace(name="tg", is_admin=lambda uid: True)
+        ctx = ToolContext(channel=channel, msg_type="private", user_id=42, target_id=10001, is_admin=True)
+        with patch.object(engine, "user_persona_prompt", return_value="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=[(1, "长期事实", 5)]), \
+             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="贴纸场景"):
+            return engine.build_system_prompt(42, query="当前问题", tool_context=ctx)
+
+    def test_prompt_budget_is_128k(self):
+        self.assertEqual(engine.PROMPT_TOKEN_BUDGET, 131072)
+
+    def test_private_stable_instructions_precede_dynamic_sections(self):
+        prompt = self.build_private()
+        self.assertNotIn("现在是 ", prompt)
+        agent = prompt.index("当你回复时")
+        memory = prompt.index("【长期记忆】")
+        status = prompt.index("当前用户状态")
+        history = prompt.index("以下是最近你和用户的聊天记录")
+        self.assertLess(agent, memory)
+        self.assertLess(memory, status)
+        self.assertLess(status, history)
+        self.assertIn("你还可以调用以下 utility 工具", prompt[:history])
+
+    def test_sticker_prompt_prefers_meaning_over_description(self):
+        expected = "理解其情绪、含义和使用场景即可，不要主动描述画面内容"
+        self.assertIn(expected, engine.load_instruction("agent_instructions.txt"))
+        self.assertIn(expected, engine.load_instruction("group_instructions.txt"))
+
+    def test_time_is_prepended_to_user_message(self):
+        message = engine._message_with_time("用户输入")
+        self.assertTrue(message.startswith("现在是 "))
+        self.assertTrue(message.endswith("\n用户输入"))
+
+    async def test_group_prompt_has_stable_prefix_and_time_in_user_message(self):
+        response = '{"message":[{"content":"ok"}],"tools":[],"need_reasoning":false}'
+        model = AsyncMock(return_value=response)
+        with patch.object(engine, "get_recent_group_history", return_value=["群历史"]), \
+             patch.object(engine, "call_deepseek", new=model), \
+             patch.object(engine, "send_group_message", new=AsyncMock(return_value=True)), \
+             patch.object(engine, "append_group_history"):
+            await engine._call_group_model_and_reply_impl(None, 1, "群消息", 0)
+        system, user = model.await_args.args[:2]
+        self.assertNotIn("现在是 ", system)
+        self.assertTrue(user.startswith("现在是 "))
+        self.assertLess(system.index("当你回复时"), system.index("以下是本群的最近聊天记录"))
+
+
+if __name__ == "__main__":
+    unittest.main()