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


class Migration(migrations.Migration):
    dependencies = [
        ("agents", "0005_agentversion_allowed_actions"),
        ("modelhub", "0002_modelprovider_image_egress_approved_and_more"),
    ]

    operations = [
        migrations.AddField(
            model_name="agentstepbinding",
            name="model_deployment",
            field=models.ForeignKey(
                blank=True,
                null=True,
                on_delete=django.db.models.deletion.PROTECT,
                related_name="agent_step_bindings",
                to="modelhub.modeldeployment",
            ),
        ),
        migrations.AddField(
            model_name="agentrunstep",
            name="model_deployment",
            field=models.ForeignKey(
                blank=True,
                null=True,
                on_delete=django.db.models.deletion.PROTECT,
                related_name="agent_run_steps",
                to="modelhub.modeldeployment",
            ),
        ),
    ]