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.