"""
ورودی RAG: لحن، پایگاه دانش و اطلاعات کاربر را embed و به Qdrant می‌فرستد.
اجرا: python manage.py rag_ingest [--recreate]
"""
from django.core.management.base import BaseCommand

from rag.ingest import ingest


class Command(BaseCommand):
    help = "Embed لحن، پایگاه دانش و اطلاعات کاربر و ذخیره در Qdrant"

    def add_arguments(self, parser):
        parser.add_argument(
            "--recreate",
            action="store_true",
            help="collection را از نو بساز (حذف و ایجاد مجدد)",
        )

    def handle(self, *args, **options):
        recreate = options.get("recreate", False)
        result = ingest(recreate=recreate)
        if "error" in result:
            self.stderr.write(self.style.ERROR(result["error"]))
            return
        self.stdout.write(
            self.style.SUCCESS(
                f"✓ {result['chunks_added']} چانک از منابع {result['sources']} ذخیره شد."
            )
        )
