def __init__()

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

• 15 lines of code
• 2 McCabe index (conditional complexity)

    def __init__(self, track, params, **kwargs):
        if len(track.indices) == 1:
            default_index = track.indices[0].name
        else:
            default_index = "_all"

        self._index_name = params.get("index", default_index)
        self._cache = params.get("cache", False)
        self._params = params
        self.infinite = True
        self._target_k = 1_000
        cwd = os.path.dirname(__file__)

        self._queries_file = os.path.join(cwd, 'queries.json')

        with open(os.path.join(cwd, QUERIES_FILENAME), 'rb') as source, open(self._queries_file, 'wb') as dest:
            dest.write(bz2.decompress(source.read()))

        self._vector_field: str = "emb"