@things-factory/integration-mlflow 9.1.19 → 10.0.0-beta.1

package/server/types/mlflow-types.ts

@@ -1,235 +0,0 @@
-import { Field, ObjectType, InputType, Int } from 'type-graphql'
-
-// ============================================================================
-// Experiment Types
-// ============================================================================
-
-@ObjectType({ description: 'MLflow Experiment' })
-export class MLflowExperiment {
-  @Field({ description: 'Experiment ID' })
-  experimentId: string
-
-  @Field({ description: 'Experiment name' })
-  name: string
-
-  @Field({ description: 'Artifact location' })
-  artifactLocation: string
-
-  @Field({ description: 'Lifecycle stage' })
-  lifecycleStage: string
-
-  @Field(type => String, { nullable: true, description: 'Tags as JSON string' })
-  tags?: string
-}
-
-@InputType({ description: 'Input for creating MLflow experiment' })
-export class CreateExperimentInput {
-  @Field({ description: 'Experiment name' })
-  name: string
-
-  @Field({ nullable: true, description: 'Artifact location' })
-  artifactLocation?: string
-
-  @Field({ nullable: true, description: 'Tags as JSON string' })
-  tags?: string
-}
-
-// ============================================================================
-// Run Types
-// ============================================================================
-
-@ObjectType({ description: 'MLflow Run' })
-export class MLflowRun {
-  @Field({ description: 'Run ID' })
-  runId: string
-
-  @Field({ description: 'Experiment ID' })
-  experimentId: string
-
-  @Field({ description: 'Run status' })
-  status: string
-
-  @Field(type => Int, { description: 'Start time (Unix timestamp)' })
-  startTime: number
-
-  @Field(type => Int, { nullable: true, description: 'End time (Unix timestamp)' })
-  endTime?: number
-
-  @Field({ description: 'Artifact URI' })
-  artifactUri: string
-
-  @Field(type => String, { nullable: true, description: 'Metrics as JSON string' })
-  metrics?: string
-
-  @Field(type => String, { nullable: true, description: 'Params as JSON string' })
-  params?: string
-
-  @Field(type => String, { nullable: true, description: 'Tags as JSON string' })
-  tags?: string
-}
-
-@InputType({ description: 'Input for creating MLflow run' })
-export class CreateRunInput {
-  @Field({ description: 'Experiment ID' })
-  experimentId: string
-
-  @Field(type => Int, { nullable: true, description: 'Start time' })
-  startTime?: number
-
-  @Field({ nullable: true, description: 'Tags as JSON string' })
-  tags?: string
-}
-
-@InputType({ description: 'Input for logging metric' })
-export class LogMetricInput {
-  @Field({ description: 'Run ID' })
-  runId: string
-
-  @Field({ description: 'Metric key' })
-  key: string
-
-  @Field({ description: 'Metric value' })
-  value: number
-
-  @Field(type => Int, { nullable: true, description: 'Timestamp' })
-  timestamp?: number
-
-  @Field(type => Int, { nullable: true, description: 'Step' })
-  step?: number
-}
-
-@InputType({ description: 'Input for logging parameter' })
-export class LogParamInput {
-  @Field({ description: 'Run ID' })
-  runId: string
-
-  @Field({ description: 'Parameter key' })
-  key: string
-
-  @Field({ description: 'Parameter value' })
-  value: string
-}
-
-@InputType({ description: 'Input for updating run' })
-export class UpdateRunInput {
-  @Field({ description: 'Run ID' })
-  runId: string
-
-  @Field({ nullable: true, description: 'Status' })
-  status?: string
-
-  @Field(type => Int, { nullable: true, description: 'End time' })
-  endTime?: number
-}
-
-// ============================================================================
-// Model Registry Types
-// ============================================================================
-
-@ObjectType({ description: 'MLflow Registered Model' })
-export class MLflowRegisteredModel {
-  @Field({ description: 'Model name' })
-  name: string
-
-  @Field(type => Int, { description: 'Creation timestamp' })
-  creationTimestamp: number
-
-  @Field(type => Int, { description: 'Last updated timestamp' })
-  lastUpdatedTimestamp: number
-
-  @Field({ nullable: true, description: 'Model description' })
-  description?: string
-
-  @Field(type => String, { nullable: true, description: 'Tags as JSON string' })
-  tags?: string
-
-  @Field(type => String, { nullable: true, description: 'Latest versions as JSON string' })
-  latestVersions?: string
-}
-
-@InputType({ description: 'Input for creating registered model' })
-export class CreateRegisteredModelInput {
-  @Field({ description: 'Model name' })
-  name: string
-
-  @Field({ nullable: true, description: 'Model description' })
-  description?: string
-
-  @Field({ nullable: true, description: 'Tags as JSON string' })
-  tags?: string
-}
-
-@ObjectType({ description: 'MLflow Model Version' })
-export class MLflowModelVersion {
-  @Field({ description: 'Model name' })
-  name: string
-
-  @Field({ description: 'Version number' })
-  version: string
-
-  @Field(type => Int, { description: 'Creation timestamp' })
-  creationTimestamp: number
-
-  @Field({ description: 'Current stage' })
-  currentStage: string
-
-  @Field({ description: 'Model source' })
-  source: string
-
-  @Field({ description: 'Version status' })
-  status: string
-
-  @Field({ nullable: true, description: 'Run ID' })
-  runId?: string
-
-  @Field({ nullable: true, description: 'Description' })
-  description?: string
-}
-
-@InputType({ description: 'Input for creating model version' })
-export class CreateModelVersionInput {
-  @Field({ description: 'Model name' })
-  name: string
-
-  @Field({ description: 'Model source URI' })
-  source: string
-
-  @Field({ nullable: true, description: 'Run ID' })
-  runId?: string
-
-  @Field({ nullable: true, description: 'Description' })
-  description?: string
-
-  @Field({ nullable: true, description: 'Tags as JSON string' })
-  tags?: string
-}
-
-@InputType({ description: 'Input for transitioning model version stage' })
-export class TransitionStageInput {
-  @Field({ description: 'Model name' })
-  name: string
-
-  @Field({ description: 'Version number' })
-  version: string
-
-  @Field({ description: 'Target stage (None, Staging, Production, Archived)' })
-  stage: string
-
-  @Field({ nullable: true, description: 'Archive existing versions in target stage' })
-  archiveExistingVersions?: boolean
-}
-
-@InputType({ description: 'Input for searching runs' })
-export class SearchRunsInput {
-  @Field(type => [String], { description: 'Experiment IDs to search' })
-  experimentIds: string[]
-
-  @Field({ nullable: true, description: 'Filter string' })
-  filter?: string
-
-  @Field(type => Int, { nullable: true, description: 'Max results' })
-  maxResults?: number
-
-  @Field(type => [String], { nullable: true, description: 'Order by' })
-  orderBy?: string[]
-}

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

@@ -1,375 +0,0 @@
-/**
- * MLflow REST API Client
- *
- * Thin wrapper around MLflow REST API for Things-Factory integration
- * Reference: https://mlflow.org/docs/latest/rest-api.html
- */
-
-import { config } from '@things-factory/env'
-
-const MLFLOW_TRACKING_URI = config.get('mlflowTrackingUri', process.env.MLFLOW_TRACKING_URI || 'http://localhost:5000')
-
-interface MLflowResponse<T> {
-  data?: T
-  error?: string
-}
-
-class MLflowApiClient {
-  private baseUrl: string
-
-  constructor(baseUrl: string = MLFLOW_TRACKING_URI) {
-    this.baseUrl = baseUrl.replace(/\/$/, '') // Remove trailing slash
-  }
-
-  /**
-   * Generic HTTP request wrapper
-   */
-  private async request<T>(endpoint: string, options: RequestInit = {}): Promise<T> {
-    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: {
-    name: string
-    artifact_location?: string
-    tags?: Array<{ key: string; value: string }>
-  }): Promise<{ experiment_id: string }> {
-    return this.request('/api/2.0/mlflow/experiments/create', {
-      method: 'POST',
-      body: JSON.stringify(params)
-    })
-  }
-
-  /**
-   * Get experiment by ID
-   */
-  async getExperiment(experimentId: string): Promise<{
-    experiment: {
-      experiment_id: string
-      name: string
-      artifact_location: string
-      lifecycle_stage: string
-      tags?: Array<{ key: string; value: string }>
-    }
-  }> {
-    return this.request(`/api/2.0/mlflow/experiments/get?experiment_id=${experimentId}`)
-  }
-
-  /**
-   * Search experiments
-   */
-  async 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 }>
-    }>
-  }> {
-    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: string): Promise<void> {
-    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: {
-    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
-    }
-  }> {
-    return this.request('/api/2.0/mlflow/runs/create', {
-      method: 'POST',
-      body: JSON.stringify(params)
-    })
-  }
-
-  /**
-   * Get run by ID
-   */
-  async 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 }>
-      }
-    }
-  }> {
-    return this.request(`/api/2.0/mlflow/runs/get?run_id=${runId}`)
-  }
-
-  /**
-   * Search runs
-   */
-  async 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
-    }>
-  }> {
-    return this.request('/api/2.0/mlflow/runs/search', {
-      method: 'POST',
-      body: JSON.stringify(params)
-    })
-  }
-
-  /**
-   * Log metric for a run
-   */
-  async logMetric(params: {
-    run_id: string
-    key: string
-    value: number
-    timestamp?: number
-    step?: number
-  }): Promise<void> {
-    return this.request('/api/2.0/mlflow/runs/log-metric', {
-      method: 'POST',
-      body: JSON.stringify(params)
-    })
-  }
-
-  /**
-   * Log parameter for a run
-   */
-  async logParam(params: { run_id: string; key: string; value: string }): Promise<void> {
-    return this.request('/api/2.0/mlflow/runs/log-parameter', {
-      method: 'POST',
-      body: JSON.stringify(params)
-    })
-  }
-
-  /**
-   * Update run status
-   */
-  async updateRun(params: {
-    run_id: string
-    status?: 'RUNNING' | 'SCHEDULED' | 'FINISHED' | 'FAILED' | 'KILLED'
-    end_time?: number
-  }): Promise<void> {
-    return this.request('/api/2.0/mlflow/runs/update', {
-      method: 'POST',
-      body: JSON.stringify(params)
-    })
-  }
-
-  // ========================================================================
-  // Model Registry API
-  // ========================================================================
-
-  /**
-   * Create registered model
-   */
-  async createRegisteredModel(params: {
-    name: string
-    tags?: Array<{ key: string; value: string }>
-    description?: string
-  }): Promise<{
-    registered_model: {
-      name: string
-      creation_timestamp: number
-      last_updated_timestamp: number
-    }
-  }> {
-    return this.request('/api/2.0/mlflow/registered-models/create', {
-      method: 'POST',
-      body: JSON.stringify(params)
-    })
-  }
-
-  /**
-   * Get registered model
-   */
-  async 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[]
-    }
-  }> {
-    return this.request(`/api/2.0/mlflow/registered-models/get?name=${encodeURIComponent(name)}`)
-  }
-
-  /**
-   * Search registered models
-   */
-  async 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[]
-    }>
-  }> {
-    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: {
-    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
-    }
-  }> {
-    return this.request('/api/2.0/mlflow/model-versions/create', {
-      method: 'POST',
-      body: JSON.stringify(params)
-    })
-  }
-
-  /**
-   * Get model version
-   */
-  async 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
-    }
-  }> {
-    return this.request(
-      `/api/2.0/mlflow/model-versions/get?name=${encodeURIComponent(params.name)}&version=${params.version}`
-    )
-  }
-
-  /**
-   * Transition model version stage
-   */
-  async 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
-    }
-  }> {
-    return this.request('/api/2.0/mlflow/model-versions/transition-stage', {
-      method: 'POST',
-      body: JSON.stringify(params)
-    })
-  }
-
-  /**
-   * Delete model version
-   */
-  async deleteModelVersion(params: { name: string; version: string }): Promise<void> {
-    return this.request('/api/2.0/mlflow/model-versions/delete', {
-      method: 'DELETE',
-      body: JSON.stringify(params)
-    })
-  }
-}
-
-// Export singleton instance
-export const mlflowApi = new MLflowApiClient()
-
-// Export class for testing or custom instances
-export { MLflowApiClient }