def params()

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

• 19 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": {
                    "emb": {
                        "vector": self._queries[self._iters],
                        "k": self._params.get("num-candidates", 10)
                    }
                }
            },
            "_source": False
        }
        
        if "filter" in self._params:
            result["body"]["knn"]["filter"] = self._params["filter"]

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