File size: 774 Bytes
4208392 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 | 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])
|