def params()

in openai_vector/track.py [0:0]

• 22 lines of code
• 4 McCabe index (conditional complexity)

    def params(self):
        result = {"index": self._index_name, "cache": self._params.get("cache", False), "size": self._params.get("k", 10)}
        num_candidates = self._params.get("num-candidates", 50)
        oversample = self._params.get("oversample", 0)
        query_vec = self._queries[self._iters]
        knn_query = {
            "knn": {
                "field": "emb",
                "query_vector": query_vec,
                "k": result["size"],
                "num_candidates": num_candidates,
            }
        }
        if "filter" in self._params:
            knn_query["knn"]["filter"] = self._params["filter"]
        if oversample > 0:
            knn_query["knn"]["rescore_vector"] = {"oversample": oversample}
        result["body"] = {"query": knn_query, "_source": False}
        self._iters += 1
        if self._iters >= self._maxIters:
            self._iters = 0
        return result