test_hybrid_search_rag.py 502 B

123456789101112131415
  1. # -*- coding: utf-8 -*-
  2. """HybridSearchRAG standalone test."""
  3. import os
  4. import sys
  5. sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
  6. from examples.rag_test_utils import install_rank_bm25_fallback
  7. install_rank_bm25_fallback()
  8. from bdirag.rag_methods.hybrid_search_rag import HybridSearchRAG
  9. from examples.rag_test_utils import build_vector_rag, run_retrieval_test
  10. if __name__ == "__main__":
  11. run_retrieval_test("HybridSearchRAG", build_vector_rag(HybridSearchRAG))