kienbui1995/ai-harness

AI Harness

When to Use

• Setting up automated evaluation pipeline for an LLM feature
• Catching quality regressions before they reach production
• Building a continuous eval CI/CD system
• Creating golden datasets for systematic testing
• Comparing prompt versions or model upgrades systematically

Core Components of an AI Harness

1. Golden Dataset Design

• Representative inputs covering: happy path, edge cases, adversarial, real user examples
• Each test case: input + expected output + evaluation criteria
• Minimum 50-100 cases before automated eval is meaningful
• Label types: exact match, semantic similarity, rubric-based, human preference
• Dataset versioning (git or DVC) — same discipline as code

2. Evaluation Layers

Three layers for comprehensive coverage:

Layer 1: Deterministic checks (fast, cheap)
  - Output schema validation (JSON structure, required fields)
  - Length constraints (too short/long)
  - Keyword presence/absence
  - PII detection in outputs

Layer 2: Model-as-judge (flexible, moderate cost)
  - Use GPT-4/Claude to score outputs on rubric
  - Criteria: accuracy, helpfulness, safety, hallucination
  - Score 1-5 with reasoning
  - Compare to baseline (previous prompt/model)

Layer 3: Human review (ground truth, expensive)
  - Sample 5-10% of outputs weekly
  - Focus on borderline model-as-judge scores (2-3 out of 5)
  - Use disagreements to improve rubric

3. CI Integration

# .github/workflows/ai-eval.yml
name: AI Eval
on: [push, pull_request]
jobs:
  eval:
    steps:
      - name: Run eval harness
        run: python eval/run_harness.py
        env:
          OPENAI_API_KEY: ${{ secrets.OPENAI_API_KEY }}
      - name: Check pass rate
        run: python eval/check_thresholds.py --min-pass-rate 0.85
      - name: Upload results
        uses: actions/upload-artifact@v3
        with:
          name: eval-results
          path: eval/results/

4. Regression Detection

• Baseline: capture current pass rates per category
• Alert when: pass rate drops >5% vs baseline, any critical test fails, cost increases >20%
• Track over time: eval results → time-series DB → dashboard

5. Prompt Version Testing

# A/B test two prompts against golden dataset
results = {
    "prompt_v1": run_eval(dataset, prompt_v1),
    "prompt_v2": run_eval(dataset, prompt_v2),
}
# Compare: accuracy, cost, latency, safety
winner = select_winner(results, primary_metric="accuracy", cost_constraint=1.2)

6. Tools and Frameworks

| Tool | Use case | |------|---------| | LangSmith | Tracing + eval for LangChain apps | | Braintrust | Eval platform, dataset management, CI integration | | PromptFoo | Open-source prompt testing, CLI-first | | Weights & Biases | Experiment tracking, model comparison | | Ragas | RAG-specific evaluation (faithfulness, answer relevance) | | pytest + custom | Lightweight eval for simple use cases |

Core Jobs

1. Golden dataset — curate representative test cases (happy path, edge cases, adversarial, real user queries)
2. Eval layers — deterministic checks → model-as-judge → human review
3. CI integration — run eval on every PR; block merge on critical failures
4. Regression detection — baseline pass rates; alert on >5% drop by category
5. Prompt A/B testing — compare prompt versions against dataset; select winner by metric

Key Outputs

• Eval report: pass rate by category + comparison vs baseline
• CI check: green/red with failure breakdown
• Dataset: versioned JSONL with input/expected/criteria
• Regression alert: which category dropped and by how much

Anti-Patterns

• Evaluating only on training distribution (test on real user queries)
• Using BLEU/ROUGE for open-ended generation (meaningless for chat)
• No baseline — "better" means nothing without comparison
• Running eval only before major releases (too late to catch drift)
• One aggregated score hiding category regressions

Statistical Rigor

Avoid declaring "improvement" without statistical evidence:

from scipy import stats

def is_statistically_significant(baseline_scores, new_scores, alpha=0.05):
    """Two-sample t-test for eval score comparison"""
    t_stat, p_value = stats.ttest_ind(baseline_scores, new_scores)
    
    effect_size = (mean(new_scores) - mean(baseline_scores)) / std(baseline_scores)
    
    return StatResult(
        significant=p_value < alpha,
        p_value=p_value,
        effect_size=effect_size,  # Cohen's d
        practical_significant=abs(effect_size) > 0.2,  # small effect
        recommendation="ship" if (p_value < alpha and effect_size > 0.2) else "no change"
    )

# Minimum sample sizes for reliable conclusions:
# Small effect (d=0.2): n ≥ 197 per group
# Medium effect (d=0.5): n ≥ 52 per group
# Large effect (d=0.8): n ≥ 26 per group

Key rules:

• Never compare single-run scores — always run 5-10 times minimum
• Report confidence intervals, not just averages
• Distinguish statistical significance (p<0.05) from practical significance (effect size matters)
• A 1% quality improvement usually isn't worth shipping complexity

Dataset Maintenance

Golden datasets degrade — keep them fresh:

class EvalDatasetManager:
    def review_test_cases(self, dataset: Dataset, threshold_days=30):
        stale = [tc for tc in dataset if tc.last_reviewed_days > threshold_days]
        low_signal = [tc for tc in dataset if tc.pass_rate in (0.0, 1.0)]
        # Pass rate 0% = always fails (broken test or impossible), 1% = always passes (too easy)
        
        return DatasetHealthReport(
            stale_count=len(stale),
            low_signal_count=len(low_signal),
            action="review_and_update" if len(stale) + len(low_signal) > len(dataset) * 0.2 else "ok"
        )
    
    def add_from_production_failures(self, prod_failures: list[Failure]):
        """Convert production failures into eval test cases"""
        for failure in prod_failures:
            if failure.confirmed_bug:  # human verified
                dataset.add(TestCase(
                    input=failure.input,
                    expected=failure.expected_output,
                    source="production_failure",
                    added_date=today()
                ))

Dataset health signals:

• 20% of tests always pass → too easy, add harder cases

• 10% of tests never pass → broken tests or capability gap, investigate

• Last reviewed >30 days → likely stale, review against recent prod distribution

Integration

• Use with llm-evaluation (frameworks) and llm-observability (production monitoring)
• Use ai-safety-guardrails to add safety checks as eval layer
• CI eval catches regressions; observability catches production drift