from django.conf import settings
from django.db import migrations, models
import django.db.models.deletion


class Migration(migrations.Migration):
    dependencies = [
        migrations.swappable_dependency(settings.AUTH_USER_MODEL),
        ("modelhub", "0003_structured_output_governance"),
    ]

    operations = [
        migrations.AlterField(
            model_name="modelprovider",
            name="name",
            field=models.CharField(max_length=120),
        ),
        migrations.AddField(
            model_name="modelprovider",
            name="archived_at",
            field=models.DateTimeField(blank=True, db_index=True, null=True),
        ),
        migrations.AddField(
            model_name="modelprovider",
            name="archived_by",
            field=models.ForeignKey(
                blank=True,
                null=True,
                on_delete=django.db.models.deletion.SET_NULL,
                related_name="archived_model_providers",
                to=settings.AUTH_USER_MODEL,
            ),
        ),
        migrations.AddField(
            model_name="modelprovider",
            name="credential_updated_at",
            field=models.DateTimeField(blank=True, null=True),
        ),
        migrations.AddField(
            model_name="modelprovider",
            name="credential_updated_by",
            field=models.ForeignKey(
                blank=True,
                null=True,
                on_delete=django.db.models.deletion.SET_NULL,
                related_name="updated_model_provider_credentials",
                to=settings.AUTH_USER_MODEL,
            ),
        ),
        migrations.AddField(
            model_name="modeldeployment",
            name="archived_at",
            field=models.DateTimeField(blank=True, db_index=True, null=True),
        ),
        migrations.AddField(
            model_name="modeldeployment",
            name="archived_by",
            field=models.ForeignKey(
                blank=True,
                null=True,
                on_delete=django.db.models.deletion.SET_NULL,
                related_name="archived_model_deployments",
                to=settings.AUTH_USER_MODEL,
            ),
        ),
        migrations.AddField(
            model_name="modeldeployment",
            name="archived_via_provider",
            field=models.BooleanField(default=False),
        ),
        migrations.AddField(
            model_name="modelinvocation",
            name="diagnostics_scrubbed_at",
            field=models.DateTimeField(blank=True, null=True),
        ),
        migrations.RemoveConstraint(
            model_name="modeldeployment",
            name="unique_provider_model_name",
        ),
        migrations.AddConstraint(
            model_name="modelprovider",
            constraint=models.UniqueConstraint(
                condition=models.Q(("archived_at__isnull", True)),
                fields=("name",),
                name="unique_active_model_provider_name",
            ),
        ),
        migrations.AddConstraint(
            model_name="modeldeployment",
            constraint=models.UniqueConstraint(
                condition=models.Q(("archived_at__isnull", True)),
                fields=("provider", "model_name"),
                name="unique_active_provider_model_name",
            ),
        ),
        migrations.AddIndex(
            model_name="modelinvocation",
            index=models.Index(
                fields=["deployment", "created_at"],
                name="modelhub_mo_deploym_844a9d_idx",
            ),
        ),
    ]