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

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@things-factory/integration-mlflow",
-  "version": "9.1.19",
+  "version": "10.0.0-beta.1",
   "main": "dist-server/index.js",
   "browser": "dist-client/index.js",
   "things-factory": true,
@@ -26,12 +26,12 @@
     "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"
+    "@operato/graphql": "^10.0.0-beta.1",
+    "@operato/layout": "^10.0.0-beta.1",
+    "@things-factory/auth-base": "^10.0.0-beta.1",
+    "@things-factory/env": "^10.0.0-beta.1",
+    "@things-factory/menu-base": "^10.0.0-beta.1",
+    "@things-factory/shell": "^10.0.0-beta.1"
   },
   "keywords": [
     "things-factory",
@@ -41,5 +41,5 @@
     "model-registry",
     "machine-learning"
   ],
-  "gitHead": "078438034dbe19915108e89ff24024f7044a85a9"
+  "gitHead": "90f40bf160fa25edf4560d7893cfd576cf474411"
 }

package/client/tsconfig.json

@@ -1,11 +0,0 @@
-{
-  "extends": "../../tsconfig-base.json",
-  "compilerOptions": {
-    "strict": true,
-    "declaration": true,
-    "module": "esnext",
-    "outDir": "../dist-client",
-    "baseUrl": "./"
-  },
-  "include": ["./**/*"]
-}

package/server/index.ts

@@ -1,5 +0,0 @@
-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

@@ -1,13 +0,0 @@
-/**
- * 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

@@ -1,117 +0,0 @@
-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}`)
-    }
-  }
-}

package/server/service/index.ts

@@ -1,17 +0,0 @@
-import { ExperimentService } from './experiment/experiment-service'
-import { RunService } from './run/run-service'
-import { ModelRegistryService } from './model-registry/model-registry-service'
-
-export const entities = []
-
-export const schema = {
-  resolverClasses: [
-    ExperimentService,
-    RunService,
-    ModelRegistryService
-  ]
-}
-
-export * from './experiment/experiment-service'
-export * from './run/run-service'
-export * from './model-registry/model-registry-service'

package/server/service/model-registry/model-registry-service.ts

@@ -1,232 +0,0 @@
-import { Resolver, Query, Mutation, Arg, Ctx, Directive } from 'type-graphql'
-import { mlflowApi } from '../../utils/mlflow-api-client'
-import {
-  MLflowRegisteredModel,
-  MLflowModelVersion,
-  CreateRegisteredModelInput,
-  CreateModelVersionInput,
-  TransitionStageInput
-} from '../../types/mlflow-types'
-
-@Resolver()
-export class ModelRegistryService {
-  /**
-   * Get all registered models
-   */
-  @Query(returns => [MLflowRegisteredModel], {
-    description: 'Get all registered models in MLflow'
-  })
-  @Directive('@privilege(category: "mlflow", privilege: "query")')
-  async mlflowRegisteredModels(@Ctx() context: ResolverContext): Promise<MLflowRegisteredModel[]> {
-    try {
-      const response = await 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
-   */
-  @Query(returns => MLflowRegisteredModel, {
-    description: 'Get registered model by name',
-    nullable: true
-  })
-  @Directive('@privilege(category: "mlflow", privilege: "query")')
-  async mlflowRegisteredModel(
-    @Arg('name') name: string,
-    @Ctx() context: ResolverContext
-  ): Promise<MLflowRegisteredModel | null> {
-    try {
-      const response = await 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
-   */
-  @Mutation(returns => MLflowRegisteredModel, {
-    description: 'Create new registered model'
-  })
-  @Directive('@privilege(category: "mlflow", privilege: "mutation")')
-  async createMLflowRegisteredModel(
-    @Arg('input') input: CreateRegisteredModelInput,
-    @Ctx() context: ResolverContext
-  ): Promise<MLflowRegisteredModel> {
-    try {
-      const tags = input.tags ? JSON.parse(input.tags) : undefined
-
-      const response = await 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
-   */
-  @Query(returns => MLflowModelVersion, {
-    description: 'Get model version',
-    nullable: true
-  })
-  @Directive('@privilege(category: "mlflow", privilege: "query")')
-  async mlflowModelVersion(
-    @Arg('name') name: string,
-    @Arg('version') version: string,
-    @Ctx() context: ResolverContext
-  ): Promise<MLflowModelVersion | null> {
-    try {
-      const response = await 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
-   */
-  @Mutation(returns => MLflowModelVersion, {
-    description: 'Create new model version'
-  })
-  @Directive('@privilege(category: "mlflow", privilege: "mutation")')
-  async createMLflowModelVersion(
-    @Arg('input') input: CreateModelVersionInput,
-    @Ctx() context: ResolverContext
-  ): Promise<MLflowModelVersion> {
-    try {
-      const tags = input.tags ? JSON.parse(input.tags) : undefined
-
-      const response = await 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
-   */
-  @Mutation(returns => MLflowModelVersion, {
-    description: 'Transition model version to target stage (None, Staging, Production, Archived)'
-  })
-  @Directive('@privilege(category: "mlflow", privilege: "mutation")')
-  async transitionMLflowModelStage(
-    @Arg('input') input: TransitionStageInput,
-    @Ctx() context: ResolverContext
-  ): Promise<MLflowModelVersion> {
-    try {
-      const response = await mlflowApi.transitionModelVersionStage({
-        name: input.name,
-        version: input.version,
-        stage: input.stage as any,
-        archive_existing_versions: input.archiveExistingVersions
-      })
-
-      // Fetch updated version
-      const version = await 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
-   */
-  @Mutation(returns => Boolean, {
-    description: 'Delete model version'
-  })
-  @Directive('@privilege(category: "mlflow", privilege: "mutation")')
-  async deleteMLflowModelVersion(
-    @Arg('name') name: string,
-    @Arg('version') version: string,
-    @Ctx() context: ResolverContext
-  ): Promise<boolean> {
-    try {
-      await 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}`)
-    }
-  }
-}

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

@@ -1,177 +0,0 @@
-import { Resolver, Query, Mutation, Arg, Ctx, Directive } from 'type-graphql'
-import { User } from '@things-factory/auth-base'
-import { mlflowApi } from '../../utils/mlflow-api-client'
-import {
-  MLflowRun,
-  CreateRunInput,
-  LogMetricInput,
-  LogParamInput,
-  UpdateRunInput,
-  SearchRunsInput
-} from '../../types/mlflow-types'
-
-@Resolver()
-export class RunService {
-  /**
-   * Get run by ID
-   */
-  @Query(returns => MLflowRun, {
-    description: 'Get MLflow run by ID',
-    nullable: true
-  })
-  @Directive('@privilege(category: "mlflow", privilege: "query")')
-  async mlflowRun(@Arg('runId') runId: string, @Ctx() context: ResolverContext): Promise<MLflowRun | null> {
-    try {
-      const response = await 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
-   */
-  @Query(returns => [MLflowRun], {
-    description: 'Search MLflow runs'
-  })
-  @Directive('@privilege(category: "mlflow", privilege: "query")')
-  async searchMLflowRuns(@Arg('input') input: SearchRunsInput, @Ctx() context: ResolverContext): Promise<MLflowRun[]> {
-    try {
-      const response = await 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
-   */
-  @Mutation(returns => MLflowRun, {
-    description: 'Create new MLflow run'
-  })
-  @Directive('@privilege(category: "mlflow", privilege: "mutation")')
-  async createMLflowRun(@Arg('input') input: CreateRunInput, @Ctx() context: ResolverContext): Promise<MLflowRun> {
-    try {
-      const tags = input.tags ? JSON.parse(input.tags) : undefined
-
-      const response = await 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
-   */
-  @Mutation(returns => Boolean, {
-    description: 'Log metric to MLflow run'
-  })
-  @Directive('@privilege(category: "mlflow", privilege: "mutation")')
-  async logMLflowMetric(@Arg('input') input: LogMetricInput, @Ctx() context: ResolverContext): Promise<boolean> {
-    try {
-      await 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
-   */
-  @Mutation(returns => Boolean, {
-    description: 'Log parameter to MLflow run'
-  })
-  @Directive('@privilege(category: "mlflow", privilege: "mutation")')
-  async logMLflowParam(@Arg('input') input: LogParamInput, @Ctx() context: ResolverContext): Promise<boolean> {
-    try {
-      await 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
-   */
-  @Mutation(returns => Boolean, {
-    description: 'Update MLflow run status'
-  })
-  @Directive('@privilege(category: "mlflow", privilege: "mutation")')
-  async updateMLflowRun(@Arg('input') input: UpdateRunInput, @Ctx() context: ResolverContext): Promise<boolean> {
-    try {
-      await mlflowApi.updateRun({
-        run_id: input.runId,
-        status: input.status as any,
-        end_time: input.endTime
-      })
-      return true
-    } catch (error) {
-      console.error('Failed to update run:', error)
-      throw new Error(`Failed to update run: ${error.message}`)
-    }
-  }
-}

package/server/tsconfig.json

@@ -1,11 +0,0 @@
-{
-  "extends": "../../tsconfig-base.json",
-  "compilerOptions": {
-    "strict": false,
-    "declaration": true,
-    "module": "commonjs",
-    "outDir": "../dist-server",
-    "baseUrl": "./"
-  },
-  "include": ["./**/*"]
-}