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

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!")

from praisonai_tools.observability import obs

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

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