@things-factory/integration-mlflow 9.1.19

package/dist-server/service/model-registry/model-registry-service.js

@@ -0,0 +1,255 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ModelRegistryService = void 0;
+const tslib_1 = require("tslib");
+const type_graphql_1 = require("type-graphql");
+const mlflow_api_client_1 = require("../../utils/mlflow-api-client");
+const mlflow_types_1 = require("../../types/mlflow-types");
+let ModelRegistryService = class ModelRegistryService {
+    /**
+     * Get all registered models
+     */
+    async mlflowRegisteredModels(context) {
+        try {
+            const response = await mlflow_api_client_1.mlflowApi.searchRegisteredModels();
+            return response.registered_models.map(model => ({
+                name: model.name,
+                creationTimestamp: model.creation_timestamp,
+                lastUpdatedTimestamp: model.last_updated_timestamp,
+                description: model.description,
+                tags: undefined,
+                latestVersions: model.latest_versions ? JSON.stringify(model.latest_versions) : undefined
+            }));
+        }
+        catch (error) {
+            console.error('Failed to fetch registered models:', error);
+            throw new Error(`Failed to fetch registered models: ${error.message}`);
+        }
+    }
+    /**
+     * Get registered model by name
+     */
+    async mlflowRegisteredModel(name, context) {
+        try {
+            const response = await mlflow_api_client_1.mlflowApi.getRegisteredModel(name);
+            return {
+                name: response.registered_model.name,
+                creationTimestamp: response.registered_model.creation_timestamp,
+                lastUpdatedTimestamp: response.registered_model.last_updated_timestamp,
+                description: response.registered_model.description,
+                tags: response.registered_model.tags ? JSON.stringify(response.registered_model.tags) : undefined,
+                latestVersions: response.registered_model.latest_versions ? JSON.stringify(response.registered_model.latest_versions) : undefined
+            };
+        }
+        catch (error) {
+            console.error(`Failed to fetch model ${name}:`, error);
+            return null;
+        }
+    }
+    /**
+     * Create registered model
+     */
+    async createMLflowRegisteredModel(input, context) {
+        try {
+            const tags = input.tags ? JSON.parse(input.tags) : undefined;
+            const response = await mlflow_api_client_1.mlflowApi.createRegisteredModel({
+                name: input.name,
+                description: input.description,
+                tags
+            });
+            return {
+                name: response.registered_model.name,
+                creationTimestamp: response.registered_model.creation_timestamp,
+                lastUpdatedTimestamp: response.registered_model.last_updated_timestamp,
+                description: input.description,
+                tags: input.tags,
+                latestVersions: undefined
+            };
+        }
+        catch (error) {
+            console.error('Failed to create registered model:', error);
+            throw new Error(`Failed to create registered model: ${error.message}`);
+        }
+    }
+    /**
+     * Get model version
+     */
+    async mlflowModelVersion(name, version, context) {
+        try {
+            const response = await mlflow_api_client_1.mlflowApi.getModelVersion({ name, version });
+            return {
+                name: response.model_version.name,
+                version: response.model_version.version,
+                creationTimestamp: response.model_version.creation_timestamp,
+                currentStage: response.model_version.current_stage,
+                source: response.model_version.source,
+                status: response.model_version.status,
+                runId: response.model_version.run_id,
+                description: undefined
+            };
+        }
+        catch (error) {
+            console.error(`Failed to fetch model version ${name}/${version}:`, error);
+            return null;
+        }
+    }
+    /**
+     * Create model version
+     */
+    async createMLflowModelVersion(input, context) {
+        try {
+            const tags = input.tags ? JSON.parse(input.tags) : undefined;
+            const response = await mlflow_api_client_1.mlflowApi.createModelVersion({
+                name: input.name,
+                source: input.source,
+                run_id: input.runId,
+                description: input.description,
+                tags
+            });
+            return {
+                name: response.model_version.name,
+                version: response.model_version.version,
+                creationTimestamp: response.model_version.creation_timestamp,
+                currentStage: response.model_version.status,
+                source: response.model_version.source,
+                status: response.model_version.status,
+                runId: input.runId,
+                description: input.description
+            };
+        }
+        catch (error) {
+            console.error('Failed to create model version:', error);
+            throw new Error(`Failed to create model version: ${error.message}`);
+        }
+    }
+    /**
+     * Transition model version stage
+     */
+    async transitionMLflowModelStage(input, context) {
+        try {
+            const response = await mlflow_api_client_1.mlflowApi.transitionModelVersionStage({
+                name: input.name,
+                version: input.version,
+                stage: input.stage,
+                archive_existing_versions: input.archiveExistingVersions
+            });
+            // Fetch updated version
+            const version = await mlflow_api_client_1.mlflowApi.getModelVersion({
+                name: input.name,
+                version: input.version
+            });
+            return {
+                name: version.model_version.name,
+                version: version.model_version.version,
+                creationTimestamp: version.model_version.creation_timestamp,
+                currentStage: version.model_version.current_stage,
+                source: version.model_version.source,
+                status: version.model_version.status,
+                runId: version.model_version.run_id,
+                description: undefined
+            };
+        }
+        catch (error) {
+            console.error('Failed to transition model stage:', error);
+            throw new Error(`Failed to transition model stage: ${error.message}`);
+        }
+    }
+    /**
+     * Delete model version
+     */
+    async deleteMLflowModelVersion(name, version, context) {
+        try {
+            await mlflow_api_client_1.mlflowApi.deleteModelVersion({ name, version });
+            return true;
+        }
+        catch (error) {
+            console.error(`Failed to delete model version ${name}/${version}:`, error);
+            throw new Error(`Failed to delete model version: ${error.message}`);
+        }
+    }
+};
+exports.ModelRegistryService = ModelRegistryService;
+tslib_1.__decorate([
+    (0, type_graphql_1.Query)(returns => [mlflow_types_1.MLflowRegisteredModel], {
+        description: 'Get all registered models in MLflow'
+    }),
+    (0, type_graphql_1.Directive)('@privilege(category: "mlflow", privilege: "query")'),
+    tslib_1.__param(0, (0, type_graphql_1.Ctx)()),
+    tslib_1.__metadata("design:type", Function),
+    tslib_1.__metadata("design:paramtypes", [Object]),
+    tslib_1.__metadata("design:returntype", Promise)
+], ModelRegistryService.prototype, "mlflowRegisteredModels", null);
+tslib_1.__decorate([
+    (0, type_graphql_1.Query)(returns => mlflow_types_1.MLflowRegisteredModel, {
+        description: 'Get registered model by name',
+        nullable: true
+    }),
+    (0, type_graphql_1.Directive)('@privilege(category: "mlflow", privilege: "query")'),
+    tslib_1.__param(0, (0, type_graphql_1.Arg)('name')),
+    tslib_1.__param(1, (0, type_graphql_1.Ctx)()),
+    tslib_1.__metadata("design:type", Function),
+    tslib_1.__metadata("design:paramtypes", [String, Object]),
+    tslib_1.__metadata("design:returntype", Promise)
+], ModelRegistryService.prototype, "mlflowRegisteredModel", null);
+tslib_1.__decorate([
+    (0, type_graphql_1.Mutation)(returns => mlflow_types_1.MLflowRegisteredModel, {
+        description: 'Create new registered model'
+    }),
+    (0, type_graphql_1.Directive)('@privilege(category: "mlflow", privilege: "mutation")'),
+    tslib_1.__param(0, (0, type_graphql_1.Arg)('input')),
+    tslib_1.__param(1, (0, type_graphql_1.Ctx)()),
+    tslib_1.__metadata("design:type", Function),
+    tslib_1.__metadata("design:paramtypes", [mlflow_types_1.CreateRegisteredModelInput, Object]),
+    tslib_1.__metadata("design:returntype", Promise)
+], ModelRegistryService.prototype, "createMLflowRegisteredModel", null);
+tslib_1.__decorate([
+    (0, type_graphql_1.Query)(returns => mlflow_types_1.MLflowModelVersion, {
+        description: 'Get model version',
+        nullable: true
+    }),
+    (0, type_graphql_1.Directive)('@privilege(category: "mlflow", privilege: "query")'),
+    tslib_1.__param(0, (0, type_graphql_1.Arg)('name')),
+    tslib_1.__param(1, (0, type_graphql_1.Arg)('version')),
+    tslib_1.__param(2, (0, type_graphql_1.Ctx)()),
+    tslib_1.__metadata("design:type", Function),
+    tslib_1.__metadata("design:paramtypes", [String, String, Object]),
+    tslib_1.__metadata("design:returntype", Promise)
+], ModelRegistryService.prototype, "mlflowModelVersion", null);
+tslib_1.__decorate([
+    (0, type_graphql_1.Mutation)(returns => mlflow_types_1.MLflowModelVersion, {
+        description: 'Create new model version'
+    }),
+    (0, type_graphql_1.Directive)('@privilege(category: "mlflow", privilege: "mutation")'),
+    tslib_1.__param(0, (0, type_graphql_1.Arg)('input')),
+    tslib_1.__param(1, (0, type_graphql_1.Ctx)()),
+    tslib_1.__metadata("design:type", Function),
+    tslib_1.__metadata("design:paramtypes", [mlflow_types_1.CreateModelVersionInput, Object]),
+    tslib_1.__metadata("design:returntype", Promise)
+], ModelRegistryService.prototype, "createMLflowModelVersion", null);
+tslib_1.__decorate([
+    (0, type_graphql_1.Mutation)(returns => mlflow_types_1.MLflowModelVersion, {
+        description: 'Transition model version to target stage (None, Staging, Production, Archived)'
+    }),
+    (0, type_graphql_1.Directive)('@privilege(category: "mlflow", privilege: "mutation")'),
+    tslib_1.__param(0, (0, type_graphql_1.Arg)('input')),
+    tslib_1.__param(1, (0, type_graphql_1.Ctx)()),
+    tslib_1.__metadata("design:type", Function),
+    tslib_1.__metadata("design:paramtypes", [mlflow_types_1.TransitionStageInput, Object]),
+    tslib_1.__metadata("design:returntype", Promise)
+], ModelRegistryService.prototype, "transitionMLflowModelStage", null);
+tslib_1.__decorate([
+    (0, type_graphql_1.Mutation)(returns => Boolean, {
+        description: 'Delete model version'
+    }),
+    (0, type_graphql_1.Directive)('@privilege(category: "mlflow", privilege: "mutation")'),
+    tslib_1.__param(0, (0, type_graphql_1.Arg)('name')),
+    tslib_1.__param(1, (0, type_graphql_1.Arg)('version')),
+    tslib_1.__param(2, (0, type_graphql_1.Ctx)()),
+    tslib_1.__metadata("design:type", Function),
+    tslib_1.__metadata("design:paramtypes", [String, String, Object]),
+    tslib_1.__metadata("design:returntype", Promise)
+], ModelRegistryService.prototype, "deleteMLflowModelVersion", null);
+exports.ModelRegistryService = ModelRegistryService = tslib_1.__decorate([
+    (0, type_graphql_1.Resolver)()
+], ModelRegistryService);
+//# sourceMappingURL=model-registry-service.js.map

package/dist-server/service/model-registry/model-registry-service.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"model-registry-service.js","sourceRoot":"","sources":["../../../server/service/model-registry/model-registry-service.ts"],"names":[],"mappings":";;;;AAAA,+CAA6E;AAC7E,qEAAyD;AACzD,2DAMiC;AAG1B,IAAM,oBAAoB,GAA1B,MAAM,oBAAoB;IAC/B;;OAEG;IAKG,AAAN,KAAK,CAAC,sBAAsB,CAAQ,OAAwB;QAC1D,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,6BAAS,CAAC,sBAAsB,EAAE,CAAA;YAEzD,OAAO,QAAQ,CAAC,iBAAiB,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;gBAC9C,IAAI,EAAE,KAAK,CAAC,IAAI;gBAChB,iBAAiB,EAAE,KAAK,CAAC,kBAAkB;gBAC3C,oBAAoB,EAAE,KAAK,CAAC,sBAAsB;gBAClD,WAAW,EAAE,KAAK,CAAC,WAAW;gBAC9B,IAAI,EAAE,SAAS;gBACf,cAAc,EAAE,KAAK,CAAC,eAAe,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC,SAAS;aAC1F,CAAC,CAAC,CAAA;QACL,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,oCAAoC,EAAE,KAAK,CAAC,CAAA;YAC1D,MAAM,IAAI,KAAK,CAAC,sCAAsC,KAAK,CAAC,OAAO,EAAE,CAAC,CAAA;QACxE,CAAC;IACH,CAAC;IAED;;OAEG;IAMG,AAAN,KAAK,CAAC,qBAAqB,CACZ,IAAY,EAClB,OAAwB;QAE/B,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,6BAAS,CAAC,kBAAkB,CAAC,IAAI,CAAC,CAAA;YAEzD,OAAO;gBACL,IAAI,EAAE,QAAQ,CAAC,gBAAgB,CAAC,IAAI;gBACpC,iBAAiB,EAAE,QAAQ,CAAC,gBAAgB,CAAC,kBAAkB;gBAC/D,oBAAoB,EAAE,QAAQ,CAAC,gBAAgB,CAAC,sBAAsB;gBACtE,WAAW,EAAE,QAAQ,CAAC,gBAAgB,CAAC,WAAW;gBAClD,IAAI,EAAE,QAAQ,CAAC,gBAAgB,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,gBAAgB,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,SAAS;gBACjG,cAAc,EAAE,QAAQ,CAAC,gBAAgB,CAAC,eAAe,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,gBAAgB,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC,SAAS;aAClI,CAAA;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,yBAAyB,IAAI,GAAG,EAAE,KAAK,CAAC,CAAA;YACtD,OAAO,IAAI,CAAA;QACb,CAAC;IACH,CAAC;IAED;;OAEG;IAKG,AAAN,KAAK,CAAC,2BAA2B,CACjB,KAAiC,EACxC,OAAwB;QAE/B,IAAI,CAAC;YACH,MAAM,IAAI,GAAG,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,SAAS,CAAA;YAE5D,MAAM,QAAQ,GAAG,MAAM,6BAAS,CAAC,qBAAqB,CAAC;gBACrD,IAAI,EAAE,KAAK,CAAC,IAAI;gBAChB,WAAW,EAAE,KAAK,CAAC,WAAW;gBAC9B,IAAI;aACL,CAAC,CAAA;YAEF,OAAO;gBACL,IAAI,EAAE,QAAQ,CAAC,gBAAgB,CAAC,IAAI;gBACpC,iBAAiB,EAAE,QAAQ,CAAC,gBAAgB,CAAC,kBAAkB;gBAC/D,oBAAoB,EAAE,QAAQ,CAAC,gBAAgB,CAAC,sBAAsB;gBACtE,WAAW,EAAE,KAAK,CAAC,WAAW;gBAC9B,IAAI,EAAE,KAAK,CAAC,IAAI;gBAChB,cAAc,EAAE,SAAS;aAC1B,CAAA;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,oCAAoC,EAAE,KAAK,CAAC,CAAA;YAC1D,MAAM,IAAI,KAAK,CAAC,sCAAsC,KAAK,CAAC,OAAO,EAAE,CAAC,CAAA;QACxE,CAAC;IACH,CAAC;IAED;;OAEG;IAMG,AAAN,KAAK,CAAC,kBAAkB,CACT,IAAY,EACT,OAAe,EACxB,OAAwB;QAE/B,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,6BAAS,CAAC,eAAe,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,CAAA;YAEnE,OAAO;gBACL,IAAI,EAAE,QAAQ,CAAC,aAAa,CAAC,IAAI;gBACjC,OAAO,EAAE,QAAQ,CAAC,aAAa,CAAC,OAAO;gBACvC,iBAAiB,EAAE,QAAQ,CAAC,aAAa,CAAC,kBAAkB;gBAC5D,YAAY,EAAE,QAAQ,CAAC,aAAa,CAAC,aAAa;gBAClD,MAAM,EAAE,QAAQ,CAAC,aAAa,CAAC,MAAM;gBACrC,MAAM,EAAE,QAAQ,CAAC,aAAa,CAAC,MAAM;gBACrC,KAAK,EAAE,QAAQ,CAAC,aAAa,CAAC,MAAM;gBACpC,WAAW,EAAE,SAAS;aACvB,CAAA;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,iCAAiC,IAAI,IAAI,OAAO,GAAG,EAAE,KAAK,CAAC,CAAA;YACzE,OAAO,IAAI,CAAA;QACb,CAAC;IACH,CAAC;IAED;;OAEG;IAKG,AAAN,KAAK,CAAC,wBAAwB,CACd,KAA8B,EACrC,OAAwB;QAE/B,IAAI,CAAC;YACH,MAAM,IAAI,GAAG,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,SAAS,CAAA;YAE5D,MAAM,QAAQ,GAAG,MAAM,6BAAS,CAAC,kBAAkB,CAAC;gBAClD,IAAI,EAAE,KAAK,CAAC,IAAI;gBAChB,MAAM,EAAE,KAAK,CAAC,MAAM;gBACpB,MAAM,EAAE,KAAK,CAAC,KAAK;gBACnB,WAAW,EAAE,KAAK,CAAC,WAAW;gBAC9B,IAAI;aACL,CAAC,CAAA;YAEF,OAAO;gBACL,IAAI,EAAE,QAAQ,CAAC,aAAa,CAAC,IAAI;gBACjC,OAAO,EAAE,QAAQ,CAAC,aAAa,CAAC,OAAO;gBACvC,iBAAiB,EAAE,QAAQ,CAAC,aAAa,CAAC,kBAAkB;gBAC5D,YAAY,EAAE,QAAQ,CAAC,aAAa,CAAC,MAAM;gBAC3C,MAAM,EAAE,QAAQ,CAAC,aAAa,CAAC,MAAM;gBACrC,MAAM,EAAE,QAAQ,CAAC,aAAa,CAAC,MAAM;gBACrC,KAAK,EAAE,KAAK,CAAC,KAAK;gBAClB,WAAW,EAAE,KAAK,CAAC,WAAW;aAC/B,CAAA;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,iCAAiC,EAAE,KAAK,CAAC,CAAA;YACvD,MAAM,IAAI,KAAK,CAAC,mCAAmC,KAAK,CAAC,OAAO,EAAE,CAAC,CAAA;QACrE,CAAC;IACH,CAAC;IAED;;OAEG;IAKG,AAAN,KAAK,CAAC,0BAA0B,CAChB,KAA2B,EAClC,OAAwB;QAE/B,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,6BAAS,CAAC,2BAA2B,CAAC;gBAC3D,IAAI,EAAE,KAAK,CAAC,IAAI;gBAChB,OAAO,EAAE,KAAK,CAAC,OAAO;gBACtB,KAAK,EAAE,KAAK,CAAC,KAAY;gBACzB,yBAAyB,EAAE,KAAK,CAAC,uBAAuB;aACzD,CAAC,CAAA;YAEF,wBAAwB;YACxB,MAAM,OAAO,GAAG,MAAM,6BAAS,CAAC,eAAe,CAAC;gBAC9C,IAAI,EAAE,KAAK,CAAC,IAAI;gBAChB,OAAO,EAAE,KAAK,CAAC,OAAO;aACvB,CAAC,CAAA;YAEF,OAAO;gBACL,IAAI,EAAE,OAAO,CAAC,aAAa,CAAC,IAAI;gBAChC,OAAO,EAAE,OAAO,CAAC,aAAa,CAAC,OAAO;gBACtC,iBAAiB,EAAE,OAAO,CAAC,aAAa,CAAC,kBAAkB;gBAC3D,YAAY,EAAE,OAAO,CAAC,aAAa,CAAC,aAAa;gBACjD,MAAM,EAAE,OAAO,CAAC,aAAa,CAAC,MAAM;gBACpC,MAAM,EAAE,OAAO,CAAC,aAAa,CAAC,MAAM;gBACpC,KAAK,EAAE,OAAO,CAAC,aAAa,CAAC,MAAM;gBACnC,WAAW,EAAE,SAAS;aACvB,CAAA;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,mCAAmC,EAAE,KAAK,CAAC,CAAA;YACzD,MAAM,IAAI,KAAK,CAAC,qCAAqC,KAAK,CAAC,OAAO,EAAE,CAAC,CAAA;QACvE,CAAC;IACH,CAAC;IAED;;OAEG;IAKG,AAAN,KAAK,CAAC,wBAAwB,CACf,IAAY,EACT,OAAe,EACxB,OAAwB;QAE/B,IAAI,CAAC;YACH,MAAM,6BAAS,CAAC,kBAAkB,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,CAAA;YACrD,OAAO,IAAI,CAAA;QACb,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,kCAAkC,IAAI,IAAI,OAAO,GAAG,EAAE,KAAK,CAAC,CAAA;YAC1E,MAAM,IAAI,KAAK,CAAC,mCAAmC,KAAK,CAAC,OAAO,EAAE,CAAC,CAAA;QACrE,CAAC;IACH,CAAC;CACF,CAAA;AA5NY,oDAAoB;AAQzB;IAJL,IAAA,oBAAK,EAAC,OAAO,CAAC,EAAE,CAAC,CAAC,oCAAqB,CAAC,EAAE;QACzC,WAAW,EAAE,qCAAqC;KACnD,CAAC;IACD,IAAA,wBAAS,EAAC,oDAAoD,CAAC;IAClC,mBAAA,IAAA,kBAAG,GAAE,CAAA;;;;kEAgBlC;AAUK;IALL,IAAA,oBAAK,EAAC,OAAO,CAAC,EAAE,CAAC,oCAAqB,EAAE;QACvC,WAAW,EAAE,8BAA8B;QAC3C,QAAQ,EAAE,IAAI;KACf,CAAC;IACD,IAAA,wBAAS,EAAC,oDAAoD,CAAC;IAE7D,mBAAA,IAAA,kBAAG,EAAC,MAAM,CAAC,CAAA;IACX,mBAAA,IAAA,kBAAG,GAAE,CAAA;;;;iEAiBP;AASK;IAJL,IAAA,uBAAQ,EAAC,OAAO,CAAC,EAAE,CAAC,oCAAqB,EAAE;QAC1C,WAAW,EAAE,6BAA6B;KAC3C,CAAC;IACD,IAAA,wBAAS,EAAC,uDAAuD,CAAC;IAEhE,mBAAA,IAAA,kBAAG,EAAC,OAAO,CAAC,CAAA;IACZ,mBAAA,IAAA,kBAAG,GAAE,CAAA;;6CADe,yCAA0B;;uEAwBhD;AAUK;IALL,IAAA,oBAAK,EAAC,OAAO,CAAC,EAAE,CAAC,iCAAkB,EAAE;QACpC,WAAW,EAAE,mBAAmB;QAChC,QAAQ,EAAE,IAAI;KACf,CAAC;IACD,IAAA,wBAAS,EAAC,oDAAoD,CAAC;IAE7D,mBAAA,IAAA,kBAAG,EAAC,MAAM,CAAC,CAAA;IACX,mBAAA,IAAA,kBAAG,EAAC,SAAS,CAAC,CAAA;IACd,mBAAA,IAAA,kBAAG,GAAE,CAAA;;;;8DAmBP;AASK;IAJL,IAAA,uBAAQ,EAAC,OAAO,CAAC,EAAE,CAAC,iCAAkB,EAAE;QACvC,WAAW,EAAE,0BAA0B;KACxC,CAAC;IACD,IAAA,wBAAS,EAAC,uDAAuD,CAAC;IAEhE,mBAAA,IAAA,kBAAG,EAAC,OAAO,CAAC,CAAA;IACZ,mBAAA,IAAA,kBAAG,GAAE,CAAA;;6CADe,sCAAuB;;oEA4B7C;AASK;IAJL,IAAA,uBAAQ,EAAC,OAAO,CAAC,EAAE,CAAC,iCAAkB,EAAE;QACvC,WAAW,EAAE,gFAAgF;KAC9F,CAAC;IACD,IAAA,wBAAS,EAAC,uDAAuD,CAAC;IAEhE,mBAAA,IAAA,kBAAG,EAAC,OAAO,CAAC,CAAA;IACZ,mBAAA,IAAA,kBAAG,GAAE,CAAA;;6CADe,mCAAoB;;sEA+B1C;AASK;IAJL,IAAA,uBAAQ,EAAC,OAAO,CAAC,EAAE,CAAC,OAAO,EAAE;QAC5B,WAAW,EAAE,sBAAsB;KACpC,CAAC;IACD,IAAA,wBAAS,EAAC,uDAAuD,CAAC;IAEhE,mBAAA,IAAA,kBAAG,EAAC,MAAM,CAAC,CAAA;IACX,mBAAA,IAAA,kBAAG,EAAC,SAAS,CAAC,CAAA;IACd,mBAAA,IAAA,kBAAG,GAAE,CAAA;;;;oEASP;+BA3NU,oBAAoB;IADhC,IAAA,uBAAQ,GAAE;GACE,oBAAoB,CA4NhC","sourcesContent":["import { Resolver, Query, Mutation, Arg, Ctx, Directive } from 'type-graphql'\nimport { mlflowApi } from '../../utils/mlflow-api-client'\nimport {\n  MLflowRegisteredModel,\n  MLflowModelVersion,\n  CreateRegisteredModelInput,\n  CreateModelVersionInput,\n  TransitionStageInput\n} from '../../types/mlflow-types'\n\n@Resolver()\nexport class ModelRegistryService {\n  /**\n   * Get all registered models\n   */\n  @Query(returns => [MLflowRegisteredModel], {\n    description: 'Get all registered models in MLflow'\n  })\n  @Directive('@privilege(category: \"mlflow\", privilege: \"query\")')\n  async mlflowRegisteredModels(@Ctx() context: ResolverContext): Promise<MLflowRegisteredModel[]> {\n    try {\n      const response = await mlflowApi.searchRegisteredModels()\n\n      return response.registered_models.map(model => ({\n        name: model.name,\n        creationTimestamp: model.creation_timestamp,\n        lastUpdatedTimestamp: model.last_updated_timestamp,\n        description: model.description,\n        tags: undefined,\n        latestVersions: model.latest_versions ? JSON.stringify(model.latest_versions) : undefined\n      }))\n    } catch (error) {\n      console.error('Failed to fetch registered models:', error)\n      throw new Error(`Failed to fetch registered models: ${error.message}`)\n    }\n  }\n\n  /**\n   * Get registered model by name\n   */\n  @Query(returns => MLflowRegisteredModel, {\n    description: 'Get registered model by name',\n    nullable: true\n  })\n  @Directive('@privilege(category: \"mlflow\", privilege: \"query\")')\n  async mlflowRegisteredModel(\n    @Arg('name') name: string,\n    @Ctx() context: ResolverContext\n  ): Promise<MLflowRegisteredModel | null> {\n    try {\n      const response = await mlflowApi.getRegisteredModel(name)\n\n      return {\n        name: response.registered_model.name,\n        creationTimestamp: response.registered_model.creation_timestamp,\n        lastUpdatedTimestamp: response.registered_model.last_updated_timestamp,\n        description: response.registered_model.description,\n        tags: response.registered_model.tags ? JSON.stringify(response.registered_model.tags) : undefined,\n        latestVersions: response.registered_model.latest_versions ? JSON.stringify(response.registered_model.latest_versions) : undefined\n      }\n    } catch (error) {\n      console.error(`Failed to fetch model ${name}:`, error)\n      return null\n    }\n  }\n\n  /**\n   * Create registered model\n   */\n  @Mutation(returns => MLflowRegisteredModel, {\n    description: 'Create new registered model'\n  })\n  @Directive('@privilege(category: \"mlflow\", privilege: \"mutation\")')\n  async createMLflowRegisteredModel(\n    @Arg('input') input: CreateRegisteredModelInput,\n    @Ctx() context: ResolverContext\n  ): Promise<MLflowRegisteredModel> {\n    try {\n      const tags = input.tags ? JSON.parse(input.tags) : undefined\n\n      const response = await mlflowApi.createRegisteredModel({\n        name: input.name,\n        description: input.description,\n        tags\n      })\n\n      return {\n        name: response.registered_model.name,\n        creationTimestamp: response.registered_model.creation_timestamp,\n        lastUpdatedTimestamp: response.registered_model.last_updated_timestamp,\n        description: input.description,\n        tags: input.tags,\n        latestVersions: undefined\n      }\n    } catch (error) {\n      console.error('Failed to create registered model:', error)\n      throw new Error(`Failed to create registered model: ${error.message}`)\n    }\n  }\n\n  /**\n   * Get model version\n   */\n  @Query(returns => MLflowModelVersion, {\n    description: 'Get model version',\n    nullable: true\n  })\n  @Directive('@privilege(category: \"mlflow\", privilege: \"query\")')\n  async mlflowModelVersion(\n    @Arg('name') name: string,\n    @Arg('version') version: string,\n    @Ctx() context: ResolverContext\n  ): Promise<MLflowModelVersion | null> {\n    try {\n      const response = await mlflowApi.getModelVersion({ name, version })\n\n      return {\n        name: response.model_version.name,\n        version: response.model_version.version,\n        creationTimestamp: response.model_version.creation_timestamp,\n        currentStage: response.model_version.current_stage,\n        source: response.model_version.source,\n        status: response.model_version.status,\n        runId: response.model_version.run_id,\n        description: undefined\n      }\n    } catch (error) {\n      console.error(`Failed to fetch model version ${name}/${version}:`, error)\n      return null\n    }\n  }\n\n  /**\n   * Create model version\n   */\n  @Mutation(returns => MLflowModelVersion, {\n    description: 'Create new model version'\n  })\n  @Directive('@privilege(category: \"mlflow\", privilege: \"mutation\")')\n  async createMLflowModelVersion(\n    @Arg('input') input: CreateModelVersionInput,\n    @Ctx() context: ResolverContext\n  ): Promise<MLflowModelVersion> {\n    try {\n      const tags = input.tags ? JSON.parse(input.tags) : undefined\n\n      const response = await mlflowApi.createModelVersion({\n        name: input.name,\n        source: input.source,\n        run_id: input.runId,\n        description: input.description,\n        tags\n      })\n\n      return {\n        name: response.model_version.name,\n        version: response.model_version.version,\n        creationTimestamp: response.model_version.creation_timestamp,\n        currentStage: response.model_version.status,\n        source: response.model_version.source,\n        status: response.model_version.status,\n        runId: input.runId,\n        description: input.description\n      }\n    } catch (error) {\n      console.error('Failed to create model version:', error)\n      throw new Error(`Failed to create model version: ${error.message}`)\n    }\n  }\n\n  /**\n   * Transition model version stage\n   */\n  @Mutation(returns => MLflowModelVersion, {\n    description: 'Transition model version to target stage (None, Staging, Production, Archived)'\n  })\n  @Directive('@privilege(category: \"mlflow\", privilege: \"mutation\")')\n  async transitionMLflowModelStage(\n    @Arg('input') input: TransitionStageInput,\n    @Ctx() context: ResolverContext\n  ): Promise<MLflowModelVersion> {\n    try {\n      const response = await mlflowApi.transitionModelVersionStage({\n        name: input.name,\n        version: input.version,\n        stage: input.stage as any,\n        archive_existing_versions: input.archiveExistingVersions\n      })\n\n      // Fetch updated version\n      const version = await mlflowApi.getModelVersion({\n        name: input.name,\n        version: input.version\n      })\n\n      return {\n        name: version.model_version.name,\n        version: version.model_version.version,\n        creationTimestamp: version.model_version.creation_timestamp,\n        currentStage: version.model_version.current_stage,\n        source: version.model_version.source,\n        status: version.model_version.status,\n        runId: version.model_version.run_id,\n        description: undefined\n      }\n    } catch (error) {\n      console.error('Failed to transition model stage:', error)\n      throw new Error(`Failed to transition model stage: ${error.message}`)\n    }\n  }\n\n  /**\n   * Delete model version\n   */\n  @Mutation(returns => Boolean, {\n    description: 'Delete model version'\n  })\n  @Directive('@privilege(category: \"mlflow\", privilege: \"mutation\")')\n  async deleteMLflowModelVersion(\n    @Arg('name') name: string,\n    @Arg('version') version: string,\n    @Ctx() context: ResolverContext\n  ): Promise<boolean> {\n    try {\n      await mlflowApi.deleteModelVersion({ name, version })\n      return true\n    } catch (error) {\n      console.error(`Failed to delete model version ${name}/${version}:`, error)\n      throw new Error(`Failed to delete model version: ${error.message}`)\n    }\n  }\n}\n"]}

package/dist-server/service/run/run-service.d.ts

@@ -0,0 +1,27 @@
+import { MLflowRun, CreateRunInput, LogMetricInput, LogParamInput, UpdateRunInput, SearchRunsInput } from '../../types/mlflow-types';
+export declare class RunService {
+    /**
+     * Get run by ID
+     */
+    mlflowRun(runId: string, context: ResolverContext): Promise<MLflowRun | null>;
+    /**
+     * Search runs
+     */
+    searchMLflowRuns(input: SearchRunsInput, context: ResolverContext): Promise<MLflowRun[]>;
+    /**
+     * Create new run
+     */
+    createMLflowRun(input: CreateRunInput, context: ResolverContext): Promise<MLflowRun>;
+    /**
+     * Log metric
+     */
+    logMLflowMetric(input: LogMetricInput, context: ResolverContext): Promise<boolean>;
+    /**
+     * Log parameter
+     */
+    logMLflowParam(input: LogParamInput, context: ResolverContext): Promise<boolean>;
+    /**
+     * Update run
+     */
+    updateMLflowRun(input: UpdateRunInput, context: ResolverContext): Promise<boolean>;
+}

package/dist-server/service/run/run-service.js

@@ -0,0 +1,214 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.RunService = void 0;
+const tslib_1 = require("tslib");
+const type_graphql_1 = require("type-graphql");
+const mlflow_api_client_1 = require("../../utils/mlflow-api-client");
+const mlflow_types_1 = require("../../types/mlflow-types");
+let RunService = class RunService {
+    /**
+     * Get run by ID
+     */
+    async mlflowRun(runId, context) {
+        try {
+            const response = await mlflow_api_client_1.mlflowApi.getRun(runId);
+            const run = response.run;
+            return {
+                runId: run.info.run_id,
+                experimentId: run.info.experiment_id,
+                status: run.info.status,
+                startTime: run.info.start_time,
+                endTime: run.info.end_time,
+                artifactUri: run.info.artifact_uri,
+                metrics: run.data.metrics ? JSON.stringify(run.data.metrics) : undefined,
+                params: run.data.params ? JSON.stringify(run.data.params) : undefined,
+                tags: run.data.tags ? JSON.stringify(run.data.tags) : undefined
+            };
+        }
+        catch (error) {
+            console.error(`Failed to fetch run ${runId}:`, error);
+            return null;
+        }
+    }
+    /**
+     * Search runs
+     */
+    async searchMLflowRuns(input, context) {
+        try {
+            const response = await mlflow_api_client_1.mlflowApi.searchRuns({
+                experiment_ids: input.experimentIds,
+                filter: input.filter,
+                max_results: input.maxResults,
+                order_by: input.orderBy
+            });
+            return response.runs.map(run => ({
+                runId: run.info.run_id,
+                experimentId: run.info.experiment_id,
+                status: run.info.status,
+                startTime: run.info.start_time,
+                endTime: run.info.end_time,
+                artifactUri: run.info.artifact_uri,
+                metrics: run.data.metrics ? JSON.stringify(run.data.metrics) : undefined,
+                params: run.data.params ? JSON.stringify(run.data.params) : undefined,
+                tags: run.data.tags ? JSON.stringify(run.data.tags) : undefined
+            }));
+        }
+        catch (error) {
+            console.error('Failed to search runs:', error);
+            throw new Error(`Failed to search runs: ${error.message}`);
+        }
+    }
+    /**
+     * Create new run
+     */
+    async createMLflowRun(input, context) {
+        try {
+            const tags = input.tags ? JSON.parse(input.tags) : undefined;
+            const response = await mlflow_api_client_1.mlflowApi.createRun({
+                experiment_id: input.experimentId,
+                start_time: input.startTime,
+                tags
+            });
+            const run = response.run;
+            return {
+                runId: run.info.run_id,
+                experimentId: run.info.experiment_id,
+                status: run.info.status,
+                startTime: run.info.start_time,
+                artifactUri: run.info.artifact_uri,
+                metrics: run.data.metrics ? JSON.stringify(run.data.metrics) : undefined,
+                params: run.data.params ? JSON.stringify(run.data.params) : undefined,
+                tags: run.data.tags ? JSON.stringify(run.data.tags) : undefined
+            };
+        }
+        catch (error) {
+            console.error('Failed to create run:', error);
+            throw new Error(`Failed to create run: ${error.message}`);
+        }
+    }
+    /**
+     * Log metric
+     */
+    async logMLflowMetric(input, context) {
+        try {
+            await mlflow_api_client_1.mlflowApi.logMetric({
+                run_id: input.runId,
+                key: input.key,
+                value: input.value,
+                timestamp: input.timestamp,
+                step: input.step
+            });
+            return true;
+        }
+        catch (error) {
+            console.error('Failed to log metric:', error);
+            throw new Error(`Failed to log metric: ${error.message}`);
+        }
+    }
+    /**
+     * Log parameter
+     */
+    async logMLflowParam(input, context) {
+        try {
+            await mlflow_api_client_1.mlflowApi.logParam({
+                run_id: input.runId,
+                key: input.key,
+                value: input.value
+            });
+            return true;
+        }
+        catch (error) {
+            console.error('Failed to log parameter:', error);
+            throw new Error(`Failed to log parameter: ${error.message}`);
+        }
+    }
+    /**
+     * Update run
+     */
+    async updateMLflowRun(input, context) {
+        try {
+            await mlflow_api_client_1.mlflowApi.updateRun({
+                run_id: input.runId,
+                status: input.status,
+                end_time: input.endTime
+            });
+            return true;
+        }
+        catch (error) {
+            console.error('Failed to update run:', error);
+            throw new Error(`Failed to update run: ${error.message}`);
+        }
+    }
+};
+exports.RunService = RunService;
+tslib_1.__decorate([
+    (0, type_graphql_1.Query)(returns => mlflow_types_1.MLflowRun, {
+        description: 'Get MLflow run by ID',
+        nullable: true
+    }),
+    (0, type_graphql_1.Directive)('@privilege(category: "mlflow", privilege: "query")'),
+    tslib_1.__param(0, (0, type_graphql_1.Arg)('runId')),
+    tslib_1.__param(1, (0, type_graphql_1.Ctx)()),
+    tslib_1.__metadata("design:type", Function),
+    tslib_1.__metadata("design:paramtypes", [String, Object]),
+    tslib_1.__metadata("design:returntype", Promise)
+], RunService.prototype, "mlflowRun", null);
+tslib_1.__decorate([
+    (0, type_graphql_1.Query)(returns => [mlflow_types_1.MLflowRun], {
+        description: 'Search MLflow runs'
+    }),
+    (0, type_graphql_1.Directive)('@privilege(category: "mlflow", privilege: "query")'),
+    tslib_1.__param(0, (0, type_graphql_1.Arg)('input')),
+    tslib_1.__param(1, (0, type_graphql_1.Ctx)()),
+    tslib_1.__metadata("design:type", Function),
+    tslib_1.__metadata("design:paramtypes", [mlflow_types_1.SearchRunsInput, Object]),
+    tslib_1.__metadata("design:returntype", Promise)
+], RunService.prototype, "searchMLflowRuns", null);
+tslib_1.__decorate([
+    (0, type_graphql_1.Mutation)(returns => mlflow_types_1.MLflowRun, {
+        description: 'Create new MLflow run'
+    }),
+    (0, type_graphql_1.Directive)('@privilege(category: "mlflow", privilege: "mutation")'),
+    tslib_1.__param(0, (0, type_graphql_1.Arg)('input')),
+    tslib_1.__param(1, (0, type_graphql_1.Ctx)()),
+    tslib_1.__metadata("design:type", Function),
+    tslib_1.__metadata("design:paramtypes", [mlflow_types_1.CreateRunInput, Object]),
+    tslib_1.__metadata("design:returntype", Promise)
+], RunService.prototype, "createMLflowRun", null);
+tslib_1.__decorate([
+    (0, type_graphql_1.Mutation)(returns => Boolean, {
+        description: 'Log metric to MLflow run'
+    }),
+    (0, type_graphql_1.Directive)('@privilege(category: "mlflow", privilege: "mutation")'),
+    tslib_1.__param(0, (0, type_graphql_1.Arg)('input')),
+    tslib_1.__param(1, (0, type_graphql_1.Ctx)()),
+    tslib_1.__metadata("design:type", Function),
+    tslib_1.__metadata("design:paramtypes", [mlflow_types_1.LogMetricInput, Object]),
+    tslib_1.__metadata("design:returntype", Promise)
+], RunService.prototype, "logMLflowMetric", null);
+tslib_1.__decorate([
+    (0, type_graphql_1.Mutation)(returns => Boolean, {
+        description: 'Log parameter to MLflow run'
+    }),
+    (0, type_graphql_1.Directive)('@privilege(category: "mlflow", privilege: "mutation")'),
+    tslib_1.__param(0, (0, type_graphql_1.Arg)('input')),
+    tslib_1.__param(1, (0, type_graphql_1.Ctx)()),
+    tslib_1.__metadata("design:type", Function),
+    tslib_1.__metadata("design:paramtypes", [mlflow_types_1.LogParamInput, Object]),
+    tslib_1.__metadata("design:returntype", Promise)
+], RunService.prototype, "logMLflowParam", null);
+tslib_1.__decorate([
+    (0, type_graphql_1.Mutation)(returns => Boolean, {
+        description: 'Update MLflow run status'
+    }),
+    (0, type_graphql_1.Directive)('@privilege(category: "mlflow", privilege: "mutation")'),
+    tslib_1.__param(0, (0, type_graphql_1.Arg)('input')),
+    tslib_1.__param(1, (0, type_graphql_1.Ctx)()),
+    tslib_1.__metadata("design:type", Function),
+    tslib_1.__metadata("design:paramtypes", [mlflow_types_1.UpdateRunInput, Object]),
+    tslib_1.__metadata("design:returntype", Promise)
+], RunService.prototype, "updateMLflowRun", null);
+exports.RunService = RunService = tslib_1.__decorate([
+    (0, type_graphql_1.Resolver)()
+], RunService);
+//# sourceMappingURL=run-service.js.map

package/dist-server/service/run/run-service.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"run-service.js","sourceRoot":"","sources":["../../../server/service/run/run-service.ts"],"names":[],"mappings":";;;;AAAA,+CAA6E;AAE7E,qEAAyD;AACzD,2DAOiC;AAG1B,IAAM,UAAU,GAAhB,MAAM,UAAU;IACrB;;OAEG;IAMG,AAAN,KAAK,CAAC,SAAS,CAAe,KAAa,EAAS,OAAwB;QAC1E,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,6BAAS,CAAC,MAAM,CAAC,KAAK,CAAC,CAAA;YAC9C,MAAM,GAAG,GAAG,QAAQ,CAAC,GAAG,CAAA;YAExB,OAAO;gBACL,KAAK,EAAE,GAAG,CAAC,IAAI,CAAC,MAAM;gBACtB,YAAY,EAAE,GAAG,CAAC,IAAI,CAAC,aAAa;gBACpC,MAAM,EAAE,GAAG,CAAC,IAAI,CAAC,MAAM;gBACvB,SAAS,EAAE,GAAG,CAAC,IAAI,CAAC,UAAU;gBAC9B,OAAO,EAAE,GAAG,CAAC,IAAI,CAAC,QAAQ;gBAC1B,WAAW,EAAE,GAAG,CAAC,IAAI,CAAC,YAAY;gBAClC,OAAO,EAAE,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,SAAS;gBACxE,MAAM,EAAE,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,SAAS;gBACrE,IAAI,EAAE,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,SAAS;aAChE,CAAA;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,uBAAuB,KAAK,GAAG,EAAE,KAAK,CAAC,CAAA;YACrD,OAAO,IAAI,CAAA;QACb,CAAC;IACH,CAAC;IAED;;OAEG;IAKG,AAAN,KAAK,CAAC,gBAAgB,CAAe,KAAsB,EAAS,OAAwB;QAC1F,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,6BAAS,CAAC,UAAU,CAAC;gBAC1C,cAAc,EAAE,KAAK,CAAC,aAAa;gBACnC,MAAM,EAAE,KAAK,CAAC,MAAM;gBACpB,WAAW,EAAE,KAAK,CAAC,UAAU;gBAC7B,QAAQ,EAAE,KAAK,CAAC,OAAO;aACxB,CAAC,CAAA;YAEF,OAAO,QAAQ,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;gBAC/B,KAAK,EAAE,GAAG,CAAC,IAAI,CAAC,MAAM;gBACtB,YAAY,EAAE,GAAG,CAAC,IAAI,CAAC,aAAa;gBACpC,MAAM,EAAE,GAAG,CAAC,IAAI,CAAC,MAAM;gBACvB,SAAS,EAAE,GAAG,CAAC,IAAI,CAAC,UAAU;gBAC9B,OAAO,EAAE,GAAG,CAAC,IAAI,CAAC,QAAQ;gBAC1B,WAAW,EAAE,GAAG,CAAC,IAAI,CAAC,YAAY;gBAClC,OAAO,EAAE,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,SAAS;gBACxE,MAAM,EAAE,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,SAAS;gBACrE,IAAI,EAAE,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,SAAS;aAChE,CAAC,CAAC,CAAA;QACL,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,wBAAwB,EAAE,KAAK,CAAC,CAAA;YAC9C,MAAM,IAAI,KAAK,CAAC,0BAA0B,KAAK,CAAC,OAAO,EAAE,CAAC,CAAA;QAC5D,CAAC;IACH,CAAC;IAED;;OAEG;IAKG,AAAN,KAAK,CAAC,eAAe,CAAe,KAAqB,EAAS,OAAwB;QACxF,IAAI,CAAC;YACH,MAAM,IAAI,GAAG,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,SAAS,CAAA;YAE5D,MAAM,QAAQ,GAAG,MAAM,6BAAS,CAAC,SAAS,CAAC;gBACzC,aAAa,EAAE,KAAK,CAAC,YAAY;gBACjC,UAAU,EAAE,KAAK,CAAC,SAAS;gBAC3B,IAAI;aACL,CAAC,CAAA;YAEF,MAAM,GAAG,GAAG,QAAQ,CAAC,GAAG,CAAA;YAExB,OAAO;gBACL,KAAK,EAAE,GAAG,CAAC,IAAI,CAAC,MAAM;gBACtB,YAAY,EAAE,GAAG,CAAC,IAAI,CAAC,aAAa;gBACpC,MAAM,EAAE,GAAG,CAAC,IAAI,CAAC,MAAM;gBACvB,SAAS,EAAE,GAAG,CAAC,IAAI,CAAC,UAAU;gBAC9B,WAAW,EAAE,GAAG,CAAC,IAAI,CAAC,YAAY;gBAClC,OAAO,EAAE,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,SAAS;gBACxE,MAAM,EAAE,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,SAAS;gBACrE,IAAI,EAAE,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,SAAS;aAChE,CAAA;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,uBAAuB,EAAE,KAAK,CAAC,CAAA;YAC7C,MAAM,IAAI,KAAK,CAAC,yBAAyB,KAAK,CAAC,OAAO,EAAE,CAAC,CAAA;QAC3D,CAAC;IACH,CAAC;IAED;;OAEG;IAKG,AAAN,KAAK,CAAC,eAAe,CAAe,KAAqB,EAAS,OAAwB;QACxF,IAAI,CAAC;YACH,MAAM,6BAAS,CAAC,SAAS,CAAC;gBACxB,MAAM,EAAE,KAAK,CAAC,KAAK;gBACnB,GAAG,EAAE,KAAK,CAAC,GAAG;gBACd,KAAK,EAAE,KAAK,CAAC,KAAK;gBAClB,SAAS,EAAE,KAAK,CAAC,SAAS;gBAC1B,IAAI,EAAE,KAAK,CAAC,IAAI;aACjB,CAAC,CAAA;YACF,OAAO,IAAI,CAAA;QACb,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,uBAAuB,EAAE,KAAK,CAAC,CAAA;YAC7C,MAAM,IAAI,KAAK,CAAC,yBAAyB,KAAK,CAAC,OAAO,EAAE,CAAC,CAAA;QAC3D,CAAC;IACH,CAAC;IAED;;OAEG;IAKG,AAAN,KAAK,CAAC,cAAc,CAAe,KAAoB,EAAS,OAAwB;QACtF,IAAI,CAAC;YACH,MAAM,6BAAS,CAAC,QAAQ,CAAC;gBACvB,MAAM,EAAE,KAAK,CAAC,KAAK;gBACnB,GAAG,EAAE,KAAK,CAAC,GAAG;gBACd,KAAK,EAAE,KAAK,CAAC,KAAK;aACnB,CAAC,CAAA;YACF,OAAO,IAAI,CAAA;QACb,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,0BAA0B,EAAE,KAAK,CAAC,CAAA;YAChD,MAAM,IAAI,KAAK,CAAC,4BAA4B,KAAK,CAAC,OAAO,EAAE,CAAC,CAAA;QAC9D,CAAC;IACH,CAAC;IAED;;OAEG;IAKG,AAAN,KAAK,CAAC,eAAe,CAAe,KAAqB,EAAS,OAAwB;QACxF,IAAI,CAAC;YACH,MAAM,6BAAS,CAAC,SAAS,CAAC;gBACxB,MAAM,EAAE,KAAK,CAAC,KAAK;gBACnB,MAAM,EAAE,KAAK,CAAC,MAAa;gBAC3B,QAAQ,EAAE,KAAK,CAAC,OAAO;aACxB,CAAC,CAAA;YACF,OAAO,IAAI,CAAA;QACb,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,uBAAuB,EAAE,KAAK,CAAC,CAAA;YAC7C,MAAM,IAAI,KAAK,CAAC,yBAAyB,KAAK,CAAC,OAAO,EAAE,CAAC,CAAA;QAC3D,CAAC;IACH,CAAC;CACF,CAAA;AAnKY,gCAAU;AASf;IALL,IAAA,oBAAK,EAAC,OAAO,CAAC,EAAE,CAAC,wBAAS,EAAE;QAC3B,WAAW,EAAE,sBAAsB;QACnC,QAAQ,EAAE,IAAI;KACf,CAAC;IACD,IAAA,wBAAS,EAAC,oDAAoD,CAAC;IAC/C,mBAAA,IAAA,kBAAG,EAAC,OAAO,CAAC,CAAA;IAAiB,mBAAA,IAAA,kBAAG,GAAE,CAAA;;;;2CAoBlD;AASK;IAJL,IAAA,oBAAK,EAAC,OAAO,CAAC,EAAE,CAAC,CAAC,wBAAS,CAAC,EAAE;QAC7B,WAAW,EAAE,oBAAoB;KAClC,CAAC;IACD,IAAA,wBAAS,EAAC,oDAAoD,CAAC;IACxC,mBAAA,IAAA,kBAAG,EAAC,OAAO,CAAC,CAAA;IAA0B,mBAAA,IAAA,kBAAG,GAAE,CAAA;;6CAAvB,8BAAe;;kDAwB1D;AASK;IAJL,IAAA,uBAAQ,EAAC,OAAO,CAAC,EAAE,CAAC,wBAAS,EAAE;QAC9B,WAAW,EAAE,uBAAuB;KACrC,CAAC;IACD,IAAA,wBAAS,EAAC,uDAAuD,CAAC;IAC5C,mBAAA,IAAA,kBAAG,EAAC,OAAO,CAAC,CAAA;IAAyB,mBAAA,IAAA,kBAAG,GAAE,CAAA;;6CAAtB,6BAAc;;iDA0BxD;AASK;IAJL,IAAA,uBAAQ,EAAC,OAAO,CAAC,EAAE,CAAC,OAAO,EAAE;QAC5B,WAAW,EAAE,0BAA0B;KACxC,CAAC;IACD,IAAA,wBAAS,EAAC,uDAAuD,CAAC;IAC5C,mBAAA,IAAA,kBAAG,EAAC,OAAO,CAAC,CAAA;IAAyB,mBAAA,IAAA,kBAAG,GAAE,CAAA;;6CAAtB,6BAAc;;iDAcxD;AASK;IAJL,IAAA,uBAAQ,EAAC,OAAO,CAAC,EAAE,CAAC,OAAO,EAAE;QAC5B,WAAW,EAAE,6BAA6B;KAC3C,CAAC;IACD,IAAA,wBAAS,EAAC,uDAAuD,CAAC;IAC7C,mBAAA,IAAA,kBAAG,EAAC,OAAO,CAAC,CAAA;IAAwB,mBAAA,IAAA,kBAAG,GAAE,CAAA;;6CAArB,4BAAa;;gDAYtD;AASK;IAJL,IAAA,uBAAQ,EAAC,OAAO,CAAC,EAAE,CAAC,OAAO,EAAE;QAC5B,WAAW,EAAE,0BAA0B;KACxC,CAAC;IACD,IAAA,wBAAS,EAAC,uDAAuD,CAAC;IAC5C,mBAAA,IAAA,kBAAG,EAAC,OAAO,CAAC,CAAA;IAAyB,mBAAA,IAAA,kBAAG,GAAE,CAAA;;6CAAtB,6BAAc;;iDAYxD;qBAlKU,UAAU;IADtB,IAAA,uBAAQ,GAAE;GACE,UAAU,CAmKtB","sourcesContent":["import { Resolver, Query, Mutation, Arg, Ctx, Directive } from 'type-graphql'\nimport { User } from '@things-factory/auth-base'\nimport { mlflowApi } from '../../utils/mlflow-api-client'\nimport {\n  MLflowRun,\n  CreateRunInput,\n  LogMetricInput,\n  LogParamInput,\n  UpdateRunInput,\n  SearchRunsInput\n} from '../../types/mlflow-types'\n\n@Resolver()\nexport class RunService {\n  /**\n   * Get run by ID\n   */\n  @Query(returns => MLflowRun, {\n    description: 'Get MLflow run by ID',\n    nullable: true\n  })\n  @Directive('@privilege(category: \"mlflow\", privilege: \"query\")')\n  async mlflowRun(@Arg('runId') runId: string, @Ctx() context: ResolverContext): Promise<MLflowRun | null> {\n    try {\n      const response = await mlflowApi.getRun(runId)\n      const run = response.run\n\n      return {\n        runId: run.info.run_id,\n        experimentId: run.info.experiment_id,\n        status: run.info.status,\n        startTime: run.info.start_time,\n        endTime: run.info.end_time,\n        artifactUri: run.info.artifact_uri,\n        metrics: run.data.metrics ? JSON.stringify(run.data.metrics) : undefined,\n        params: run.data.params ? JSON.stringify(run.data.params) : undefined,\n        tags: run.data.tags ? JSON.stringify(run.data.tags) : undefined\n      }\n    } catch (error) {\n      console.error(`Failed to fetch run ${runId}:`, error)\n      return null\n    }\n  }\n\n  /**\n   * Search runs\n   */\n  @Query(returns => [MLflowRun], {\n    description: 'Search MLflow runs'\n  })\n  @Directive('@privilege(category: \"mlflow\", privilege: \"query\")')\n  async searchMLflowRuns(@Arg('input') input: SearchRunsInput, @Ctx() context: ResolverContext): Promise<MLflowRun[]> {\n    try {\n      const response = await mlflowApi.searchRuns({\n        experiment_ids: input.experimentIds,\n        filter: input.filter,\n        max_results: input.maxResults,\n        order_by: input.orderBy\n      })\n\n      return response.runs.map(run => ({\n        runId: run.info.run_id,\n        experimentId: run.info.experiment_id,\n        status: run.info.status,\n        startTime: run.info.start_time,\n        endTime: run.info.end_time,\n        artifactUri: run.info.artifact_uri,\n        metrics: run.data.metrics ? JSON.stringify(run.data.metrics) : undefined,\n        params: run.data.params ? JSON.stringify(run.data.params) : undefined,\n        tags: run.data.tags ? JSON.stringify(run.data.tags) : undefined\n      }))\n    } catch (error) {\n      console.error('Failed to search runs:', error)\n      throw new Error(`Failed to search runs: ${error.message}`)\n    }\n  }\n\n  /**\n   * Create new run\n   */\n  @Mutation(returns => MLflowRun, {\n    description: 'Create new MLflow run'\n  })\n  @Directive('@privilege(category: \"mlflow\", privilege: \"mutation\")')\n  async createMLflowRun(@Arg('input') input: CreateRunInput, @Ctx() context: ResolverContext): Promise<MLflowRun> {\n    try {\n      const tags = input.tags ? JSON.parse(input.tags) : undefined\n\n      const response = await mlflowApi.createRun({\n        experiment_id: input.experimentId,\n        start_time: input.startTime,\n        tags\n      })\n\n      const run = response.run\n\n      return {\n        runId: run.info.run_id,\n        experimentId: run.info.experiment_id,\n        status: run.info.status,\n        startTime: run.info.start_time,\n        artifactUri: run.info.artifact_uri,\n        metrics: run.data.metrics ? JSON.stringify(run.data.metrics) : undefined,\n        params: run.data.params ? JSON.stringify(run.data.params) : undefined,\n        tags: run.data.tags ? JSON.stringify(run.data.tags) : undefined\n      }\n    } catch (error) {\n      console.error('Failed to create run:', error)\n      throw new Error(`Failed to create run: ${error.message}`)\n    }\n  }\n\n  /**\n   * Log metric\n   */\n  @Mutation(returns => Boolean, {\n    description: 'Log metric to MLflow run'\n  })\n  @Directive('@privilege(category: \"mlflow\", privilege: \"mutation\")')\n  async logMLflowMetric(@Arg('input') input: LogMetricInput, @Ctx() context: ResolverContext): Promise<boolean> {\n    try {\n      await mlflowApi.logMetric({\n        run_id: input.runId,\n        key: input.key,\n        value: input.value,\n        timestamp: input.timestamp,\n        step: input.step\n      })\n      return true\n    } catch (error) {\n      console.error('Failed to log metric:', error)\n      throw new Error(`Failed to log metric: ${error.message}`)\n    }\n  }\n\n  /**\n   * Log parameter\n   */\n  @Mutation(returns => Boolean, {\n    description: 'Log parameter to MLflow run'\n  })\n  @Directive('@privilege(category: \"mlflow\", privilege: \"mutation\")')\n  async logMLflowParam(@Arg('input') input: LogParamInput, @Ctx() context: ResolverContext): Promise<boolean> {\n    try {\n      await mlflowApi.logParam({\n        run_id: input.runId,\n        key: input.key,\n        value: input.value\n      })\n      return true\n    } catch (error) {\n      console.error('Failed to log parameter:', error)\n      throw new Error(`Failed to log parameter: ${error.message}`)\n    }\n  }\n\n  /**\n   * Update run\n   */\n  @Mutation(returns => Boolean, {\n    description: 'Update MLflow run status'\n  })\n  @Directive('@privilege(category: \"mlflow\", privilege: \"mutation\")')\n  async updateMLflowRun(@Arg('input') input: UpdateRunInput, @Ctx() context: ResolverContext): Promise<boolean> {\n    try {\n      await mlflowApi.updateRun({\n        run_id: input.runId,\n        status: input.status as any,\n        end_time: input.endTime\n      })\n      return true\n    } catch (error) {\n      console.error('Failed to update run:', error)\n      throw new Error(`Failed to update run: ${error.message}`)\n    }\n  }\n}\n"]}