mlflow-assistant
is an MLflow plugin that enables natural language conversations with your MLflow server using LLM providers like OpenAI and Ollama.
- Interact with your MLflow server using natural language.
- Powered by large language models (LLMs).
- Easy integration with MLflow.
To install the package, use:
pip install mlflow-assistant
- Python >= 3.9
- MLflow >= 2.21.0, < 3.0.0
You can use the get_mlflow_client
function to initialize an MLflow client:
from mlflow_assistant.core.core import get_mlflow_client
client = get_mlflow_client()
print(client)