test_tfidf_rag.py 434 B

1234567891011121314
  1. # -*- coding: utf-8 -*-
  2. """TFIDFRAG 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.tfidf_rag import TFIDFRAG
  7. from examples.rag_test_utils import FakeLLMClient, install_sklearn_fallback, run_retrieval_test
  8. install_sklearn_fallback()
  9. if __name__ == "__main__":
  10. run_retrieval_test("TFIDFRAG", TFIDFRAG(llm_client=FakeLLMClient()))