Import AITURK IDE 1.0.0-beta.1 from Hermes 63279301; preserve MIT license

tests/computer_use/test_cua_perf_knobs.py

@@ -0,0 +1,48 @@
+"""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