from pathlib import Path

import yaml


def test_models_config_maps_roles_to_same_qwen_non_mtp_model():
    config = yaml.safe_load(Path("config/models.yaml").read_text())

    assert config["default_provider"] == "llama_server"
    roles = config["models"]
    for role in ["thinker", "critic", "coder", "action", "summary"]:
        assert roles[role]["base_url"] == "http://127.0.0.1:8081/v1"
        assert roles[role]["model"] == "local-main"

    assert roles["action"]["structured_output"] is True
    assert roles["thinker"]["max_output_tokens"] == 8192


def test_action_prompt_forbids_tools_for_simple_arithmetic():
    prompt = Path("prompts/roles/action.md").read_text()

    assert "Return actions=[] for simple arithmetic" in prompt
    assert "Do not run shell commands like" in prompt