init: first commit to Gitea mirror, update README with Docker quick start and new repo URL

Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>

test_tools.py

@@ -0,0 +1,159 @@
+"""Ombre Brain MCP tool-level end-to-end test: direct calls to @mcp.tool() functions
+   Ombre Brain MCP 工具层端到端测试：直接调用 @mcp.tool() 函数"""
+import asyncio
+import os
+import sys
+
+sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
+
+from utils import load_config, setup_logging
+
+config = load_config()
+setup_logging("INFO")
+
+# Must import after config is set, since server.py does module-level init
+# 必须在配置好后导入，因为 server.py 有模块级初始化
+from server import breath, hold, trace, pulse, grow
+
+
+async def main():
+    passed = 0
+    failed = 0
+
+    # ===== pulse =====
+    print("=== [1/6] pulse ===")
+    try:
+        r = await pulse()
+        assert "Ombre Brain" in r
+        print(f"  {r.splitlines()[0]}")
+        print("  [OK]")
+        passed += 1
+    except Exception as e:
+        print(f"  [FAIL] {e}")
+        failed += 1
+    print()
+
+    # ===== hold =====
+    print("=== [2/6] hold ===")
+    try:
+        r = await hold(content="P酱最喜欢的编程语言是 Python，喜欢用 FastAPI 写后端", tags="编程,偏好", importance=8)
+        print(f"  {r.splitlines()[0]}")
+        assert any(kw in r for kw in ["新建", "合并", "📌"])
+        print("  [OK]")
+        passed += 1
+    except Exception as e:
+        print(f"  [FAIL] {e}")
+        failed += 1
+    print()
+
+    # ===== hold (merge test / 合并测试) =====
+    print("=== [2b/6] hold (合并测试) ===")
+    try:
+        r = await hold(content="P酱也喜欢用 Python 写爬虫和数据分析", tags="编程", importance=6)
+        print(f"  {r.splitlines()[0]}")
+        print("  [OK]")
+        passed += 1
+    except Exception as e:
+        print(f"  [FAIL] {e}")
+        failed += 1
+    print()
+
+    # ===== breath =====
+    print("=== [3/6] breath ===")
+    try:
+        r = await breath(query="Python 编程", max_results=3)
+        print(f"  结果前80字: {r[:80]}...")
+        assert "未找到" not in r
+        print("  [OK]")
+        passed += 1
+    except Exception as e:
+        print(f"  [FAIL] {e}")
+        failed += 1
+    print()
+
+    # ===== breath (emotion resonance / 情感共鸣) =====
+    print("=== [3b/6] breath (情感共鸣检索) ===")
+    try:
+        r = await breath(query="编程", domain="编程", valence=0.8, arousal=0.5)
+        print(f"  结果前80字: {r[:80]}...")
+        print("  [OK]")
+        passed += 1
+    except Exception as e:
+        print(f"  [FAIL] {e}")
+        failed += 1
+    print()
+
+    # --- Get a bucket ID for subsequent tests / 取一个桶 ID 用于后续测试 ---
+    bucket_id = None
+    from bucket_manager import BucketManager
+    bm = BucketManager(config)
+    all_buckets = await bm.list_all()
+    if all_buckets:
+        bucket_id = all_buckets[0]["id"]
+
+    # ===== trace =====
+    print("=== [4/6] trace ===")
+    if bucket_id:
+        try:
+            r = await trace(bucket_id=bucket_id, domain="编程,创作", importance=9)
+            print(f"  {r}")
+            assert "已修改" in r
+            print("  [OK]")
+            passed += 1
+        except Exception as e:
+            print(f"  [FAIL] {e}")
+            failed += 1
+    else:
+        print("  [SKIP] 没有可编辑的桶")
+    print()
+
+    # ===== grow =====
+    print("=== [5/6] grow ===")
+    try:
+        diary = (
+            "今天早上复习了线性代数，搞懂了特征值分解。"
+            "中午和室友去吃了拉面，聊了聊暑假实习的事。"
+            "下午写了一个 Flask 项目的 API 接口。"
+            "晚上看了部电影叫《星际穿越》，被结尾感动哭了。"
+        )
+        r = await grow(content=diary)
+        print(f"  {r.splitlines()[0]}")
+        for line in r.splitlines()[1:]:
+            if line.strip():
+                print(f"  {line}")
+        assert "条|新" in r or "整理" in r
+        print("  [OK]")
+        passed += 1
+    except Exception as e:
+        print(f"  [FAIL] {e}")
+        failed += 1
+    print()
+
+    # ===== cleanup via trace(delete=True) / 清理测试数据 =====
+    print("=== [6/6] cleanup (清理全部测试数据) ===")
+    try:
+        all_buckets = await bm.list_all()
+        for b in all_buckets:
+            r = await trace(bucket_id=b["id"], delete=True)
+            print(f"  {r}")
+        print("  [OK]")
+        passed += 1
+    except Exception as e:
+        print(f"  [FAIL] {e}")
+        failed += 1
+    print()
+
+    # ===== Confirm cleanup / 确认清理干净 =====
+    final = await pulse()
+    print(f"清理后: {final.splitlines()[0]}")
+    print()
+    print("=" * 50)
+    print(f"MCP tool test complete / 工具测试完成: {passed} passed / {failed} failed")
+    if failed == 0:
+        print("All passed ✓")
+    else:
+        print(f"{failed} failed ✗")
+
+
+if __name__ == "__main__":
+    asyncio.run(main())