FROM python:3.11

# Install Milvus dependencies
USER root
RUN apt-get update && apt-get install -y wget ffmpeg libsm6 libxext6 libaio1

# Download and install Milvus
RUN wget https://github.com/milvus-io/milvus/releases/download/v2.3.7/milvus_2.3.7-1_amd64.deb && \
    dpkg -i milvus_2.3.7-1_amd64.deb && \
    apt-get -f install && \
    apt-get clean && \
    rm milvus_2.3.7-1_amd64.deb

# Create a directory for Milvus data
RUN mkdir -p /milvus/data

# Set up Milvus user
RUN useradd -m -u 1000 milvus
USER milvus

# Set Milvus environment variables
ENV MILVUS_HOME=/home/milvus
ENV PATH=$MILVUS_HOME/bin:$PATH

# Set working directory
WORKDIR $MILVUS_HOME

# Expose Milvus ports
EXPOSE 19530

# Start Milvus server
CMD ["milvus", "run", "standalone", "-d", "/milvus/data"]