def params()

in rally-custom/custom_tracks/elasticsearch/openai_vector/track.py [0:0]

• 18 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)}

        result["body"] = {
            "query": {
                "knn": {
                    "field": "emb",
                    "query_vector": self._queries[self._iters],
                    "num_candidates": self._params.get("num-candidates", 50)
                }
            },
            "_source": False
        }

        if "filter" in self._params:
            #add the prefilter to the knn query
            result["body"]["query"]["knn"]["filter"] = self._params["filter"]

        self._iters += 1
        if self._iters >= self._maxIters:
            self._iters = 0
        return result