@things-factory/integration-mlflow 9.1.19

package/dist-server/utils/mlflow-api-client.d.ts

@@ -0,0 +1,277 @@
+/**
+ * MLflow REST API Client
+ *
+ * Thin wrapper around MLflow REST API for Things-Factory integration
+ * Reference: https://mlflow.org/docs/latest/rest-api.html
+ */
+declare class MLflowApiClient {
+    private baseUrl;
+    constructor(baseUrl?: string);
+    /**
+     * Generic HTTP request wrapper
+     */
+    private request;
+    /**
+     * Create a new experiment
+     */
+    createExperiment(params: {
+        name: string;
+        artifact_location?: string;
+        tags?: Array<{
+            key: string;
+            value: string;
+        }>;
+    }): Promise<{
+        experiment_id: string;
+    }>;
+    /**
+     * Get experiment by ID
+     */
+    getExperiment(experimentId: string): Promise<{
+        experiment: {
+            experiment_id: string;
+            name: string;
+            artifact_location: string;
+            lifecycle_stage: string;
+            tags?: Array<{
+                key: string;
+                value: string;
+            }>;
+        };
+    }>;
+    /**
+     * Search experiments
+     */
+    searchExperiments(params?: {
+        max_results?: number;
+        filter?: string;
+        order_by?: string[];
+    }): Promise<{
+        experiments: Array<{
+            experiment_id: string;
+            name: string;
+            artifact_location: string;
+            lifecycle_stage: string;
+            tags?: Array<{
+                key: string;
+                value: string;
+            }>;
+        }>;
+    }>;
+    /**
+     * Delete experiment
+     */
+    deleteExperiment(experimentId: string): Promise<void>;
+    /**
+     * Create a new run
+     */
+    createRun(params: {
+        experiment_id: string;
+        start_time?: number;
+        tags?: Array<{
+            key: string;
+            value: string;
+        }>;
+    }): Promise<{
+        run: {
+            info: {
+                run_id: string;
+                experiment_id: string;
+                status: string;
+                start_time: number;
+                artifact_uri: string;
+            };
+            data: any;
+        };
+    }>;
+    /**
+     * Get run by ID
+     */
+    getRun(runId: string): Promise<{
+        run: {
+            info: {
+                run_id: string;
+                experiment_id: string;
+                status: string;
+                start_time: number;
+                end_time?: number;
+                artifact_uri: string;
+            };
+            data: {
+                metrics: Array<{
+                    key: string;
+                    value: number;
+                    timestamp: number;
+                    step: number;
+                }>;
+                params: Array<{
+                    key: string;
+                    value: string;
+                }>;
+                tags: Array<{
+                    key: string;
+                    value: string;
+                }>;
+            };
+        };
+    }>;
+    /**
+     * Search runs
+     */
+    searchRuns(params: {
+        experiment_ids: string[];
+        filter?: string;
+        max_results?: number;
+        order_by?: string[];
+    }): Promise<{
+        runs: Array<{
+            info: {
+                run_id: string;
+                experiment_id: string;
+                status: string;
+                start_time: number;
+                end_time?: number;
+                artifact_uri: string;
+            };
+            data: any;
+        }>;
+    }>;
+    /**
+     * Log metric for a run
+     */
+    logMetric(params: {
+        run_id: string;
+        key: string;
+        value: number;
+        timestamp?: number;
+        step?: number;
+    }): Promise<void>;
+    /**
+     * Log parameter for a run
+     */
+    logParam(params: {
+        run_id: string;
+        key: string;
+        value: string;
+    }): Promise<void>;
+    /**
+     * Update run status
+     */
+    updateRun(params: {
+        run_id: string;
+        status?: 'RUNNING' | 'SCHEDULED' | 'FINISHED' | 'FAILED' | 'KILLED';
+        end_time?: number;
+    }): Promise<void>;
+    /**
+     * Create registered model
+     */
+    createRegisteredModel(params: {
+        name: string;
+        tags?: Array<{
+            key: string;
+            value: string;
+        }>;
+        description?: string;
+    }): Promise<{
+        registered_model: {
+            name: string;
+            creation_timestamp: number;
+            last_updated_timestamp: number;
+        };
+    }>;
+    /**
+     * Get registered model
+     */
+    getRegisteredModel(name: string): Promise<{
+        registered_model: {
+            name: string;
+            creation_timestamp: number;
+            last_updated_timestamp: number;
+            description?: string;
+            tags?: Array<{
+                key: string;
+                value: string;
+            }>;
+            latest_versions?: any[];
+        };
+    }>;
+    /**
+     * Search registered models
+     */
+    searchRegisteredModels(params?: {
+        filter?: string;
+        max_results?: number;
+        order_by?: string[];
+    }): Promise<{
+        registered_models: Array<{
+            name: string;
+            creation_timestamp: number;
+            last_updated_timestamp: number;
+            description?: string;
+            latest_versions?: any[];
+        }>;
+    }>;
+    /**
+     * Create model version
+     */
+    createModelVersion(params: {
+        name: string;
+        source: string;
+        run_id?: string;
+        tags?: Array<{
+            key: string;
+            value: string;
+        }>;
+        description?: string;
+    }): Promise<{
+        model_version: {
+            name: string;
+            version: string;
+            creation_timestamp: number;
+            last_updated_timestamp: number;
+            source: string;
+            status: string;
+        };
+    }>;
+    /**
+     * Get model version
+     */
+    getModelVersion(params: {
+        name: string;
+        version: string;
+    }): Promise<{
+        model_version: {
+            name: string;
+            version: string;
+            creation_timestamp: number;
+            current_stage: string;
+            source: string;
+            status: string;
+            run_id?: string;
+        };
+    }>;
+    /**
+     * Transition model version stage
+     */
+    transitionModelVersionStage(params: {
+        name: string;
+        version: string;
+        stage: 'None' | 'Staging' | 'Production' | 'Archived';
+        archive_existing_versions?: boolean;
+    }): Promise<{
+        model_version: {
+            name: string;
+            version: string;
+            current_stage: string;
+        };
+    }>;
+    /**
+     * Delete model version
+     */
+    deleteModelVersion(params: {
+        name: string;
+        version: string;
+    }): Promise<void>;
+}
+export declare const mlflowApi: MLflowApiClient;
+export { MLflowApiClient };

package/dist-server/utils/mlflow-api-client.js

@@ -0,0 +1,201 @@
+"use strict";
+/**
+ * MLflow REST API Client
+ *
+ * Thin wrapper around MLflow REST API for Things-Factory integration
+ * Reference: https://mlflow.org/docs/latest/rest-api.html
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.MLflowApiClient = exports.mlflowApi = void 0;
+const env_1 = require("@things-factory/env");
+const MLFLOW_TRACKING_URI = env_1.config.get('mlflowTrackingUri', process.env.MLFLOW_TRACKING_URI || 'http://localhost:5000');
+class MLflowApiClient {
+    constructor(baseUrl = MLFLOW_TRACKING_URI) {
+        this.baseUrl = baseUrl.replace(/\/$/, ''); // Remove trailing slash
+    }
+    /**
+     * Generic HTTP request wrapper
+     */
+    async request(endpoint, options = {}) {
+        const url = `${this.baseUrl}${endpoint}`;
+        const defaultHeaders = {
+            'Content-Type': 'application/json'
+        };
+        try {
+            const response = await fetch(url, {
+                ...options,
+                headers: {
+                    ...defaultHeaders,
+                    ...options.headers
+                }
+            });
+            if (!response.ok) {
+                const errorText = await response.text();
+                throw new Error(`MLflow API error: ${response.status} - ${errorText}`);
+            }
+            return await response.json();
+        }
+        catch (error) {
+            console.error(`MLflow API request failed: ${endpoint}`, error);
+            throw error;
+        }
+    }
+    // ========================================================================
+    // Experiments API
+    // ========================================================================
+    /**
+     * Create a new experiment
+     */
+    async createExperiment(params) {
+        return this.request('/api/2.0/mlflow/experiments/create', {
+            method: 'POST',
+            body: JSON.stringify(params)
+        });
+    }
+    /**
+     * Get experiment by ID
+     */
+    async getExperiment(experimentId) {
+        return this.request(`/api/2.0/mlflow/experiments/get?experiment_id=${experimentId}`);
+    }
+    /**
+     * Search experiments
+     */
+    async searchExperiments(params) {
+        const queryParams = new URLSearchParams();
+        if (params?.max_results)
+            queryParams.append('max_results', String(params.max_results));
+        if (params?.filter)
+            queryParams.append('filter', params.filter);
+        return this.request(`/api/2.0/mlflow/experiments/search?${queryParams}`);
+    }
+    /**
+     * Delete experiment
+     */
+    async deleteExperiment(experimentId) {
+        return this.request('/api/2.0/mlflow/experiments/delete', {
+            method: 'POST',
+            body: JSON.stringify({ experiment_id: experimentId })
+        });
+    }
+    // ========================================================================
+    // Runs API
+    // ========================================================================
+    /**
+     * Create a new run
+     */
+    async createRun(params) {
+        return this.request('/api/2.0/mlflow/runs/create', {
+            method: 'POST',
+            body: JSON.stringify(params)
+        });
+    }
+    /**
+     * Get run by ID
+     */
+    async getRun(runId) {
+        return this.request(`/api/2.0/mlflow/runs/get?run_id=${runId}`);
+    }
+    /**
+     * Search runs
+     */
+    async searchRuns(params) {
+        return this.request('/api/2.0/mlflow/runs/search', {
+            method: 'POST',
+            body: JSON.stringify(params)
+        });
+    }
+    /**
+     * Log metric for a run
+     */
+    async logMetric(params) {
+        return this.request('/api/2.0/mlflow/runs/log-metric', {
+            method: 'POST',
+            body: JSON.stringify(params)
+        });
+    }
+    /**
+     * Log parameter for a run
+     */
+    async logParam(params) {
+        return this.request('/api/2.0/mlflow/runs/log-parameter', {
+            method: 'POST',
+            body: JSON.stringify(params)
+        });
+    }
+    /**
+     * Update run status
+     */
+    async updateRun(params) {
+        return this.request('/api/2.0/mlflow/runs/update', {
+            method: 'POST',
+            body: JSON.stringify(params)
+        });
+    }
+    // ========================================================================
+    // Model Registry API
+    // ========================================================================
+    /**
+     * Create registered model
+     */
+    async createRegisteredModel(params) {
+        return this.request('/api/2.0/mlflow/registered-models/create', {
+            method: 'POST',
+            body: JSON.stringify(params)
+        });
+    }
+    /**
+     * Get registered model
+     */
+    async getRegisteredModel(name) {
+        return this.request(`/api/2.0/mlflow/registered-models/get?name=${encodeURIComponent(name)}`);
+    }
+    /**
+     * Search registered models
+     */
+    async searchRegisteredModels(params) {
+        const queryParams = new URLSearchParams();
+        if (params?.filter)
+            queryParams.append('filter', params.filter);
+        if (params?.max_results)
+            queryParams.append('max_results', String(params.max_results));
+        return this.request(`/api/2.0/mlflow/registered-models/search?${queryParams}`);
+    }
+    /**
+     * Create model version
+     */
+    async createModelVersion(params) {
+        return this.request('/api/2.0/mlflow/model-versions/create', {
+            method: 'POST',
+            body: JSON.stringify(params)
+        });
+    }
+    /**
+     * Get model version
+     */
+    async getModelVersion(params) {
+        return this.request(`/api/2.0/mlflow/model-versions/get?name=${encodeURIComponent(params.name)}&version=${params.version}`);
+    }
+    /**
+     * Transition model version stage
+     */
+    async transitionModelVersionStage(params) {
+        return this.request('/api/2.0/mlflow/model-versions/transition-stage', {
+            method: 'POST',
+            body: JSON.stringify(params)
+        });
+    }
+    /**
+     * Delete model version
+     */
+    async deleteModelVersion(params) {
+        return this.request('/api/2.0/mlflow/model-versions/delete', {
+            method: 'DELETE',
+            body: JSON.stringify(params)
+        });
+    }
+}
+exports.MLflowApiClient = MLflowApiClient;
+// Export singleton instance
+exports.mlflowApi = new MLflowApiClient();
+//# sourceMappingURL=mlflow-api-client.js.map

package/dist-server/utils/mlflow-api-client.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"mlflow-api-client.js","sourceRoot":"","sources":["../../server/utils/mlflow-api-client.ts"],"names":[],"mappings":";AAAA;;;;;GAKG;;;AAEH,6CAA4C;AAE5C,MAAM,mBAAmB,GAAG,YAAM,CAAC,GAAG,CAAC,mBAAmB,EAAE,OAAO,CAAC,GAAG,CAAC,mBAAmB,IAAI,uBAAuB,CAAC,CAAA;AAOvH,MAAM,eAAe;IAGnB,YAAY,UAAkB,mBAAmB;QAC/C,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC,KAAK,EAAE,EAAE,CAAC,CAAA,CAAC,wBAAwB;IACpE,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,OAAO,CAAI,QAAgB,EAAE,UAAuB,EAAE;QAClE,MAAM,GAAG,GAAG,GAAG,IAAI,CAAC,OAAO,GAAG,QAAQ,EAAE,CAAA;QAExC,MAAM,cAAc,GAAG;YACrB,cAAc,EAAE,kBAAkB;SACnC,CAAA;QAED,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,EAAE;gBAChC,GAAG,OAAO;gBACV,OAAO,EAAE;oBACP,GAAG,cAAc;oBACjB,GAAG,OAAO,CAAC,OAAO;iBACnB;aACF,CAAC,CAAA;YAEF,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;gBACjB,MAAM,SAAS,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAA;gBACvC,MAAM,IAAI,KAAK,CAAC,qBAAqB,QAAQ,CAAC,MAAM,MAAM,SAAS,EAAE,CAAC,CAAA;YACxE,CAAC;YAED,OAAO,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAA;QAC9B,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,8BAA8B,QAAQ,EAAE,EAAE,KAAK,CAAC,CAAA;YAC9D,MAAM,KAAK,CAAA;QACb,CAAC;IACH,CAAC;IAED,2EAA2E;IAC3E,kBAAkB;IAClB,2EAA2E;IAE3E;;OAEG;IACH,KAAK,CAAC,gBAAgB,CAAC,MAItB;QACC,OAAO,IAAI,CAAC,OAAO,CAAC,oCAAoC,EAAE;YACxD,MAAM,EAAE,MAAM;YACd,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC;SAC7B,CAAC,CAAA;IACJ,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,aAAa,CAAC,YAAoB;QAStC,OAAO,IAAI,CAAC,OAAO,CAAC,iDAAiD,YAAY,EAAE,CAAC,CAAA;IACtF,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,iBAAiB,CAAC,MAAuE;QAS7F,MAAM,WAAW,GAAG,IAAI,eAAe,EAAE,CAAA;QACzC,IAAI,MAAM,EAAE,WAAW;YAAE,WAAW,CAAC,MAAM,CAAC,aAAa,EAAE,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC,CAAA;QACtF,IAAI,MAAM,EAAE,MAAM;YAAE,WAAW,CAAC,MAAM,CAAC,QAAQ,EAAE,MAAM,CAAC,MAAM,CAAC,CAAA;QAE/D,OAAO,IAAI,CAAC,OAAO,CAAC,sCAAsC,WAAW,EAAE,CAAC,CAAA;IAC1E,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,gBAAgB,CAAC,YAAoB;QACzC,OAAO,IAAI,CAAC,OAAO,CAAC,oCAAoC,EAAE;YACxD,MAAM,EAAE,MAAM;YACd,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,aAAa,EAAE,YAAY,EAAE,CAAC;SACtD,CAAC,CAAA;IACJ,CAAC;IAED,2EAA2E;IAC3E,WAAW;IACX,2EAA2E;IAE3E;;OAEG;IACH,KAAK,CAAC,SAAS,CAAC,MAIf;QAYC,OAAO,IAAI,CAAC,OAAO,CAAC,6BAA6B,EAAE;YACjD,MAAM,EAAE,MAAM;YACd,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC;SAC7B,CAAC,CAAA;IACJ,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,MAAM,CAAC,KAAa;QAiBxB,OAAO,IAAI,CAAC,OAAO,CAAC,mCAAmC,KAAK,EAAE,CAAC,CAAA;IACjE,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,UAAU,CAAC,MAKhB;QAaC,OAAO,IAAI,CAAC,OAAO,CAAC,6BAA6B,EAAE;YACjD,MAAM,EAAE,MAAM;YACd,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC;SAC7B,CAAC,CAAA;IACJ,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,SAAS,CAAC,MAMf;QACC,OAAO,IAAI,CAAC,OAAO,CAAC,iCAAiC,EAAE;YACrD,MAAM,EAAE,MAAM;YACd,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC;SAC7B,CAAC,CAAA;IACJ,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,QAAQ,CAAC,MAAsD;QACnE,OAAO,IAAI,CAAC,OAAO,CAAC,oCAAoC,EAAE;YACxD,MAAM,EAAE,MAAM;YACd,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC;SAC7B,CAAC,CAAA;IACJ,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,SAAS,CAAC,MAIf;QACC,OAAO,IAAI,CAAC,OAAO,CAAC,6BAA6B,EAAE;YACjD,MAAM,EAAE,MAAM;YACd,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC;SAC7B,CAAC,CAAA;IACJ,CAAC;IAED,2EAA2E;IAC3E,qBAAqB;IACrB,2EAA2E;IAE3E;;OAEG;IACH,KAAK,CAAC,qBAAqB,CAAC,MAI3B;QAOC,OAAO,IAAI,CAAC,OAAO,CAAC,0CAA0C,EAAE;YAC9D,MAAM,EAAE,MAAM;YACd,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC;SAC7B,CAAC,CAAA;IACJ,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,kBAAkB,CAAC,IAAY;QAUnC,OAAO,IAAI,CAAC,OAAO,CAAC,8CAA8C,kBAAkB,CAAC,IAAI,CAAC,EAAE,CAAC,CAAA;IAC/F,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,sBAAsB,CAAC,MAAuE;QASlG,MAAM,WAAW,GAAG,IAAI,eAAe,EAAE,CAAA;QACzC,IAAI,MAAM,EAAE,MAAM;YAAE,WAAW,CAAC,MAAM,CAAC,QAAQ,EAAE,MAAM,CAAC,MAAM,CAAC,CAAA;QAC/D,IAAI,MAAM,EAAE,WAAW;YAAE,WAAW,CAAC,MAAM,CAAC,aAAa,EAAE,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC,CAAA;QAEtF,OAAO,IAAI,CAAC,OAAO,CAAC,4CAA4C,WAAW,EAAE,CAAC,CAAA;IAChF,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,kBAAkB,CAAC,MAMxB;QAUC,OAAO,IAAI,CAAC,OAAO,CAAC,uCAAuC,EAAE;YAC3D,MAAM,EAAE,MAAM;YACd,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC;SAC7B,CAAC,CAAA;IACJ,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,eAAe,CAAC,MAAyC;QAW7D,OAAO,IAAI,CAAC,OAAO,CACjB,2CAA2C,kBAAkB,CAAC,MAAM,CAAC,IAAI,CAAC,YAAY,MAAM,CAAC,OAAO,EAAE,CACvG,CAAA;IACH,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,2BAA2B,CAAC,MAKjC;QAOC,OAAO,IAAI,CAAC,OAAO,CAAC,iDAAiD,EAAE;YACrE,MAAM,EAAE,MAAM;YACd,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC;SAC7B,CAAC,CAAA;IACJ,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,kBAAkB,CAAC,MAAyC;QAChE,OAAO,IAAI,CAAC,OAAO,CAAC,uCAAuC,EAAE;YAC3D,MAAM,EAAE,QAAQ;YAChB,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC;SAC7B,CAAC,CAAA;IACJ,CAAC;CACF;AAMQ,0CAAe;AAJxB,4BAA4B;AACf,QAAA,SAAS,GAAG,IAAI,eAAe,EAAE,CAAA","sourcesContent":["/**\n * MLflow REST API Client\n *\n * Thin wrapper around MLflow REST API for Things-Factory integration\n * Reference: https://mlflow.org/docs/latest/rest-api.html\n */\n\nimport { config } from '@things-factory/env'\n\nconst MLFLOW_TRACKING_URI = config.get('mlflowTrackingUri', process.env.MLFLOW_TRACKING_URI || 'http://localhost:5000')\n\ninterface MLflowResponse<T> {\n  data?: T\n  error?: string\n}\n\nclass MLflowApiClient {\n  private baseUrl: string\n\n  constructor(baseUrl: string = MLFLOW_TRACKING_URI) {\n    this.baseUrl = baseUrl.replace(/\\/$/, '') // Remove trailing slash\n  }\n\n  /**\n   * Generic HTTP request wrapper\n   */\n  private async request<T>(endpoint: string, options: RequestInit = {}): Promise<T> {\n    const url = `${this.baseUrl}${endpoint}`\n\n    const defaultHeaders = {\n      'Content-Type': 'application/json'\n    }\n\n    try {\n      const response = await fetch(url, {\n        ...options,\n        headers: {\n          ...defaultHeaders,\n          ...options.headers\n        }\n      })\n\n      if (!response.ok) {\n        const errorText = await response.text()\n        throw new Error(`MLflow API error: ${response.status} - ${errorText}`)\n      }\n\n      return await response.json()\n    } catch (error) {\n      console.error(`MLflow API request failed: ${endpoint}`, error)\n      throw error\n    }\n  }\n\n  // ========================================================================\n  // Experiments API\n  // ========================================================================\n\n  /**\n   * Create a new experiment\n   */\n  async createExperiment(params: {\n    name: string\n    artifact_location?: string\n    tags?: Array<{ key: string; value: string }>\n  }): Promise<{ experiment_id: string }> {\n    return this.request('/api/2.0/mlflow/experiments/create', {\n      method: 'POST',\n      body: JSON.stringify(params)\n    })\n  }\n\n  /**\n   * Get experiment by ID\n   */\n  async getExperiment(experimentId: string): Promise<{\n    experiment: {\n      experiment_id: string\n      name: string\n      artifact_location: string\n      lifecycle_stage: string\n      tags?: Array<{ key: string; value: string }>\n    }\n  }> {\n    return this.request(`/api/2.0/mlflow/experiments/get?experiment_id=${experimentId}`)\n  }\n\n  /**\n   * Search experiments\n   */\n  async searchExperiments(params?: { max_results?: number; filter?: string; order_by?: string[] }): Promise<{\n    experiments: Array<{\n      experiment_id: string\n      name: string\n      artifact_location: string\n      lifecycle_stage: string\n      tags?: Array<{ key: string; value: string }>\n    }>\n  }> {\n    const queryParams = new URLSearchParams()\n    if (params?.max_results) queryParams.append('max_results', String(params.max_results))\n    if (params?.filter) queryParams.append('filter', params.filter)\n\n    return this.request(`/api/2.0/mlflow/experiments/search?${queryParams}`)\n  }\n\n  /**\n   * Delete experiment\n   */\n  async deleteExperiment(experimentId: string): Promise<void> {\n    return this.request('/api/2.0/mlflow/experiments/delete', {\n      method: 'POST',\n      body: JSON.stringify({ experiment_id: experimentId })\n    })\n  }\n\n  // ========================================================================\n  // Runs API\n  // ========================================================================\n\n  /**\n   * Create a new run\n   */\n  async createRun(params: {\n    experiment_id: string\n    start_time?: number\n    tags?: Array<{ key: string; value: string }>\n  }): Promise<{\n    run: {\n      info: {\n        run_id: string\n        experiment_id: string\n        status: string\n        start_time: number\n        artifact_uri: string\n      }\n      data: any\n    }\n  }> {\n    return this.request('/api/2.0/mlflow/runs/create', {\n      method: 'POST',\n      body: JSON.stringify(params)\n    })\n  }\n\n  /**\n   * Get run by ID\n   */\n  async getRun(runId: string): Promise<{\n    run: {\n      info: {\n        run_id: string\n        experiment_id: string\n        status: string\n        start_time: number\n        end_time?: number\n        artifact_uri: string\n      }\n      data: {\n        metrics: Array<{ key: string; value: number; timestamp: number; step: number }>\n        params: Array<{ key: string; value: string }>\n        tags: Array<{ key: string; value: string }>\n      }\n    }\n  }> {\n    return this.request(`/api/2.0/mlflow/runs/get?run_id=${runId}`)\n  }\n\n  /**\n   * Search runs\n   */\n  async searchRuns(params: {\n    experiment_ids: string[]\n    filter?: string\n    max_results?: number\n    order_by?: string[]\n  }): Promise<{\n    runs: Array<{\n      info: {\n        run_id: string\n        experiment_id: string\n        status: string\n        start_time: number\n        end_time?: number\n        artifact_uri: string\n      }\n      data: any\n    }>\n  }> {\n    return this.request('/api/2.0/mlflow/runs/search', {\n      method: 'POST',\n      body: JSON.stringify(params)\n    })\n  }\n\n  /**\n   * Log metric for a run\n   */\n  async logMetric(params: {\n    run_id: string\n    key: string\n    value: number\n    timestamp?: number\n    step?: number\n  }): Promise<void> {\n    return this.request('/api/2.0/mlflow/runs/log-metric', {\n      method: 'POST',\n      body: JSON.stringify(params)\n    })\n  }\n\n  /**\n   * Log parameter for a run\n   */\n  async logParam(params: { run_id: string; key: string; value: string }): Promise<void> {\n    return this.request('/api/2.0/mlflow/runs/log-parameter', {\n      method: 'POST',\n      body: JSON.stringify(params)\n    })\n  }\n\n  /**\n   * Update run status\n   */\n  async updateRun(params: {\n    run_id: string\n    status?: 'RUNNING' | 'SCHEDULED' | 'FINISHED' | 'FAILED' | 'KILLED'\n    end_time?: number\n  }): Promise<void> {\n    return this.request('/api/2.0/mlflow/runs/update', {\n      method: 'POST',\n      body: JSON.stringify(params)\n    })\n  }\n\n  // ========================================================================\n  // Model Registry API\n  // ========================================================================\n\n  /**\n   * Create registered model\n   */\n  async createRegisteredModel(params: {\n    name: string\n    tags?: Array<{ key: string; value: string }>\n    description?: string\n  }): Promise<{\n    registered_model: {\n      name: string\n      creation_timestamp: number\n      last_updated_timestamp: number\n    }\n  }> {\n    return this.request('/api/2.0/mlflow/registered-models/create', {\n      method: 'POST',\n      body: JSON.stringify(params)\n    })\n  }\n\n  /**\n   * Get registered model\n   */\n  async getRegisteredModel(name: string): Promise<{\n    registered_model: {\n      name: string\n      creation_timestamp: number\n      last_updated_timestamp: number\n      description?: string\n      tags?: Array<{ key: string; value: string }>\n      latest_versions?: any[]\n    }\n  }> {\n    return this.request(`/api/2.0/mlflow/registered-models/get?name=${encodeURIComponent(name)}`)\n  }\n\n  /**\n   * Search registered models\n   */\n  async searchRegisteredModels(params?: { filter?: string; max_results?: number; order_by?: string[] }): Promise<{\n    registered_models: Array<{\n      name: string\n      creation_timestamp: number\n      last_updated_timestamp: number\n      description?: string\n      latest_versions?: any[]\n    }>\n  }> {\n    const queryParams = new URLSearchParams()\n    if (params?.filter) queryParams.append('filter', params.filter)\n    if (params?.max_results) queryParams.append('max_results', String(params.max_results))\n\n    return this.request(`/api/2.0/mlflow/registered-models/search?${queryParams}`)\n  }\n\n  /**\n   * Create model version\n   */\n  async createModelVersion(params: {\n    name: string\n    source: string\n    run_id?: string\n    tags?: Array<{ key: string; value: string }>\n    description?: string\n  }): Promise<{\n    model_version: {\n      name: string\n      version: string\n      creation_timestamp: number\n      last_updated_timestamp: number\n      source: string\n      status: string\n    }\n  }> {\n    return this.request('/api/2.0/mlflow/model-versions/create', {\n      method: 'POST',\n      body: JSON.stringify(params)\n    })\n  }\n\n  /**\n   * Get model version\n   */\n  async getModelVersion(params: { name: string; version: string }): Promise<{\n    model_version: {\n      name: string\n      version: string\n      creation_timestamp: number\n      current_stage: string\n      source: string\n      status: string\n      run_id?: string\n    }\n  }> {\n    return this.request(\n      `/api/2.0/mlflow/model-versions/get?name=${encodeURIComponent(params.name)}&version=${params.version}`\n    )\n  }\n\n  /**\n   * Transition model version stage\n   */\n  async transitionModelVersionStage(params: {\n    name: string\n    version: string\n    stage: 'None' | 'Staging' | 'Production' | 'Archived'\n    archive_existing_versions?: boolean\n  }): Promise<{\n    model_version: {\n      name: string\n      version: string\n      current_stage: string\n    }\n  }> {\n    return this.request('/api/2.0/mlflow/model-versions/transition-stage', {\n      method: 'POST',\n      body: JSON.stringify(params)\n    })\n  }\n\n  /**\n   * Delete model version\n   */\n  async deleteModelVersion(params: { name: string; version: string }): Promise<void> {\n    return this.request('/api/2.0/mlflow/model-versions/delete', {\n      method: 'DELETE',\n      body: JSON.stringify(params)\n    })\n  }\n}\n\n// Export singleton instance\nexport const mlflowApi = new MLflowApiClient()\n\n// Export class for testing or custom instances\nexport { MLflowApiClient }\n"]}

package/package.json

@@ -0,0 +1,45 @@
+{
+  "name": "@things-factory/integration-mlflow",
+  "version": "9.1.19",
+  "main": "dist-server/index.js",
+  "browser": "dist-client/index.js",
+  "things-factory": true,
+  "author": "Things-Factory Team",
+  "description": "Integration module for MLflow - experiment tracking and model registry",
+  "license": "MIT",
+  "publishConfig": {
+    "access": "public",
+    "@things-factory:registry": "https://registry.npmjs.org"
+  },
+  "repository": {
+    "type": "git",
+    "url": "git+https://github.com/hatiolab/things-factory.git",
+    "directory": "packages/integration-mlflow"
+  },
+  "scripts": {
+    "build": "npm run build:server && npm run build:client",
+    "copy:files": "copyfiles -e \"./client/**/*.{ts,js,json}\" -u 1 \"./client/**/*\" dist-client",
+    "build:client": "npm run copy:files && npx tsc --p ./client/tsconfig.json",
+    "build:server": "npx tsc --p ./server/tsconfig.json",
+    "clean:client": "npx rimraf dist-client",
+    "clean:server": "npx rimraf dist-server",
+    "clean": "npm run clean:server && npm run clean:client"
+  },
+  "dependencies": {
+    "@operato/graphql": "^9.0.0",
+    "@operato/layout": "^9.0.0",
+    "@things-factory/auth-base": "^9.1.19",
+    "@things-factory/env": "^9.1.0",
+    "@things-factory/menu-base": "^9.1.19",
+    "@things-factory/shell": "^9.1.19"
+  },
+  "keywords": [
+    "things-factory",
+    "mlflow",
+    "integration",
+    "experiment-tracking",
+    "model-registry",
+    "machine-learning"
+  ],
+  "gitHead": "078438034dbe19915108e89ff24024f7044a85a9"
+}

package/server/index.ts

@@ -0,0 +1,5 @@
+export * from './service/index.js'
+export * from './types/mlflow-types.js'
+export * from './utils/mlflow-api-client.js'
+
+import './route.js'

package/server/route.ts

@@ -0,0 +1,13 @@
+/**
+ * MLflow Integration Routes
+ *
+ * Register additional routes if needed (beyond GraphQL)
+ */
+
+// Example: Proxy to MLflow UI, webhooks, etc.
+// import Koa from 'koa'
+// import Router from 'koa-router'
+
+// process.on('bootstrap-module-domain-private-route', (app: Koa, router: Router) => {
+//   // Add custom routes here
+// })

package/server/service/experiment/experiment-service.ts

@@ -0,0 +1,117 @@
+import { Resolver, Query, Mutation, Arg, Ctx, Directive } from 'type-graphql'
+import { mlflowApi } from '../../utils/mlflow-api-client'
+import {
+  MLflowExperiment,
+  CreateExperimentInput
+} from '../../types/mlflow-types'
+
+@Resolver()
+export class ExperimentService {
+  /**
+   * Get all experiments
+   */
+  @Query(returns => [MLflowExperiment], {
+    description: 'Get all MLflow experiments'
+  })
+  @Directive('@privilege(category: "mlflow", privilege: "query")')
+  async mlflowExperiments(@Ctx() context: ResolverContext): Promise<MLflowExperiment[]> {
+    try {
+      const response = await mlflowApi.searchExperiments()
+
+      return response.experiments.map(exp => ({
+        experimentId: exp.experiment_id,
+        name: exp.name,
+        artifactLocation: exp.artifact_location,
+        lifecycleStage: exp.lifecycle_stage,
+        tags: exp.tags ? JSON.stringify(exp.tags) : undefined
+      }))
+    } catch (error) {
+      console.error('Failed to fetch MLflow experiments:', error)
+      throw new Error(`Failed to fetch experiments: ${error.message}`)
+    }
+  }
+
+  /**
+   * Get experiment by ID
+   */
+  @Query(returns => MLflowExperiment, {
+    description: 'Get MLflow experiment by ID',
+    nullable: true
+  })
+  @Directive('@privilege(category: "mlflow", privilege: "query")')
+  async mlflowExperiment(
+    @Arg('experimentId') experimentId: string,
+    @Ctx() context: ResolverContext
+  ): Promise<MLflowExperiment | null> {
+    try {
+      const response = await mlflowApi.getExperiment(experimentId)
+
+      return {
+        experimentId: response.experiment.experiment_id,
+        name: response.experiment.name,
+        artifactLocation: response.experiment.artifact_location,
+        lifecycleStage: response.experiment.lifecycle_stage,
+        tags: response.experiment.tags ? JSON.stringify(response.experiment.tags) : undefined
+      }
+    } catch (error) {
+      console.error(`Failed to fetch experiment ${experimentId}:`, error)
+      return null
+    }
+  }
+
+  /**
+   * Create new experiment
+   */
+  @Mutation(returns => MLflowExperiment, {
+    description: 'Create new MLflow experiment'
+  })
+  @Directive('@privilege(category: "mlflow", privilege: "mutation")')
+  async createMLflowExperiment(
+    @Arg('input') input: CreateExperimentInput,
+    @Ctx() context: ResolverContext
+  ): Promise<MLflowExperiment> {
+    try {
+      const tags = input.tags ? JSON.parse(input.tags) : undefined
+
+      const response = await mlflowApi.createExperiment({
+        name: input.name,
+        artifact_location: input.artifactLocation,
+        tags
+      })
+
+      // Fetch created experiment
+      const experiment = await mlflowApi.getExperiment(response.experiment_id)
+
+      return {
+        experimentId: experiment.experiment.experiment_id,
+        name: experiment.experiment.name,
+        artifactLocation: experiment.experiment.artifact_location,
+        lifecycleStage: experiment.experiment.lifecycle_stage,
+        tags: experiment.experiment.tags ? JSON.stringify(experiment.experiment.tags) : undefined
+      }
+    } catch (error) {
+      console.error('Failed to create MLflow experiment:', error)
+      throw new Error(`Failed to create experiment: ${error.message}`)
+    }
+  }
+
+  /**
+   * Delete experiment
+   */
+  @Mutation(returns => Boolean, {
+    description: 'Delete MLflow experiment'
+  })
+  @Directive('@privilege(category: "mlflow", privilege: "mutation")')
+  async deleteMLflowExperiment(
+    @Arg('experimentId') experimentId: string,
+    @Ctx() context: ResolverContext
+  ): Promise<boolean> {
+    try {
+      await mlflowApi.deleteExperiment(experimentId)
+      return true
+    } catch (error) {
+      console.error(`Failed to delete experiment ${experimentId}:`, error)
+      throw new Error(`Failed to delete experiment: ${error.message}`)
+    }
+  }
+}