Agent/knowhub/scripts/version_store_smoke_test.py

#!/usr/bin/env python3
"""Smoke-test: 确认 store 读路径按 contextvar 过滤版本。"""
import sys
from pathlib import Path
sys.path.insert(0, str(Path(__file__).parent.parent.parent))
from knowhub.knowhub_db.pg_strategy_store import PostgreSQLStrategyStore
from knowhub.knowhub_db.pg_capability_store import PostgreSQLCapabilityStore
from knowhub.knowhub_db.pg_store import PostgreSQLStore
from knowhub.knowhub_db.version_context import set_version


def main():
    strat = PostgreSQLStrategyStore()
    cap = PostgreSQLCapabilityStore()
    know = PostgreSQLStore()
    try:
        for v in ('dev_abstract', 'dev_dedup'):
            set_version(v)
            print(f'\n=== active_version = {v} ===', flush=True)
            print(f'  strategy.count()   : {strat.count()}', flush=True)
            print(f'  capability.count() : {cap.count()}', flush=True)
            print(f'  knowledge.count()  : {know.count()}', flush=True)
            sample = strat.list_all(limit=2)
            print(f'  strategy sample    : {[(s["id"], s["version"]) for s in sample]}', flush=True)
    finally:
        strat.close(); cap.close(); know.close()


if __name__ == '__main__':
    main()