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

from examples.rag_test_utils import (
    FakeLLMClient,
    install_rank_bm25_fallback,
    install_sklearn_fallback,
    run_retrieval_test,
)
install_rank_bm25_fallback()
install_sklearn_fallback()

from bdirag.rag_methods.keyword_rag import KeywordRAG


if __name__ == "__main__":
    run_retrieval_test("KeywordRAG-BM25", KeywordRAG(search_method="bm25", llm_client=FakeLLMClient()))
    print("\n")
    run_retrieval_test("KeywordRAG-TFIDF", KeywordRAG(search_method="tfidf", llm_client=FakeLLMClient()))