Files
zk-data-agent/tests/test_bootstrap.py
T
2026-07-26 18:50:51 +08:00

17 lines
776 B
Python

from agent_platform.bootstrap import LEGACY_MODEL_IDS, _model_payload
def test_bootstrap_models_are_agent_only_and_public() -> None:
models = _model_payload()
assert {model["id"] for model in models} == {"luna", "terra", "sol", "deepseek-v4-pro"}
assert all(model["base_model_id"] is None for model in models)
assert all(model["params"] == {} for model in models)
assert all(model["meta"]["toolIds"] == [] for model in models)
assert all(model["meta"]["tags"] == [{"name": "Agent"}] for model in models)
assert all(
model["access_grants"] == [{"principal_type": "user", "principal_id": "*", "permission": "read"}]
for model in models
)
assert "chat-medium" in LEGACY_MODEL_IDS
assert "work-medium" in LEGACY_MODEL_IDS