mirror of
https://github.com/headroomlabs-ai/headroom.git
synced 2026-08-27 14:17:10 -04:00
test: add fluent Headroom harness (#2650)
## Description Adds `headroom.testing`, a fluent, contractual test harness for building Headroom scenarios and suites that can be simulated locally, orchestrated, deployed through the proxy, and handed off to `headroom-bench` / `agent-evals` with bench-native manifests. Closes # ## Type of Change - [x] Bug fix (non-breaking change that fixes an issue) - [x] New feature (non-breaking change that adds functionality) - [ ] Breaking change (fix or feature that would cause existing functionality to change) - [x] Documentation update - [ ] Performance improvement - [ ] Code refactoring (no functional changes) ## Changes Made - Add `headroom.testing.Headroom` fluent scenario builder with provider/platform/configuration facets such as `WithBedrock`, `OnAppleSilicon`, `Configure`, `WithCompression`, `WithCCR`, `WithCache`, `WithPrefixFreeze`, `WithReadMaturation`, and `WithMemory`. - Add contractual coverage over the current `HeadroomConfig` and `ProxyConfig` dataclass surfaces, including full JSON-ready proxy deployment payloads. - Add no-key local simulations, scenario/suite orchestration, guarantee evaluation, deployment plans, and a local proxy lifecycle context manager. - Add `headroom-bench` handoff artifacts, including `agent_evals.models.RunManifest`-compatible JSON without taking a runtime dependency on `agent-evals`. - Add demonstration tests for providers, feature facets, manifests, suites, guarantees, deployment payloads, and the no-key simulation path. - Fix unversioned OTEL meter lookup typing so `mypy headroom` remains green on current `main`. ## Testing - [x] Unit tests pass (`pytest`) - [x] Linting passes (`ruff check .`) - [x] Type checking passes (`mypy headroom`) - [x] New tests added for new functionality - [x] Manual testing performed ### Test Output ```text python -m ruff check . All checks passed! python -m mypy headroom headroom\proxy\server.py:1680: note: By default the bodies of untyped functions are not checked, consider using --check-untyped-defs [annotation-unchecked] headroom\proxy\server.py:1691: note: By default the bodies of untyped functions are not checked, consider using --check-untyped-defs [annotation-unchecked] Success: no issues found in 512 source files python -m pytest tests/test_cli/test_subprocess_utf8_encoding.py tests/test_testing_harness.py -q 24 passed, 1 warning in 4.68s ``` ## Real Behavior Proof - Environment: Windows, Python 3.13.13, branch `feat/headroom-test-harness` rebased on `headroomlabs-ai/main`. - Exact command / steps: built a `Headroom.WithOpenAI().WithCompression(mode="cache", kompress=False).Build()` scenario and entered `scenario.deploy_local(port=19192, timeout_s=20)`. - Observed result: proxy launched, `/readyz` succeeded, handle returned `http://127.0.0.1:19192`, `OPENAI_BASE_URL=http://127.0.0.1:19192/v1`, and context-manager teardown completed. - Exact command / steps: emitted `scenario.agent_evals_manifest(...).to_dict()` and validated it with the current cloned `headroom-bench` `agent_evals.models.RunManifest` pydantic model. - Observed result: validation succeeded with arms `a0_direct`, `a1_passthrough`, and `b_headroom` for provider `openai`. - Not tested: upstream-provider API calls requiring real OpenAI/Anthropic/Bedrock keys; phase-1 validation intentionally stays no-key/local. ## Review Readiness - [x] I have performed a self-review - [x] This PR is ready for human review ## Checklist - [x] My code follows the project's style guidelines - [x] I have performed a self-review of my code - [x] I have commented my code, particularly in hard-to-understand areas - [x] I have made corresponding changes to the documentation - [x] My changes generate no new warnings - [x] I have added tests that prove my fix is effective or that my feature works - [x] New and existing unit tests pass locally with my changes - [x] I did **not** edit `CHANGELOG.md` — it is generated by release-please from my Conventional Commit PR title (a CI guard enforces this) ## Screenshots (if applicable) N/A. ## Additional Notes The pytest warning shown above is the existing OpenAI pricing-data staleness warning from cost estimation. The harness does not call upstream providers during local simulation.
This commit is contained in:
parent
e0d2cd0c5a
commit
2dc7e4ab27
5 changed files with 2155 additions and 0 deletions
|
|
@ -321,7 +321,11 @@ class HeadroomOtelMetrics:
|
|||
"""
|
||||
|
||||
if self._meter_provider is None:
|
||||
if version is None:
|
||||
return metrics.get_meter(name)
|
||||
return metrics.get_meter(name, version)
|
||||
if version is None:
|
||||
return self._meter_provider.get_meter(name)
|
||||
return self._meter_provider.get_meter(name, version)
|
||||
|
||||
@staticmethod
|
||||
|
|
|
|||
72
headroom/testing/README.md
Normal file
72
headroom/testing/README.md
Normal file
|
|
@ -0,0 +1,72 @@
|
|||
# Headroom Testing Harness
|
||||
|
||||
`headroom.testing` is the fluent scenario contract for local simulations,
|
||||
deployment planning, and `headroom-bench` handoff.
|
||||
|
||||
## Scenario
|
||||
|
||||
```python
|
||||
from headroom.testing import Headroom, ScenarioTask
|
||||
|
||||
scenario = (
|
||||
Headroom.WithBedrock(region="us-east-1", profile="bench")
|
||||
.OnAppleSilicon()
|
||||
.WithCompression(mode="cache", kompress=False, savings_profile="coding")
|
||||
.WithCCR(enabled=True, inject_tool=False, inject_marker=True)
|
||||
.WithMemory(enabled=True, mode="tool", top_k=4)
|
||||
.Configure(lambda c: setattr(c, "default_mode", "optimize"))
|
||||
.Build()
|
||||
)
|
||||
|
||||
task = ScenarioTask(
|
||||
task_id="smoke",
|
||||
messages=[{"role": "user", "content": "Summarize this payload."}],
|
||||
)
|
||||
|
||||
report = scenario.orchestrate([task])
|
||||
assert report.passed
|
||||
```
|
||||
|
||||
## Suite
|
||||
|
||||
```python
|
||||
suite = (
|
||||
Headroom.Suite("phase-1")
|
||||
.Add(Headroom.WithOpenAI().named("openai-cache").WithCompression(mode="cache"))
|
||||
.Add(Headroom.WithBedrock(region="us-east-1").named("bedrock-token").WithCompression(mode="token"))
|
||||
)
|
||||
|
||||
suite.write_manifest_bundle("headroom-testing-bundle.json", provider="openai", port_start=19000)
|
||||
suite.write_agent_evals_manifests(
|
||||
"agent-evals-manifests",
|
||||
benchmark="mini_swebench",
|
||||
benchmark_ref="mini@abc123",
|
||||
provider="openai",
|
||||
)
|
||||
```
|
||||
|
||||
## Contract
|
||||
|
||||
Every scenario builds real `HeadroomConfig` and `ProxyConfig` objects. The harness
|
||||
exports the complete dataclass constructor surface, including compatibility
|
||||
`InitVar` fields, through:
|
||||
|
||||
- `scenario.audit_contract()`
|
||||
- `scenario.deployment_plan()`
|
||||
- `scenario.bench_manifest_fragment()`
|
||||
- `scenario.agent_evals_manifest()`
|
||||
- `suite.manifest_bundle()`
|
||||
- `suite.agent_evals_manifests()`
|
||||
|
||||
Phase-1 validation is local and no-key: `simulate` and `orchestrate` run the SDK
|
||||
transform pipeline without calling upstream providers. Live proxy deployment is
|
||||
available through `scenario.deploy_local(...)`; deployment plans carry the full
|
||||
proxy config in `HEADROOM_PROXY_CONFIG_JSON`.
|
||||
|
||||
## headroom-bench
|
||||
|
||||
`agent_evals_manifest(...)` emits the same top-level field names as
|
||||
`agent_evals.models.RunManifest`, including `arms` entries shaped like
|
||||
`ArmSpec`. The harness keeps this adapter dependency-free: `headroom-ai` can
|
||||
write bench-native JSON without importing `agent-evals`, while `headroom-bench`
|
||||
can validate the artifact with its own pydantic model.
|
||||
68
headroom/testing/__init__.py
Normal file
68
headroom/testing/__init__.py
Normal file
|
|
@ -0,0 +1,68 @@
|
|||
"""Fluent testing harness for Headroom scenarios.
|
||||
|
||||
The testing package is intentionally small at import time. It builds real
|
||||
``HeadroomConfig`` and ``ProxyConfig`` instances and exposes the same surface to
|
||||
bench runners, smoke tests, and local simulations.
|
||||
"""
|
||||
|
||||
from __future__ import annotations
|
||||
|
||||
from .harness import (
|
||||
AgentEvalsManifest,
|
||||
AgentEvalsPricing,
|
||||
ArmName,
|
||||
BenchArm,
|
||||
BenchManifestFragment,
|
||||
Configurator,
|
||||
ContractAudit,
|
||||
DeploymentHandle,
|
||||
FieldContract,
|
||||
Guarantee,
|
||||
GuaranteeResult,
|
||||
HarnessScenario,
|
||||
Headroom,
|
||||
HeadroomSuite,
|
||||
LocalProxyDeployment,
|
||||
PlatformTarget,
|
||||
ProviderTarget,
|
||||
ProxyDeploymentPlan,
|
||||
ScenarioCaseResult,
|
||||
ScenarioContract,
|
||||
ScenarioOrchestrator,
|
||||
ScenarioResult,
|
||||
ScenarioRunReport,
|
||||
ScenarioTask,
|
||||
SuiteManifestBundle,
|
||||
guarantee_messages_remain_non_empty,
|
||||
guarantee_tokens_do_not_increase,
|
||||
)
|
||||
|
||||
__all__ = [
|
||||
"AgentEvalsManifest",
|
||||
"AgentEvalsPricing",
|
||||
"ArmName",
|
||||
"BenchArm",
|
||||
"BenchManifestFragment",
|
||||
"Configurator",
|
||||
"ContractAudit",
|
||||
"DeploymentHandle",
|
||||
"FieldContract",
|
||||
"Guarantee",
|
||||
"GuaranteeResult",
|
||||
"Headroom",
|
||||
"HarnessScenario",
|
||||
"HeadroomSuite",
|
||||
"LocalProxyDeployment",
|
||||
"PlatformTarget",
|
||||
"ProviderTarget",
|
||||
"ProxyDeploymentPlan",
|
||||
"ScenarioCaseResult",
|
||||
"ScenarioContract",
|
||||
"ScenarioOrchestrator",
|
||||
"ScenarioResult",
|
||||
"ScenarioRunReport",
|
||||
"ScenarioTask",
|
||||
"SuiteManifestBundle",
|
||||
"guarantee_messages_remain_non_empty",
|
||||
"guarantee_tokens_do_not_increase",
|
||||
]
|
||||
1552
headroom/testing/harness.py
Normal file
1552
headroom/testing/harness.py
Normal file
File diff suppressed because it is too large
Load diff
459
tests/test_testing_harness.py
Normal file
459
tests/test_testing_harness.py
Normal file
|
|
@ -0,0 +1,459 @@
|
|||
from __future__ import annotations
|
||||
|
||||
import json
|
||||
from datetime import datetime, timezone
|
||||
from pathlib import Path
|
||||
|
||||
import pytest
|
||||
|
||||
from headroom.config import HeadroomConfig
|
||||
from headroom.proxy.models import ProxyConfig
|
||||
from headroom.testing import (
|
||||
AgentEvalsPricing,
|
||||
ArmName,
|
||||
Configurator,
|
||||
GuaranteeResult,
|
||||
Headroom,
|
||||
ProviderTarget,
|
||||
ScenarioOrchestrator,
|
||||
ScenarioTask,
|
||||
)
|
||||
|
||||
AGENT_EVALS_RUN_MANIFEST_FIELDS = {
|
||||
"experiment_id",
|
||||
"created_at",
|
||||
"headroom_git_sha",
|
||||
"agent_evals_git_sha",
|
||||
"model_snapshot",
|
||||
"provider",
|
||||
"auth_mode",
|
||||
"benchmark",
|
||||
"benchmark_ref",
|
||||
"harness",
|
||||
"harness_version",
|
||||
"docker_digests",
|
||||
"arms",
|
||||
"k_runs",
|
||||
"temperature",
|
||||
"seeds",
|
||||
"alpha",
|
||||
"margins",
|
||||
"pricing",
|
||||
}
|
||||
|
||||
|
||||
def _configure_bedrock_apple(c: Configurator) -> None:
|
||||
c.kompress_enabled = False
|
||||
c.mode = "cache"
|
||||
c.default_mode = "optimize"
|
||||
|
||||
|
||||
def test_contract_covers_current_headroom_and_proxy_config_fields() -> None:
|
||||
scenario = Headroom.scenario("contract").build()
|
||||
|
||||
assert scenario.contract.field_names("headroom") == set(HeadroomConfig.__dataclass_fields__)
|
||||
assert scenario.contract.field_names("proxy") == set(ProxyConfig.__dataclass_fields__)
|
||||
|
||||
|
||||
def test_fluent_builder_configures_real_proxy_and_sdk_configs() -> None:
|
||||
scenario = (
|
||||
Headroom.WithBedrock(region="us-east-1", profile="bench")
|
||||
.named("bedrock-apple")
|
||||
.OnAppleSilicon()
|
||||
.configure(_configure_bedrock_apple)
|
||||
.configure_proxy(savings_profile="coding", min_tokens_to_crush=10)
|
||||
.build()
|
||||
)
|
||||
|
||||
assert scenario.provider is ProviderTarget.BEDROCK
|
||||
assert scenario.proxy_config.backend == "bedrock"
|
||||
assert scenario.proxy_config.bedrock_region == "us-east-1"
|
||||
assert scenario.proxy_config.bedrock_profile == "bench"
|
||||
assert scenario.proxy_config.disable_kompress is True
|
||||
assert scenario.proxy_config.mode == "cache"
|
||||
assert scenario.headroom_config.default_mode.value == "optimize"
|
||||
|
||||
command = scenario.proxy_command(port=18800)
|
||||
assert command[:4] == ("headroom", "proxy", "--port", "18800")
|
||||
assert "--backend" in command
|
||||
assert "bedrock" in command
|
||||
assert "--disable-kompress" in command
|
||||
assert "--savings-profile" not in command
|
||||
|
||||
env = scenario.env()
|
||||
assert env["HEADROOM_BACKEND"] == "bedrock"
|
||||
assert env["HEADROOM_DISABLE_KOMPRESS"] == "1"
|
||||
assert env["HEADROOM_BEDROCK_REGION"] == "us-east-1"
|
||||
assert env["AWS_PROFILE"] == "bench"
|
||||
|
||||
|
||||
def test_unknown_config_field_fails_fast() -> None:
|
||||
with pytest.raises(AttributeError, match="unknown Headroom harness config field"):
|
||||
Headroom.scenario().configure(not_a_real_knob=True)
|
||||
|
||||
|
||||
def test_bench_manifest_fragment_matches_headroom_bench_arm_shape() -> None:
|
||||
scenario = (
|
||||
Headroom.with_openai()
|
||||
.configure(mode="cache", kompress_enabled=False)
|
||||
.configure_proxy(savings_profile="coding")
|
||||
.build()
|
||||
)
|
||||
|
||||
fragment = scenario.bench_manifest_fragment(provider="openai").to_dict()
|
||||
|
||||
assert fragment["harness"] == "headroom.testing"
|
||||
assert [arm["name"] for arm in fragment["arms"]] == [
|
||||
ArmName.A0_DIRECT.value,
|
||||
ArmName.A1_PASSTHROUGH.value,
|
||||
ArmName.B_HEADROOM.value,
|
||||
]
|
||||
assert fragment["arms"][0]["proxy_mode"] is None
|
||||
assert fragment["arms"][1]["proxy_mode"] == "off"
|
||||
assert fragment["arms"][2]["proxy_mode"] == "cache"
|
||||
assert fragment["arms"][2]["proxy_flags"] == ["--disable-kompress"]
|
||||
assert fragment["env"]["HEADROOM_MODE"] == "cache"
|
||||
assert fragment["env"]["HEADROOM_SAVINGS_PROFILE"] == "coding"
|
||||
assert fragment["deployment_plan"]["config_env_var"] == "HEADROOM_PROXY_CONFIG_JSON"
|
||||
assert fragment["contract_audit"]["passed"] is True
|
||||
json.dumps(fragment)
|
||||
|
||||
|
||||
def test_agent_evals_manifest_matches_run_manifest_contract() -> None:
|
||||
now = datetime(2026, 6, 15, 9, 30, tzinfo=timezone.utc)
|
||||
scenario = (
|
||||
Headroom.with_openai()
|
||||
.named("openai-cache")
|
||||
.WithCompression(mode="cache", kompress=False)
|
||||
.Build()
|
||||
)
|
||||
|
||||
manifest = scenario.agent_evals_manifest(
|
||||
benchmark="mini_swebench",
|
||||
benchmark_ref="mini@abc123",
|
||||
provider="openai",
|
||||
now=now,
|
||||
model_snapshot="openai/gpt-4o",
|
||||
headroom_repo_path="/nonexistent-headroom",
|
||||
agent_evals_repo_path="/nonexistent-agent-evals",
|
||||
k_runs=3,
|
||||
pricing=AgentEvalsPricing(input_usd_per_1m=2.5, output_usd_per_1m=10.0),
|
||||
)
|
||||
payload = manifest.to_dict()
|
||||
|
||||
assert set(payload) == AGENT_EVALS_RUN_MANIFEST_FIELDS
|
||||
assert payload["experiment_id"] == "mini_swebench-openai-cache-20260615T093000Z"
|
||||
assert payload["created_at"] == "2026-06-15T09:30:00+00:00"
|
||||
assert payload["headroom_git_sha"] == "unknown"
|
||||
assert payload["agent_evals_git_sha"] == "unknown"
|
||||
assert payload["provider"] == "openai"
|
||||
assert payload["benchmark"] == "mini_swebench"
|
||||
assert payload["benchmark_ref"] == "mini@abc123"
|
||||
assert payload["harness"] == "headroom.testing"
|
||||
assert payload["model_snapshot"] == "openai/gpt-4o"
|
||||
assert payload["seeds"] == [0, 1, 2]
|
||||
assert payload["margins"] == {"ccr": 0.0, "lossy": 2.0}
|
||||
assert payload["pricing"] == {"input_usd_per_1m": 2.5, "output_usd_per_1m": 10.0}
|
||||
assert [arm["name"] for arm in payload["arms"]] == [
|
||||
"a0_direct",
|
||||
"a1_passthrough",
|
||||
"b_headroom",
|
||||
]
|
||||
assert payload["arms"][2]["proxy_mode"] == "cache"
|
||||
assert payload["arms"][2]["proxy_flags"] == ["--disable-kompress"]
|
||||
json.dumps(payload)
|
||||
|
||||
|
||||
def test_sdk_simulation_runs_without_provider_api_keys() -> None:
|
||||
scenario = Headroom.with_openai().configure(default_mode="optimize").build()
|
||||
messages = [
|
||||
{"role": "system", "content": "You are concise."},
|
||||
{"role": "user", "content": "Summarize this small payload."},
|
||||
]
|
||||
|
||||
result = scenario.simulate(messages, model="gpt-4o")
|
||||
|
||||
assert result.tokens_before >= result.tokens_after
|
||||
assert result.tokens_saved >= 0
|
||||
assert result.messages
|
||||
|
||||
|
||||
def test_deployment_plan_carries_full_proxy_config_payload_through_env() -> None:
|
||||
scenario = (
|
||||
Headroom.WithBedrock(region="us-east-2", profile="bench")
|
||||
.Configure(mode="cache", kompress_enabled=False)
|
||||
.ConfigureProxy(memory_enabled=True, memory_top_k=3, offline=True)
|
||||
.Build()
|
||||
)
|
||||
|
||||
plan = scenario.deployment_plan(port=18888)
|
||||
payload_from_env = json.loads(plan.env["HEADROOM_PROXY_CONFIG_JSON"])
|
||||
|
||||
assert plan.command[:4] == ("headroom", "proxy", "--port", "18888")
|
||||
assert payload_from_env == plan.config_payload
|
||||
assert plan.env["HEADROOM_SKIP_UPSTREAM_CHECK"] == "1"
|
||||
assert plan.config_payload["backend"] == "bedrock"
|
||||
assert plan.config_payload["memory_enabled"] is True
|
||||
assert plan.config_payload["memory_top_k"] == 3
|
||||
assert plan.config_payload["offline"] is True
|
||||
assert set(plan.config_payload) == set(ProxyConfig.__dataclass_fields__)
|
||||
|
||||
|
||||
def test_contract_audit_reports_full_payload_coverage_and_proxy_only_notes() -> None:
|
||||
scenario = (
|
||||
Headroom.WithBedrock(region="us-east-1")
|
||||
.WithReadMaturation(enabled=True, quiesce_turns=2)
|
||||
.Build()
|
||||
)
|
||||
|
||||
audit = scenario.audit_contract()
|
||||
|
||||
assert audit.passed is True
|
||||
assert audit.missing_headroom_payload_fields == ()
|
||||
assert audit.missing_proxy_payload_fields == ()
|
||||
assert audit.extra_headroom_payload_fields == ()
|
||||
assert audit.extra_proxy_payload_fields == ()
|
||||
assert audit.headroom_fields_total == len(HeadroomConfig.__dataclass_fields__)
|
||||
assert audit.proxy_fields_total == len(ProxyConfig.__dataclass_fields__)
|
||||
assert "read_maturation is currently a proxy-only surface" in audit.notes
|
||||
json.dumps(audit.to_dict())
|
||||
|
||||
|
||||
def test_write_manifest_fragment_outputs_json_file(tmp_path: Path) -> None:
|
||||
path = tmp_path / "headroom-manifest-fragment.json"
|
||||
scenario = (
|
||||
Headroom.WithOpenAI(api_url="https://openai.internal")
|
||||
.WithCompression(mode="cache", kompress=False)
|
||||
.Build()
|
||||
)
|
||||
|
||||
written = scenario.write_manifest_fragment(path, provider="openai")
|
||||
payload = json.loads(written.read_text(encoding="utf-8"))
|
||||
|
||||
assert written == path
|
||||
assert payload["harness"] == "headroom.testing"
|
||||
assert payload["provider"] == "openai"
|
||||
assert (
|
||||
payload["deployment_plan"]["config_payload"]["openai_api_url"] == "https://openai.internal"
|
||||
)
|
||||
assert payload["contract_audit"]["passed"] is True
|
||||
|
||||
|
||||
def test_feature_facets_configure_authoritative_scenario_surfaces() -> None:
|
||||
scenario = (
|
||||
Headroom.WithAnthropic(api_url="https://anthropic.internal")
|
||||
.named("enterprise-feature-matrix")
|
||||
.WithCompression(
|
||||
mode="cache",
|
||||
kompress=False,
|
||||
lossless=True,
|
||||
compressors=["smart_crusher", "log", "diff"],
|
||||
min_tokens=25,
|
||||
max_items=9,
|
||||
savings_profile="coding",
|
||||
)
|
||||
.WithCCR(
|
||||
enabled=True,
|
||||
inject_tool=False,
|
||||
inject_marker=True,
|
||||
handle_responses=True,
|
||||
proactive_expansion=False,
|
||||
max_retrieval_rounds=1,
|
||||
)
|
||||
.WithCache(enabled=True, semantic=True, ttl_seconds=120, max_entries=33)
|
||||
.WithPrefixFreeze(enabled=False, session_ttl_seconds=42)
|
||||
.WithReadMaturation(enabled=True, quiesce_turns=2, max_hold_turns=8, min_size_bytes=512)
|
||||
.WithMemory(
|
||||
enabled=True,
|
||||
backend="local",
|
||||
mode="tool",
|
||||
top_k=4,
|
||||
min_similarity=0.5,
|
||||
inject_tools=False,
|
||||
inject_context=False,
|
||||
storage_mode="project",
|
||||
)
|
||||
.Build()
|
||||
)
|
||||
|
||||
assert scenario.proxy_config.anthropic_api_url == "https://anthropic.internal"
|
||||
assert scenario.proxy_config.disable_kompress is True
|
||||
assert scenario.proxy_config.lossless is True
|
||||
assert scenario.proxy_config.compressors == {"smart_crusher", "log", "diff"}
|
||||
assert scenario.proxy_config.min_tokens_to_crush == 25
|
||||
assert scenario.proxy_config.max_items_after_crush == 9
|
||||
assert scenario.headroom_config.smart_crusher.lossless_only is True
|
||||
assert scenario.headroom_config.smart_crusher.min_tokens_to_crush == 25
|
||||
assert scenario.headroom_config.smart_crusher.max_items_after_crush == 9
|
||||
assert scenario.proxy_config.ccr_inject_tool is False
|
||||
assert scenario.proxy_config.ccr_inject_marker is True
|
||||
assert scenario.proxy_config.ccr_proactive_expansion is False
|
||||
assert scenario.proxy_config.ccr_max_retrieval_rounds == 1
|
||||
assert scenario.headroom_config.ccr.enabled is True
|
||||
assert scenario.headroom_config.ccr.inject_tool is False
|
||||
assert scenario.headroom_config.ccr.inject_retrieval_marker is True
|
||||
assert scenario.proxy_config.cache_ttl_seconds == 120
|
||||
assert scenario.proxy_config.cache_max_entries == 33
|
||||
assert scenario.headroom_config.cache_optimizer.enable_semantic_cache is True
|
||||
assert scenario.proxy_config.prefix_freeze_enabled is False
|
||||
assert scenario.headroom_config.prefix_freeze.enabled is False
|
||||
assert scenario.proxy_config.read_maturation is True
|
||||
assert scenario.metadata["read_maturation"]["quiesce_turns"] == 2
|
||||
assert scenario.proxy_config.memory_enabled is True
|
||||
assert scenario.proxy_config.memory_mode == "tool"
|
||||
assert scenario.proxy_config.memory_top_k == 4
|
||||
assert scenario.proxy_config.memory_min_similarity == 0.5
|
||||
assert scenario.proxy_config.memory_inject_tools is False
|
||||
assert scenario.proxy_config.memory_inject_context is False
|
||||
|
||||
payload = scenario.deployment_plan(port=18889).config_payload
|
||||
assert payload["compressors"] == ["diff", "log", "smart_crusher"]
|
||||
assert payload["read_maturation"] is True
|
||||
assert payload["memory_mode"] == "tool"
|
||||
|
||||
|
||||
@pytest.mark.parametrize(
|
||||
("builder", "expected_provider", "expected_backend"),
|
||||
[
|
||||
(
|
||||
lambda: Headroom.WithAnthropic(api_url="https://anthropic.internal"),
|
||||
"anthropic",
|
||||
"anthropic",
|
||||
),
|
||||
(lambda: Headroom.WithOpenAI(api_url="https://openai.internal"), "openai", "anthropic"),
|
||||
(lambda: Headroom.WithGemini(api_url="https://gemini.internal"), "gemini", "anthropic"),
|
||||
(
|
||||
lambda: Headroom.WithCloudCode(api_url="https://cloudcode.internal"),
|
||||
"cloudcode",
|
||||
"anthropic",
|
||||
),
|
||||
(
|
||||
lambda: Headroom.WithVertex(api_url="https://vertex.internal"),
|
||||
"vertex",
|
||||
"litellm-vertex",
|
||||
),
|
||||
(lambda: Headroom.WithBedrock(region="us-east-1"), "bedrock", "bedrock"),
|
||||
(lambda: Headroom.WithAnyLLM(provider="mistral"), "anyllm", "anyllm"),
|
||||
(lambda: Headroom.WithLiteLLM(provider="openrouter"), "litellm", "litellm-openrouter"),
|
||||
],
|
||||
)
|
||||
def test_provider_builders_cover_current_proxy_targets(
|
||||
builder: object,
|
||||
expected_provider: str,
|
||||
expected_backend: str,
|
||||
) -> None:
|
||||
scenario = builder().WithCompression(mode="cache").Build() # type: ignore[operator]
|
||||
plan = scenario.deployment_plan(port=18901)
|
||||
|
||||
assert scenario.provider.value == expected_provider
|
||||
assert plan.config_payload["backend"] == expected_backend
|
||||
assert plan.command[:4] == ("headroom", "proxy", "--port", "18901")
|
||||
assert json.loads(plan.env["HEADROOM_PROXY_CONFIG_JSON"]) == plan.config_payload
|
||||
|
||||
|
||||
def test_scenario_orchestrator_runs_multiple_scenarios_and_reports_guarantees() -> None:
|
||||
passthrough = (
|
||||
Headroom.with_openai()
|
||||
.named("passthrough")
|
||||
.Configure(optimize=False, default_mode="audit")
|
||||
.Build()
|
||||
)
|
||||
optimized = (
|
||||
Headroom.with_openai()
|
||||
.named("optimized")
|
||||
.Configure(mode="cache", default_mode="optimize")
|
||||
.Build()
|
||||
)
|
||||
task = ScenarioTask(
|
||||
task_id="tiny-chat",
|
||||
messages=[
|
||||
{"role": "system", "content": "You are concise."},
|
||||
{"role": "user", "content": "Summarize this small payload."},
|
||||
],
|
||||
model="gpt-4o",
|
||||
)
|
||||
|
||||
report = ScenarioOrchestrator([passthrough, optimized]).run([task])
|
||||
|
||||
assert report.passed is True
|
||||
assert len(report.cases) == 2
|
||||
assert report.total_tokens_before >= report.total_tokens_after
|
||||
payload = report.to_dict()
|
||||
assert payload["total_cases"] == 2
|
||||
assert payload["cases"][0]["guarantees"]
|
||||
|
||||
|
||||
def test_orchestrator_surfaces_custom_guarantee_failures() -> None:
|
||||
scenario = Headroom.with_openai().named("guarded").Build()
|
||||
task = ScenarioTask(
|
||||
task_id="expected-failure",
|
||||
messages=[{"role": "user", "content": "hello"}],
|
||||
)
|
||||
|
||||
def always_fail(*_args: object) -> GuaranteeResult:
|
||||
return GuaranteeResult(name="always_fail", passed=False, detail="demonstration failure")
|
||||
|
||||
report = ScenarioOrchestrator([scenario], guarantees=[always_fail]).run([task])
|
||||
|
||||
assert report.passed is False
|
||||
assert report.cases[0].passed is False
|
||||
assert report.to_dict()["cases"][0]["guarantees"] == [
|
||||
{"name": "always_fail", "passed": False, "detail": "demonstration failure"}
|
||||
]
|
||||
|
||||
|
||||
def test_headroom_suite_orchestrates_matrix_and_assigns_deployment_ports(tmp_path: Path) -> None:
|
||||
suite = (
|
||||
Headroom.Suite("phase-1-matrix")
|
||||
.Add(Headroom.WithOpenAI().named("openai-cache").WithCompression(mode="cache"))
|
||||
.Add(
|
||||
Headroom.WithBedrock(region="us-east-1")
|
||||
.named("bedrock-token")
|
||||
.WithCompression(mode="token")
|
||||
)
|
||||
)
|
||||
task = ScenarioTask(
|
||||
task_id="suite-smoke",
|
||||
messages=[{"role": "user", "content": "hello"}],
|
||||
)
|
||||
|
||||
report = suite.Orchestrate([task])
|
||||
plans = suite.DeploymentPlans(port_start=19000)
|
||||
bundle = suite.ManifestBundle(provider="openai", port_start=19000).to_dict()
|
||||
path = suite.WriteManifestBundle(tmp_path / "suite.json", provider="openai", port_start=19000)
|
||||
agent_paths = suite.WriteAgentEvalsManifests(
|
||||
tmp_path / "agent-evals",
|
||||
benchmark="mini_swebench",
|
||||
benchmark_ref="mini@abc123",
|
||||
provider="openai",
|
||||
now=datetime(2026, 6, 15, 9, 30, tzinfo=timezone.utc),
|
||||
)
|
||||
written = json.loads(path.read_text(encoding="utf-8"))
|
||||
|
||||
assert report.passed is True
|
||||
assert len(report.cases) == 2
|
||||
assert set(plans) == {"openai-cache", "bedrock-token"}
|
||||
assert plans["openai-cache"].command[:4] == ("headroom", "proxy", "--port", "19000")
|
||||
assert plans["bedrock-token"].command[:4] == ("headroom", "proxy", "--port", "19001")
|
||||
assert bundle["name"] == "phase-1-matrix"
|
||||
assert [scenario["suite_port"] for scenario in bundle["scenarios"]] == [19000, 19001]
|
||||
assert written == bundle
|
||||
assert {path.name for path in agent_paths} == {
|
||||
"openai-cache.agent-evals.json",
|
||||
"bedrock-token.agent-evals.json",
|
||||
}
|
||||
first_agent_payload = json.loads(agent_paths[0].read_text(encoding="utf-8"))
|
||||
assert set(first_agent_payload) == AGENT_EVALS_RUN_MANIFEST_FIELDS
|
||||
assert first_agent_payload["provider"] == "openai"
|
||||
|
||||
|
||||
def test_headroom_suite_rejects_duplicate_scenario_names() -> None:
|
||||
suite = Headroom.Suite("duplicates").Add(Headroom.WithOpenAI().named("same"))
|
||||
|
||||
with pytest.raises(ValueError, match="duplicate scenario name"):
|
||||
suite.Add(Headroom.WithBedrock(region="us-east-1").named("same"))
|
||||
|
||||
|
||||
def test_empty_suite_fails_loudly() -> None:
|
||||
with pytest.raises(ValueError, match="requires at least one scenario"):
|
||||
Headroom.Suite("empty").DeploymentPlans()
|
||||
Loading…
Add table
Add a link
Reference in a new issue