Don't select vector on infinity (#11151)

### What problem does this PR solve?

Don't select vector on infinity

### Type of change

- [x] Performance Improvement
This commit is contained in:
Zhichang Yu
2025-11-10 18:01:40 +08:00
committed by GitHub
parent 0879b6af2c
commit 68b952abb1
3 changed files with 45 additions and 2 deletions

View File

@ -156,6 +156,7 @@ 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<vector_size>\d+)_vec")
logger.info(f"Infinity {infinity_uri} is healthy.")
def _migrate_db(self, inf_conn):
@ -323,7 +324,8 @@ 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"]
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}")
if limit <= 0:
# ElasticSearch default limit is 10000
limit = 10000