feat(config): add RSAConfig + named profiles (default-16gb, paper-16k, paper-headline-40k)

This commit is contained in:
transcrilive
2026-05-10 02:36:51 +02:00
parent e2237f788c
commit 08ae956986
2 changed files with 131 additions and 0 deletions

60
tests/test_config.py Normal file
View File

@@ -0,0 +1,60 @@
import pytest
from markovian_rsa_mlx.config import RSAConfig
def test_default_config_values():
cfg = RSAConfig()
assert cfg.rounds == 2
assert cfg.parallel == 4
assert cfg.aggregation_subsample == 4
assert cfg.chunk_tokens == 16384
assert cfg.tail_tokens == 4096
assert cfg.final_tokens is None
assert cfg.temperature == 1.0
assert cfg.top_p == 0.95
assert cfg.top_k == -1
assert cfg.seed is None
assert cfg.serial is False
assert cfg.auto_serial is True
assert cfg.memory_fraction == 0.80
assert cfg.max_context_tokens is None
assert cfg.aggregation_template == "zaya_v1"
assert cfg.answer_selection == "first_final_candidate"
def test_validation_rejects_negative_tokens():
with pytest.raises(ValueError, match="chunk_tokens must be positive"):
RSAConfig(chunk_tokens=0)
def test_validation_rejects_K_greater_than_N():
with pytest.raises(ValueError, match="aggregation_subsample.*<= parallel"):
RSAConfig(parallel=2, aggregation_subsample=4)
def test_default_16gb_profile():
cfg = RSAConfig.default_16gb()
assert cfg.parallel == 2
assert cfg.chunk_tokens == 16384
def test_paper_16k_profile():
cfg = RSAConfig.paper_16k()
assert cfg.parallel == 4
assert cfg.chunk_tokens == 16384
assert cfg.top_p == 1.0
def test_paper_headline_40k_profile():
cfg = RSAConfig.paper_headline_40k()
assert cfg.parallel == 16
assert cfg.chunk_tokens == 40960
assert cfg.final_tokens == 40960
assert cfg.top_p == 1.0
def test_effective_final_tokens_falls_back_to_chunk():
cfg = RSAConfig(chunk_tokens=8192, final_tokens=None)
assert cfg.effective_final_tokens() == 8192
cfg2 = RSAConfig(chunk_tokens=8192, final_tokens=20000)
assert cfg2.effective_final_tokens() == 20000