| @@ -496,6 +496,8 @@ class DatasetRetrieval: | |||
| all_documents = self.calculate_keyword_score(query, all_documents, top_k) | |||
| elif index_type == "high_quality": | |||
| all_documents = self.calculate_vector_score(all_documents, top_k, score_threshold) | |||
| else: | |||
| all_documents = all_documents[:top_k] if top_k else all_documents | |||
| self._on_query(query, dataset_ids, app_id, user_from, user_id) | |||