Spaces:

MrSimple01
/

RAG_AIEXP_01

Sleeping

MrSimple07 commited on Sep 30, 2025

Commit

e875be5

1 Parent(s): 730c607

hybrid = 30, bm = 25

Files changed (1) hide show

index_retriever.py CHANGED Viewed

@@ -16,7 +16,7 @@ def create_query_engine(vector_index):
     try:
         bm25_retriever = BM25Retriever.from_defaults(
             docstore=vector_index.docstore,
-            similarity_top_k=20
         )
         vector_retriever = VectorIndexRetriever(
@@ -27,7 +27,7 @@ def create_query_engine(vector_index):
         hybrid_retriever = QueryFusionRetriever(
             [vector_retriever, bm25_retriever],
-            similarity_top_k=45,
             num_queries=1
         )

     try:
         bm25_retriever = BM25Retriever.from_defaults(
             docstore=vector_index.docstore,
+            similarity_top_k=25
         )
         vector_retriever = VectorIndexRetriever(
         hybrid_retriever = QueryFusionRetriever(
             [vector_retriever, bm25_retriever],
+            similarity_top_k=30,
             num_queries=1
         )