@things-factory/integration-mlflow 9.2.5 → 10.0.0-beta.10

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": ["./**/*"]
-}

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[]
-}