24 lines
801 B
Python
24 lines
801 B
Python
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
|