# -*- coding: utf-8 -*-
"""RerankRAG standalone test."""
import os
import sys
sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))

from bdirag.rag_methods.rerank_rag import RerankRAG
from examples.rag_test_utils import FakeRerankModel, build_vector_rag, run_retrieval_test


if __name__ == "__main__":
    rag = build_vector_rag(RerankRAG, rerank_model=FakeRerankModel(), rerank_top_k=5)
    run_retrieval_test("RerankRAG", rag)