scripts/_speed_test.py

import sys, os, time, warnings, logging
os.environ["TOKENIZERS_PARALLELISM"] = "false"
warnings.filterwarnings("ignore"); logging.disable(logging.CRITICAL)
sys.path.insert(0, str(__import__("pathlib").Path(__file__).parent.parent))
from dotenv import load_dotenv; load_dotenv()

from core.agent import run_agent

queries = ["내 주식 얼마야?", "배당금 언제 들어와?", "주문 취소하고 싶어"]
for q in queries:
    t = time.time()
    result = run_agent(q, top_n=5, threshold=0.3, use_reranker=False, use_hyde=True)
    elapsed = time.time() - t
    results = result.get("search_results", [])
    top1 = results[0]["menu_name"] if results else "없음"
    retry = result.get("retry_count", 0)
    print(f"[{elapsed:.1f}s] {q[:15]} → {top1} (결과:{len(results)}개, 재시도:{retry}회)")