fix: add logo_annotation.server_type field

This is the same as ImageModel.server_type, but we store it here to
avoid performing a double join with ImageModel table when we need to
filter logos by server_type
(LogoAnnotation > ImagePrediction > ImageModel)

migrations/006_add_logo_annotation_server_type.py

@@ -0,0 +1,23 @@
+import peewee as pw
+from peewee_migrate import Migrator
+
+
+def migrate(migrator: Migrator, database: pw.Database, *, fake=False):
+    """Write your migrations here."""
+
+    migrator.add_fields(
+        "logo_annotation",
+        server_type=pw.CharField(
+            null=True,
+            max_length=10,
+            help_text="project associated with the logo annotation, "
+            "one of 'off', 'obf', 'opff', 'opf', 'off-pro'",
+            index=False,
+        ),
+    )
+
+
+def rollback(migrator: Migrator, database: pw.Database, *, fake=False):
+    """Write your rollback migrations here."""
+
+    migrator.remove_fields("logo_annotation", "server_type")

robotoff/models.py

@@ -224,8 +224,8 @@ class Prediction(BaseModel):
     server_type = peewee.CharField(
         null=False,
         max_length=10,
-        help_text="project associated with the insight, "
-        "one of 'off', 'obf', 'opff', 'opf'",
+        help_text="project associated with the prediction, "
+        "one of 'off', 'obf', 'opff', 'opf', 'off-pro'",
         index=True,
         default="off",
     )
@@ -343,6 +343,17 @@ class LogoAnnotation(BaseModel):
     source_image = peewee.TextField(null=True, index=True)
     # The logo text extracted from the image using OCR
     text = peewee.TextField(null=True)
+    # This is the same as ImageModel.server_type, but we store it here to
+    # avoid performing a double join with ImageModel table when we need to
+    # filter logos by server_type
+    # (LogoAnnotation > ImagePrediction > ImageModel)
+    server_type = peewee.CharField(
+        null=True,
+        max_length=10,
+        help_text="project associated with the logo annotation, "
+        "one of 'off', 'obf', 'opff', 'opf', 'off-pro'",
+        index=False,
+    )
 
     class Meta:
         constraints = [peewee.SQL("UNIQUE(image_prediction_id, index)")]