Extend service subtypes with single cell build operations

Author: pgetta

alembic/versions/20250219_154931_a4cab64aa586_.py

@@ -0,0 +1,62 @@
+"""empty message
+
+Revision ID: a4cab64aa586
+Revises: b974ebded6ab
+Create Date: 2025-02-19 15:49:31.212536
+
+"""
+
+from collections.abc import Sequence
+
+from alembic_postgresql_enum import TableReference
+
+from alembic import op
+
+# revision identifiers, used by Alembic.
+revision: str = "a4cab64aa586"
+down_revision: str | None = "b974ebded6ab"
+branch_labels: str | Sequence[str] | None = None
+depends_on: str | Sequence[str] | None = None
+
+
+def upgrade() -> None:
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.sync_enum_values(
+        "public",
+        "servicesubtype",
+        [
+            "STORAGE",
+            "SINGLE_CELL_SIM",
+            "SINGLE_CELL_BUILD",
+            "SYNAPTOME_SIM",
+            "SYNAPTOME_BUILD",
+            "ML_RETRIEVAL",
+            "ML_LLM",
+            "ML_RAG",
+        ],
+        [
+            TableReference(table_schema="public", table_name="job", column_name="service_subtype"),
+            TableReference(
+                table_schema="public", table_name="price", column_name="service_subtype"
+            ),
+        ],
+        enum_values_to_rename=[],
+    )
+    # ### end Alembic commands ###
+
+
+def downgrade() -> None:
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.sync_enum_values(
+        "public",
+        "servicesubtype",
+        ["STORAGE", "SINGLE_CELL_SIM", "SYNAPTOME_SIM", "ML_RETRIEVAL", "ML_LLM", "ML_RAG"],
+        [
+            TableReference(table_schema="public", table_name="job", column_name="service_subtype"),
+            TableReference(
+                table_schema="public", table_name="price", column_name="service_subtype"
+            ),
+        ],
+        enum_values_to_rename=[],
+    )
+    # ### end Alembic commands ###

app/constants.py

@@ -61,7 +61,9 @@ class ServiceSubtype(HyphenStrEnum):
 
     STORAGE = auto()
     SINGLE_CELL_SIM = auto()
+    SINGLE_CELL_BUILD = auto()
     SYNAPTOME_SIM = auto()
+    SYNAPTOME_BUILD = auto()
     ML_RETRIEVAL = auto()
     ML_LLM = auto()
     ML_RAG = auto()