Update Dockerfile

Dockerfile

@@ -6,6 +6,26 @@ RUN apt-get update \
     && pip install psycopg2
 # Install uvicorn
 RUN pip install uvicorn
+
+# Use an official PostgreSQL image
+FROM postgres:16
+
+# Set environment variables
+ENV POSTGRES_DB ai
+ENV POSTGRES_USER ai
+ENV POSTGRES_PASSWORD ai
+
+# Mount a volume for storing data
+VOLUME pgvolume
+
+# Expose port 5532
+EXPOSE 5532
+
+# Run the container in detached mode
+CMD ["docker", "run", "-d", "--name", "pgvector", "phidata/pgvector:16"]
+
+RUN pip install ollama
+RUN ollama pull nomic-embed-text
 # Install dependencies
 RUN pip install -r requirements.txt
 COPY . /app