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

2026-09-05 13:26:46 +03:00
commit 03634b1ca3
11340 changed files with 3442369 additions and 0 deletions
@@ -0,0 +1,97 @@
+"""Regression tests for parallel image-generation tool batches."""
+
+import json
+from types import SimpleNamespace
+from unittest.mock import MagicMock, patch
+
+import run_agent
+from agent import tool_executor
+
+
+def _tool_call(name: str, args: dict, call_id: str) -> SimpleNamespace:
+    return SimpleNamespace(
+        id=call_id,
+        function=SimpleNamespace(
+            name=name,
+            arguments=json.dumps(args),
+        ),
+    )
+
+
+def test_image_generate_batch_routes_to_concurrent_executor():
+    agent = SimpleNamespace()
+    agent._execute_tool_calls = run_agent.AIAgent._execute_tool_calls.__get__(agent)
+    agent._execute_tool_calls_concurrent = MagicMock()
+    agent._execute_tool_calls_sequential = MagicMock()
+    assistant_message = SimpleNamespace(
+        tool_calls=[
+            _tool_call("image_generate", {"prompt": "variation one"}, "img_1"),
+            _tool_call("image_generate", {"prompt": "variation two"}, "img_2"),
+        ],
+    )
+
+    agent._execute_tool_calls(assistant_message, [], "task-image-batch")
+
+    agent._execute_tool_calls_concurrent.assert_called_once()
+    agent._execute_tool_calls_sequential.assert_not_called()
+
+
+def test_image_generate_parallel_worker_cap_defaults_to_four():
+    runnable_calls = [
+        (
+            0,
+            _tool_call("image_generate", {"prompt": "one"}, "img_1"),
+            "image_generate",
+            {},
+        ),
+        (
+            1,
+            _tool_call("image_generate", {"prompt": "two"}, "img_2"),
+            "image_generate",
+            {},
+        ),
+        (
+            2,
+            _tool_call("image_generate", {"prompt": "three"}, "img_3"),
+            "image_generate",
+            {},
+        ),
+        (
+            3,
+            _tool_call("image_generate", {"prompt": "four"}, "img_4"),
+            "image_generate",
+            {},
+        ),
+        (
+            4,
+            _tool_call("image_generate", {"prompt": "five"}, "img_5"),
+            "image_generate",
+            {},
+        ),
+    ]
+
+    with patch("hermes_cli.config.load_config", return_value={}):
+        assert tool_executor._max_workers_for_tool_batch(runnable_calls) == 4
+
+
+def test_image_generate_parallel_worker_cap_can_be_configured_lower():
+    runnable_calls = [
+        (
+            0,
+            _tool_call("image_generate", {"prompt": "one"}, "img_1"),
+            "image_generate",
+            {},
+        ),
+        (
+            1,
+            _tool_call("image_generate", {"prompt": "two"}, "img_2"),
+            "image_generate",
+            {},
+        ),
+    ]
+
+    with patch(
+        "hermes_cli.config.load_config",
+        return_value={"image_gen": {"max_parallel_requests": 1}},
+    ):
+        assert tool_executor._max_workers_for_tool_batch(runnable_calls) == 1