adding new management command for creating collections and fixing ini…

…t script (#1911)

docker-compose.apps.yml

@@ -20,6 +20,8 @@ services:
         condition: service_healthy
       redis:
         condition: service_healthy
+      qdrant:
+        condition: service_healthy
     volumes:
       - .:/src
       - django_media:/var/media

docker-compose.services.yml

@@ -38,6 +38,11 @@ services:
       - "6333:6333"
     volumes:
       - qdrant-data:/qdrant/storage
+    healthcheck:
+      test: ["CMD", "cat", ".qdrant-initialized"]
+      interval: 1s
+      timeout: 3s
+      retries: 10
   nginx:
     profiles:
       - backend

scripts/run-django-dev.sh

@@ -17,7 +17,10 @@ RUN_DATA_MIGRATIONS=true python3 manage.py migrate --noinput
 echo "Loading fixtures!"
 python3 manage.py loaddata platforms schools departments offered_by
 
+# create initial qdrant collections
+python manage.py create_qdrant_collections
+
 # consolidate user subscriptions and remove duplicate percolate instances
-python $MANAGE_FILE prune_subscription_queries 2>&1 | indent
+python manage.py prune_subscription_queries 2>&1
 
 uwsgi uwsgi.ini --honour-stdin

vector_search/management/commands/create_qdrant_collections.py

@@ -0,0 +1,33 @@
+"""Management command to create Qdrant collections"""
+
+from django.core.management.base import BaseCommand
+
+from vector_search.utils import (
+    create_qdrand_collections,
+)
+
+
+class Command(BaseCommand):
+    """Creates Qdrant collections"""
+
+    help = "Create Qdrant collections"
+
+    def add_arguments(self, parser):
+        parser.add_argument(
+            "--force",
+            dest="force",
+            action="store_true",
+            help="delete existing collections and force recreate",
+        )
+
+        super().add_arguments(parser)
+
+    def handle(self, *args, **options):  # noqa: ARG002
+        """Create Qdrant collections"""
+
+        if options["force"]:
+            create_qdrand_collections(force_recreate=True)
+        else:
+            create_qdrand_collections(force_recreate=False)
+
+        self.stdout.write("Created Qdrant collections")