query_engine = index.as_query_engine(similarity_top_k=3) retriever = index.as_retriever(similarity_top_k=1) source_node = retriever.retrieve(query_str)[0] if source_node.score > 0.9: # return answer from source_node.node.text else: return query_engine.query(query_str)