| 1234567891011121314 |
- # -*- coding: utf-8 -*-
- """TFIDFRAG standalone test."""
- import os
- import sys
- sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
- from bdirag.rag_methods.tfidf_rag import TFIDFRAG
- from examples.rag_test_utils import FakeLLMClient, install_sklearn_fallback, run_retrieval_test
- install_sklearn_fallback()
- if __name__ == "__main__":
- run_retrieval_test("TFIDFRAG", TFIDFRAG(llm_client=FakeLLMClient()))
|