From 0b5753a514db0adf6ee768745b4afcd1925a57a2 Mon Sep 17 00:00:00 2001
From: Niklas Schmidtmer <niklas@crate.io>
Date: Thu, 14 Sep 2023 09:15:42 +0200
Subject: [PATCH 1/2] Add a few missing columns to make the Models tab load in
 the UI

---
 mlflow_cratedb/adapter/ddl/cratedb.sql | 11 ++++++++---
 1 file changed, 8 insertions(+), 3 deletions(-)

diff --git a/mlflow_cratedb/adapter/ddl/cratedb.sql b/mlflow_cratedb/adapter/ddl/cratedb.sql
index bead087..16d749f 100644
--- a/mlflow_cratedb/adapter/ddl/cratedb.sql
+++ b/mlflow_cratedb/adapter/ddl/cratedb.sql
@@ -67,7 +67,7 @@ CREATE TABLE IF NOT EXISTS "model_versions" (
    "name" TEXT NOT NULL,
    "version" INTEGER NOT NULL,
    "creation_time" BIGINT,  -- default=get_current_time_millis
-   "last_update_time" BIGINT,  -- default=get_current_time_millis
+   "last_updated_time" BIGINT,  -- default=get_current_time_millis
    "description" TEXT,
    "user_id" TEXT,
    "current_stage" TEXT,
@@ -94,8 +94,11 @@ CREATE TABLE IF NOT EXISTS "params" (
 
 CREATE TABLE IF NOT EXISTS "registered_models" (
    "name" TEXT NOT NULL,
-   "key" TEXT NOT NULL,
-   "value" TEXT
+   "key" TEXT DEFAULT GEN_RANDOM_TEXT_UUID() NOT NULL,
+   "value" TEXT,
+   "creation_time" BIGINT,
+   "last_updated_time" BIGINT,
+   "description" TEXT
 );
 
 CREATE TABLE IF NOT EXISTS "registered_model_aliases" (
@@ -106,6 +109,8 @@ CREATE TABLE IF NOT EXISTS "registered_model_aliases" (
 
 CREATE TABLE IF NOT EXISTS "registered_model_tags" (
    "name" TEXT NOT NULL,
+   "key" TEXT NOT NULL,
+   "value" TEXT NOT NULL,
    "creation_time" BIGINT,  -- default=get_current_time_millis
    "last_update_time" BIGINT,  -- default=get_current_time_millis
    "description" TEXT

From 424b51ba017031911d0b1d7760ea0b50fdfece5a Mon Sep 17 00:00:00 2001
From: Niklas Schmidtmer <niklas@crate.io>
Date: Thu, 14 Sep 2023 09:24:45 +0200
Subject: [PATCH 2/2] Ignore `build` and `model` folders

---
 .gitignore | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/.gitignore b/.gitignore
index 0a3dedd..592bf4b 100644
--- a/.gitignore
+++ b/.gitignore
@@ -4,6 +4,7 @@
 /.ruff_cache
 /.vagrant
 /.venv*
+/build
 
 /dist
 
@@ -15,3 +16,4 @@ __pycache__
 
 /mlartifacts
 /mlruns
+/model