0d3cb498a3
CI / Shell Format Check (push) Has been cancelled
CI / Check Ruby (3.4) (push) Has been cancelled
CI / CI Config (push) Has been cancelled
CI / Test on Node ${{ matrix.node }} and ${{ matrix.os }}${{ matrix.shard && format(' (shard {0}/3)', matrix.shard) || '' }} (push) Has been cancelled
CI / Build on Node ${{ matrix.node }} (push) Has been cancelled
CI / Style Check (push) Has been cancelled
CI / Generate Assets (push) Has been cancelled
CI / Check Python (3.14) (push) Has been cancelled
CI / Check Python (3.9) (push) Has been cancelled
CI / Build Docs (push) Has been cancelled
CI / Code Scan Action (push) Has been cancelled
CI / Site tests (push) Has been cancelled
CI / webui tests (push) Has been cancelled
CI / Run Integration Tests (push) Has been cancelled
CI / Run Smoke Tests (push) Has been cancelled
CI / Go Tests (push) Has been cancelled
CI / Share Test (push) Has been cancelled
CI / Redteam (Production API) (push) Has been cancelled
CI / Redteam (Staging API) (push) Has been cancelled
CI / GitHub Actions Lint (push) Has been cancelled
CI / Check Ruby (3.0) (push) Has been cancelled
release-please / release-please (push) Has been cancelled
release-please / build (push) Has been cancelled
release-please / publish-npm (push) Has been cancelled
release-please / publish-npm-backfill (push) Has been cancelled
release-please / docker (push) Has been cancelled
release-please / publish-code-scan-action (push) Has been cancelled
release-please / attest-code-scan-action (push) Has been cancelled
Deploy local.promptfoo.app / Deploy to Cloudflare Pages (push) Has been cancelled
Test and Publish Multi-arch Docker Image / test (push) Has been cancelled
Test and Publish Multi-arch Docker Image / build-docker-and-push-digests (map[digest-suffix:linux-amd64 platform:linux/amd64 runner:ubuntu-latest]) (push) Has been cancelled
Test and Publish Multi-arch Docker Image / build-docker-and-push-digests (map[digest-suffix:linux-arm64 platform:linux/arm64 runner:ubuntu-24.04-arm]) (push) Has been cancelled
Test and Publish Multi-arch Docker Image / merge-docker-digests (push) Has been cancelled
Test and Publish Multi-arch Docker Image / Attest Multi-arch Image (push) Has been cancelled
Validate Renovate Config / Validate Renovate Configuration (push) Has been cancelled
40 lines
1.0 KiB
Markdown
40 lines
1.0 KiB
Markdown
# provider-mlflow-gateway (MLflow AI Gateway)
|
|
|
|
This example demonstrates how to use [MLflow AI Gateway](https://mlflow.org/docs/latest/genai/governance/ai-gateway/) as an LLM provider in promptfoo.
|
|
|
|
To get started:
|
|
|
|
```bash
|
|
npx promptfoo@latest init --example provider-mlflow-gateway
|
|
```
|
|
|
|
## Setup
|
|
|
|
1. Install and start MLflow:
|
|
|
|
```bash
|
|
pip install mlflow[genai]
|
|
mlflow server --host 127.0.0.1 --port 5000
|
|
```
|
|
|
|
2. Create a gateway endpoint in the MLflow UI at http://localhost:5000 (AI Gateway → Create Endpoint).
|
|
|
|
3. Set environment variables:
|
|
|
|
```bash
|
|
export MLFLOW_GATEWAY_URL=http://localhost:5000
|
|
```
|
|
|
|
4. Run the evaluation:
|
|
|
|
```bash
|
|
promptfoo eval -c promptfooconfig.yaml
|
|
```
|
|
|
|
## Configuration
|
|
|
|
Update `my-chat-endpoint` in `promptfooconfig.yaml` with the name of the gateway endpoint you created.
|
|
The example also uses that endpoint as the `llm-rubric` grader, so it runs without a separate OpenAI API key.
|
|
|
|
See the [MLflow Gateway provider docs](https://www.promptfoo.dev/docs/providers/mlflow-gateway/) for all configuration options.
|