| 12345678910111213 |
- # -*- coding: utf-8 -*-
- """RerankRAG standalone test."""
- import os
- import sys
- sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
- from bdirag.rag_methods.rerank_rag import RerankRAG
- from examples.rag_test_utils import FakeRerankModel, build_vector_rag, run_retrieval_test
- if __name__ == "__main__":
- rag = build_vector_rag(RerankRAG, rerank_model=FakeRerankModel(), rerank_top_k=5)
- run_retrieval_test("RerankRAG", rag)
|
