[REVIEW] Add OpenSearch backend to cuvs-bench (#2012)

Would still like to refine a bit but pushing up for early feedback 

cc @singhmanas1 @janakivamaraju @afourniernv

xref #1907 which does something similar but for Elastic


EDIT: Okay, this is now ready for review (cc @jnke2016)

Authors:
  - James Bourbeau (https://github.com/jrbourbeau)
  - Corey J. Nolet (https://github.com/cjnolet)

Approvers:
  - Joseph Nke (https://github.com/jnke2016)
  - Corey J. Nolet (https://github.com/cjnolet)
  - James Lamb (https://github.com/jameslamb)

URL: #2012

Author: jrbourbeau

conda/environments/bench_ann_cuda-129_arch-aarch64.yaml

@@ -37,6 +37,7 @@ dependencies:
 - ninja
 - nlohmann_json>=3.12.0
 - openblas
+- opensearch-py>=2.4.0
 - pandas
 - pylibraft==26.8.*,>=0.0.0a0
 - pyyaml

conda/environments/bench_ann_cuda-129_arch-x86_64.yaml

@@ -40,6 +40,7 @@ dependencies:
 - ninja
 - nlohmann_json>=3.12.0
 - openblas
+- opensearch-py>=2.4.0
 - pandas
 - pylibraft==26.8.*,>=0.0.0a0
 - pyyaml

conda/environments/bench_ann_cuda-132_arch-aarch64.yaml

@@ -37,6 +37,7 @@ dependencies:
 - ninja
 - nlohmann_json>=3.12.0
 - openblas
+- opensearch-py>=2.4.0
 - pandas
 - pylibraft==26.8.*,>=0.0.0a0
 - pyyaml

conda/environments/bench_ann_cuda-132_arch-x86_64.yaml

@@ -40,6 +40,7 @@ dependencies:
 - ninja
 - nlohmann_json>=3.12.0
 - openblas
+- opensearch-py>=2.4.0
 - pandas
 - pylibraft==26.8.*,>=0.0.0a0
 - pyyaml

conda/recipes/cuvs-bench-cpu/recipe.yaml

@@ -94,6 +94,7 @@ requirements:
     - h5py ${{ h5py_version }}
     - matplotlib-base>=3.9
     - numpy >=1.23,<3.0
+    - opensearch-py>=2.4.0
     - pandas
     - pyyaml
     - python

conda/recipes/cuvs-bench/recipe.yaml

@@ -47,6 +47,7 @@ requirements:
     - h5py ${{ h5py_version }}
     - libcuvs-bench-ann =${{ version }}
     - matplotlib-base>=3.9
+    - opensearch-py>=2.4.0
     - pandas
     - pylibraft =${{ minor_version }}
     - python

dependencies.yaml

@@ -35,6 +35,7 @@ files:
     includes:
       - bench
       - bench_python
+      - bench_python_opensearch
       - build_cython
       - build_py_cuvs
       - clang
@@ -213,6 +214,14 @@ files:
       table: project
     includes:
       - bench_python
+  py_opensearch_cuvs_bench:
+    output: pyproject
+    pyproject_dir: python/cuvs_bench
+    extras:
+      table: project.optional-dependencies
+      key: opensearch
+    includes:
+      - bench_python_opensearch
 channels:
   - rapidsai-nightly
   - rapidsai
@@ -577,6 +586,11 @@ dependencies:
       - output_types: [requirements, pyproject]
         packages:
           - matplotlib>=3.9
+  bench_python_opensearch:
+    common:
+      - output_types: [conda, pyproject]
+        packages:
+          - opensearch-py>=2.4.0
   depends_on_cuda_python:
     specific:
       - output_types: [conda, requirements, pyproject]

python/cuvs_bench/cuvs_bench/backends/__init__.py

@@ -25,10 +25,12 @@
 )
 
 from .cpp_gbench import CppGoogleBenchmarkBackend
+from .opensearch import OpenSearchBackend
 
 # Auto-register built-in backends
 _registry = get_registry()
 _registry.register("cpp_gbench", CppGoogleBenchmarkBackend)
+_registry.register("opensearch", OpenSearchBackend)
 
 __all__ = [
     # Base classes and data structures
@@ -43,4 +45,5 @@
     "get_backend",
     # Built-in backends
     "CppGoogleBenchmarkBackend",
+    "OpenSearchBackend",
 ]