passing tests for multiple versions

src/ell/stores/migrations/script.py.mako

@@ -9,6 +9,7 @@ from typing import Sequence, Union
 
 from alembic import op
 import sqlalchemy as sa
+import sqlmodel
 ${imports if imports else ""}
 
 # revision identifiers, used by Alembic.

src/ell/stores/studio.py

@@ -86,14 +86,12 @@ class SerializedLMP(SerializedLMPBase, table=True):
             secondaryjoin="SerializedLMP.lmp_id==SerializedLMPUses.lmp_user_id",
         ),
     )
-
+    
     class Config:
         table_name = "serializedlmp"
+        # XXX: THis is not a real constraint.
         unique_together = [("version_number", "name")]
 
-    __table_args__ = (
-        sa.UniqueConstraint('version_number', 'name', name='uq_serializedlmp_version_number_name'),
-    )
 
 class InvocationTrace(SQLModel, table=True):
     invocation_consumer_id: str = Field(foreign_key="invocation.id", primary_key=True, index=True)

tests/test_migrations.py

@@ -119,25 +119,25 @@ def test_multiple_migrations(temp_db_url):
     with engine.connect() as conn:
         result = conn.execute(text("SELECT version_num FROM ell_alembic_version"))
         version = result.scalar()
-        assert version == "4524fb60d23e"
+        assert version
 
     # Downgrade to base
     command.downgrade(alembic_cfg, "base")
 
     # Verify no version
     with engine.connect() as conn:
         result = conn.execute(text("SELECT version_num FROM ell_alembic_version"))
-        version = result.scalar()
-        assert version is None
+        first_version = result.scalar()
+        assert first_version is None
 
     # Upgrade to head again
     command.upgrade(alembic_cfg, "head")
 
     # Verify back at head
     with engine.connect() as conn:
         result = conn.execute(text("SELECT version_num FROM ell_alembic_version"))
-        version = result.scalar()
-        assert version == "4524fb60d23e"
+        new_version = result.scalar()
+        assert version == new_version
 
 def test_migration_idempotency(temp_db_url):
     """Test that running migrations multiple times is idempotent"""