MrSimple07 commited on
Commit
f9516ca
·
1 Parent(s): 07d69bf

bm25 full corpuse + COMPACT + NodePostprocessor

Browse files
Files changed (1) hide show
  1. 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 LlamaIndexNodePostprocessor
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(LlamaIndexNodePostprocessor(reranker))
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,