INNER CODE UNIT · Python

create_query_engine

wzdavid/ThinkRAG · server/engine.py:9

def create_query_engine(index, 
                        top_k=config.TOP_K, 
                        response_mode=config.RESPONSE_MODE, 
                        use_reranker=config.USE_RERANKER, 
                        top_n=config.RERANKER_MODEL_TOP_N, 
                        reranker=config.DEFAULT_RERANKER_MODEL):
    # Customized query engine with hybrid search and reranker
    node_postprocessors = [create_reranker_model(model_name=reranker, top_n=top_n)] if use_reranker else []
    retriever = SimpleFusionRetriever(vector_index=index, top_k=top_k)

    query_engine = RetrieverQueryEngine.from_args(
        retriever=retriever,
        text_qa_template=text_qa_template,
        refine_template=refine_template,
        node_postprocessors=node_postprocessors,
        response_mode=response_mode, # https://docs.llamaindex.ai/en/stable/api_reference/response_synthesizers/
        verbose=True,
        streaming=True,

View source record →