红尘
d73ff9b0fb
Deploy Site / deploy-vercel (push) Has been cancelled
Deploy Site / deploy-docs (push) Has been cancelled
Docker / shell lint / Lint Dockerfile (hadolint) (push) Has been cancelled
Docker / shell lint / Lint docker/ shell scripts (shellcheck) (push) Has been cancelled
Docker Build and Publish / build-amd64 (push) Has been cancelled
Docker Build and Publish / build-arm64 (push) Has been cancelled
Lint (ruff + ty) / ruff + ty diff (push) Has been cancelled
Lint (ruff + ty) / ruff enforcement (blocking) (push) Has been cancelled
Lint (ruff + ty) / Windows footguns (blocking) (push) Has been cancelled
Nix Lockfile Fix / auto-fix-main (push) Has been cancelled
Nix Lockfile Fix / fix (push) Has been cancelled
Nix / nix (macos-latest) (push) Has been cancelled
Nix / nix (ubuntu-latest) (push) Has been cancelled
OSV-Scanner / Scan lockfiles (push) Has been cancelled
Build Skills Index / build-index (push) Has been cancelled
Tests / test (1) (push) Has been cancelled
Tests / test (2) (push) Has been cancelled
Tests / test (3) (push) Has been cancelled
Tests / test (4) (push) Has been cancelled
Tests / test (5) (push) Has been cancelled
Tests / test (6) (push) Has been cancelled
Tests / e2e (push) Has been cancelled
uv.lock check / uv lock --check (push) Has been cancelled
Docker Build and Publish / merge (push) Has been cancelled
Build Skills Index / trigger-deploy (push) Has been cancelled
Tests / save-durations (push) Has been cancelled
77 lines
2.6 KiB
Python
77 lines
2.6 KiB
Python
"""Tests for Arcee Trinity Large Thinking per-model overrides.
|
|
|
|
Arcee Trinity Large Thinking is a reasoning model that wants:
|
|
- Fixed temperature=0.5 (vs the global default)
|
|
- Compression threshold=0.75 (delay compression to preserve reasoning context)
|
|
|
|
The helpers must match the bare model name, including when it arrives via
|
|
OpenRouter as ``arcee-ai/trinity-large-thinking``, but must NOT hit sibling
|
|
Arcee models like trinity-large-preview or trinity-mini.
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
import pytest
|
|
|
|
from agent.auxiliary_client import (
|
|
_compression_threshold_for_model,
|
|
_fixed_temperature_for_model,
|
|
_is_arcee_trinity_thinking,
|
|
)
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"model",
|
|
[
|
|
"trinity-large-thinking",
|
|
"arcee-ai/trinity-large-thinking",
|
|
"Arcee-AI/Trinity-Large-Thinking", # case-insensitive
|
|
" trinity-large-thinking ", # whitespace tolerant
|
|
],
|
|
)
|
|
def test_is_arcee_trinity_thinking_matches(model: str) -> None:
|
|
assert _is_arcee_trinity_thinking(model) is True
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"model",
|
|
[
|
|
None,
|
|
"",
|
|
"trinity-large-preview",
|
|
"arcee-ai/trinity-large-preview:free",
|
|
"trinity-mini",
|
|
"arcee-ai/trinity-mini",
|
|
"trinity-large", # prefix-only must not match
|
|
"claude-sonnet-4.6",
|
|
"gpt-5.4",
|
|
],
|
|
)
|
|
def test_is_arcee_trinity_thinking_rejects_non_matches(model) -> None:
|
|
assert _is_arcee_trinity_thinking(model) is False
|
|
|
|
|
|
def test_fixed_temperature_for_trinity_thinking() -> None:
|
|
assert _fixed_temperature_for_model("trinity-large-thinking") == 0.5
|
|
assert _fixed_temperature_for_model("arcee-ai/trinity-large-thinking") == 0.5
|
|
|
|
|
|
def test_fixed_temperature_sibling_arcee_models_unaffected() -> None:
|
|
# Preview and mini do not pin temperature — caller chooses its default.
|
|
assert _fixed_temperature_for_model("trinity-large-preview") is None
|
|
assert _fixed_temperature_for_model("trinity-mini") is None
|
|
|
|
|
|
def test_compression_threshold_for_trinity_thinking() -> None:
|
|
assert _compression_threshold_for_model("trinity-large-thinking") == 0.75
|
|
assert _compression_threshold_for_model("arcee-ai/trinity-large-thinking") == 0.75
|
|
|
|
|
|
def test_compression_threshold_default_none_for_other_models() -> None:
|
|
# None means "leave the user's config value unchanged".
|
|
assert _compression_threshold_for_model(None) is None
|
|
assert _compression_threshold_for_model("") is None
|
|
assert _compression_threshold_for_model("trinity-large-preview") is None
|
|
assert _compression_threshold_for_model("claude-sonnet-4.6") is None
|
|
assert _compression_threshold_for_model("kimi-k2") is None
|