Skip to content
Closed
Show file tree
Hide file tree
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
49 changes: 49 additions & 0 deletions packages/components/credentials/MLflowApi.credential.ts
Original file line number Diff line number Diff line change
@@ -0,0 +1,49 @@
import { INodeParams, INodeCredential } from '../src/Interface'

class MLflowApi implements INodeCredential {
label: string
name: string
version: number
description: string
inputs: INodeParams[]

constructor() {
this.label = 'MLflow API'
this.name = 'mlflowApi'
this.version = 1.0
this.description =
'Refer to <a target="_blank" href="https://mlflow.org/docs/latest/index.html">official guide</a> on how to set up MLflow tracking server.'
this.inputs = [
{
label: 'Tracking URI',
name: 'mlflowTrackingUri',
type: 'string',
description: 'MLflow tracking server URI (e.g., http://localhost:5000)',
placeholder: 'http://localhost:5000'
},
{
label: 'API Token',
name: 'mlflowApiToken',
type: 'password',
optional: true,
description: 'API token for authenticated MLflow servers (e.g., Databricks, managed MLflow)'
},
{
label: 'Username',
name: 'mlflowUsername',
type: 'string',
optional: true,
description: 'Username for basic auth (if required)'
},
{
label: 'Password',
name: 'mlflowPassword',
type: 'password',
optional: true,
description: 'Password for basic auth (if required)'
}
]
}
}

module.exports = { credClass: MLflowApi }
33 changes: 33 additions & 0 deletions packages/components/nodes/analytic/MLflow/MLflow.ts
Original file line number Diff line number Diff line change
@@ -0,0 +1,33 @@
import { INode, INodeParams } from '../../../src/Interface'

class MLflow_Analytic implements INode {
label: string
name: string
version: number
description: string
type: string
icon: string
category: string
baseClasses: string[]
inputs?: INodeParams[]
credential: INodeParams

constructor() {
this.label = 'MLflow'
this.name = 'mlflow'
this.version = 1.0
this.type = 'MLflow'
this.icon = 'mlflow.svg'
this.category = 'Analytic'
this.baseClasses = [this.type]
this.inputs = []
this.credential = {
label: 'Connect Credential',
name: 'credential',
type: 'credential',
credentialNames: ['mlflowApi']
}
}
}

module.exports = { nodeClass: MLflow_Analytic }
17 changes: 17 additions & 0 deletions packages/components/nodes/analytic/MLflow/mlflow.svg
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
1 change: 1 addition & 0 deletions packages/components/package.json
Original file line number Diff line number Diff line change
Expand Up @@ -136,6 +136,7 @@
"lunary": "^0.7.12",
"mammoth": "^1.5.1",
"meilisearch": "^0.41.0",
"mlflow-tracing": "^0.1.2",
Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

medium

The mlflow-tracing dependency is added but it doesn't appear to be used. The MLflow integration is implemented via a custom MLflowClient that uses the REST API directly. To keep dependencies clean and avoid unnecessary package bloat, this unused dependency should be removed.

"moment": "^2.29.3",
"mongodb": "6.3.0",
"mysql2": "^3.11.3",
Expand Down
Loading