databricks-mlflow-evaluation

Evaluate GenAI agents with MLflow 3 scorers, traces, and judge alignment workflows.

Updated Mar 24, 2026
One-click install
npx skills add https://github.com/AarushiShah/coding-agents-databricks-apps --skill databricks-mlflow-evaluation-aarushishah
Or copy as Structured Prompt for Agent▼
Please help me install this Agent Skill.
Skill: databricks-mlflow-evaluation
Source: https://github.com/AarushiShah/coding-agents-databricks-apps/tree/main/.claude/skills/databricks-mlflow-evaluation
Command: npx skills add https://github.com/AarushiShah/coding-agents-databricks-apps --skill databricks-mlflow-evaluation-aarushishah

SYSTEM DOCUMENTATION & REQUIREMENTS

💡 This Skill requires mlflow, databricks-connect, openai, and includes references (resource) components.

What problem does it solve? Evaluating GenAI agents requires correct use of the MLflow 3 GenAI APIs, and common mistakes like wrong data formats, misconfigured scorers, or misaligned judges cause silent failures and misleading quality signals. ## Core Features & Use Cases - Evaluation Workflows: Eight guided workflows covering first-time setup, production trace datasets, regression detection, custom scorer development, and performance optimization. - Judge Alignment & Prompt Optimization: Align LLM judges with domain expert feedback using MemAlign, then run optimize_prompts() with GEPA for automated prompt improvement. - Production Monitoring: Configure Unity Catalog trace ingestion, register and start monitoring scorers with sampling rates, and query UC trace tables. - Use Case: You built a RAG agent on Databricks and need to measure groundedness and correctness before launch. Follow Workflow 1 to build an eval dataset from traces, apply RetrievalGroundedness and Correctness scorers, and compare runs to catch regressions. ## Quick Start Use the databricks-mlflow-evaluation skill to write an mlflow.genai.evaluate() script with Guidelines and Correctness scorers for my agent.

Frequently Asked Questions about databricks-mlflow-evaluation

High-intent search queries and answers about installing and using this skill.

FAQPage Schema
How do I evaluate a GenAI agent with MLflow 3?▼

Use mlflow.genai.evaluate() with a dataset containing an inputs key per record, a predict_fn that receives unpacked keyword arguments, and a list of scorers such as Guidelines, Correctness, or Safety. Do not use the older mlflow.evaluate() API.

How do I create a custom scorer in MLflow GenAI?▼

Decorate a function with @scorer from mlflow.genai.scorers and return a boolean, float, string, or Feedback object. The function can accept inputs, outputs, expectations, and the full trace object for deeper analysis.

Why does the Correctness scorer fail on my evaluation data?▼

Correctness requires ground truth in each record's expectations field, either expected_facts or expected_response. Without expectations, the scorer has nothing to compare the agent output against and will fail.

Does RetrievalGroundedness work without a retriever span?▼

No. RetrievalGroundedness requires a span marked with span_type="RETRIEVER" in the trace. Decorate your retrieval function with @mlflow.trace(span_type="RETRIEVER") so the scorer can find the retrieved documents.

Why did my judge scores drop after MemAlign alignment?▼

Lower scores after alignment are expected and indicate the judge now applies domain-expert standards instead of generic criteria. The unaligned judge was likely giving inflated scores, so the lower aligned score is a more accurate signal.

What MLflow version is required for Unity Catalog trace ingestion?▼

Unity Catalog trace ingestion requires mlflow[databricks]>=3.9.0, while optimize_prompts() requires MLflow 3.5.0 or later. You must also set MLFLOW_TRACING_SQL_WAREHOUSE_ID before linking a UC schema to an experiment.