Skip to content

Commit 1fa677f

Browse files
committed
Fix database schema DDL statements after update to MLflow 2.13
New models / tables have been added. - trace_info - trace_tags - trace_request_metadata
1 parent 04674c1 commit 1fa677f

File tree

2 files changed

+28
-0
lines changed

2 files changed

+28
-0
lines changed

mlflow_cratedb/adapter/ddl/cratedb.sql

+25
Original file line numberDiff line numberDiff line change
@@ -140,3 +140,28 @@ CREATE TABLE IF NOT EXISTS "tags" (
140140
"run_uuid" TEXT NOT NULL,
141141
PRIMARY KEY ("key", "run_uuid")
142142
);
143+
144+
CREATE TABLE IF NOT EXISTS "trace_info" (
145+
"request_id" TEXT NOT NULL,
146+
"experiment_id" BIGINT,
147+
"timestamp_ms" BIGINT NOT NULL,
148+
"execution_time_ms" BIGINT NOT NULL,
149+
"status" TEXT,
150+
PRIMARY KEY ("request_id")
151+
);
152+
153+
CREATE TABLE IF NOT EXISTS "trace_tags" (
154+
"key" TEXT,
155+
"value" TEXT NOT NULL,
156+
"request_id" TEXT NOT NULL,
157+
"trace_info" BIGINT,
158+
PRIMARY KEY ("request_id", "key")
159+
);
160+
161+
CREATE TABLE IF NOT EXISTS "trace_request_metadata" (
162+
"key" TEXT,
163+
"value" TEXT NOT NULL,
164+
"request_id" TEXT NOT NULL,
165+
"trace_info" BIGINT,
166+
PRIMARY KEY ("request_id", "key")
167+
);

mlflow_cratedb/adapter/ddl/drop.sql

+3
Original file line numberDiff line numberDiff line change
@@ -13,3 +13,6 @@ DROP TABLE IF EXISTS "registered_model_aliases";
1313
DROP TABLE IF EXISTS "registered_model_tags";
1414
DROP TABLE IF EXISTS "runs";
1515
DROP TABLE IF EXISTS "tags";
16+
DROP TABLE IF EXISTS "trace_info";
17+
DROP TABLE IF EXISTS "trace_tags";
18+
DROP TABLE IF EXISTS "trace_request_metadata";

0 commit comments

Comments
 (0)