| 12345678910111213141516171819202122 |
- # -*- 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()))
|
