deploy: own deploy

Signed-off-by: cutecutecat <[email protected]>

.github/workflows/docker.yml

@@ -0,0 +1,31 @@
+name: Release Nightly
+
+on:
+  workflow_dispatch:
+  pull_request:
+
+permissions:
+  actions: write
+
+jobs:
+  trigger:
+    runs-on: ubuntu-latest
+    steps:
+      - name: Checkout
+        uses: actions/checkout@v4
+      - name: Set up QEMU
+        uses: docker/setup-qemu-action@v3
+      - name: Set up Docker Buildx
+        uses: docker/setup-buildx-action@v3
+      - name: Login to Docker Hub
+        uses: docker/login-action@v2
+        with:
+          username: ${{ secrets.DOCKERIO_USERNAME }}
+          password: ${{ secrets.DOCKERIO_TOKEN }}
+      - name: Push binary release to Docker Registry
+        uses: docker/build-push-action@v4
+        with:
+          context: .
+          push: true
+          file: .Dockerfile
+          tags: ${{ secrets.DOCKERIO_USERNAME }}/vectordbbench:latest

Dockerfile

@@ -14,5 +14,7 @@ COPY --from=builder-image /usr/local/lib/python3.11/site-packages /usr/local/lib
 WORKDIR /opt/code
 COPY . .
 ENV PYTHONPATH /opt/code
+RUN sed -i 's/deb.debian.org/mirrors.ustc.edu.cn/g' /etc/apt/sources.list
+RUN apt-get update && apt-get install libpq5 -y
 
 ENTRYPOINT ["python3", "-m", "vectordb_bench"]

install/requirements_py3.11.txt

@@ -1,13 +1,6 @@
 grpcio==1.53.0
 grpcio-tools==1.53.0
-qdrant-client
-pinecone-client
-weaviate-client
-elasticsearch
-pgvector
-sqlalchemy
-redis
-chromadb
+pgvecto.rs @ git+https://github.com/cutecutecat/pgvecto.rs-py@django-types
 pytz
 streamlit-autorefresh
 streamlit>=1.23.0
@@ -21,3 +14,4 @@ environs
 pydantic<v2
 scikit-learn
 pymilvus
+psycopg

vectordb_bench/__init__.py

@@ -46,14 +46,14 @@ class config:
     LOAD_TIMEOUT_1536D_500K     = 2.5 * 3600 # 2.5h
     LOAD_TIMEOUT_1536D_5M       =  25 * 3600 # 25h
 
-    OPTIMIZE_TIMEOUT_DEFAULT    = 30 * 60   # 30min
-    OPTIMIZE_TIMEOUT_768D_1M    =  30 * 60   # 30min
-    OPTIMIZE_TIMEOUT_768D_10M   = 5 * 3600 # 5h
-    OPTIMIZE_TIMEOUT_768D_100M  =  50 * 3600 # 50h
+    OPTIMIZE_TIMEOUT_DEFAULT    = LOAD_TIMEOUT_DEFAULT
+    OPTIMIZE_TIMEOUT_768D_1M    =  LOAD_TIMEOUT_768D_1M
+    OPTIMIZE_TIMEOUT_768D_10M   = LOAD_TIMEOUT_768D_10M
+    OPTIMIZE_TIMEOUT_768D_100M  =  LOAD_TIMEOUT_768D_100M
 
 
-    OPTIMIZE_TIMEOUT_1536D_500K =  15 * 60   # 15min
-    OPTIMIZE_TIMEOUT_1536D_5M   =   2.5 * 3600 # 2.5h
+    OPTIMIZE_TIMEOUT_1536D_500K =  LOAD_TIMEOUT_1536D_500K
+    OPTIMIZE_TIMEOUT_1536D_5M   =   LOAD_TIMEOUT_1536D_5M
     def display(self) -> str:
         tmp = [
             i for i in inspect.getmembers(self)