import sys
from pathlib import Path

root = Path(__file__).resolve().parents[1]
sys.path.append(str(root / 'runtime'))
from nzfc_hybrid_exact_recall import NZFCHybridExactRecall10M, ContextGovernor, TokenBudget

mem = NZFCHybridExactRecall10M(root)
query = '라그랑주-베셀 NZFC 기억정리에서 T_mem(q), K(q), 핵노름 사영, rank_eff 조건을 설명한 원문 passage를 정확히 다시 가져와.'
strict, selected, diag = mem.query(query, tau_trace=0.3)
print('RID:', strict[0]['rid'])
print('Exact:', strict[0]['exact_text_match'], strict[0]['exact_target_sha_match'])
governor = ContextGovernor(TokenBudget())
prompt = governor.build_prompts(query, strict, diag)
print('Prompt tokens:', prompt['combined_prompt_tokens'])
print(prompt['memory_pack'][:5000])