为何collection.num_entities超1000且limit=1000时仍仅返回10条结果？

问题场景 #

当集合实体数collection.num_entities为8503（大于1000）时，调用collection.search方法并设置limit=1000，但查询结果始终仅返回10条数据。

用户执行脚本 #

collection = Collection(name="test_collection")

collection.load()
print(collection.num_entities)
search_params = {
"metric_type": "IP",
"params": {"nprobe": 10}
}

np_query_embeddings = np.array(query_embeddings, dtype=np.float32)
norms = np.linalg.norm([np_query_embeddings], axis=1, keepdims=True)
normalized_query_embeddings = np_query_embeddings / norms

query_vectors = normalized_query_embeddings.tolist()
limit = 1000

results = collection.search(
data=query_vectors,
anns_field="embedding",
param=search_params,
limit=limit,
consistency_level="Strong",
)
print(results)

查询输出示例 #

8503 ["['id: 449617239426872838, distance: 0.7159370183944702, entity: {}', 'id: 449617239426877164, distance: 0.7090992331504822, entity: {}', 'id: 449617239426880450, distance: 0.699631929397583, entity: {}', 'id: 449617239426876554, distance: 0.6782248020172119, entity: {}', 'id: 449617239426876178, distance: 0.6769057512283325, entity: {}', 'id: 449617239426880350, distance: 0.6754627823829651, entity: {}', 'id: 449617239426875291, distance: 0.6698116064071655, entity: {}', 'id: 449617239426877126, distance: 0.6647711396217346, entity: {}', 'id: 449617239426876516, distance: 0.6610612869262695, entity: {}', 'id: 449617239426880437, distance: 0.6571118831634521, entity: {}']"]

原因分析 #

Milvus存在全局配置项query.max_return_row，默认值为10。这个配置会限制单条查询返回的最大结果数，优先级高于search方法中设置的limit参数，因此即使设置了limit=1000，最终返回结果也会被限制为10条。

解决方案 #

方法1：修改配置文件 #

1. 找到Milvus的配置文件milvus.yaml，定位到query模块下的max_return_row配置；
2. 将其值修改为你需要的最大返回数（例如1000或更大）：

query:
  max_return_row: 1000

3. 重启Milvus服务，使配置生效。

方法2：通过环境变量临时修改 #

启动Milvus前，设置环境变量调整最大返回行数：

export MILVUS_QUERY_MAX_RETURN_ROW=1000

然后再启动Milvus服务。

注意事项 #

如果使用Milvus集群版本，需要确保所有查询节点的配置都同步修改，避免出现不一致的情况。

修改完成后重新执行查询脚本，limit=1000即可返回对应数量的结果。

内容的提问来源于stack exchange，提问作者rachel song