def extract_vector_operations_count()

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

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

def extract_vector_operations_count(knn_result):
    vector_operations_count = 0
    profile = knn_result["profile"]
    for shard in profile["shards"]:
        assert len(shard["dfs"]["knn"]) == 1
        knn_search = shard["dfs"]["knn"][0]
        if "vector_operations_count" in knn_search:
            vector_operations_count += knn_search["vector_operations_count"]
    return vector_operations_count