Enable mlfow model deployments with a click (#584)

* enable mlfow deployments

* update actions

* use ubuntu 22

* remove old python version

* ch: try version 3.9

* remove typo

* remove 3.11 version

Author: khalifan-kfan

.github/workflows/test.yml

@@ -12,7 +12,7 @@ env:
 jobs:
   build:
 
-    runs-on: ubuntu-20.04
+    runs-on: ubuntu-24.04
 
     services:
       postgres:
@@ -38,7 +38,7 @@ jobs:
 
     strategy:
       matrix:
-        python-version: ["3.6", "3.7", "3.10"]
+        python-version: [ "3.9", "3.10"]
 
     steps:
     - uses: actions/checkout@v2

api_docs.yml

@@ -2870,6 +2870,12 @@ paths:
               model_image_uri:
                 type: string
                 example: ""
+              mlflow_artifact_uri:
+                type: string
+                example: "run:/658456547698779878/model"
+              is_mlflow:
+                type: boolean
+                example: false
               api_type:
                 type: string
                 example: "REST"

app/schemas/app.py

@@ -85,6 +85,8 @@ class MLAppDeploySchema(Schema):
     is_notebook = fields.Boolean(required=True)
     is_modal = fields.Bool(required=False)
     model_image_uri = fields.Str(required=False)
+    mlflow_artifact_uri = fields.Str(required=False)
+    is_mlflow = fields.Boolean(required=False)
     api_type = fields.Str(required=False, default="REST", validate=validate.OneOf(
         ["REST", "GRPC"]
     ))

migrations/versions/d4fe6198fe3d_.py

@@ -0,0 +1,45 @@
+"""empty message
+
+Revision ID: d4fe6198fe3d
+Revises: c7eb8227cead
+Create Date: 2025-04-08 11:21:22.937379
+
+"""
+from alembic import op
+import sqlalchemy as sa
+from sqlalchemy.dialects import postgresql
+
+# revision identifiers, used by Alembic.
+revision = 'd4fe6198fe3d'
+down_revision = 'c7eb8227cead'
+branch_labels = None
+depends_on = None
+
+
+def upgrade():
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.drop_table('project_database')
+    op.add_column('user', sa.Column('last_reminder_sent', sa.DateTime(), nullable=True))
+    # ### end Alembic commands ###
+
+
+def downgrade():
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.drop_column('user', 'last_reminder_sent')
+    op.create_table('project_database',
+    sa.Column('id', postgresql.UUID(), server_default=sa.text('uuid_generate_v4()'), autoincrement=False, nullable=False),
+    sa.Column('project_id', postgresql.UUID(), autoincrement=False, nullable=True),
+    sa.Column('date_created', postgresql.TIMESTAMP(), autoincrement=False, nullable=True),
+    sa.Column('host', sa.VARCHAR(length=256), autoincrement=False, nullable=True),
+    sa.Column('name', sa.VARCHAR(length=256), autoincrement=False, nullable=False),
+    sa.Column('password', sa.VARCHAR(length=256), autoincrement=False, nullable=False),
+    sa.Column('user', sa.VARCHAR(length=256), autoincrement=False, nullable=False),
+    sa.Column('port', sa.INTEGER(), autoincrement=False, nullable=True),
+    sa.Column('database_flavour_name', sa.VARCHAR(length=256), autoincrement=False, nullable=True),
+    sa.Column('deleted', sa.BOOLEAN(), autoincrement=False, nullable=True),
+    sa.Column('disabled', sa.BOOLEAN(), autoincrement=False, nullable=True),
+    sa.Column('admin_disabled', sa.BOOLEAN(), autoincrement=False, nullable=True),
+    sa.ForeignKeyConstraint(['project_id'], ['project.id'], name='project_database_project_id_fkey'),
+    sa.PrimaryKeyConstraint('id', name='project_database_pkey')
+    )
+    # ### end Alembic commands ###