feat(moa): add reference model toggles

This commit is contained in:
oppenheimor 2026-07-07 00:30:22 +08:00 committed by Teknium
parent ad6a2ae401
commit ca294d3e62
12 changed files with 181 additions and 22 deletions

View file

@ -880,6 +880,7 @@ def aggregate_moa_context(
so the provider default applies matching single-model agent behavior.
Presets may still pin explicit values.
"""
reference_models = [slot for slot in reference_models if slot.get("enabled", True)]
reference_outputs: list[tuple[str, str, Any]] = []
ref_messages = _reference_messages(api_messages)
reference_outputs = _run_references_parallel(
@ -1267,7 +1268,10 @@ class MoAChatCompletions:
privacy_mode = _moa_privacy_mode(_moa_raw)
self._privacy_mode = privacy_mode
messages = list(api_kwargs.get("messages") or [])
reference_models = preset.get("reference_models") or []
reference_models = [
slot for slot in (preset.get("reference_models") or [])
if slot.get("enabled", True)
]
aggregator = preset.get("aggregator") or {}
# Expose the resolved aggregator slot so session cost accounting can
# price the aggregator's acting turn at its REAL model/provider. The