use crate::cache::vcr::VcrCache;
use crate::metrics_api::Metric;
use crate::model::{EvalConfig, LlmResponse, TestCase, TestResultRow, TestStatus};
use crate::providers::llm::LlmClient;
use crate::quarantine::QuarantineMode;
use crate::report::progress::ProgressSink;
use crate::report::RunArtifacts;
use crate::storage::store::Store;
use std::sync::Arc;
#[path = "runner_next/mod.rs"]
mod runner_next;
#[derive(Debug, Clone)]
pub struct RunPolicy {
pub rerun_failures: u32,
pub quarantine_mode: QuarantineMode,
pub replay_strict: bool,
}
impl Default for RunPolicy {
fn default() -> Self {
Self {
rerun_failures: 1,
quarantine_mode: QuarantineMode::Warn,
replay_strict: false,
}
}
}
pub struct Runner {
pub store: Store,
pub cache: VcrCache,
pub client: Arc<dyn LlmClient>,
pub metrics: Vec<Arc<dyn Metric>>,
pub policy: RunPolicy,
pub _network_guard: Option<crate::providers::network::NetworkPolicyGuard>,
pub embedder: Option<Arc<dyn crate::providers::embedder::Embedder>>,
pub refresh_embeddings: bool,
pub incremental: bool,
pub refresh_cache: bool,
pub judge: Option<crate::judge::JudgeService>,
pub baseline: Option<crate::baseline::Baseline>,
}
impl Runner {
pub async fn run_suite(
&self,
cfg: &EvalConfig,
progress: Option<ProgressSink>,
) -> anyhow::Result<RunArtifacts> {
runner_next::execute::run_suite_impl(self, cfg, progress).await
}
fn apply_agent_assertions(
&self,
run_id: i64,
tc: &TestCase,
final_row: &mut TestResultRow,
) -> anyhow::Result<()> {
runner_next::assertions::apply_agent_assertions_impl(self, run_id, tc, final_row)
}
async fn run_test_once(
&self,
cfg: &EvalConfig,
tc: &TestCase,
) -> anyhow::Result<(TestResultRow, LlmResponse)> {
runner_next::single::run_test_once_impl(self, cfg, tc).await
}
async fn call_llm(&self, cfg: &EvalConfig, tc: &TestCase) -> anyhow::Result<LlmResponse> {
runner_next::execute::call_llm_impl(self, cfg, tc).await
}
fn check_baseline_regressions(
&self,
tc: &TestCase,
cfg: &EvalConfig,
details: &serde_json::Value,
metrics: &[Arc<dyn Metric>],
baseline: &crate::baseline::Baseline,
) -> Option<(TestStatus, String)> {
runner_next::baseline::check_baseline_regressions_impl(
self, tc, cfg, details, metrics, baseline,
)
}
async fn enrich_semantic(
&self,
_cfg: &EvalConfig,
tc: &TestCase,
resp: &mut LlmResponse,
) -> anyhow::Result<()> {
runner_next::scoring::enrich_semantic_impl(self, _cfg, tc, resp).await
}
pub async fn embed_text(
&self,
model_id: &str,
embedder: &dyn crate::providers::embedder::Embedder,
text: &str,
) -> anyhow::Result<(Vec<f32>, &'static str)> {
runner_next::cache::embed_text_impl(self, model_id, embedder, text).await
}
async fn enrich_judge(
&self,
cfg: &EvalConfig,
tc: &TestCase,
resp: &mut LlmResponse,
) -> anyhow::Result<()> {
runner_next::scoring::enrich_judge_impl(self, cfg, tc, resp).await
}
}
#[cfg(test)]
mod tests {
use super::*;
use crate::metrics_api::{Metric, MetricResult};
use crate::model::{Expected, Settings, TestInput};
use crate::on_error::ErrorPolicy;
use crate::providers::llm::fake::FakeClient;
use crate::providers::llm::LlmClient;
use async_trait::async_trait;
use std::sync::atomic::{AtomicUsize, Ordering};
#[derive(Clone, Copy)]
enum MetricMode {
FailThenPass,
AlwaysFail,
AlwaysPass,
}
struct ScriptedMetric {
mode: MetricMode,
calls: AtomicUsize,
}
impl ScriptedMetric {
fn fail_then_pass() -> Self {
Self {
mode: MetricMode::FailThenPass,
calls: AtomicUsize::new(0),
}
}
fn always_fail() -> Self {
Self {
mode: MetricMode::AlwaysFail,
calls: AtomicUsize::new(0),
}
}
fn always_pass() -> Self {
Self {
mode: MetricMode::AlwaysPass,
calls: AtomicUsize::new(0),
}
}
}
#[async_trait]
impl Metric for ScriptedMetric {
fn name(&self) -> &'static str {
"scripted"
}
async fn evaluate(
&self,
_tc: &TestCase,
_expected: &Expected,
_resp: &LlmResponse,
) -> anyhow::Result<MetricResult> {
let n = self.calls.fetch_add(1, Ordering::SeqCst);
match self.mode {
MetricMode::FailThenPass => {
if n == 0 {
Ok(MetricResult::fail(0.0, "scripted_fail_once"))
} else {
Ok(MetricResult::pass(1.0))
}
}
MetricMode::AlwaysFail => Ok(MetricResult::fail(0.0, "scripted_fail")),
MetricMode::AlwaysPass => Ok(MetricResult::pass(1.0)),
}
}
}
struct ErrorClient;
#[async_trait]
impl LlmClient for ErrorClient {
async fn complete(
&self,
_prompt: &str,
_context: Option<&[String]>,
) -> anyhow::Result<LlmResponse> {
Err(anyhow::anyhow!("scripted provider error"))
}
fn provider_name(&self) -> &'static str {
"error_client"
}
}
fn runner_for_contract_tests(
client: Arc<dyn LlmClient>,
metrics: Vec<Arc<dyn Metric>>,
rerun_failures: u32,
) -> Runner {
let store = Store::memory().expect("in-memory store");
store.init_schema().expect("schema init");
Runner {
store: store.clone(),
cache: VcrCache::new(store),
client,
metrics,
policy: RunPolicy {
rerun_failures,
quarantine_mode: QuarantineMode::Off,
replay_strict: false,
},
_network_guard: None,
embedder: None,
refresh_embeddings: false,
incremental: false,
refresh_cache: false,
judge: None,
baseline: None,
}
}
fn single_test_config(on_error: ErrorPolicy) -> EvalConfig {
EvalConfig {
version: 1,
suite: "runner-contract".to_string(),
model: "fake-model".to_string(),
settings: Settings {
parallel: Some(1),
cache: Some(false),
seed: Some(1234),
on_error,
..Default::default()
},
thresholds: Default::default(),
otel: Default::default(),
tests: vec![TestCase {
id: "t1".to_string(),
input: TestInput {
prompt: "contract prompt".to_string(),
context: None,
},
expected: Expected::MustContain {
must_contain: vec!["ok".to_string()],
},
assertions: None,
on_error: None,
tags: vec![],
metadata: None,
}],
}
}
fn config_with_test_ids(ids: &[&str], on_error: ErrorPolicy) -> EvalConfig {
EvalConfig {
version: 1,
suite: "runner-contract".to_string(),
model: "fake-model".to_string(),
settings: Settings {
parallel: Some(1),
cache: Some(false),
seed: Some(1234),
on_error,
..Default::default()
},
thresholds: Default::default(),
otel: Default::default(),
tests: ids
.iter()
.map(|id| TestCase {
id: (*id).to_string(),
input: TestInput {
prompt: format!("prompt-{id}"),
context: None,
},
expected: Expected::MustContain {
must_contain: vec!["ok".to_string()],
},
assertions: None,
on_error: None,
tags: vec![],
metadata: None,
})
.collect(),
}
}
#[tokio::test]
async fn runner_contract_flake_fail_then_pass_classified_flaky() -> anyhow::Result<()> {
let cfg = single_test_config(ErrorPolicy::Block);
let client = Arc::new(FakeClient::new("fake-model".to_string()).with_response("ok".into()));
let metric = Arc::new(ScriptedMetric::fail_then_pass());
let runner = runner_for_contract_tests(client, vec![metric], 1);
let artifacts = runner.run_suite(&cfg, None).await?;
let row = artifacts
.results
.iter()
.find(|r| r.test_id == "t1")
.expect("result for t1");
assert_eq!(row.status, TestStatus::Flaky);
assert_eq!(row.message, "flake detected (rerun passed)");
let attempts = row.attempts.as_ref().expect("attempts");
assert_eq!(attempts.len(), 2);
assert_eq!(attempts[0].status, TestStatus::Fail);
assert_eq!(attempts[1].status, TestStatus::Pass);
Ok(())
}
#[tokio::test]
async fn runner_contract_fail_after_retries_stays_fail() -> anyhow::Result<()> {
let cfg = single_test_config(ErrorPolicy::Block);
let client = Arc::new(FakeClient::new("fake-model".to_string()).with_response("ok".into()));
let metric = Arc::new(ScriptedMetric::always_fail());
let runner = runner_for_contract_tests(client, vec![metric], 1);
let artifacts = runner.run_suite(&cfg, None).await?;
let row = artifacts
.results
.iter()
.find(|r| r.test_id == "t1")
.expect("result for t1");
assert_eq!(row.status, TestStatus::Fail);
assert!(
row.message.contains("failed: scripted"),
"expected stable failure reason, got: {}",
row.message
);
let attempts = row.attempts.as_ref().expect("attempts");
assert_eq!(attempts.len(), 2);
assert_eq!(attempts[0].status, TestStatus::Fail);
assert_eq!(attempts[1].status, TestStatus::Fail);
Ok(())
}
#[tokio::test]
async fn runner_contract_on_error_allow_marks_allowed_and_policy_applied() -> anyhow::Result<()>
{
let cfg = single_test_config(ErrorPolicy::Allow);
let client = Arc::new(ErrorClient);
let runner = runner_for_contract_tests(client, vec![], 2);
let artifacts = runner.run_suite(&cfg, None).await?;
let row = artifacts
.results
.iter()
.find(|r| r.test_id == "t1")
.expect("result for t1");
assert_eq!(row.status, TestStatus::AllowedOnError);
assert_eq!(row.error_policy_applied, Some(ErrorPolicy::Allow));
assert_eq!(row.details["policy_applied"], serde_json::json!("allow"));
let attempts = row.attempts.as_ref().expect("attempts");
assert_eq!(attempts.len(), 1);
assert_eq!(attempts[0].status, TestStatus::AllowedOnError);
Ok(())
}
#[tokio::test]
async fn runner_contract_results_sorted_by_test_id() -> anyhow::Result<()> {
let mut cfg = config_with_test_ids(&["t3", "t1", "t2"], ErrorPolicy::Block);
cfg.settings.parallel = Some(3);
let client = Arc::new(FakeClient::new("fake-model".to_string()).with_response("ok".into()));
let metric = Arc::new(ScriptedMetric::always_pass());
let runner = runner_for_contract_tests(client, vec![metric], 0);
let artifacts = runner.run_suite(&cfg, None).await?;
let ids: Vec<_> = artifacts
.results
.iter()
.map(|r| r.test_id.as_str())
.collect();
assert_eq!(ids, vec!["t1", "t2", "t3"]);
Ok(())
}
#[tokio::test]
async fn runner_contract_progress_sink_reports_done_total() -> anyhow::Result<()> {
let cfg = config_with_test_ids(&["p1", "p2", "p3"], ErrorPolicy::Block);
let client = Arc::new(FakeClient::new("fake-model".to_string()).with_response("ok".into()));
let metric = Arc::new(ScriptedMetric::always_pass());
let runner = runner_for_contract_tests(client, vec![metric], 0);
let events = Arc::new(std::sync::Mutex::new(Vec::<(usize, usize)>::new()));
let sink = {
let events = Arc::clone(&events);
Arc::new(move |ev: crate::report::progress::ProgressEvent| {
events
.lock()
.expect("progress lock")
.push((ev.done, ev.total));
}) as crate::report::progress::ProgressSink
};
let artifacts = runner.run_suite(&cfg, Some(sink)).await?;
assert_eq!(artifacts.results.len(), 3);
let observed = events.lock().expect("progress lock");
assert_eq!(observed.len(), 3);
assert_eq!(observed.last(), Some(&(3, 3)));
assert!(observed.windows(2).all(|w| w[0].0 < w[1].0));
Ok(())
}
#[tokio::test]
async fn runner_contract_relative_baseline_missing_warns_in_helper() -> anyhow::Result<()> {
let mut cfg = single_test_config(ErrorPolicy::Block);
cfg.settings.thresholding = Some(crate::model::ThresholdingSettings {
mode: Some("relative".to_string()),
max_drop: Some(0.05),
min_floor: None,
});
let client = Arc::new(FakeClient::new("fake-model".to_string()).with_response("ok".into()));
let metric = Arc::new(ScriptedMetric::always_pass());
let runner = runner_for_contract_tests(client, vec![], 0);
let baseline = crate::baseline::Baseline {
schema_version: 1,
suite: "runner-contract".to_string(),
assay_version: env!("CARGO_PKG_VERSION").to_string(),
created_at: "2026-01-01T00:00:00Z".to_string(),
config_fingerprint: "md5:test".to_string(),
git_info: None,
entries: vec![],
};
let tc = cfg.tests.first().cloned().expect("single test case");
let details = serde_json::json!({
"metrics": {
"scripted": {
"score": 1.0,
"passed": true,
"unstable": false,
"details": {}
}
}
});
let verdict = runner.check_baseline_regressions(&tc, &cfg, &details, &[metric], &baseline);
let (status, message) = verdict.expect("relative baseline decision");
assert_eq!(status, TestStatus::Warn);
assert_eq!(message, "missing baseline for t1/scripted");
Ok(())
}
}