diff --git a/rag/utils/infinity_conn.py b/rag/utils/infinity_conn.py index 46d2c4054..03251e72c 100644 --- a/rag/utils/infinity_conn.py +++ b/rag/utils/infinity_conn.py @@ -156,7 +156,6 @@ class InfinityConnection(DocStoreConnection): msg = f"Infinity {infinity_uri} is unhealthy in 120s." logger.error(msg) raise Exception(msg) - self.column_vec_patt = re.compile(r"q_(?P\d+)_vec") logger.info(f"Infinity {infinity_uri} is healthy.") def _migrate_db(self, inf_conn): @@ -324,8 +323,7 @@ class InfinityConnection(DocStoreConnection): output.append(score_func) if PAGERANK_FLD not in output: output.append(PAGERANK_FLD) - output = [f for f in output if f != "_score" and self.column_vec_patt.match(f) is None] - logger.info(f"infinity output fields: {output}") + output = [f for f in output if f != "_score"] if limit <= 0: # ElasticSearch default limit is 10000 limit = 10000