def params()

in rally-custom/custom_tracks/opensearch/dense_vector/track.py [0:0]

• 36 lines of code
• 3 McCabe index (conditional complexity)

    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