add query_top_k