in rally-custom/custom_tracks/opensearch/dense_vector/track.py [0:0]
def params(self):
result = {"index": self._index_name, "cache": self._params.get("cache", False), "size": self._params.get("k", 10)}
if self._exact_scan:
result["body"] = {
"_source": False,
"query": {
"script_score": {
"query": {"match_all": {}},
"script": {
"source": "knn_score",
"lang": "knn",
"params": {
"field": self._vector_field,
"query_value": self._queries[self._iters],
"space_type": "cosinesimil"
}
}
}
}
}
else:
result["body"] = {
"query": {
"knn": {
self._vector_field: {
"vector": self._queries[self._iters],
"k": self._params.get("num-candidates", 10)
}
}
},
"_source": False
}
self._iters += 1
if self._iters >= len(self._queries):
self._iters = 0
return result