Skip to main content

MLflow Integration

MLflow provides experiment tracking and model management.

Setup

pip install mlflow
export MLFLOW_TRACKING_URI=http://localhost:5000

Usage

from praisonai_tools.observability import obs

obs.init(provider="mlflow")

with obs.trace("experiment"):
    # Your agent code
    pass