File size: 1,321 Bytes
108e227 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 | """Behavior contracts for computer_use latency knobs."""
from unittest.mock import patch
from tools.computer_use import cua_backend
from tools.computer_use import tool as cu_tool
def test_max_image_dimension_default():
with patch("hermes_cli.config.load_config", return_value={}):
assert cua_backend._computer_use_max_image_dimension() == 1456
def test_capture_after_mode_default_som():
with patch("hermes_cli.config.load_config", return_value={}):
assert cu_tool._capture_after_mode() == "som"
def test_aux_vision_route_caches_per_provider_model(monkeypatch):
cu_tool._AUX_VISION_ROUTE_CACHE.clear()
calls = {"n": 0}
monkeypatch.setattr(
"agent.auxiliary_client._read_main_provider", lambda: "openai"
)
monkeypatch.setattr(
"agent.auxiliary_client._read_main_model", lambda: "gpt-test"
)
def fake_load():
calls["n"] += 1
return {"auxiliary": {"vision": {}}}
monkeypatch.setattr("hermes_cli.config.load_config", fake_load)
monkeypatch.setattr(
"tools.computer_use.vision_routing.should_route_capture_to_aux_vision",
lambda *a, **k: True,
)
assert cu_tool._should_route_through_aux_vision() is True
assert cu_tool._should_route_through_aux_vision() is True
assert calls["n"] == 1
|