Spaces:
Sleeping
Sleeping
Commit ·
f9516ca
1
Parent(s): 07d69bf
bm25 full corpuse + COMPACT + NodePostprocessor
Browse files- index_retriever.py +2 -2
index_retriever.py
CHANGED
|
@@ -7,7 +7,7 @@ from llama_index.retrievers.bm25 import BM25Retriever
|
|
| 7 |
from llama_index.core.retrievers import QueryFusionRetriever
|
| 8 |
from my_logging import log_message
|
| 9 |
from config import CUSTOM_PROMPT, PROMPT_SIMPLE_POISK
|
| 10 |
-
from llama_index.core.postprocessor import
|
| 11 |
|
| 12 |
|
| 13 |
def create_vector_index(documents):
|
|
@@ -44,7 +44,7 @@ def create_query_engine(vector_index, reranker=None):
|
|
| 44 |
# Add reranker as a NodePostprocessor if provided
|
| 45 |
node_postprocessors = []
|
| 46 |
if reranker is not None:
|
| 47 |
-
node_postprocessors.append(
|
| 48 |
|
| 49 |
query_engine = RetrieverQueryEngine(
|
| 50 |
retriever=hybrid_retriever,
|
|
|
|
| 7 |
from llama_index.core.retrievers import QueryFusionRetriever
|
| 8 |
from my_logging import log_message
|
| 9 |
from config import CUSTOM_PROMPT, PROMPT_SIMPLE_POISK
|
| 10 |
+
from llama_index.core.postprocessor import BaseNodePostprocessor
|
| 11 |
|
| 12 |
|
| 13 |
def create_vector_index(documents):
|
|
|
|
| 44 |
# Add reranker as a NodePostprocessor if provided
|
| 45 |
node_postprocessors = []
|
| 46 |
if reranker is not None:
|
| 47 |
+
node_postprocessors.append(BaseNodePostprocessor(reranker))
|
| 48 |
|
| 49 |
query_engine = RetrieverQueryEngine(
|
| 50 |
retriever=hybrid_retriever,
|