zenml-nightly 0.75.0.dev20250312__py3-none-any.whl → 0.75.0.dev20250314__py3-none-any.whl

zenml/zen_stores/migrations/versions/b557b2871693_update_step_run_input_types.py

@@ -17,17 +17,21 @@ depends_on = None
 
 def upgrade() -> None:
     """Upgrade database schema and/or data, creating a new revision."""
-    op.execute("""
+    op.execute(
+        """
         UPDATE step_run_input_artifact
         SET type = 'step_output'
         WHERE type = 'default'
-    """)
+    """
+    )
 
 
 def downgrade() -> None:
     """Downgrade database schema and/or data back to the previous revision."""
-    op.execute("""
+    op.execute(
+        """
         UPDATE step_run_input_artifact
         SET type = 'default'
         WHERE type = 'step_output'
-    """)
+    """
+    )

zenml/zen_stores/migrations/versions/cbc6acd71f92_add_workspace_display_name.py

@@ -0,0 +1,58 @@
+"""add workspace display name [cbc6acd71f92].
+
+Revision ID: cbc6acd71f92
+Revises: 9e7bf0970266
+Create Date: 2025-03-12 19:38:57.126846
+
+"""
+
+import sqlalchemy as sa
+import sqlmodel
+from alembic import op
+
+# revision identifiers, used by Alembic.
+revision = "cbc6acd71f92"
+down_revision = "9e7bf0970266"
+branch_labels = None
+depends_on = None
+
+
+def upgrade() -> None:
+    """Upgrade database schema and/or data, creating a new revision."""
+    with op.batch_alter_table("workspace", schema=None) as batch_op:
+        batch_op.add_column(
+            sa.Column(
+                "display_name",
+                sqlmodel.sql.sqltypes.AutoString(),
+                nullable=True,
+            )
+        )
+
+    # Migrate existing workspace names to display_name
+    connection = op.get_bind()
+    meta = sa.MetaData()
+    meta.reflect(bind=connection, only=("workspace",))
+    workspace_table = sa.Table("workspace", meta)
+
+    for workspace_id, workspace_name in connection.execute(
+        sa.select(
+            workspace_table.c.id,
+            workspace_table.c.name,
+        )
+    ):
+        connection.execute(
+            sa.update(workspace_table)
+            .where(workspace_table.c.id == workspace_id)
+            .values(display_name=workspace_name)
+        )
+
+    with op.batch_alter_table("workspace", schema=None) as batch_op:
+        batch_op.alter_column(
+            "display_name", existing_type=sa.String(50), nullable=False
+        )
+
+
+def downgrade() -> None:
+    """Downgrade database schema and/or data back to the previous revision."""
+    with op.batch_alter_table("workspace", schema=None) as batch_op:
+        batch_op.drop_column("display_name")

zenml/zen_stores/migrations/versions/cc269488e5a9_separate_run_metadata.py

@@ -41,10 +41,12 @@ def upgrade() -> None:
     connection = op.get_bind()
 
     run_metadata_data = connection.execute(
-        sa.text("""
+        sa.text(
+            """
             SELECT id, resource_id, resource_type
             FROM run_metadata
-        """)
+        """
+        )
     ).fetchall()
 
     # Prepare data with new UUIDs for bulk insert
@@ -107,20 +109,24 @@ def downgrade() -> None:
 
     # Fetch data from `run_metadata_resource`
     run_metadata_resource_data = connection.execute(
-        sa.text("""
+        sa.text(
+            """
             SELECT resource_id, resource_type, run_metadata_id
             FROM run_metadata_resource
-        """)
+        """
+        )
     ).fetchall()
 
     # Update `run_metadata` with the data from `run_metadata_resource`
     for row in run_metadata_resource_data:
         connection.execute(
-            sa.text("""
+            sa.text(
+                """
                 UPDATE run_metadata
                 SET resource_id = :resource_id, resource_type = :resource_type
                 WHERE id = :run_metadata_id
-            """),
+            """
+            ),
             {
                 "resource_id": row.resource_id,
                 "resource_type": row.resource_type,

zenml/zen_stores/migrations/versions/f1d723fd723b_add_secret_private_attr.py

@@ -0,0 +1,61 @@
+"""add secret private attr [f1d723fd723b].
+
+Revision ID: f1d723fd723b
+Revises: 41b28cae31ce
+Create Date: 2025-02-21 19:40:14.596681
+
+"""
+
+import sqlalchemy as sa
+from alembic import op
+
+# revision identifiers, used by Alembic.
+revision = "f1d723fd723b"
+down_revision = "41b28cae31ce"
+branch_labels = None
+depends_on = None
+
+
+def upgrade() -> None:
+    """Upgrade database schema and/or data, creating a new revision."""
+    # Step 1: Add private column as nullable
+    with op.batch_alter_table("secret", schema=None) as batch_op:
+        batch_op.add_column(sa.Column("private", sa.Boolean(), nullable=True))
+
+    # Step 2: Update existing records based on scope
+    connection = op.get_bind()
+    connection.execute(
+        sa.text("UPDATE secret SET private = TRUE WHERE scope = 'user';")
+    )
+    connection.execute(
+        sa.text("UPDATE secret SET private = FALSE WHERE scope != 'user';")
+    )
+
+    # Step 3: Make private column non-nullable and drop scope
+    with op.batch_alter_table("secret", schema=None) as batch_op:
+        batch_op.alter_column(
+            "private", existing_type=sa.Boolean(), nullable=False
+        )
+        batch_op.drop_column("scope")
+
+
+def downgrade() -> None:
+    """Downgrade database schema and/or data back to the previous revision."""
+    with op.batch_alter_table("secret", schema=None) as batch_op:
+        batch_op.add_column(sa.Column("scope", sa.VARCHAR(), nullable=True))
+
+    # Restore scope values based on private flag
+    connection = op.get_bind()
+    connection.execute(
+        sa.text("UPDATE secret SET scope = 'user' WHERE private = TRUE;")
+    )
+    connection.execute(
+        sa.text("UPDATE secret SET scope = 'workspace' WHERE private = FALSE;")
+    )
+
+    # Make scope non-nullable and drop private
+    with op.batch_alter_table("secret", schema=None) as batch_op:
+        batch_op.alter_column(
+            "scope", existing_type=sa.VARCHAR(), nullable=False
+        )
+        batch_op.drop_column("private")

zenml/zen_stores/migrations/versions/f76a368a25a5_add_stack_description.py

@@ -0,0 +1,35 @@
+"""add stack description [f76a368a25a5].
+
+Revision ID: f76a368a25a5
+Revises: f1d723fd723b
+Create Date: 2025-02-22 19:03:32.568076
+
+"""
+
+import sqlalchemy as sa
+import sqlmodel
+from alembic import op
+
+# revision identifiers, used by Alembic.
+revision = "f76a368a25a5"
+down_revision = "f1d723fd723b"
+branch_labels = None
+depends_on = None
+
+
+def upgrade() -> None:
+    """Upgrade database schema and/or data, creating a new revision."""
+    with op.batch_alter_table("stack", schema=None) as batch_op:
+        batch_op.add_column(
+            sa.Column(
+                "description",
+                sqlmodel.sql.sqltypes.AutoString(),
+                nullable=True,
+            )
+        )
+
+
+def downgrade() -> None:
+    """Downgrade database schema and/or data back to the previous revision."""
+    with op.batch_alter_table("stack", schema=None) as batch_op:
+        batch_op.drop_column("description")