diff --git a/.github/workflows/docker.yml b/.github/workflows/docker.yml new file mode 100644 index 000000000..ba1db45c8 --- /dev/null +++ b/.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 diff --git a/Dockerfile b/Dockerfile index 5c57ceb03..a5b9b1ec2 100644 --- a/Dockerfile +++ b/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"] diff --git a/install/requirements_py3.11.txt b/install/requirements_py3.11.txt index c55601934..2c3d4f0c6 100644 --- a/install/requirements_py3.11.txt +++ b/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 str: tmp = [ i for i in inspect.getmembers(self)