Import AITURK IDE 1.0.0-beta.1 from Hermes 63279301; preserve MIT license
This commit is contained in:
@@ -0,0 +1,67 @@
|
||||
"""Reasoning-effort resolution for LM Studio.
|
||||
|
||||
Covers the contract that Hermes' generic effort ladder must stay monotonic
|
||||
once it is mapped onto LM Studio's narrower vocabulary: a stronger requested
|
||||
level may resolve to an equal-or-stronger LM Studio level, never a weaker one.
|
||||
"""
|
||||
|
||||
from __future__ import annotations
|
||||
|
||||
import pytest
|
||||
|
||||
from agent.lmstudio_reasoning import resolve_lmstudio_effort
|
||||
from hermes_constants import VALID_REASONING_EFFORTS
|
||||
|
||||
# Rank of each value LM Studio accepts, weakest to strongest. Used to assert
|
||||
# the resolved ladder never inverts.
|
||||
_LM_RANK = {"minimal": 0, "low": 1, "medium": 2, "high": 3, "xhigh": 4}
|
||||
|
||||
|
||||
|
||||
|
||||
def test_effort_ladder_is_monotonic():
|
||||
"""Resolving Hermes' canonical ladder never produces an inversion."""
|
||||
resolved = [
|
||||
resolve_lmstudio_effort({"enabled": True, "effort": effort}, None)
|
||||
for effort in VALID_REASONING_EFFORTS
|
||||
]
|
||||
ranks = [_LM_RANK[value] for value in resolved]
|
||||
assert ranks == sorted(ranks), dict(zip(VALID_REASONING_EFFORTS, resolved))
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
@pytest.mark.parametrize("effort", ["max", "ultra"])
|
||||
def test_clamped_effort_is_still_checked_against_allowed_options(effort):
|
||||
"""A clamped value stays subject to the model's published allowed set.
|
||||
|
||||
"max" resolves to "xhigh"; a model that does not publish "xhigh" gets the
|
||||
field omitted (``None``) so LM Studio applies the model's own default --
|
||||
exactly how a directly-requested "xhigh" already behaves.
|
||||
"""
|
||||
assert (
|
||||
resolve_lmstudio_effort(
|
||||
{"enabled": True, "effort": effort}, ["off", "minimal", "low"]
|
||||
)
|
||||
is None
|
||||
)
|
||||
assert (
|
||||
resolve_lmstudio_effort(
|
||||
{"enabled": True, "effort": effort}, ["low", "medium", "high", "xhigh"]
|
||||
)
|
||||
== "xhigh"
|
||||
)
|
||||
|
||||
|
||||
def test_clamp_does_not_rewrite_published_allowed_options():
|
||||
"""The clamp must not leak into allowed_options normalization.
|
||||
|
||||
A model publishing "max" is not claiming LM Studio's request vocabulary
|
||||
accepts it; allowed_options passes through untouched, so a resolved
|
||||
"xhigh" that the model does not publish is still omitted.
|
||||
"""
|
||||
assert resolve_lmstudio_effort({"enabled": True, "effort": "max"}, ["max"]) is None
|
||||
Reference in New Issue
Block a user