test_bm25_html_tree_rag.py 355 B

123456789101112
  1. # -*- coding: utf-8 -*-
  2. """BM25HTMLTreeRAG 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 bdirag.rag_methods.bm25_html_tree_rag import BM25HTMLTreeRAG
  7. from examples.rag_test_utils import run_html_tree_test
  8. if __name__ == "__main__":
  9. run_html_tree_test(BM25HTMLTreeRAG)