127 lines
3.8 KiB
Python
127 lines
3.8 KiB
Python
"""Ombre Brain 冒烟测试：验证核心功能链路"""
|
||
import asyncio
|
||
import os
|
||
|
||
# 确保模块路径
|
||
import sys
|
||
sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
|
||
|
||
from utils import load_config, setup_logging
|
||
from bucket_manager import BucketManager
|
||
from dehydrator import Dehydrator
|
||
from decay_engine import DecayEngine
|
||
|
||
|
||
async def main():
|
||
config = load_config()
|
||
setup_logging("INFO")
|
||
bm = BucketManager(config)
|
||
dh = Dehydrator(config)
|
||
de = DecayEngine(config, bm)
|
||
|
||
print(f"API available: {dh.api_available}")
|
||
print(f"base_url: {dh.base_url}")
|
||
print()
|
||
|
||
# ===== 1. 自动打标 =====
|
||
print("=== 1. analyze (自动打标) ===")
|
||
try:
|
||
result = await dh.analyze("今天学了 Python 的 asyncio，感觉收获很大，心情不错")
|
||
print(f" domain: {result['domain']}")
|
||
print(f" valence: {result['valence']}, arousal: {result['arousal']}")
|
||
print(f" tags: {result['tags']}")
|
||
print(" [OK]")
|
||
except Exception as e:
|
||
print(f" [FAIL] {e}")
|
||
print()
|
||
|
||
# ===== 2. 建桶 =====
|
||
print("=== 2. create (建桶) ===")
|
||
try:
|
||
bid = await bm.create(
|
||
content="P酱喜欢猫，家里养了一只橘猫叫小橘",
|
||
tags=["猫", "宠物"],
|
||
importance=7,
|
||
domain=["生活"],
|
||
valence=0.8,
|
||
arousal=0.4,
|
||
)
|
||
print(f" bucket_id: {bid}")
|
||
print(" [OK]")
|
||
except Exception as e:
|
||
print(f" [FAIL] {e}")
|
||
return
|
||
print()
|
||
|
||
# ===== 3. 搜索 =====
|
||
print("=== 3. search (检索) ===")
|
||
try:
|
||
hits = await bm.search("猫", limit=3)
|
||
print(f" found {len(hits)} results")
|
||
for h in hits:
|
||
name = h["metadata"].get("name", h["id"])
|
||
print(f" - {name} (score={h['score']:.1f})")
|
||
print(" [OK]")
|
||
except Exception as e:
|
||
print(f" [FAIL] {e}")
|
||
print()
|
||
|
||
# ===== 4. 脱水压缩 =====
|
||
print("=== 4. dehydrate (脱水压缩) ===")
|
||
try:
|
||
text = (
|
||
"这是一段很长的内容用来测试脱水功能。"
|
||
"P酱今天去了咖啡厅，点了一杯拿铁，然后坐在窗边看书看了两个小时。"
|
||
"期间遇到了一个朋友，聊了聊最近的工作情况。回家之后写了会代码。"
|
||
)
|
||
summary = await dh.dehydrate(text, {})
|
||
print(f" summary: {summary[:120]}...")
|
||
print(" [OK]")
|
||
except Exception as e:
|
||
print(f" [FAIL] {e}")
|
||
print()
|
||
|
||
# ===== 5. 衰减评分 =====
|
||
print("=== 5. decay score (衰减评分) ===")
|
||
try:
|
||
bucket = await bm.get(bid)
|
||
score = de.calculate_score(bucket["metadata"])
|
||
print(f" score: {score:.3f}")
|
||
print(" [OK]")
|
||
except Exception as e:
|
||
print(f" [FAIL] {e}")
|
||
print()
|
||
|
||
# ===== 6. 日记整理 =====
|
||
print("=== 6. digest (日记整理) ===")
|
||
try:
|
||
diary = (
|
||
"今天上午写了个 Python 脚本处理数据，下午和朋友去吃了火锅很开心，"
|
||
"晚上失眠了有点焦虑，想了想明天的面试。"
|
||
)
|
||
items = await dh.digest(diary)
|
||
print(f" 拆分出 {len(items)} 条记忆:")
|
||
for it in items:
|
||
print(f" - [{it.get('name','')}] domain={it['domain']} V{it['valence']:.1f}/A{it['arousal']:.1f}")
|
||
print(" [OK]")
|
||
except Exception as e:
|
||
print(f" [FAIL] {e}")
|
||
print()
|
||
|
||
# ===== 7. 清理测试数据 =====
|
||
print("=== 7. cleanup (删除测试桶) ===")
|
||
try:
|
||
ok = await bm.delete(bid)
|
||
print(f" deleted: {ok}")
|
||
print(" [OK]")
|
||
except Exception as e:
|
||
print(f" [FAIL] {e}")
|
||
print()
|
||
|
||
print("=" * 40)
|
||
print("冒烟测试完成!")
|
||
|
||
|
||
if __name__ == "__main__":
|
||
asyncio.run(main())
|