Add missing fields to backend, fix migration error after merging with…

… main

ami/jobs/migrations/0013_merge_0011_alter_job_limit_0012_alter_job_limit.py

@@ -0,0 +1,12 @@
+# Generated by Django 4.2.10 on 2024-12-17 22:28
+
+from django.db import migrations
+
+
+class Migration(migrations.Migration):
+    dependencies = [
+        ("jobs", "0011_alter_job_limit"),
+        ("jobs", "0012_alter_job_limit"),
+    ]
+
+    operations = []

ami/ml/migrations/0013_backend_description_backend_name_backend_slug_and_more.py

@@ -0,0 +1,39 @@
+# Generated by Django 4.2.10 on 2024-12-17 22:37
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+    dependencies = [
+        ("ml", "0012_backend_last_checked_backend_last_checked_live"),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name="backend",
+            name="description",
+            field=models.TextField(blank=True),
+        ),
+        migrations.AddField(
+            model_name="backend",
+            name="name",
+            field=models.CharField(default="ml_backend", max_length=255),
+            preserve_default=False,
+        ),
+        migrations.AddField(
+            model_name="backend",
+            name="slug",
+            field=models.SlugField(default="ml_backend", max_length=255, unique=True),
+            preserve_default=False,
+        ),
+        migrations.AddField(
+            model_name="backend",
+            name="version",
+            field=models.IntegerField(default=1),
+        ),
+        migrations.AddField(
+            model_name="backend",
+            name="version_name",
+            field=models.CharField(blank=True, max_length=255),
+        ),
+    ]

ami/ml/models/backend.py

@@ -19,6 +19,11 @@
 class Backend(BaseModel):
     """An ML processing backend"""
 
+    name = models.CharField(max_length=255)
+    slug = models.SlugField(max_length=255, unique=True)
+    description = models.TextField(blank=True)
+    version = models.IntegerField(default=1)
+    version_name = models.CharField(max_length=255, blank=True)
     projects = models.ManyToManyField("main.Project", related_name="backends", blank=True)
     endpoint_url = models.CharField(max_length=1024, null=True, blank=True)
     pipelines = models.ManyToManyField("ml.Pipeline", related_name="backends", blank=True)

ami/ml/serializers.py

@@ -34,6 +34,8 @@ class Meta:
             "key",
             "version",
             "version_name",
+            "created_at",
+            "updated_at",
         ]
 
 
@@ -46,6 +48,8 @@ class Meta:
             "endpoint_url",
             "last_checked",
             "last_checked_live",
+            "created_at",
+            "updated_at",
         ]
 
 
@@ -83,6 +87,8 @@ class Meta:
             "description",
             "version",
             "version_name",
+            "created_at",
+            "updated_at",
         ]