Merge pull request #31 from cococo2000/main

Add redis

.github/workflows/bvb-run.yml

@@ -34,6 +34,7 @@ jobs:
           - milvus
           - qdrant
           - weaviate
+          - redis
           # - vearch # Vearch test is disabled temporarily due to disk space constraints.
           # GPU support is not available in the free tier
           # - milvus_gpu

README.md

@@ -97,11 +97,13 @@ python run.py --dataset app_reviews-384-euclidean-filter --count 100 --runs 3 --
   - vearch-ivfflat
   - vearch-ivfpq
   - vearch-hnsw
+- [Redis](https://redis.io/)
+  - redis-flat
+  - redis-hnsw
 
 **TODO**
 
 - [Elasitcsearch](https://www.elastic.co/)
-- [Redis](https://redis.io/)
 - [Vespa](https://vespa.ai/)
 - [SPTAG](https://github.com/microsoft/SPTAG)
 - [pgvector](https://github.com/pgvector/pgvector)

bigvectorbench/algorithms/redis/Dockerfile

@@ -0,0 +1,6 @@
+FROM bigvectorbench-base
+
+RUN pip install redisvl==0.3.2
+RUN rvl version
+
+COPY bigvectorbench/algorithms/redis/docker-compose.yml ./

bigvectorbench/algorithms/redis/config.yml

@@ -0,0 +1,23 @@
+float:
+  euclidean:
+    - base_args: ["@metric", "@dimension"]
+      constructor: RedisFlat
+      disabled: False
+      docker_tag: bigvectorbench-redis
+      module: bigvectorbench.algorithms.redis
+      name: redis-flat
+      run_groups:
+        FLAT:
+          args: []
+    - base_args: ["@metric", "@dimension"]
+      constructor: RedisHNSW
+      disabled: False
+      docker_tag: bigvectorbench-redis
+      module: bigvectorbench.algorithms.redis
+      name: redis-hnsw
+      run_groups:
+        HNSW:
+          args:
+            m: [4, 8, 12, 16, 24, 36, 48, 64, 96]
+            ef_construction: [200, 500]
+            ef_runtime: [100, 120, 150, 200, 500, 600, 800]

bigvectorbench/algorithms/redis/docker-compose.yml

@@ -0,0 +1,14 @@
+services:
+  redis:
+    image: redis/redis-stack-server:7.4.0-v0
+    container_name: redis
+    ports:
+      - "6379:6379"
+    environment:
+      - "REDIS_ARGS=--save '' --appendonly no"
+    deploy:
+      replicas: 1
+      restart_policy:
+        condition: on-failure
+    labels:
+      - "com.docker.compose.publishers=redis,6379,6379"