MLflow

Integrate MLflow observability with PraisonAI agents

MLflow Integration

MLflow provides experiment tracking and model management.

Setup

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

Usage

Auto-Instrumentation (Recommended)

from praisonai_tools.observability import obs
from praisonaiagents import Agent

obs.init(provider="mlflow")

# Everything is auto-traced!
agent = Agent(name="Assistant", instructions="You are helpful.", model="gpt-4o-mini")
response = agent.chat("Hello!")

Explicit Tracing

from praisonai_tools.observability import obs

obs.init(provider="mlflow", auto_instrument=False)

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